Convolutional_nn.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "opening-graphics",
   "metadata": {},
   "outputs": [],
   "source": [
    "import torch\n",
    "import torch.nn as nn\n",
    "import torch.nn.functional as F\n",
    "import torch.optim as optim\n",
    "from torch.utils.data import DataLoader\n",
    "\n",
    "from torchvision import datasets\n",
    "from torchvision.transforms import ToTensor\n",
    "\n",
    "from sklearn.metrics import accuracy_score\n",
    "\n",
    "import matplotlib.pyplot as plt\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "authorized-syntax",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "cpu\n"
     ]
    }
   ],
   "source": [
    "# device for all computations\n",
    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
    "\n",
    "print(device)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "together-forward",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "CNN(\n",
      "  (conv0): Conv2d(1, 8, kernel_size=(7, 7), stride=(1, 1))\n",
      "  (conv1): Conv2d(8, 32, kernel_size=(5, 5), stride=(1, 1))\n",
      "  (fc0): Linear(in_features=288, out_features=128, bias=True)\n",
      "  (fc1): Linear(in_features=128, out_features=50, bias=True)\n",
      "  (fc2): Linear(in_features=50, out_features=10, bias=True)\n",
      ") \n",
      "\n",
      "Total Parameters: 50784\n",
      "Using Device: cpu\n",
      "\n",
      "Training...\n",
      "\n",
      "Epoch:  0\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.8120\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.5848 \tTest Accuracy: 0.7736\n",
      "\n",
      "Epoch:  1\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.5215\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.5039 \tTest Accuracy: 0.8151\n",
      "\n",
      "Epoch:  2\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.4468\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.4587 \tTest Accuracy: 0.8327\n",
      "\n",
      "Epoch:  3\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.3976\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.4106 \tTest Accuracy: 0.8526\n",
      "\n",
      "Epoch:  4\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.3652\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.4024 \tTest Accuracy: 0.8514\n",
      "\n",
      "Epoch:  5\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.3419\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3543 \tTest Accuracy: 0.8699\n",
      "\n",
      "Epoch:  6\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.3232\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3576 \tTest Accuracy: 0.8690\n",
      "\n",
      "Epoch:  7\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.3076\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3347 \tTest Accuracy: 0.8771\n",
      "\n",
      "Epoch:  8\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2929\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3334 \tTest Accuracy: 0.8781\n",
      "\n",
      "Epoch:  9\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2831\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3279 \tTest Accuracy: 0.8801\n",
      "\n",
      "Epoch: 10\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2764\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3159 \tTest Accuracy: 0.8848\n",
      "\n",
      "Epoch: 11\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2646\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3152 \tTest Accuracy: 0.8855\n",
      "\n",
      "Epoch: 12\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2578\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3100 \tTest Accuracy: 0.8881\n",
      "\n",
      "Epoch: 13\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2498\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3179 \tTest Accuracy: 0.8838\n",
      "\n",
      "Epoch: 14\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([96, 288])\n",
      "Train Loss: 0.2437\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([128, 288])\n",
      "torch.Size([16, 288])\n",
      "Test Loss:  0.3080 \tTest Accuracy: 0.8874\n"
     ]
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAXoAAAEWCAYAAABollyxAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjMuNCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8QVMy6AAAACXBIWXMAAAsTAAALEwEAmpwYAAA77UlEQVR4nO3dd3hc1bXw4d/SjKRRl6zmbrkbIxsbjA2GEBsnYMDgJEAIEOpNCFwSSEhCSe5NCCkfScilBIIJhJKE0DEl9GYDobiAce+WbbmoWr3MaLS+P86RPJIlW7bKSKP1Ps88c+bUNWN57X322WcfUVWMMcZErqhwB2CMMaZ7WaI3xpgIZ4neGGMinCV6Y4yJcJbojTEmwlmiN8aYCGeJ3hhjIpwlehPxRCRPRL4S7jiMCRdL9MYYE+Es0Zt+SURiReQuEdntvu4SkVh3WYaI/FtEykSkVEQ+EJEod9lNIrJLRCpFZIOIzHHnR4nIzSKyRURKRORpERngLvOJyD/d+WUislREssP37U1/Y4ne9Fc/B04ApgDHANOB/3GX/RjIBzKBbOBngIrIeOD7wPGqmgScDuS521wHfA34MjAY2Afc5y67DEgBhgHpwNVAbXd9MWNas0Rv+quLgdtUtVBVi4BfAZe4ywLAIGCEqgZU9QN1BoUKArHARBGJVtU8Vd3ibvM94Oeqmq+q9cCtwHki4nX3lw6MUdWgqi5X1Yoe+6am37NEb/qrwcD2kM/b3XkAfwQ2A2+KyFYRuRlAVTcDP8RJ4oUi8qSING0zAljoNs2UAetwCoZs4B/AG8CTbjPRH0Qkuju/nDGhLNGb/mo3TnJuMtydh6pWquqPVXUUcDZwQ1NbvKr+S1VPdrdV4Pfu9juBM1Q1NeTlU9Vd7lnBr1R1IjATmAdc2iPf0hgs0Zv+I9q9KOoTER/wBPA/IpIpIhnAL4B/AojIPBEZIyICVODUzIMiMl5ETnUv2tbhtLMH3f0vAH4rIiPcfWSKyHx3eraITBIRj7u/QMh2xnQ7S/Smv3gVJzE3vXzAMmAlsAr4DPiNu+5Y4G2gCvgY+IuqLsJpn78dKAb2Alk4F2oB7gZewmnuqQQ+AWa4ywYCz+Ik+XXAYtxCxZieIPbgEWOMiWxWozfGmAhnid4YYyKcJXpjjIlwluiNMSbCecMdQFsyMjI0Jycn3GEYY0yfsXz58mJVzWxrWa9M9Dk5OSxbtizcYRhjTJ8hItvbW2ZNN8YYE+EiKtE3Nir+hsZwh2GMMb1KpxK9iMx1x+Te3DTwU6vlKSLysoh8ISJrROSKzhzvYCrqApx21/s8+tG27jqEMcb0SUfcRu+O23Ef8FWcsbuXishLqro2ZLVrgbWqeraIZAIbRORxVfV3Kuo2JPuiyU6O5aEPtnHZzBxivZ6uPoQxESEQCJCfn09dXV24QzFHwOfzMXToUKKjOz4Aamcuxk4HNqvqVgAReRKYD4QmegWS3MGhEoFSoKETxzyo/541hosf+pTnlu/iohnDu+swxvRp+fn5JCUlkZOTg/Nf0/QVqkpJSQn5+fmMHDmyw9t1pulmCM7QrE3y3Xmh7gWOwhn+dRVwvaq22YguIleJyDIRWVZUVHREAc0cnc4xQ1NYsHgLDUFrqzemLXV1daSnp1uS74NEhPT09MM+G+tMom/rr6T1CGmnAytwHugwBbhXRJLb2pmq/lVVp6nqtMzMNruCHjogEa6ZNYYdpTW8smrPEe3DmP7AknzfdST/dp1J9Pk4z8BsMhT3wQ0hrgCeV8dmYBswoRPHPKTTJmYzJiuR+xdtwUbmNMaYziX6pcBYERkpIjHAt3DG4w61A5gD4D71fjywtRPHPKSoKOGaL49m/d5K3ttQ2J2HMsYcgZKSEqZMmcKUKVMYOHAgQ4YMaf7s9x+8n8ayZcu47rrrDut4OTk5FBcXdybkPu+IL8aqaoOIfB/nWZge4GFVXSMiV7vLFwC/Bh4VkVU4TT03qWq3/+LnTBnM/721kb+8t4VTJ2R39+GMMYchPT2dFStWAHDrrbeSmJjIT37yk+blDQ0NeL1tp6Zp06Yxbdq0nggzonSqH72qvqqq41R1tKr+1p23wE3yqOpuVT1NVSepaq6q9shTdaI9UVx1yiiWbd/Hkm2lPXFIY0wnXH755dxwww3Mnj2bm266iSVLljBz5kymTp3KzJkz2bBhAwCLFi1i3rx5gFNIXHnllcyaNYtRo0Zxzz33dPh427dvZ86cOUyePJk5c+awY8cOAJ555hlyc3M55phjOOWUUwBYs2YN06dPZ8qUKUyePJlNmzZ18bfvfr1yrJuu8M1pw7jnnU3c995mpo+cHu5wjOmVfvXyGtburujSfU4cnMwvzz76sLfbuHEjb7/9Nh6Ph4qKCt5//328Xi9vv/02P/vZz3juuecO2Gb9+vW89957VFZWMn78eK655poO9S///ve/z6WXXspll13Gww8/zHXXXccLL7zAbbfdxhtvvMGQIUMoKysDYMGCBVx//fVcfPHF+P1+gsG+97jfiBoCIVRcjIcrTx7J4o1FrN5VHu5wjDGHcP755+PxODc6lpeXc/7555Obm8uPfvQj1qxZ0+Y2Z511FrGxsWRkZJCVlUVBQUGHjvXxxx9z0UUXAXDJJZfw4YcfAnDSSSdx+eWX8+CDDzYn9BNPPJHf/e53/P73v2f79u3ExcV19qv2uIit0QN8+4QRLFi0hfsXb+G+i44NdzjG9DpHUvPuLgkJCc3T//u//8vs2bNZuHAheXl5zJo1q81tYmNjm6c9Hg8NDUd2P2ZTl8UFCxbw6aef8sorrzBlyhRWrFjBRRddxIwZM3jllVc4/fTTeeihhzj11FOP6DjhErE1eoCUuGi+feIIXlu1h23F1eEOxxjTQeXl5QwZ4tx/+eijj3b5/mfOnMmTTz4JwOOPP87JJ58MwJYtW5gxYwa33XYbGRkZ7Ny5k61btzJq1Ciuu+46zjnnHFauXNnl8XS3iE70AFeeNJJoTxQPLN4S7lCMMR104403csstt3DSSSd1SZv45MmTGTp0KEOHDuWGG27gnnvu4ZFHHmHy5Mn84x//4O677wbgpz/9KZMmTSI3N5dTTjmFY445hqeeeorc3FymTJnC+vXrufTSSzsdT0+T3nhT0bRp07QrHzzyvy+s5smlO/jgxlMZmOLrsv0a0xetW7eOo446KtxhmE5o699QRJarapt9TyO+Rg9w1SmjaFR48INuvVfLGGN6pX6R6IcNiOecYwbzxJId7Kvu8hGSjTGmV+sXiR7gmlmjqfEHefSjvHCHYowxParfJPpx2Ul8dWI2j36UR3V9tw2Jb4wxvU6/SfQA/z1rNOW1AZ5YsiPcoRhjTI/pV4l+6vA0ThyVzoMfbKW+oe/dxmyMMUeiXyV6gP+ePZqCinqe/2xXuEMxpt/yeDzNQxNPmTKF22+//Yj2M2vWLI60K/aiRYv46KOPmj8vWLCAv//970e0r1B5eXnk5uZ2ej9dKaKHQGjLyWMymDQkhQcWb+Gb04bhibIn7RjT0+Li4pqHKg6XRYsWkZiYyMyZMwG4+uqrwxpPd+p3NXoR4drZo8krqeFVe9ygMb3Ga6+9xje/+c3mz4sWLeLss88G4JprrmHatGkcffTR/PKXv2xz+8TExObpZ599lssvvxyAl19+mRkzZjB16lS+8pWvUFBQQF5eHgsWLODOO+9kypQpfPDBB9x6663ccccdAKxYsYITTjiByZMn8/Wvf519+/YBzhnETTfdxPTp0xk3bhwffPBBh7/fO++8w9SpU5k0aRJXXnkl9fX1ANx8881MnDiRyZMnN4/L39ZwyZ3R72r0AKdNHMjozAT+smgL8yYPsudnmv7rtZth76qu3efASXDGwZtiamtrmTJlSvPnW265hXPPPZfvfe97VFdXk5CQwFNPPcUFF1wAwG9/+1sGDBhAMBhkzpw5rFy5ksmTJ3conJNPPplPPvkEEeGhhx7iD3/4A3/605+4+uqrWzz05J133mne5tJLL+XPf/4zX/7yl/nFL37Br371K+666y7AeTDKkiVLePXVV/nVr37F22+/fcgY6urquPzyy3nnnXcYN24cl156Kffffz+XXnopCxcuZP369YhI89DIbQ2X3Bn9rkYPzuMGr/7yaNbtqWDRxqJwh2NMv9PUdNP0uuCCC/B6vcydO5eXX36ZhoYGXnnlFebPnw/A008/zbHHHsvUqVNZs2YNa9eu7fCx8vPzOf3005k0aRJ//OMf2x3yuEl5eTllZWV8+ctfBuCyyy7j/fffb17+jW98A4DjjjuOvLy8DsWwYcMGRo4cybhx41rsMzk5GZ/Px3e+8x2ef/554uPjgbaHS+6MflmjB5g/ZQh3vrWR+9/bwuzxWeEOx5jwOETNu6ddcMEF3HfffQwYMIDjjz+epKQktm3bxh133MHSpUtJS0vj8ssvp66u7oBtQ8/MQ5f/4Ac/4IYbbuCcc85h0aJF3HrrrZ2KsWlo5MMZFrm9McW8Xi9LlizhnXfe4cknn+Tee+/l3XffbXO45PT09COOuV/W6AFivFF895RRLMkrZWmePW7QmN5g1qxZfPbZZzz44IPNzTYVFRUkJCSQkpJCQUEBr732WpvbZmdns27dOhobG1m4cGHz/NAhjx977LHm+UlJSVRWVh6wn5SUFNLS0prb3//xj3801+6P1IQJE8jLy2Pz5s0t9llVVUV5eTlnnnkmd911V/MF6raGS+6MflujB/jW8cP587ub+ct7m3nkCnvcoDE9pXUb/dy5c7n99tvxeDzMmzePRx99tDkpH3PMMUydOpWjjz6aUaNGcdJJJ7W5z9tvv5158+YxbNgwcnNzqaqqApxny55//vkMGTKEE044gW3btgFw9tlnc9555/Hiiy/y5z//ucW+HnvsMa6++mpqamoYNWoUjzzyyGF9vw0bNjB06NDmz3feeSePPPII559/Pg0NDRx//PFcffXVlJaWMn/+fOrq6lBV7rzzTsAZLnnTpk2oKnPmzOGYY445rOO31i+GKT6Ye9/dxB1vbuTV677ExMHJPXJMY8LJhinu+2yY4sN0yYk5JMZ6ud8eTGKMiVCdSvQiMldENojIZhG5uZ11ZonIChFZIyKLO3O87pASF83FJwznlZW7ybPHDRpjItARJ3oR8QD3AWcAE4ELRWRiq3VSgb8A56jq0cD5Rx5q9/mvk0fi9UTxwPv2YBLTP/TGJlvTMUfyb9eZGv10YLOqblVVP/AkML/VOhcBz6vqDjfAwk4cr9tkJfk4/7ihPLc8n4KKA7ttGRNJfD4fJSUlluz7IFWlpKQEn+/wHonamV43Q4DQPj/5wIxW64wDokVkEZAE3K2qbY4aJCJXAVcBDB8+vBNhHZnvnTKaJ5fu5KEPtvLzsyYeegNj+qihQ4eSn59PUZHdLNgX+Xy+Fj16OqIzib6tcQNaVxG8wHHAHCAO+FhEPlHVjQdsqPpX4K/g9Lo5ooiqiyEh44g2HZ4ez9mTB/H4pzu4dvYYUuNjjmg/xvR20dHRjBw5MtxhmB7UmaabfGBYyOehwO421nldVatVtRh4H+hch9D2+KvhgVPgiQuhdNsR7eKaWWOo8Qd57KPtXRycMcaET2cS/VJgrIiMFJEY4FvAS63WeRH4koh4RSQep2lnXSeO2b6oaJj+Xdi6GO6bAe/+xkn+h2H8wCS+clQWj360jRq/PW7QGBMZjjjRq2oD8H3gDZzk/bSqrhGRq0XkaneddcDrwEpgCfCQqq7ufNht8MbAyT+CHyyHifPh/T/CvdNh9fNwGBedrpk1hn01AZ5Y0rlbjo0xpreI3Dtjt38Mr/3UGYI150twxu8h++gObXrBAx+zvaSGxTfOItbr6VwcxhjTA/rnnbEjToSrFsNZf4KC1bDgS/DqjVC775CbXjt7DHsr6njhc3vcoDGm74vcRA8Q5YHjvwM/+AyOuxyWPgh/Pg6WPwaN7Y/x/KWxGeQOSWbB4q0EG3vfGY8xxhyOyE70TeIHwLz/c2r4GePg5evgwVNh59I2VxcR/nvWGLYVV/P66r09HKwxxnSt/pHomwyaDFe8Bt94CKoK4G9fgYXXQGXBAauefvRARmUk8JdFm+0OQmNMn9a/Ej2ACEw+H76/zOmls+oZpznno3shGGhezeM+bnDN7gre31QcxoCNMaZz+l+ibxKbCF+5Fa791Llw++bP4f6ZsOXd5lW+NnUIg1J83Pfe5vDFaYwxndR/E32T9NFw8TNw4VNOjf4fX4cnL4Z9ec7jBr80iiXbSvnNv9dSXhM49P6MMaaXsUTfZPxcp3Y/5xdOrf6+GfDe77jo2Ay+OW0of/vPNr70h3d5YPEW6gKdfyq7Mcb0lMi9YaozynfBW7+A1c9CyjA47TesHzCL37++kfc2FDE4xccNp43n61OH4Ilqa2w3Y4zpWQe7YcoS/cHk/Qdeu9G54SpxIBx1NqtTZ/E/nyWxYlcVEwYmcdMZE5g1LhMRS/jGmPCxRN8ZwQZY9xKsWQib3oKGWjQhk+1Zp3LX7om8XD6K6aOyuOXMCUwemhruaI0x/ZQl+q7ir3aS/doXYeMbEKimLjqVVxuO44X6aaTlfpUbTp/IiPSEcEdqjOlnLNF3B38NbHkH1r6IbngN8VdRpgm8o9OoG3s2c8/5FukpSeGO0hjTT1ii726BOtj6HrUrnoMNrxHXWEWFxrM7exYjT7mI2PFfhejDe8ajMcYcjoMl+s48StA0ifbB+DOIG38GNNSz+/PX2Lr4CXILFhP77KsEPPF4JpxBVO7XYcxXIDou3BEbY/oRq9F3o+VbC3jlxacYW/IuZ0YvI0Ur0egEZNxpzsNRxp4GMdaeb4zpPGu6CSNV5Y01Bdzx+hqySpdxecoKZuunRNeVgDcOxp8Bk853avpeeyC5MebIWKLvBRqCjTy1bCd3vb2Jksparh1VwHfSVpCy9RWoLQVfilPLzz0Pck52xtI3xpgOskTfi1TXN/C3D7fxwOItVPuDnDAiiauG7uCk2kXEbnoVAtXOzVm534BJ58HgY50RN40x5iAs0fdCxVX1PLV0Jws/38XmwiqiPcJpY5P5r6yNHFP2Np4tb0HQDwNGObX8SedB5vhwh22M6aUs0fdiqsqa3RW88PkuXvpiN4WV9STFevn6xAQuTVnF6ILXkLwPQBth4CQn6eeeC6nDwh26MaYXsUTfRwQblY+3lLDw8128vnoP1f4gA5N9XDgxhgvilpG949/ILvd3GX6iU8uf+DVIyAhr3MaY8LNE3wfV+oO8va6AFz7fxeKNRTQ0KuOzk7hkQpCzPZ+QsukFKFoP4oHRpzpJf8JZEGt34xrTH3VboheRucDdgAd4SFVvb2e944FPgAtU9dlD7dcSfUul1X5eWbmbF1bsZvn2fQDMyEnj8jHVzA58gG/9QijfAV4fjJvr9NrxpYIv2enNE5u8fzom0S7uGhOBuiXRi4gH2Ah8FcgHlgIXquraNtZ7C6gDHrZE3znbS6p5ccVuXvh8F1uLq4nxRDFnfAaXDS/k+Kp38ax9AWoO8oxbidqf+GNTnOTvS3bnpbRdODStN2Ckdfs0ppfqrkR/InCrqp7ufr4FQFX/X6v1fggEgOOBf1ui7xqqyqpd5bzw+W5e+mI3xVX1JPu8zJuUxbzRMUzOFBK1BurKnVd9BdRVtDNd7nyud99p528iIcvp9pl7HgydZmcGxvQi3TXWzRBgZ8jnfGBGqwMPAb4OnIqT6A8W5FXAVQDDhw/vRFj9g4gweWgqk4em8rMzJ/CfLSW8+PkuXvhiL/9aGiRK4KhByRyfM5BpOUdx/NgBZCd3YGC1xkbwVx1YINQUw8bXYdkj8OkCSB3hXBeYdD5kHdX9X9gYc8Q6U6M/HzhdVb/jfr4EmK6qPwhZ5xngT6r6iYg8itXou12tP8hnO/axZFspy7aX8tn2MmrdZ9wOGxDH8TkD3FcaozMTD//JWHXlsO7fzmMWty5yun1mHe0k/dxzIW1E138pY8whha3pRkS2AU2ZJAOoAa5S1RcOtm9L9F0nEGxk3Z4KJ/Hn7WPZ9lKKq/wApMVHM81N+tNyBpA7OIUY72E8L76qENa8AKuegfwlzrxhM5ymnaO/BolZXf59jDFt665E78W5GDsH2IVzMfYiVV3TzvqPYjX6sFNVthVXsyxvH0vzSlm2fR/biqsB8EVHMWVYKsfnDGBazgCOHZ5Kki+6Yzvelwern4NVz0HhGuei76hZTtI/ap5zMdcY0226s3vlmcBdON0rH1bV34rI1QCquqDVuo9iib5XKqysY3nePpa6yX/N7nIalZB2/gFMy0kjd3AKwwbE44k6RHNPwVqnaWfVM1C2AzyxMO40pz1/7Gk2Hr8x3cBumDKHpaq+gRU7yliaV8rSvFI+37G/nd8XHcWYrETGZSUxbmAS47ITGZuVxJDUOKJaFwCqkL/MSfqrn4fqQohJgqPOhknnwshZ4OnCZ9+oQmPQGSMoygPe2K7btzG9nCV60ymBYCNrd1ewfm8FGwuq2FhQycaCSgoq6pvXSYjxMCY7iXFZiYwfmMTYbKcQGJjscy74Bhsg7wNY9Syse9npypmQ6YzHH5PoJOegHxr8+6eDgVbv9SHTIfNDt2nqGipRkJYDmUc5g8FlHQWZEyBjrJ1RmIhkid50i/KaAJsKK9lQUMmmkAKg6WIvQJLPyzg36TvvSYxN95K5531k9XOw5T1AwRPjvqIPMR3t1NQPta6/Boo3QOF6KN0CjQ1OQKEFQNYEJ/lnToCMcfZcX9OnWaI3Paq02t+c9DcWVLJxbxUbCyspqwk0r5MWH91c689JT2D4gHhGuO9xMV18922DH0o2O2MDFa2HwnVQtMGZp06TlFMAjHSSftaE/WcCVgCYPsISvQk7VaWoqt5J+gWVzpnA3ko2FVZRWdfQYt3MpFhGDIhneHo8IwYkMCI9nmED4hmRHk96Qszh9/1vT3MB4Cb+wnVOQVCy5cACIMtN/InZzrzQV5TnwHntLvM4dxS3Xu6NhegEp1kpOn7/e9RhdHc1/ZoletNrqSplNQG2l9awo7SGHSXVbC+pcT6X1LC3oq7F+gkxHoanJzDCTfxNBcCIAQkMTvXh9XRBYmyod5J90Tqn6afpTCC0AOgpXp+b9JsKAbcAiIlvWSCETse40zGJEJ/uXAtJyHSGs/Z0sLus6XO6awgEYzpNREhLiCEtIYYpw1IPWF4XCJK/r8ZJ/iVuYVBaw6bCSt7dUIi/obF5XW+UMCQtjuEDnAJgcIqPQSlxDEr1MTgljoEpPnzRHWgW8sZC9kTnFaqhHuqrnLuBtdFJ+s3TjU6PH9WDLG9s+blpeWPQ2XegBgK17rs77a9254XOr4H6SueGtRbLq/dfi2hPXJozZlFCJiRmhhQC7isxyykQEjJtpNMIYone9Gq+aA9jspIYk3XgOPuNjcreijr3TKCG7aXVzYXBmtV7Ka32H7BNRmKMk/xTfAxOdd4HpcY5hUJqHNlJse2fFXhje3+XzWBgfyFRX+WMUVRVCNVFLV9VRbB3tdPlta687X1549ouEHwpbtOTALK/KappunnewZa3sb0n2jlmdFzLs5fms5p45/fvysJH1fm96iqcsZ3qK91xnir3j/XUYtp9Beqc0V3jBjiFZ7z73mLafY9NCnuBaU03JmLV+oPsKa9lT3kdu8uc9z3ltewuc973lNVRWd+yBhwlkJXkaz4LGJzqnBU0vQ9K8ZGeGHvom8b6kgZ/SCFQ7CT/6iK3gAj9XOQUHIc6a+hW4iR9ry+kuSpkunVB4fW5hV5TEg9J1k1JvCPNcTFJ7nDeSc4Q3t5YZx81+6B2H/gr2982ytsy8R+sUEjIgOyjj+yXsTZ6Y9pWWRdoWRCU1bK7fH9BsLu8lrpAY4ttvFFCVlIs2Sk+Bib7GNjGe3ZyB5uJ+prGRqcGjDq14aZ3dX+jA+aFrtfY/nJVaGw6G6nb31TVUBfSpOW+Gmpbfj5g3VbbR8ftT9C+lJDp5FbTyW3Pj0k89HMYGvxQVwY1pU7ir3Xf2/3sTjfUttxPfAbcuOWI/mmsjd6YdiT5oknyRTMuu+1HMDZdLN7lFgR7K+rYW17L3vJ6Cirq2FhQyQebiqmqP7CWmxYfTXYbBUF2io9B7nRKXHTX9SLqCVFREJsY7ih6H2+Mc33jcAfyC9S2TPzB+kNvcyThdctejYkQoReLc4e0PzBbZV2Agoo69pbX7y8M3M8FFXWs3lVBSXU9rU+gY71RDEzxMSAhhrT4GFLjokmJj3am46NJiYsmNT6GtPhoUuNiSImPJtnn7VuFg2lfUxNT8uBuPYwlemO6QNOZQVsXjZsEgo0UVtY3nxHsLxDqKavxU1jpnCGU1wQOuHYQyhMlTgEQWii0U0AkxnqIj/GSGOslPsZDQqyXWG+UFRT9jCV6Y3pItCeKIalxDEk99Fg7gWAj5bUBymoClNf62VcdoKw2QFmNn/LaAPtq/O6yQIcLiCbeKCE+xuMk/1gvCbFeEtxCoOndKRi8JMS680OWJfm8ZCTGMiAhhuiuuG/BdDtL9Mb0QtGeKDISY8lIPLzunK0LiOr6INX1DVTVN1DjD7rvDc3zq/0NVNUHqalvoLS6hhr//vmtL0K3JSUumvTEGDISnMSfnhhDemIsGYkxpCfEOsvc6ZS46ANHODU9whK9MRHkSAuItjQEG6kJuIm/fn/hUFEXoKTaT0mVn5KqeoqrnfctRVUsyfOzr8Z/wLUIcJqc0uLdxN+iIIglPSGGlLho4mO9LZqbEtwmJ2tu6hxL9MaYNnk9USR7okju6FPGXMFGZV/NgQVBSZWfkup6it35K/eVUVLl73BzU2jzUVOBkBDjNiu1mA5pior1uNdPvM3viTHefndmYYneGNOlPFESclbR/sXpJnWBICXVfipqA9SENCVVNZ1JuM1JTU1PofNKqmqo9jdQU+8sq284dHOTCCTGeFsk/5bTzntyG/OaphNjvX3qpjlL9MaYsPJFezp8kfpQAsHG5usMNf4GKuucAqOyroHKugCVdQ1UhEw3vRdX+dlWXO3Oa8AfPHSB4YuOIj7GaVqKj/EQF+MlPtpDQuz+6fhYj7v8wPXiY1vOb5rujhvtLNEbYyJGtCeKlLgoUuI6N0pnXSDYoiBoWVA477WBIDV+50yj1h+k2h+k1t/AnvIANf79y2r8QYKNHRuBID0hhuX/+9VOxd4WS/TGGNOKL9qpWWcmdf6itqriDzZS6yb90ALAKSD2Fxbdde3AEr0xxnQjESHW6yHW6yE1Pjwx2N0OxhgT4SzRG2NMhOuVwxSLSBGw/Qg3zwCKuzCc7tSXYoW+FW9fihX6Vrx9KVboW/F2JtYRqprZ1oJemeg7Q0SWtTcmc2/Tl2KFvhVvX4oV+la8fSlW6Fvxdles1nRjjDERzhK9McZEuEhM9H8NdwCHoS/FCn0r3r4UK/StePtSrNC34u2WWCOujd5EBhFZBPxTVR/qhn3/DBilqt/p6n0b0xtFYo3e9CARyRORWhGpCnndG+64mojILBHJD52nqr/rziTvHlNF5MbuOoYxh8MSvekKZ6tqYsjr++EOKMwuA0rd9x4jDvs/bQ5gfxSmW4hIrIiUiUhuyLxMt/afJSJpIvJvESkSkX3u9NB29nWriPwz5HOOW2P2up+vEJF1IlIpIltF5Hvu/ATgNWBwyNnG4Db2d46IrHHjXSQiR4UsyxORn4jIShEpF5GnRMR3kO8dD5wHXAuMFZFprZZ/NyTWtSJyrDt/mIg87/4eJU1nRR347otE5Lci8h+gBhjV3u8Rso/5IrJCRCpEZIuIzBWR80Vkeav1fiwiL7T3XU3fYYnedAtVrQeeBy4Mmf1NYLGqFuL87T0CjACGA7XAkTb5FALzgGTgCuBOETlWVauBM4DdIWcbu0M3FJFxwBPAD4FM4FXgZRGJaRX3XGAkMBm4/CCxnAtUAc8AbwCXhhzrfOBWd14ycA5QIiIe4N84NwnmAEOAJw/j+18CXIUz+Pv29n4PN4bpwN+BnwKpwClAHvASMDK0kAO+DfzjMOIwvZQletMVXnBrw02v77rz/0XLRH+ROw9VLVHV51S1RlUrgd8CXz6Sg6vqK6q6RR2LgTeBL3Vw8wuAV1T1LVUNAHcAccDMkHXuUdXdqloKvAxMOcj+LgOeUtUg7vcXkaYxc78D/EFVl7qxblbV7cB0YDDwU1WtVtU6Vf2wg/EDPKqqa1S1QVUDh/g9/gt42P2+jaq6S1XXuwXzUzjJHRE5GqfQ+fdhxGF6KUv0pit8TVVTQ14PuvPfBeJEZIaIjMBJkAvBaeIQkQdEZLuIVADvA6lu7fawiMgZIvKJiJSKSBlwJs6t5B0xmJDhNlS1EdiJU6tusjdkugZIbCeOYcBs4HF31ouADzjL/TwM2NLGpsOA7ap66GfqtW1nqzgO9nu0FwPAY8BFIiI4ZwlPuwWA6eMs0Ztu4ybNp3Fq9RcB/3Zr7wA/BsYDM1Q1GacJAaCtAbmrgdABXgc2TYhILPAcTk08W1VTcZpfmvZzqP7Du3Gaj5r2JzjJcNchtmvLJTj/p14Wkb3AVpxE39R8sxMY3cZ2O4HhTe3urbT73UM0f8cO/B7txYCqfgL4cWr/F2HNNhHDEr3pbv/CaR652J1ukoTTLl8mIgOAXx5kHyuAU0RkuIikALeELIsBYoEioEFEzgBOC1leAKS727XlaeAsEZnjNrH8GKgHPurg9wt1KfArnDOXpte57v7TgYeAn4jIceIY457pLAH2ALeLSIKI+ETkpA5897Yc6vf4G3CF+32jRGSIiEwIWf53nGslDYfZfGR6MUv0piu8LC370S9sWqCqn+LUSgfj9IBpchdOW3gx8Anwens7V9W3cNqPVwLLCWk3ds8QrsNJ2PtwaqIvhSxfj3Oxdat7/WBwq31vwGmX/rMby9k43UX9h/MDiMgJOG3a96nq3pDXS8Bm4EJVfQbnWsS/gErgBWCA255/NjAG2AHk4xSOB/3u7fxWh/o9luBeoAXKgcWEnNHg1OJzsdp8RLE7Y40xzUQkDqfXzrGquinc8ZiuYTV6Y0yoa4ClluQjiz0z1hgDODeH4Vy0/Vp4IzFdzZpujDEmwlnTjTHGRLhe2XSTkZGhOTk54Q7DGGP6jOXLlxe398zYXpnoc3JyWLZsWbjDMMaYPkNEtre3zJpujDEmwvXKGr0xxhyphmAjpdV+iqrqKa32k5EYy8iMBHzRhz2MUsSwRG+M6fVUlbKaAEVV9RRVOq/ikOmm+cVV9ZRU+2ndmVAEhqTGMSozkVEZCYzOSmS0+56VFIszxFHPqQsEyd9Xw87SWnbuq2FnqTOtKA9cMu3QOzhMluiNMd1CVQk2Kv5gI4EG592Zdt79DY0Emt+V8toARZV1FFXVU1zpPyCpNzQe2BU8xhtFZmIsmUmxDBsQz7Ej0shwP2cmxpIWH01hZT1bi6rZUlTF1uIqluWVUuMPNu8jIcbDqMxERmcmOAVBZgKjMxM7dRYQCDayp6xufxIPSer5+2opqmw5KGisN4qhaXGMzmxzYNROs0RvjGlXY6Oyq6yWjQWVbCyoYlNhJXvK6pwEHZqsQ5J5cyIPNh5Qs+4IT5SQkRjTnKwnDExypt1XcyJPiiUp1nvYtXFVZW9FHVuLqtlaVMUWtxBYmrePF1bsfy6NCAxOiWN0lnsW4BYAozKds4CiqvqWSTxkem9FHcGQgskTJQxO9TEsLZ7Z4zMZlhbPsAHxDBsQx7C0eDISY4mK6r6zCkv0xhhUnYS+qaCqRVLfXFjVovabnRzLsLR4YqOjSPJ5ifZEEe2NItYTRbQnihhvlDtP2pjnrucVYjweoj0SMi+KZF80GYkxpMXHdGvSExEGpcQxKCWOk8a0fGxBrT/ItuJqthZXsaXQed9aVM0zeaVUh/wOIhxQiGW5ZxXH56Q5STwtnqFuIh+U4sPrCV/fF0v0xvQjqsqe8jo2FlTuT+qFVWwuqGyRyDKTYhmXncg3pw1jXHYS47ITGZuVREp89EH23vfFxXiYODiZiYOTW8xXVQoq6p0zgOJqCsrryE7xMSwtjqFp8QxNi+vVF3st0RsTgQLBRoqr6puT+aaCKjYWVrK5oIrK+v0PsspIjGFsVhLnHTeUsdlJzUk9NT7mIHvvf0SEgSk+Bqb4mDmmow8v6z0s0RvTy9X6g5TW+NlX7aesJsC+Gr/zqnamy2r87AuZX1YdaJHMAQYkxDA2K5GvTR3i1M7dpD4gwRJ6f2CJ3pgu1hBspDYQpC7QSF0gSH3D/unQ+XWBIHUNjdT5g5TXNiXtAKXV/ubpfTV+6hsa2z1WUqyX1IRo0uJjSI2PYWRGAmnxTjv3gMQYRmcmMC47iYzE2B78BUxvY4nemEOoCwT5dFsp760vJH9fzf5E3RCk1u8k7tBk3lY3wEOJEkiNjyEt3knaQ9PimTQkmrSEGDdxR+9fnhBDanw0qXExxHjt5nZzaJbojWnDnvJa3ltfxLvrC/nP5mJqA0FivVGMykwkLjoKX7SH5Lho4qI9xLqffV4PPnc6LtqZjo32uMuiiIvxHLCez10vIcbbrT1NTP9mid4YINiorNi5j3fXF/Lu+iLW7akAnLspzztuKKdOyOLE0em9umeFMe2xRG/6rbIaP4s3FvHe+kIWbyxiX00AT5Rw3Ig0bj5jAqdOyGJsVmKP3x5vTFezRG/6DVVlQ0El764v5L31hSzfvo9GdXqkzB6fxewJWZwyNjPi+4qb/scSvQkrVe3WGnOtP8hHW4qbk/vu8joAjh6czLWzxzB7QhbHDE3FY+3jJoJZojdhEQg2ct97m7l/0RYaVfF5nYuasV4PsV7nImasN6p52td6nntBM7Z5O+fiZtOy4qp63ttQyMdbSqhvaCQ+xsPJYzK4bs5YZk/IIjvZF+6fwJgeY4ne9Li1uyv4yTNfsHZPBXOPHsjIzATqA43UNQSpd7sq1jc09UFvpLw2QGEgiD9knvMKEgi235UxJz2ei2YM59QJWUwfOYBYr11INf2TJXrTYwLBRv7y3hb+/O4mUuNjeOCS4zj96IGd2meDO0qiU0DsLwjiYzwMGxDfRZEb07dZojc9Yt0epxa/ZncF5xwzmF+dczRpXXD7vdcThdcThQ3NYkz7LNGbbhUINnL/IqcWnxIXzYJvH8fc3M7V4o0xh8cSvek26/c6tfjVuyo4263F2yBaxvQ8S/SmywWCjSxYtIV7mmvxxzI3d1C4wzKm37JEb7pUaC1+3uRB3DY/12rxxoSZJXrTJRqCjSxYvIW739lEsi+a+y8+ljMmWS3emN7AEr3ptA17K/nJM1+wale51eKN6YUs0Zsj1hBs5IH3t3L325tI8nn5y8XHcqbV4o3pdSzRmyOyscCpxa/ML+esSYO4bf7RpNtTjIzplSzRm8MSWotP9Hm576JjOWuy1eKN6c06lOhFZC5wN+ABHlLV21stTwH+CQx393mHqj7iLssDKoEg0KCq07osenOA1bvKWZlfTnyMh7gYD/HuKy7au386xkN8jPewR2zc5Nbiv7BavDF9yiETvYh4gPuArwL5wFIReUlV14asdi2wVlXPFpFMYIOIPK6qfnf5bFUt7urgzX4NwUbueXcz9767iY4+sjTGG+Uk/+j9yb+9wqG+oZF/fbqDRJ+Xey+ayrzJg7v3CxljukxHavTTgc2quhVARJ4E5gOhiV6BJHEGFk8ESoGGLo7VtGNnaQ3XP/k5n+0o4xvHDuGHc8YRaGyk1h+kxh+kxt9AXaBpOrh/fqChebrWXa/GH6S02k/+vpbz/MFGzsgdyG3zc8mwWrwxfUpHEv0QYGfI53xgRqt17gVeAnYDScAFqtroLlPgTRFR4AFV/WtbBxGRq4CrAIYPH97hL9DfvbhiF/+zcDUAd39rCvOnDOmW43T3A0KMMd2nI4m+rf/drRsHTgdWAKcCo4G3ROQDVa0ATlLV3SKS5c5fr6rvH7BDpwD4K8C0adM62PjQf1XWBfjli2t4/vNdHDcijbsumNKtw/Jakjem7+pIos8HhoV8HopTcw91BXC7qiqwWUS2AROAJaq6G0BVC0VkIU5T0AGJ3nTc5zv2cf2TK8jfV8P1c8byg1PH4PVEhTssY0wv1ZHssBQYKyIjRSQG+BZOM02oHcAcABHJBsYDW0UkQUSS3PkJwGnA6q4Kvr8JNir3vruJ8xZ8TLBReep7J/Kjr46zJG+MOahD1uhVtUFEvg+8gdO98mFVXSMiV7vLFwC/Bh4VkVU4TT03qWqxiIwCFrqn/V7gX6r6ejd9l4i2u6yWHz61giXbSpk3eRC//fokUuKiwx2WMaYPEKe1pXeZNm2aLlu2LNxh9BqvrtrDLc+vIhBs5Lb5uZx77BBrMzfGtCAiy9u7T8nujO3FavwN3PbyWp5cupPJQ1O451tTyclICHdYxpg+xhJ9L7V6VznXPfE520qquWbWaH70lXHEeK0t3hhz+CzR9zKNjcpDH27lj29sID0hlse/M4OZozPCHZYxpg+zRN+LFFbUccPTX/Dh5mJOPzqb278xmTQb190Y00mW6HuJt9cWcONzK6nxN/C7r0/iwunD7IKrMaZLWKIPs7pAkN++so5/fLKdiYOSuefCqYzJSgx3WMaYCGKJPozW7ang+ic/Z2NBFd85eSQ/nTueWK8n3GEZYyKMJfowaAg28rcPt/GntzaS7Ivm71dO55RxmeEOyxgToSzR97BV+eXc/PxK1uyu4KsTs/l/35hkw/4aY7qVJfoeUuNv4P/e3MjD/9lGemIs9198LHNzB9oFV2NMt7NE3wMWbyzi5wtXkb+vlgunD+fmMybYODXGmB5jib4blVTV8+t/r+WFFbsZlZnAU1edwIxR6eEOyxjTz1ii7waqynOf7eI3r6ylur6B6+aM5drZo61HjTEmLCzRd7HtJdX8fOFqPtxczHEj0vh/35jEuOykcIdljOnHLNF3kYZgIw99uI273t6INyqKX38tl4unDycqyi62GmPCyxJ9F1iZX8bNz61i7Z4KTpuYzW3zcxmY4gt3WMYYA1ii75QafwN/enMjj/xnGxmJsSz49rHMzR0U7rCMMaYFS/RHaNGGQn6+cDW7ymq5eMZwbjpjAsk+6zJpjOl9LNEfpmK3y+SLK3YzOjOBZ64+keNzBoQ7LGOMaZcl+g5q3WXyh18ZyzWzrMukMab3s0TfAdtLqvnZwlX8Z3MJ00akcfu5kxiTZV0mjTF9gyX6Q1iZX8Y3H/iY6KgofvO1XC6yLpPGmD7GEv1B1AWC/PjpL0iNi+GFa0+yLpPGmD7JEv1B3P3OJjYVVvHIFcdbkjfG9FlR4Q6gt1qxs4wHFm/hgmnDmD0+K9zhGGPMEbNE34a6QJCfPPMF2ck+fj7vqHCHY4wxnWJNN2248+2NbC6s4rErp9tNUMaYPs9q9K18tmMfD76/lW8dP4wv23NcjTERwBJ9iLpAkJ8+8wUDk338/CxrsjHGRAZruglx51sb2VJUzd+vnE6SNdkYYyJEh2r0IjJXRDaIyGYRubmN5Ski8rKIfCEia0Tkio5u21t8tmMfD36wlQunD+MUa7IxxkSQQyZ6EfEA9wFnABOBC0VkYqvVrgXWquoxwCzgTyIS08Ftw66pl82glDh+dqY12RhjIktHavTTgc2qulVV/cCTwPxW6yiQJCICJAKlQEMHtw27/3trI1uLqrn93EnWZGOMiTgdSfRDgJ0hn/PdeaHuBY4CdgOrgOtVtbGD24bV8u1NTTbD+dJYa7IxxkSejiT6tkbw0lafTwdWAIOBKcC9IpLcwW2dg4hcJSLLRGRZUVFRB8LqvKZeNoNT4vjZmRN65JjGGNPTOpLo84FhIZ+H4tTcQ10BPK+OzcA2YEIHtwVAVf+qqtNUdVpmZs/UrP/05ga2Flfz+3MnW5ONMSZidSTRLwXGishIEYkBvgW81GqdHcAcABHJBsYDWzu4bVgs317KQx9u46IZwzl5bEa4wzHGmG5zyH70qtogIt8H3gA8wMOqukZErnaXLwB+DTwqIqtwmmtuUtVigLa27Z6v0nFOk81Kt8nGetkYYyJbh26YUtVXgVdbzVsQMr0bOK2j24bbHW84TTaPf2cGibF2z5gxJrL1uyEQluWV8rf/bOPiGcM5aYw12RhjIl+/SvS1/iA/fdZpsrnFmmyMMf1Ev2q3uOPNDWwrruZf1mRjjOlH+k2NfmleKQ//ZxvfPmE4M63JxhjTj/SLRF/rd26MGpIaxy1nWJONMaZ/6RftF398YwN5JTX867szSLAmG2NMPxPxNfol20p55KNtXHriCGaOtiYbY0z/E9GJvtYf5MZnv2BoWhw3zbWxbIwx/VNEt2P84Y315JXU8MR3T7AmG2NMvxWxNfol20p59KM8LjtxBCeOTg93OMYYEzYRmehr/A389NkvGJYWz01nWJONMaZ/i8j2jD+8voHtJTU8edUJxMdE5Fc0xpgOi7ga/adbS3j0ozwun5nDCaOsycYYYyIq0TtNNisZkR7PjXPHhzscY4zpFSKqXeMPr29gR2kNT1mTjTHGNIuYGn15TYCXvtjN5TNzmGFNNsYY0yxiqr0p8dG8/sMv2aiUxhjTSkRlxawkX7hDMMaYXidimm6MMca0zRK9McZEOFHVcMdwABEpArYf4eYZQHEXhtOd+lKs0Lfi7UuxQt+Kty/FCn0r3s7EOkJVM9ta0CsTfWeIyDJVnRbuODqiL8UKfSvevhQr9K14+1Ks0Lfi7a5YrenGGGMinCV6Y4yJcJGY6P8a7gAOQ1+KFfpWvH0pVuhb8falWKFvxdstsUZcG70xxpiWIrFGb4wxJoQlemOMiXARk+hFZK6IbBCRzSJyc7jjORgRGSYi74nIOhFZIyLXhzumQxERj4h8LiL/DncshyIiqSLyrIisd3/jE8MdU3tE5Efu38BqEXlCRHrVOB4i8rCIFIrI6pB5A0TkLRHZ5L6nhTPGJu3E+kf372CliCwUkdQwhthCW/GGLPuJiKiIZHTFsSIi0YuIB7gPOAOYCFwoIhPDG9VBNQA/VtWjgBOAa3t5vADXA+vCHUQH3Q28rqoTgGPopXGLyBDgOmCaquYCHuBb4Y3qAI8Cc1vNuxl4R1XHAu+4n3uDRzkw1reAXFWdDGwEbunpoA7iUQ6MFxEZBnwV2NFVB4qIRA9MBzar6lZV9QNPAvPDHFO7VHWPqn7mTlfiJKIh4Y2qfSIyFDgLeCjcsRyKiCQDpwB/A1BVv6qWhTWog/MCcSLiBeKB3WGOpwVVfR8obTV7PvCYO/0Y8LWejKk9bcWqqm+qaoP78RNgaI8H1o52fluAO4EbgS7rKRMpiX4IsDPkcz69OHGGEpEcYCrwaZhDOZi7cP7wGsMcR0eMAoqAR9ympodEJCHcQbVFVXcBd+DU3PYA5ar6Znij6pBsVd0DTqUFyApzPB11JfBauIM4GBE5B9ilql905X4jJdFLG/N6fb9REUkEngN+qKoV4Y6nLSIyDyhU1eXhjqWDvMCxwP2qOhWopvc0LbTgtm3PB0YCg4EEEfl2eKOKTCLyc5wm08fDHUt7RCQe+Dnwi67ed6Qk+nxgWMjnofSyU+DWRCQaJ8k/rqrPhzuegzgJOEdE8nCaxE4VkX+GN6SDygfyVbXpDOlZnMTfG30F2KaqRaoaAJ4HZoY5po4oEJFBAO57YZjjOSgRuQyYB1ysvfvGodE4hf4X7v+3ocBnIjKwszuOlES/FBgrIiNFJAbngtZLYY6pXSIiOG3I61T1/8Idz8Go6i2qOlRVc3B+13dVtdfWOlV1L7BTRJqeDj8HWBvGkA5mB3CCiMS7fxNz6KUXjlt5CbjMnb4MeDGMsRyUiMwFbgLOUdWacMdzMKq6SlWzVDXH/f+WDxzr/k13SkQkevdiy/eBN3D+ozytqmvCG9VBnQRcglM7XuG+zgx3UBHkB8DjIrISmAL8LrzhtM0963gW+AxYhfP/sVfdri8iTwAfA+NFJF9E/gu4HfiqiGzC6R1yezhjbNJOrPcCScBb7v+zBWENMkQ78XbPsXr3mYwxxpjOiogavTHGmPZZojfGmAhnid4YYyKcJXpjjIlwluiNMSbCWaI3/ZKIBEO6tq7oyhFPRSSnrREJjQkXb7gDMCZMalV1SriDMKYnWI3emBAikicivxeRJe5rjDt/hIi8445r/o6IDHfnZ7vjnH/hvpqGMPCIyIPuWPNvikhc2L6U6fcs0Zv+Kq5V080FIcsqVHU6zl2Vd7nz7gX+7o5r/jhwjzv/HmCxqh6DM6ZO0x3ZY4H7VPVooAw4t1u/jTEHYXfGmn5JRKpUNbGN+XnAqaq61R14bq+qpotIMTBIVQPu/D2qmiEiRcBQVa0P2UcO8Jb7YA5E5CYgWlV/0wNfzZgDWI3emANpO9PtrdOW+pDpIHY9zISRJXpjDnRByPvH7vRH7H/M38XAh+70O8A10Pxc3eSeCtKYjrJahumv4kRkRcjn11W1qYtlrIh8ilMRutCddx3wsIj8FOcJVle4868H/uqOPBjESfp7ujt4Yw6HtdEbE8Jto5+mqsXhjsWYrmJNN8YYE+GsRm+MMRHOavTGGBPhLNEbY0yEs0RvjDERzhK9McZEOEv0xhgT4f4/nqfWjkDPr48AAAAASUVORK5CYII=\n",
      "text/plain": [
       "<Figure size 432x288 with 2 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "# Το δίκτυο που πρέπει να σχεδιάσετε\n",
    "class CNN(nn.Module):\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "        # εδώ γράφουμε τα επίπεδα του δικτύου μας\n",
    "        self.conv0 = nn.Conv2d(1, 8, 7) #1 gia grayscale, 8 arithmos filtron, 7 to megethos\n",
    "        self.conv1 = nn.Conv2d(8, 32, 5)\n",
    "        self.fc0 = nn.Linear(288, 128) #288=Height*Weight*Depth apo x = torch.flatten(x, start_dim=1) \n",
    "        self.fc1 = nn.Linear(128, 50)\n",
    "        self.fc2 = nn.Linear(50, 10)\n",
    "\n",
    "    def forward(self, x):\n",
    "        x = F.relu(self.conv0(x))\n",
    "        x = F.max_pool2d(x, kernel_size=2, stride=2) #megethos filtrou 2 me vima 2\n",
    "        \n",
    "        x = F.relu(self.conv1(x))\n",
    "        x = F.max_pool2d(x, kernel_size=2, stride=2)\n",
    "        \n",
    "        x = torch.flatten(x, start_dim=1)\n",
    "        print(x.shape)\n",
    "        \n",
    "        x = F.relu(self.fc0(x))\n",
    "        x = F.relu(self.fc1(x))\n",
    "        x = self.fc2(x)\n",
    "        x = F.log_softmax(x, dim=1)\n",
    "        return x\n",
    "\n",
    "\n",
    "class Trainer():\n",
    "    def __init__(self, model, train_dataset, test_dataset, batch_size):\n",
    "        # device to put tensors on\n",
    "        self.device = torch.device(\n",
    "            \"cuda\" if torch.cuda.is_available() else \"cpu\"\n",
    "        )\n",
    "        # model\n",
    "        self.model = model.to(self.device)\n",
    "        # dataloaders\n",
    "        self.train_loader = DataLoader(train_dataset,\n",
    "                                       batch_size=batch_size,\n",
    "                                       shuffle=True)\n",
    "        self.test_loader = DataLoader(test_dataset,\n",
    "                                      batch_size=batch_size,\n",
    "                                      shuffle=True)\n",
    "        # optimizer\n",
    "        self.optimizer = optim.Adam(self.model.parameters(), lr=0.001)\n",
    "\n",
    "    def count_parameters(self):\n",
    "        params = sum((p.numel() for p in self.model.parameters()\n",
    "                     if p.requires_grad))\n",
    "        return params\n",
    "\n",
    "    def train(self):\n",
    "        # set model to train mode\n",
    "        self.model.train()\n",
    "\n",
    "        # variable to store all batch losses\n",
    "        epoch_losses = 0\n",
    "\n",
    "        # train one epoch\n",
    "        for batch, label in self.train_loader:\n",
    "            # move tensors to device\n",
    "            batch = batch.to(self.device)\n",
    "            label = label.to(self.device)\n",
    "\n",
    "            # clear gradients\n",
    "            self.optimizer.zero_grad()\n",
    "            # forward pass\n",
    "            output = self.model(batch)\n",
    "            # calculate loss\n",
    "            loss = F.nll_loss(output, label)\n",
    "            # backpropagation\n",
    "            loss.backward()\n",
    "            # optimization step\n",
    "            self.optimizer.step()\n",
    "\n",
    "            epoch_losses += loss.item()\n",
    "\n",
    "        # mean epoch loss\n",
    "        epoch_losses /= len(self.train_loader)\n",
    "\n",
    "        return epoch_losses\n",
    "\n",
    "    @torch.no_grad()\n",
    "    def evaluate(self):\n",
    "        # evaluation mode\n",
    "        self.model.eval()\n",
    "\n",
    "        test_losses = 0\n",
    "        test_accuracy = 0\n",
    "\n",
    "        for batch, label in self.test_loader:\n",
    "            # move tensors to device\n",
    "            batch = batch.to(self.device)\n",
    "            label = label.to(self.device)\n",
    "\n",
    "            # forward pass\n",
    "            output = self.model(batch)\n",
    "            # calculate test loss\n",
    "            loss = F.nll_loss(output, label)\n",
    "            test_losses += loss.item()\n",
    "\n",
    "            # calculate test accuracy\n",
    "            preds = output.argmax(1).cpu().numpy()\n",
    "            label = label.cpu().numpy()\n",
    "            accuracy = accuracy_score(label, preds)\n",
    "            test_accuracy += accuracy\n",
    "\n",
    "        # mean test loss / mean test accuracy\n",
    "        test_losses /= len(self.test_loader)\n",
    "        test_accuracy /= len(self.test_loader)\n",
    "\n",
    "        return test_losses, test_accuracy\n",
    "\n",
    "    def run(self, epochs):\n",
    "        print(self.model, \"\\n\")\n",
    "\n",
    "        params = self.count_parameters()\n",
    "        print(f\"Total Parameters: {params}\")\n",
    "        print(f\"Using Device: {self.device}\")\n",
    "\n",
    "        train_losses = []\n",
    "        test_losses = []\n",
    "\n",
    "        test_accuracy = []\n",
    "\n",
    "        print(\"\\nTraining...\")\n",
    "\n",
    "        for epoch in range(epochs):\n",
    "            print(f\"\\nEpoch: {epoch:2d}\")\n",
    "\n",
    "            train_loss = self.train()\n",
    "            print(f\"Train Loss: {train_loss:.4f}\")\n",
    "\n",
    "            test_loss, test_acc = self.evaluate()\n",
    "            print(f\"Test Loss:  {test_loss:.4f} \\tTest Accuracy: {test_acc:.4f}\")\n",
    "\n",
    "            train_losses.append(train_loss)\n",
    "            test_losses.append(test_loss)\n",
    "            test_accuracy.append(test_acc)\n",
    "\n",
    "        return train_losses, test_losses, test_accuracy\n",
    "\n",
    "\n",
    "def plot_stats(train_loss, test_loss, accuracy):\n",
    "    fig, (ax1, ax2) = plt.subplots(2)\n",
    "\n",
    "    ax1.plot(train_loss, label=\"Train Loss\")\n",
    "    ax1.plot(test_loss, label=\"Evaluation Loss\")\n",
    "    ax1.set_title(\"Losses\")\n",
    "    ax1.set(xlabel=\"Epoch\", ylabel=\"\")\n",
    "    ax1.legend()\n",
    "    ax1.label_outer()\n",
    "\n",
    "    ax2.plot(accuracy)\n",
    "    ax2.set_title(\"Evaluation Accuracy\")\n",
    "    ax2.set(xlabel=\"Epoch\", ylabel=\"\")\n",
    "    ax2.label_outer()\n",
    "\n",
    "    plt.show()\n",
    "\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    # dataset\n",
    "    # for training\n",
    "    train_dataset = datasets.FashionMNIST(root=\"data\",\n",
    "                                          train=True,\n",
    "                                          download=True,\n",
    "                                          transform=ToTensor())\n",
    "    # for test / evaluation\n",
    "    test_dataset = datasets.FashionMNIST(root=\"data\",\n",
    "                                         train=False,\n",
    "                                         download=True,\n",
    "                                         transform=ToTensor())\n",
    "\n",
    "    # hyperparameters\n",
    "    epochs = 15\n",
    "    batch_size = 128\n",
    "\n",
    "    # initialize model\n",
    "    model = CNN()\n",
    "\n",
    "    # create trainer object\n",
    "    trainer = Trainer(model, train_dataset, test_dataset, batch_size)\n",
    "\n",
    "    # train model and get losses and test accuracy\n",
    "    train_loss, test_loss, accuracy = trainer.run(epochs)\n",
    "\n",
    "    # plot losses and accuracy\n",
    "    plot_stats(train_loss, test_loss, accuracy)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "packed-excuse",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}