notebooks/model_fine_tuning_Vit_CUB_50.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Finetuning ViT with CUB dataset 50 Episodes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Defaulting to user installation because normal site-packages is not writeable\n",
      "Requirement already satisfied: gdown in c:\\users\\rperera23\\appdata\\roaming\\python\\python39\\site-packages (4.7.1)\n",
      "Requirement already satisfied: filelock in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from gdown) (3.9.0)\n",
      "Requirement already satisfied: six in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from gdown) (1.16.0)\n",
      "Requirement already satisfied: beautifulsoup4 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from gdown) (4.11.1)\n",
      "Requirement already satisfied: tqdm in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from gdown) (4.64.1)\n",
      "Requirement already satisfied: requests[socks] in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from gdown) (2.28.1)\n",
      "Requirement already satisfied: soupsieve>1.2 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from beautifulsoup4->gdown) (2.3.2.post1)\n",
      "Requirement already satisfied: charset-normalizer<3,>=2 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from requests[socks]->gdown) (2.0.4)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from requests[socks]->gdown) (2022.9.24)\n",
      "Requirement already satisfied: idna<4,>=2.5 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from requests[socks]->gdown) (3.4)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from requests[socks]->gdown) (1.26.12)\n",
      "Requirement already satisfied: PySocks!=1.5.7,>=1.5.6 in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from requests[socks]->gdown) (1.7.1)\n",
      "Requirement already satisfied: colorama in c:\\program files\\arcgis\\pro\\bin\\python\\envs\\arcgispro-py3\\lib\\site-packages (from tqdm->gdown) (0.4.5)\n"
     ]
    }
   ],
   "source": [
    "!pip install gdown"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<contextlib.ExitStack at 0x214f355adc0>"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import sys\n",
    "sys.path.append('..')\n",
    "import torch\n",
    "import torch.nn as nn\n",
    "import torch.backends.cudnn as cudnn\n",
    "import matplotlib.pyplot as plt\n",
    "import os\n",
    "import torchvision\n",
    "from torchvision import datasets, models, transforms\n",
    "from torch.optim.lr_scheduler import StepLR\n",
    "\n",
    "\n",
    "from src.models import VisionTransformerTimm\n",
    "from src.dataloader import EuroSatDownloader\n",
    "from src.training import train_fine_tuning, eval_method, eval_func\n",
    "from src.dataloader import MiniImageNetDataSet, createEuroSatDataLoaders, EuroSatDownloader\n",
    "from src.modelvis import visualize_models\n",
    "\n",
    "plt.ion()\n",
    "\n",
    "'''\n",
    "The line cudnn.benchmark = True is typically used in deep learning projects that utilize the CUDA Deep Neural Network (cuDNN) library.\n",
    "When cudnn.benchmark is set to True, it enables cuDNN to automatically find the best algorithm configuration for the specific \n",
    "input sizes and hardware being used. This can result in improved performance during training and inference.By enabling benchmarking, \n",
    "cuDNN will run a short benchmarking phase during the first iteration of the model to determine the optimal algorithm configuration. \n",
    "This configuration is then cached and used for subsequent iterations, leading to faster execution times.\n",
    "It's important to note that enabling benchmarking may introduce some overhead during the initial benchmarking phase, so it is typically \n",
    "recommended to use it when the input sizes are consistent throughout the training process.\n",
    "Overall, setting cudnn.benchmark to True can help optimize the performance of deep learning models that use cuDNN.\n",
    "'''\n",
    "torch.backends.cudnn.benchmark = True\n",
    "\n",
    "'''\n",
    "The line plt.ion() is a function call that activates interactive mode in matplotlib.\n",
    "When interactive mode is enabled, any plot that is created will be displayed immediately \n",
    "and can be updated dynamically. This means that you can modify the plot after it is displayed, \n",
    "such as changing the data or adding annotations, and the changes will be reflected in real-time.\n",
    "'''\n",
    "plt.ion()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0.9.2\n",
      "1.8.2\n"
     ]
    }
   ],
   "source": [
    "print(torchvision.__version__)\n",
    "print(torch.__version__)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "cuda:0\n"
     ]
    }
   ],
   "source": [
    "device = torch.device(\"cuda:0\" if torch.cuda.is_available() else \"cpu\")\n",
    "print(device)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Download CUB Database\n",
    "extract CUB images to the path\n",
    "\n",
    "dataDir = '..\\\\data\\\\CUB_200_2011\\\\CUB_200_2011\\\\images'"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Hyperparameters"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "LEARNING_RATE = 0.01\n",
    "WEIGHT_DECAY = 0.001\n",
    "NUM_EPOCHS = 20\n",
    "BATCH_SIZE = 25\n",
    "MOMENTUM = 0.09\n",
    "STEP_SIZE = 5\n",
    "num_of_classes = 64\n",
    "EPISODES = 50\n",
    "GAMMA = 0.5"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "data_transforms = {\n",
    "    'train': transforms.Compose([\n",
    "        transforms.Resize((224, 224)),\n",
    "        transforms.ColorJitter(brightness=0.4, contrast=0.4, saturation=0.4),\n",
    "        transforms.RandomHorizontalFlip(),\n",
    "        transforms.GaussianBlur(3, sigma=(0.1, 2.0)),\n",
    "        transforms.ToTensor(),\n",
    "        transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),\n",
    "    ]),\n",
    "    'test': transforms.Compose([\n",
    "        transforms.Resize((224, 224)),\n",
    "        transforms.ToTensor(),\n",
    "        transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),\n",
    "    ])\n",
    "}\n",
    "\n",
    "dataDir = '..\\\\data\\\\CUB_200_2011\\\\CUB_200_2011\\\\images'\n",
    "modelPath = '..\\\\data\\\\models\\\\best_model_VistionTransformerTimm_pc18_fixedTransformers.pth'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Please wait patiently, it may take some seconds...\n",
      "\n",
      "Training Episode: 1...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.491617, test accuracy: 0.360000\n",
      "epoch: 1, accuracy: 0.560000, avg. loss: 1.578376, test accuracy: 0.493333\n",
      "epoch: 2, accuracy: 0.680000, avg. loss: 0.666878, test accuracy: 0.813333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.079662, test accuracy: 0.853333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.011055, test accuracy: 0.853333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.009975, test accuracy: 0.853333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.010790, test accuracy: 0.853333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.009762, test accuracy: 0.853333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.013016, test accuracy: 0.853333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.010062, test accuracy: 0.853333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.008619, test accuracy: 0.880000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.008091, test accuracy: 0.880000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.008222, test accuracy: 0.880000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.008534, test accuracy: 0.880000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.009942, test accuracy: 0.880000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.007472, test accuracy: 0.880000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.007584, test accuracy: 0.880000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.007297, test accuracy: 0.880000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.015670, test accuracy: 0.880000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.009689, test accuracy: 0.880000\n",
      "{'171.Myrtle_Warbler', '040.Olive_sided_Flycatcher', '054.Blue_Grosbeak', '161.Blue_winged_Warbler', '029.American_Crow'}\n",
      "Episode 1: Test Accuracy: 0.8800\n",
      "\n",
      "Training Episode: 2...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 1.743244, test accuracy: 0.680000\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 0.491565, test accuracy: 0.746667\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.262854, test accuracy: 0.813333\n",
      "epoch: 3, accuracy: 0.960000, avg. loss: 0.141238, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.078747, test accuracy: 0.906667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.051830, test accuracy: 0.906667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.042691, test accuracy: 0.946667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.032940, test accuracy: 0.946667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.038799, test accuracy: 0.946667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.033095, test accuracy: 0.946667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.034010, test accuracy: 0.946667\n",
      "epoch: 11, accuracy: 0.960000, avg. loss: 0.078587, test accuracy: 0.946667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.081282, test accuracy: 0.933333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.030891, test accuracy: 0.946667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.058497, test accuracy: 0.933333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.027346, test accuracy: 0.933333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.040254, test accuracy: 0.933333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.026873, test accuracy: 0.933333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.063175, test accuracy: 0.933333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.043184, test accuracy: 0.933333\n",
      "{'144.Common_Tern', '044.Frigatebird', '167.Hooded_Warbler', '117.Clay_colored_Sparrow', '156.White_eyed_Vireo'}\n",
      "Episode 2: Test Accuracy: 0.9333\n",
      "\n",
      "Training Episode: 3...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.040000, avg. loss: 2.495950, test accuracy: 0.586667\n",
      "epoch: 1, accuracy: 0.880000, avg. loss: 0.408038, test accuracy: 0.586667\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.251184, test accuracy: 0.626667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.137217, test accuracy: 0.773333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.042781, test accuracy: 0.786667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.037495, test accuracy: 0.800000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.035137, test accuracy: 0.840000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.024853, test accuracy: 0.840000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.024313, test accuracy: 0.840000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.030990, test accuracy: 0.840000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.022271, test accuracy: 0.866667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.025899, test accuracy: 0.866667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.023866, test accuracy: 0.866667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.020977, test accuracy: 0.866667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020818, test accuracy: 0.866667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.021013, test accuracy: 0.866667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.023776, test accuracy: 0.866667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.019790, test accuracy: 0.866667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019783, test accuracy: 0.866667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.018021, test accuracy: 0.866667\n",
      "{'052.Pied_billed_Grebe', '138.Tree_Swallow', '069.Rufous_Hummingbird', '137.Cliff_Swallow', '099.Ovenbird'}\n",
      "Episode 3: Test Accuracy: 0.8667\n",
      "\n",
      "Training Episode: 4...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.080000, avg. loss: 2.196806, test accuracy: 0.373333\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 0.584377, test accuracy: 0.826667\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.102161, test accuracy: 0.946667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.030561, test accuracy: 0.933333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.027664, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.019673, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.018570, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.016319, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.017675, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.024210, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.014798, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.016758, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.017284, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.015360, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.015693, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.014048, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.012569, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.013948, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.012299, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.015475, test accuracy: 0.960000\n",
      "{'026.Bronzed_Cowbird', '061.Heermann_Gull', '025.Pelagic_Cormorant', '195.Carolina_Wren', '175.Pine_Warbler'}\n",
      "Episode 4: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 5...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.240000, avg. loss: 1.858879, test accuracy: 0.680000\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 0.665278, test accuracy: 0.960000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.086090, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.045446, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.030695, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.030855, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.028069, test accuracy: 0.973333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.020662, test accuracy: 0.973333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.029032, test accuracy: 0.973333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.020466, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.029246, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.043035, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.017231, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.017408, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020072, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.017048, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.016853, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.017425, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.017513, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.017450, test accuracy: 0.973333\n",
      "{'088.Western_Meadowlark', '119.Field_Sparrow', '141.Artic_Tern', '008.Rhinoceros_Auklet', '181.Worm_eating_Warbler'}\n",
      "Episode 5: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 6...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.892179, test accuracy: 0.493333\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 0.881668, test accuracy: 0.760000\n",
      "epoch: 2, accuracy: 0.800000, avg. loss: 0.460179, test accuracy: 0.906667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.057831, test accuracy: 0.920000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.049183, test accuracy: 0.906667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.031879, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.025965, test accuracy: 0.920000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.023757, test accuracy: 0.920000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.022480, test accuracy: 0.920000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.020167, test accuracy: 0.920000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.022955, test accuracy: 0.920000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.017046, test accuracy: 0.920000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.018494, test accuracy: 0.920000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.018269, test accuracy: 0.920000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.018612, test accuracy: 0.920000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.022438, test accuracy: 0.920000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.017963, test accuracy: 0.920000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.017313, test accuracy: 0.920000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.017960, test accuracy: 0.920000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021201, test accuracy: 0.920000\n",
      "{'043.Yellow_bellied_Flycatcher', '050.Eared_Grebe', '009.Brewer_Blackbird', '099.Ovenbird', '119.Field_Sparrow'}\n",
      "Episode 6: Test Accuracy: 0.9200\n",
      "\n",
      "Training Episode: 7...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.040000, avg. loss: 2.290261, test accuracy: 0.613333\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 0.658102, test accuracy: 0.960000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.218308, test accuracy: 0.933333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.089138, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.092413, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.054676, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.046033, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.054356, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.041267, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.039823, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.042537, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.039662, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.040638, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.030018, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.032541, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.028889, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.031912, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.032411, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.028629, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.036560, test accuracy: 0.986667\n",
      "{'191.Red_headed_Woodpecker', '011.Rusty_Blackbird', '153.Philadelphia_Vireo', '132.White_crowned_Sparrow', '042.Vermilion_Flycatcher'}\n",
      "Episode 7: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 8...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 1.871181, test accuracy: 0.373333\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.907990, test accuracy: 0.893333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.103364, test accuracy: 0.933333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.045377, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.028466, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.021961, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.019224, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.016138, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.021803, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.027382, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.022500, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.016355, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.017011, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.015753, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.044535, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.015905, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.031826, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.014571, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.015680, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.014054, test accuracy: 0.960000\n",
      "{'066.Western_Gull', '176.Prairie_Warbler', '187.American_Three_toed_Woodpecker', '178.Swainson_Warbler', '112.Great_Grey_Shrike'}\n",
      "Episode 8: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 9...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 1.930801, test accuracy: 0.693333\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 1.118236, test accuracy: 0.960000\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.217285, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.037553, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.029847, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.022697, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.026777, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.019492, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.032624, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.017684, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.019400, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.018949, test accuracy: 1.000000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.014536, test accuracy: 1.000000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.020973, test accuracy: 1.000000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.018503, test accuracy: 1.000000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.020782, test accuracy: 1.000000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.015314, test accuracy: 1.000000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.015519, test accuracy: 1.000000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.014953, test accuracy: 1.000000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.020193, test accuracy: 1.000000\n",
      "{'140.Summer_Tanager', '061.Heermann_Gull', '004.Groove_billed_Ani', '190.Red_cockaded_Woodpecker', '072.Pomarine_Jaeger'}\n",
      "Episode 9: Test Accuracy: 1.0000\n",
      "\n",
      "Training Episode: 10...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 1.688972, test accuracy: 0.600000\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.543400, test accuracy: 0.893333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.076298, test accuracy: 0.920000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.066090, test accuracy: 0.920000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.027557, test accuracy: 0.920000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.021793, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.024577, test accuracy: 0.920000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.021437, test accuracy: 0.920000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.023871, test accuracy: 0.920000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.017711, test accuracy: 0.920000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.024496, test accuracy: 0.920000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.019244, test accuracy: 0.920000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.019554, test accuracy: 0.920000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.016501, test accuracy: 0.920000\n",
      "epoch: 14, accuracy: 0.960000, avg. loss: 0.065044, test accuracy: 0.933333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.020601, test accuracy: 0.946667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.014661, test accuracy: 0.946667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.015513, test accuracy: 0.933333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.013023, test accuracy: 0.933333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.016480, test accuracy: 0.933333\n",
      "{'014.Indigo_Bunting', '171.Myrtle_Warbler', '063.Ivory_Gull', '120.Fox_Sparrow', '047.American_Goldfinch'}\n",
      "Episode 10: Test Accuracy: 0.9333\n",
      "\n",
      "Training Episode: 11...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.280000, avg. loss: 2.042552, test accuracy: 0.560000\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.629952, test accuracy: 0.826667\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.174881, test accuracy: 0.973333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.041236, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.038070, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.029959, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.029090, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.025527, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.022457, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.027986, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.019423, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.024208, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.023178, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.018318, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017691, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019743, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.019608, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018136, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019065, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.017436, test accuracy: 0.986667\n",
      "{'049.Boat_tailed_Grackle', '055.Evening_Grosbeak', '040.Olive_sided_Flycatcher', '141.Artic_Tern', '180.Wilson_Warbler'}\n",
      "Episode 11: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 12...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 1.850764, test accuracy: 0.600000\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.490254, test accuracy: 0.653333\n",
      "epoch: 2, accuracy: 0.920000, avg. loss: 0.280716, test accuracy: 0.760000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.054853, test accuracy: 0.773333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.028521, test accuracy: 0.773333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.028591, test accuracy: 0.773333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.028647, test accuracy: 0.773333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.019134, test accuracy: 0.773333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.019012, test accuracy: 0.773333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.021642, test accuracy: 0.773333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.018555, test accuracy: 0.773333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.019936, test accuracy: 0.773333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.018729, test accuracy: 0.773333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.020478, test accuracy: 0.773333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.027568, test accuracy: 0.773333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019033, test accuracy: 0.773333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.020307, test accuracy: 0.773333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018022, test accuracy: 0.773333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.015946, test accuracy: 0.773333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.029531, test accuracy: 0.773333\n",
      "{'168.Kentucky_Warbler', '159.Black_and_white_Warbler', '055.Evening_Grosbeak', '070.Green_Violetear', '122.Harris_Sparrow'}\n",
      "Episode 12: Test Accuracy: 0.7733\n",
      "\n",
      "Training Episode: 13...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.120000, avg. loss: 1.976461, test accuracy: 0.546667\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 1.068286, test accuracy: 0.920000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.104703, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.054735, test accuracy: 0.973333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.038683, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.029872, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.026389, test accuracy: 0.973333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.027406, test accuracy: 0.973333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.029612, test accuracy: 0.973333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.025029, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.022971, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.023861, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.022072, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.026850, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017087, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.018952, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.016554, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.023422, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.023064, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.018600, test accuracy: 0.973333\n",
      "{'083.White_breasted_Kingfisher', '183.Northern_Waterthrush', '185.Bohemian_Waxwing', '078.Gray_Kingbird', '134.Cape_Glossy_Starling'}\n",
      "Episode 13: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 14...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.005211, test accuracy: 0.653333\n",
      "epoch: 1, accuracy: 0.920000, avg. loss: 0.517932, test accuracy: 0.840000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.173934, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.052906, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.037558, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.043356, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.046877, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.027111, test accuracy: 0.973333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.033849, test accuracy: 0.973333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.027245, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.026029, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.023892, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.024899, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.024042, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.032737, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019068, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.023553, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.023351, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018256, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021333, test accuracy: 0.973333\n",
      "{'186.Cedar_Waxwing', '159.Black_and_white_Warbler', '049.Boat_tailed_Grackle', '078.Gray_Kingbird', '143.Caspian_Tern'}\n",
      "Episode 14: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 15...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.080000, avg. loss: 1.910846, test accuracy: 0.413333\n",
      "epoch: 1, accuracy: 0.680000, avg. loss: 1.095275, test accuracy: 0.506667\n",
      "epoch: 2, accuracy: 0.760000, avg. loss: 0.533273, test accuracy: 0.733333\n",
      "epoch: 3, accuracy: 0.800000, avg. loss: 0.895705, test accuracy: 0.693333\n",
      "epoch: 4, accuracy: 0.840000, avg. loss: 0.413224, test accuracy: 0.813333\n",
      "epoch: 5, accuracy: 0.920000, avg. loss: 0.188901, test accuracy: 0.826667\n",
      "epoch: 6, accuracy: 0.960000, avg. loss: 0.064888, test accuracy: 0.853333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.030998, test accuracy: 0.866667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.032893, test accuracy: 0.853333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.022290, test accuracy: 0.853333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.049234, test accuracy: 0.853333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.024312, test accuracy: 0.853333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.020085, test accuracy: 0.853333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.025060, test accuracy: 0.853333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020533, test accuracy: 0.853333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.023181, test accuracy: 0.853333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.017304, test accuracy: 0.853333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.020579, test accuracy: 0.853333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.024861, test accuracy: 0.853333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.022725, test accuracy: 0.853333\n",
      "{'136.Barn_Swallow', '103.Sayornis', '164.Cerulean_Warbler', '048.European_Goldfinch', '102.Western_Wood_Pewee'}\n",
      "Episode 15: Test Accuracy: 0.8533\n",
      "\n",
      "Training Episode: 16...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 1.817614, test accuracy: 0.293333\n",
      "epoch: 1, accuracy: 0.440000, avg. loss: 1.295762, test accuracy: 0.586667\n",
      "epoch: 2, accuracy: 0.680000, avg. loss: 0.850930, test accuracy: 0.760000\n",
      "epoch: 3, accuracy: 0.880000, avg. loss: 0.202614, test accuracy: 0.853333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.097369, test accuracy: 0.826667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.047147, test accuracy: 0.826667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.050183, test accuracy: 0.853333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.057744, test accuracy: 0.853333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.038888, test accuracy: 0.853333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.039739, test accuracy: 0.853333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.044710, test accuracy: 0.853333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.038602, test accuracy: 0.853333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.036343, test accuracy: 0.853333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.034154, test accuracy: 0.853333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.028170, test accuracy: 0.853333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.027042, test accuracy: 0.853333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.040577, test accuracy: 0.853333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.033078, test accuracy: 0.853333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.025162, test accuracy: 0.853333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.034681, test accuracy: 0.853333\n",
      "{'109.American_Redstart', '027.Shiny_Cowbird', '191.Red_headed_Woodpecker', '004.Groove_billed_Ani', '154.Red_eyed_Vireo'}\n",
      "Episode 16: Test Accuracy: 0.8533\n",
      "\n",
      "Training Episode: 17...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.440000, avg. loss: 1.705399, test accuracy: 0.573333\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 0.752376, test accuracy: 0.800000\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.238742, test accuracy: 0.733333\n",
      "epoch: 3, accuracy: 0.960000, avg. loss: 0.133107, test accuracy: 0.866667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.050697, test accuracy: 0.866667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.039515, test accuracy: 0.906667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.041478, test accuracy: 0.920000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.028124, test accuracy: 0.920000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.035205, test accuracy: 0.920000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.028896, test accuracy: 0.920000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.036281, test accuracy: 0.920000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.024808, test accuracy: 0.920000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.023811, test accuracy: 0.920000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.022512, test accuracy: 0.920000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.039046, test accuracy: 0.920000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.024105, test accuracy: 0.920000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.021116, test accuracy: 0.920000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.022263, test accuracy: 0.920000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.023419, test accuracy: 0.920000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.034878, test accuracy: 0.920000\n",
      "{'116.Chipping_Sparrow', '059.California_Gull', '153.Philadelphia_Vireo', '156.White_eyed_Vireo', '102.Western_Wood_Pewee'}\n",
      "Episode 17: Test Accuracy: 0.9200\n",
      "\n",
      "Training Episode: 18...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.120000, avg. loss: 1.605628, test accuracy: 0.546667\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.524489, test accuracy: 0.920000\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.254689, test accuracy: 0.893333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.048143, test accuracy: 0.893333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.030551, test accuracy: 0.946667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.028058, test accuracy: 0.946667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.041254, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.023910, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.018190, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.020770, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.023352, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.018605, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.020106, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.023448, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017942, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.016800, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.023866, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.014452, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.014921, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.018117, test accuracy: 0.960000\n",
      "{'075.Green_Jay', '184.Louisiana_Waterthrush', '007.Parakeet_Auklet', '170.Mourning_Warbler', '067.Anna_Hummingbird'}\n",
      "Episode 18: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 19...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.080000, avg. loss: 2.106793, test accuracy: 0.480000\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 0.811445, test accuracy: 0.933333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.057709, test accuracy: 0.933333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.036173, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.034790, test accuracy: 0.933333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.037519, test accuracy: 0.933333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.020763, test accuracy: 0.933333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.020344, test accuracy: 0.933333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.019387, test accuracy: 0.933333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.015853, test accuracy: 0.946667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.014750, test accuracy: 0.946667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.017896, test accuracy: 0.946667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.018997, test accuracy: 0.946667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.014356, test accuracy: 0.946667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.032719, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019937, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.021750, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.014729, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019098, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.014453, test accuracy: 0.960000\n",
      "{'027.Shiny_Cowbird', '035.Purple_Finch', '069.Rufous_Hummingbird', '143.Caspian_Tern', '165.Chestnut_sided_Warbler'}\n",
      "Episode 19: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 20...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.280000, avg. loss: 1.887164, test accuracy: 0.800000\n",
      "epoch: 1, accuracy: 0.880000, avg. loss: 0.364456, test accuracy: 0.973333\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.159280, test accuracy: 0.986667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.060771, test accuracy: 0.986667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.049884, test accuracy: 0.986667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.032036, test accuracy: 0.986667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.053598, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.031250, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.027253, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.030863, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.021820, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 0.960000, avg. loss: 0.146884, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.027557, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.020691, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.019404, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.021487, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.037129, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.020758, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019656, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.036019, test accuracy: 0.986667\n",
      "{'009.Brewer_Blackbird', '068.Ruby_throated_Hummingbird', '016.Painted_Bunting', '035.Purple_Finch', '087.Mallard'}\n",
      "Episode 20: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 21...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.280000, avg. loss: 1.722348, test accuracy: 0.440000\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.667100, test accuracy: 0.706667\n",
      "epoch: 2, accuracy: 0.920000, avg. loss: 0.304166, test accuracy: 0.853333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.086896, test accuracy: 0.920000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.046894, test accuracy: 0.920000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.036441, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.041139, test accuracy: 0.920000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.032132, test accuracy: 0.920000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.026314, test accuracy: 0.920000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.029506, test accuracy: 0.933333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.029467, test accuracy: 0.933333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.026017, test accuracy: 0.933333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.026666, test accuracy: 0.933333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.028054, test accuracy: 0.933333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.027728, test accuracy: 0.933333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.021313, test accuracy: 0.933333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.023045, test accuracy: 0.933333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.026238, test accuracy: 0.933333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.021590, test accuracy: 0.933333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.025063, test accuracy: 0.933333\n",
      "{'009.Brewer_Blackbird', '187.American_Three_toed_Woodpecker', '137.Cliff_Swallow', '092.Nighthawk', '095.Baltimore_Oriole'}\n",
      "Episode 21: Test Accuracy: 0.9333\n",
      "\n",
      "Training Episode: 22...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.731935, test accuracy: 0.626667\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 1.866837, test accuracy: 0.666667\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.431059, test accuracy: 0.933333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.063309, test accuracy: 0.920000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.024390, test accuracy: 0.933333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.026021, test accuracy: 0.933333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.032744, test accuracy: 0.933333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.022598, test accuracy: 0.933333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.024094, test accuracy: 0.933333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.021972, test accuracy: 0.933333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.019437, test accuracy: 0.933333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.018549, test accuracy: 0.933333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.025934, test accuracy: 0.933333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.019162, test accuracy: 0.933333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020900, test accuracy: 0.933333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019211, test accuracy: 0.933333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.020213, test accuracy: 0.933333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.019794, test accuracy: 0.933333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.017579, test accuracy: 0.933333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.022910, test accuracy: 0.933333\n",
      "{'198.Rock_Wren', '062.Herring_Gull', '150.Sage_Thrasher', '087.Mallard', '002.Laysan_Albatross'}\n",
      "Episode 22: Test Accuracy: 0.9333\n",
      "\n",
      "Training Episode: 23...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.240000, avg. loss: 1.941490, test accuracy: 0.640000\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.537069, test accuracy: 0.933333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.087762, test accuracy: 0.946667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.043999, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.026169, test accuracy: 0.946667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.027598, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.029960, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.026261, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.024429, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.018624, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.020892, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.017625, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.023116, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.015898, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.016105, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.018302, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.020405, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018276, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018076, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.020082, test accuracy: 0.973333\n",
      "{'160.Black_throated_Blue_Warbler', '194.Cactus_Wren', '067.Anna_Hummingbird', '089.Hooded_Merganser', '143.Caspian_Tern'}\n",
      "Episode 23: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 24...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.120000, avg. loss: 2.499855, test accuracy: 0.560000\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 1.105040, test accuracy: 0.746667\n",
      "epoch: 2, accuracy: 0.920000, avg. loss: 0.339998, test accuracy: 0.893333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.050159, test accuracy: 0.906667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.041139, test accuracy: 0.906667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.042166, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.028558, test accuracy: 0.933333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.024149, test accuracy: 0.933333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.022888, test accuracy: 0.933333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.022518, test accuracy: 0.933333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.023550, test accuracy: 0.933333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.023977, test accuracy: 0.933333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.035293, test accuracy: 0.933333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.037777, test accuracy: 0.933333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017538, test accuracy: 0.946667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.024929, test accuracy: 0.946667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.022527, test accuracy: 0.946667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018216, test accuracy: 0.946667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.017674, test accuracy: 0.946667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.020740, test accuracy: 0.946667\n",
      "{'088.Western_Meadowlark', '082.Ringed_Kingfisher', '077.Tropical_Kingbird', '158.Bay_breasted_Warbler', '076.Dark_eyed_Junco'}\n",
      "Episode 24: Test Accuracy: 0.9467\n",
      "\n",
      "Training Episode: 25...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.559019, test accuracy: 0.160000\n",
      "epoch: 1, accuracy: 0.600000, avg. loss: 1.927757, test accuracy: 0.586667\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.415936, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.028248, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.013750, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.017527, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.010435, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.010118, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.013005, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.008845, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.009638, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.010026, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.014137, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.010442, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.009411, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.009529, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.014891, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.011485, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.007848, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.010438, test accuracy: 0.973333\n",
      "{'009.Brewer_Blackbird', '066.Western_Gull', '093.Clark_Nutcracker', '197.Marsh_Wren', '165.Chestnut_sided_Warbler'}\n",
      "Episode 25: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 26...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 1.761152, test accuracy: 0.480000\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 0.661508, test accuracy: 0.946667\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.059005, test accuracy: 0.946667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.034961, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.026702, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.020854, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.018407, test accuracy: 0.973333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.019211, test accuracy: 0.973333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.018905, test accuracy: 0.973333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.012380, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.016049, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.014509, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.016042, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.013173, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.014647, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.017751, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.011009, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.011976, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.012442, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.012366, test accuracy: 0.973333\n",
      "{'064.Ring_billed_Gull', '032.Mangrove_Cuckoo', '156.White_eyed_Vireo', '053.Western_Grebe', '074.Florida_Jay'}\n",
      "Episode 26: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 27...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.280000, avg. loss: 1.739781, test accuracy: 0.346667\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 0.897598, test accuracy: 0.813333\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.319628, test accuracy: 0.840000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.093104, test accuracy: 0.866667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.054475, test accuracy: 0.880000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.052330, test accuracy: 0.880000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.046958, test accuracy: 0.880000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.040998, test accuracy: 0.880000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.040460, test accuracy: 0.880000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.049312, test accuracy: 0.893333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.036398, test accuracy: 0.893333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.037485, test accuracy: 0.893333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.036080, test accuracy: 0.893333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.042384, test accuracy: 0.893333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.038191, test accuracy: 0.893333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.035629, test accuracy: 0.893333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.024990, test accuracy: 0.893333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.030302, test accuracy: 0.893333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.033501, test accuracy: 0.893333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.070115, test accuracy: 0.893333\n",
      "{'193.Bewick_Wren', '026.Bronzed_Cowbird', '078.Gray_Kingbird', '122.Harris_Sparrow', '056.Pine_Grosbeak'}\n",
      "Episode 27: Test Accuracy: 0.8933\n",
      "\n",
      "Training Episode: 28...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.280000, avg. loss: 1.849423, test accuracy: 0.693333\n",
      "epoch: 1, accuracy: 0.880000, avg. loss: 0.413128, test accuracy: 0.826667\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.176998, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.054004, test accuracy: 0.973333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.035734, test accuracy: 0.986667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.039693, test accuracy: 1.000000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.031384, test accuracy: 1.000000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.030445, test accuracy: 1.000000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.026158, test accuracy: 1.000000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.022009, test accuracy: 1.000000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.025266, test accuracy: 1.000000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.027969, test accuracy: 1.000000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.022586, test accuracy: 1.000000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.017888, test accuracy: 1.000000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017539, test accuracy: 1.000000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.024546, test accuracy: 1.000000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.018989, test accuracy: 1.000000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.021554, test accuracy: 1.000000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.020319, test accuracy: 1.000000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021459, test accuracy: 1.000000\n",
      "{'145.Elegant_Tern', '006.Least_Auklet', '023.Brandt_Cormorant', '032.Mangrove_Cuckoo', '190.Red_cockaded_Woodpecker'}\n",
      "Episode 28: Test Accuracy: 1.0000\n",
      "\n",
      "Training Episode: 29...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.320000, avg. loss: 1.976521, test accuracy: 0.506667\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 0.990304, test accuracy: 0.826667\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.188562, test accuracy: 0.920000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.052619, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.050093, test accuracy: 0.946667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.051182, test accuracy: 0.946667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.031173, test accuracy: 0.946667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.036693, test accuracy: 0.946667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.028653, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.043427, test accuracy: 0.973333\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.025451, test accuracy: 0.973333\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.032239, test accuracy: 0.973333\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.019420, test accuracy: 0.973333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.021912, test accuracy: 0.973333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020890, test accuracy: 0.973333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.021408, test accuracy: 0.973333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.021743, test accuracy: 0.973333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.036081, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.024658, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.018856, test accuracy: 0.973333\n",
      "{'186.Cedar_Waxwing', '098.Scott_Oriole', '048.European_Goldfinch', '135.Bank_Swallow', '192.Downy_Woodpecker'}\n",
      "Episode 29: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 30...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.296113, test accuracy: 0.453333\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 1.560208, test accuracy: 0.613333\n",
      "epoch: 2, accuracy: 0.720000, avg. loss: 0.566139, test accuracy: 0.773333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.119009, test accuracy: 0.880000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.054937, test accuracy: 0.893333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.035152, test accuracy: 0.880000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.042175, test accuracy: 0.893333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.038030, test accuracy: 0.880000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.027646, test accuracy: 0.880000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.023246, test accuracy: 0.880000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.024345, test accuracy: 0.880000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.020879, test accuracy: 0.880000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.027759, test accuracy: 0.893333\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.024599, test accuracy: 0.893333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.022306, test accuracy: 0.880000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.026848, test accuracy: 0.893333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.021750, test accuracy: 0.893333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.024920, test accuracy: 0.880000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018394, test accuracy: 0.880000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.025523, test accuracy: 0.880000\n",
      "{'084.Red_legged_Kittiwake', '059.California_Gull', '058.Pigeon_Guillemot', '164.Cerulean_Warbler', '200.Common_Yellowthroat'}\n",
      "Episode 30: Test Accuracy: 0.8800\n",
      "\n",
      "Training Episode: 31...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.010329, test accuracy: 0.533333\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 0.603136, test accuracy: 0.586667\n",
      "epoch: 2, accuracy: 0.720000, avg. loss: 0.673790, test accuracy: 0.586667\n",
      "epoch: 3, accuracy: 0.800000, avg. loss: 1.411444, test accuracy: 0.693333\n",
      "epoch: 4, accuracy: 0.920000, avg. loss: 0.236273, test accuracy: 0.733333\n",
      "epoch: 5, accuracy: 0.920000, avg. loss: 0.218373, test accuracy: 0.746667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.068857, test accuracy: 0.746667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.099851, test accuracy: 0.760000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.051091, test accuracy: 0.746667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.045539, test accuracy: 0.746667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.054613, test accuracy: 0.746667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.038559, test accuracy: 0.746667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.034374, test accuracy: 0.746667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.054522, test accuracy: 0.746667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.039705, test accuracy: 0.746667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.033688, test accuracy: 0.760000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.052192, test accuracy: 0.746667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.042925, test accuracy: 0.746667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.038756, test accuracy: 0.760000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.030836, test accuracy: 0.760000\n",
      "{'144.Common_Tern', '196.House_Wren', '038.Great_Crested_Flycatcher', '141.Artic_Tern', '157.Yellow_throated_Vireo'}\n",
      "Episode 31: Test Accuracy: 0.7600\n",
      "\n",
      "Training Episode: 32...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.240000, avg. loss: 1.938562, test accuracy: 0.293333\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.771089, test accuracy: 0.640000\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.305956, test accuracy: 0.880000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.118351, test accuracy: 0.800000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.055526, test accuracy: 0.800000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.050445, test accuracy: 0.813333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.049945, test accuracy: 0.826667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.036489, test accuracy: 0.813333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.040271, test accuracy: 0.813333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.031175, test accuracy: 0.826667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.039137, test accuracy: 0.826667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.037653, test accuracy: 0.826667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.034214, test accuracy: 0.826667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.027471, test accuracy: 0.826667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.033040, test accuracy: 0.826667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.026568, test accuracy: 0.826667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.032138, test accuracy: 0.826667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.025844, test accuracy: 0.826667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.025631, test accuracy: 0.826667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.023631, test accuracy: 0.826667\n",
      "{'136.Barn_Swallow', '075.Green_Jay', '164.Cerulean_Warbler', '180.Wilson_Warbler', '181.Worm_eating_Warbler'}\n",
      "Episode 32: Test Accuracy: 0.8267\n",
      "\n",
      "Training Episode: 33...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.040000, avg. loss: 2.627214, test accuracy: 0.360000\n",
      "epoch: 1, accuracy: 0.800000, avg. loss: 1.098486, test accuracy: 0.680000\n",
      "epoch: 2, accuracy: 0.920000, avg. loss: 0.181659, test accuracy: 0.733333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.072387, test accuracy: 0.786667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.025959, test accuracy: 0.746667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.021158, test accuracy: 0.746667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.025170, test accuracy: 0.773333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.018411, test accuracy: 0.773333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.016759, test accuracy: 0.773333\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.015795, test accuracy: 0.800000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.015117, test accuracy: 0.800000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.015398, test accuracy: 0.800000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.015206, test accuracy: 0.800000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.016459, test accuracy: 0.800000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017098, test accuracy: 0.800000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.015917, test accuracy: 0.800000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.014819, test accuracy: 0.800000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.012634, test accuracy: 0.800000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.017417, test accuracy: 0.813333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.019972, test accuracy: 0.813333\n",
      "{'001.Black_footed_Albatross', '085.Horned_Lark', '067.Anna_Hummingbird', '139.Scarlet_Tanager', '008.Rhinoceros_Auklet'}\n",
      "Episode 33: Test Accuracy: 0.8133\n",
      "\n",
      "Training Episode: 34...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.032705, test accuracy: 0.573333\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 0.752367, test accuracy: 0.773333\n",
      "epoch: 2, accuracy: 0.800000, avg. loss: 1.037650, test accuracy: 0.813333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.123412, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.046337, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.042002, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.035302, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.033283, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.027744, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.025444, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.021017, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.022472, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.025322, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.024215, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.026702, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.022762, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.022314, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.020357, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.022199, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.025917, test accuracy: 0.960000\n",
      "{'052.Pied_billed_Grebe', '196.House_Wren', '059.California_Gull', '044.Frigatebird', '053.Western_Grebe'}\n",
      "Episode 34: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 35...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.019478, test accuracy: 0.586667\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.450058, test accuracy: 0.853333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.161662, test accuracy: 0.853333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.071561, test accuracy: 0.906667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.049759, test accuracy: 0.893333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.037508, test accuracy: 0.893333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.046957, test accuracy: 0.893333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.031256, test accuracy: 0.906667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.039629, test accuracy: 0.906667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.030198, test accuracy: 0.906667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.029737, test accuracy: 0.906667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.024925, test accuracy: 0.906667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.022642, test accuracy: 0.906667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.026701, test accuracy: 0.906667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.027411, test accuracy: 0.906667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.024459, test accuracy: 0.906667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.025318, test accuracy: 0.906667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.019182, test accuracy: 0.906667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.022053, test accuracy: 0.906667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.028443, test accuracy: 0.906667\n",
      "{'191.Red_headed_Woodpecker', '063.Ivory_Gull', '004.Groove_billed_Ani', '104.American_Pipit', '092.Nighthawk'}\n",
      "Episode 35: Test Accuracy: 0.9067\n",
      "\n",
      "Training Episode: 36...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.333281, test accuracy: 0.546667\n",
      "epoch: 1, accuracy: 0.560000, avg. loss: 1.619120, test accuracy: 0.600000\n",
      "epoch: 2, accuracy: 0.680000, avg. loss: 0.827799, test accuracy: 0.906667\n",
      "epoch: 3, accuracy: 0.960000, avg. loss: 0.289333, test accuracy: 0.973333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.053219, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.041313, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.032243, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.026037, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.027266, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.022948, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 0.960000, avg. loss: 0.073950, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.021324, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.020820, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.030982, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.019775, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.023124, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.037813, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018016, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.036633, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021275, test accuracy: 0.986667\n",
      "{'149.Brown_Thrasher', '021.Eastern_Towhee', '032.Mangrove_Cuckoo', '069.Rufous_Hummingbird', '010.Red_winged_Blackbird'}\n",
      "Episode 36: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 37...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.160000, avg. loss: 2.418569, test accuracy: 0.440000\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.561835, test accuracy: 0.893333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.224353, test accuracy: 0.946667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.061016, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.042288, test accuracy: 0.946667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.041128, test accuracy: 0.946667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.035226, test accuracy: 0.946667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.029943, test accuracy: 0.946667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.033667, test accuracy: 0.946667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.028893, test accuracy: 0.946667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.024554, test accuracy: 0.946667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.020344, test accuracy: 0.946667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.024591, test accuracy: 0.946667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.022048, test accuracy: 0.946667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020933, test accuracy: 0.946667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.027385, test accuracy: 0.946667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.024205, test accuracy: 0.946667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.018070, test accuracy: 0.946667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019265, test accuracy: 0.946667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.022470, test accuracy: 0.946667\n",
      "{'101.White_Pelican', '022.Chuck_will_Widow', '132.White_crowned_Sparrow', '111.Loggerhead_Shrike', '074.Florida_Jay'}\n",
      "Episode 37: Test Accuracy: 0.9467\n",
      "\n",
      "Training Episode: 38...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.360000, avg. loss: 1.619406, test accuracy: 0.773333\n",
      "epoch: 1, accuracy: 0.960000, avg. loss: 0.268021, test accuracy: 0.906667\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.102406, test accuracy: 0.893333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.073277, test accuracy: 0.933333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.046439, test accuracy: 0.946667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.050326, test accuracy: 0.946667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.040774, test accuracy: 0.946667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.035722, test accuracy: 0.946667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.029287, test accuracy: 0.946667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.031036, test accuracy: 0.946667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.029022, test accuracy: 0.946667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.026380, test accuracy: 0.946667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.027062, test accuracy: 0.946667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.026580, test accuracy: 0.946667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.026176, test accuracy: 0.946667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.021573, test accuracy: 0.946667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.026235, test accuracy: 0.946667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.026014, test accuracy: 0.946667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.022062, test accuracy: 0.946667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021719, test accuracy: 0.946667\n",
      "{'001.Black_footed_Albatross', '133.White_throated_Sparrow', '093.Clark_Nutcracker', '127.Savannah_Sparrow', '177.Prothonotary_Warbler'}\n",
      "Episode 38: Test Accuracy: 0.9467\n",
      "\n",
      "Training Episode: 39...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.000000, avg. loss: 2.371350, test accuracy: 0.493333\n",
      "epoch: 1, accuracy: 0.680000, avg. loss: 0.751844, test accuracy: 0.986667\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.082490, test accuracy: 0.986667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.044407, test accuracy: 0.973333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.031587, test accuracy: 0.986667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.028525, test accuracy: 0.986667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.023988, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.020310, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.023692, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.020526, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.015370, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.018054, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.020524, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.015959, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.019159, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.015697, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.016585, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.017320, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.013611, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.014184, test accuracy: 0.986667\n",
      "{'145.Elegant_Tern', '067.Anna_Hummingbird', '131.Vesper_Sparrow', '106.Horned_Puffin', '087.Mallard'}\n",
      "Episode 39: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 40...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.471493, test accuracy: 0.280000\n",
      "epoch: 1, accuracy: 0.560000, avg. loss: 0.901152, test accuracy: 0.720000\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.384922, test accuracy: 0.800000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.116954, test accuracy: 0.853333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.040242, test accuracy: 0.866667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.037097, test accuracy: 0.880000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.034381, test accuracy: 0.880000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.025748, test accuracy: 0.880000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.027392, test accuracy: 0.880000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.025810, test accuracy: 0.880000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.034917, test accuracy: 0.880000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.046284, test accuracy: 0.880000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.026928, test accuracy: 0.880000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.025781, test accuracy: 0.880000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.021731, test accuracy: 0.880000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.024470, test accuracy: 0.880000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.020747, test accuracy: 0.880000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.023323, test accuracy: 0.880000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.021799, test accuracy: 0.880000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021032, test accuracy: 0.880000\n",
      "{'193.Bewick_Wren', '083.White_breasted_Kingfisher', '024.Red_faced_Cormorant', '158.Bay_breasted_Warbler', '162.Canada_Warbler'}\n",
      "Episode 40: Test Accuracy: 0.8800\n",
      "\n",
      "Training Episode: 41...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.320000, avg. loss: 1.658686, test accuracy: 0.666667\n",
      "epoch: 1, accuracy: 0.840000, avg. loss: 0.383115, test accuracy: 0.853333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.119913, test accuracy: 0.920000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.071665, test accuracy: 0.906667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.045086, test accuracy: 0.906667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.037614, test accuracy: 0.906667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.034549, test accuracy: 0.906667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.030873, test accuracy: 0.906667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.031706, test accuracy: 0.906667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.032429, test accuracy: 0.920000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.037180, test accuracy: 0.920000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.025592, test accuracy: 0.920000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.021172, test accuracy: 0.920000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.026189, test accuracy: 0.933333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.022324, test accuracy: 0.933333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.022843, test accuracy: 0.933333\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.022279, test accuracy: 0.933333\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.023699, test accuracy: 0.933333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.019741, test accuracy: 0.933333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.021371, test accuracy: 0.933333\n",
      "{'103.Sayornis', '058.Pigeon_Guillemot', '064.Ring_billed_Gull', '158.Bay_breasted_Warbler', '105.Whip_poor_Will'}\n",
      "Episode 41: Test Accuracy: 0.9333\n",
      "\n",
      "Training Episode: 42...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.366871, test accuracy: 0.426667\n",
      "epoch: 1, accuracy: 0.480000, avg. loss: 1.327059, test accuracy: 0.786667\n",
      "epoch: 2, accuracy: 0.880000, avg. loss: 0.317481, test accuracy: 0.960000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.067566, test accuracy: 0.986667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.035754, test accuracy: 0.986667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.030506, test accuracy: 0.986667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.022469, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.025347, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.024864, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.020419, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.026887, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.018072, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.018803, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.029217, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.020421, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.016865, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.018132, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.022576, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018711, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.037731, test accuracy: 0.986667\n",
      "{'116.Chipping_Sparrow', '027.Shiny_Cowbird', '191.Red_headed_Woodpecker', '120.Fox_Sparrow', '146.Forsters_Tern'}\n",
      "Episode 42: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 43...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 1.815521, test accuracy: 0.346667\n",
      "epoch: 1, accuracy: 0.600000, avg. loss: 0.890839, test accuracy: 0.773333\n",
      "epoch: 2, accuracy: 0.800000, avg. loss: 0.463932, test accuracy: 0.826667\n",
      "epoch: 3, accuracy: 0.960000, avg. loss: 0.135105, test accuracy: 0.960000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.027920, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.024675, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.017318, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.025891, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.025604, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.017068, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.020256, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.015777, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.019997, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.012822, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.015926, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.017423, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.015089, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.015793, test accuracy: 0.960000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018521, test accuracy: 0.960000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.015728, test accuracy: 0.960000\n",
      "{'186.Cedar_Waxwing', '006.Least_Auklet', '189.Red_bellied_Woodpecker', '175.Pine_Warbler', '181.Worm_eating_Warbler'}\n",
      "Episode 43: Test Accuracy: 0.9600\n",
      "\n",
      "Training Episode: 44...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.120000, avg. loss: 2.233518, test accuracy: 0.533333\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 0.766328, test accuracy: 0.760000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.192139, test accuracy: 0.906667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.105336, test accuracy: 0.906667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.057469, test accuracy: 0.920000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.047627, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.047360, test accuracy: 0.906667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.066229, test accuracy: 0.893333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.045536, test accuracy: 0.906667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.036442, test accuracy: 0.906667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.027517, test accuracy: 0.906667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.039395, test accuracy: 0.906667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.027925, test accuracy: 0.906667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.026057, test accuracy: 0.906667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.025208, test accuracy: 0.906667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.034865, test accuracy: 0.920000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.029155, test accuracy: 0.906667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.028976, test accuracy: 0.906667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.025190, test accuracy: 0.906667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.024909, test accuracy: 0.920000\n",
      "{'026.Bronzed_Cowbird', '014.Indigo_Bunting', '079.Belted_Kingfisher', '178.Swainson_Warbler', '174.Palm_Warbler'}\n",
      "Episode 44: Test Accuracy: 0.9200\n",
      "\n",
      "Training Episode: 45...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.246291, test accuracy: 0.773333\n",
      "epoch: 1, accuracy: 0.720000, avg. loss: 1.182023, test accuracy: 1.000000\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.096464, test accuracy: 0.986667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.051977, test accuracy: 0.986667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.044056, test accuracy: 0.986667\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.036044, test accuracy: 0.986667\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.035363, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.032235, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.025061, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.024178, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.020757, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.022066, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.023495, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.019960, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.021045, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.020198, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.036902, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.020647, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018846, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.019180, test accuracy: 0.986667\n",
      "{'186.Cedar_Waxwing', '049.Boat_tailed_Grackle', '117.Clay_colored_Sparrow', '180.Wilson_Warbler', '017.Cardinal'}\n",
      "Episode 45: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 46...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.440000, avg. loss: 1.509590, test accuracy: 0.520000\n",
      "epoch: 1, accuracy: 0.880000, avg. loss: 0.446377, test accuracy: 0.720000\n",
      "epoch: 2, accuracy: 0.840000, avg. loss: 0.267619, test accuracy: 0.853333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.122579, test accuracy: 0.920000\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.029951, test accuracy: 0.920000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.030328, test accuracy: 0.920000\n",
      "epoch: 6, accuracy: 0.960000, avg. loss: 0.074685, test accuracy: 0.920000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.022278, test accuracy: 0.920000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.020062, test accuracy: 0.920000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.019843, test accuracy: 0.920000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.020313, test accuracy: 0.920000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.021578, test accuracy: 0.920000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.019440, test accuracy: 0.920000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.019017, test accuracy: 0.920000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.016108, test accuracy: 0.920000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.020991, test accuracy: 0.920000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.017829, test accuracy: 0.920000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.024764, test accuracy: 0.920000\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.020719, test accuracy: 0.920000\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.017282, test accuracy: 0.920000\n",
      "{'168.Kentucky_Warbler', '129.Song_Sparrow', '023.Brandt_Cormorant', '163.Cape_May_Warbler', '056.Pine_Grosbeak'}\n",
      "Episode 46: Test Accuracy: 0.9200\n",
      "\n",
      "Training Episode: 47...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.120000, avg. loss: 2.215473, test accuracy: 0.386667\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 1.165450, test accuracy: 0.760000\n",
      "epoch: 2, accuracy: 0.960000, avg. loss: 0.265283, test accuracy: 0.973333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.056066, test accuracy: 0.973333\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.037979, test accuracy: 0.973333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.023101, test accuracy: 0.973333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.046560, test accuracy: 0.986667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.022194, test accuracy: 0.986667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.023596, test accuracy: 0.986667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.024634, test accuracy: 0.986667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.017458, test accuracy: 0.986667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.022648, test accuracy: 0.986667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.025927, test accuracy: 0.986667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.017251, test accuracy: 0.986667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.017896, test accuracy: 0.986667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.014832, test accuracy: 0.986667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.021222, test accuracy: 0.986667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.017378, test accuracy: 0.986667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.018405, test accuracy: 0.986667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.023011, test accuracy: 0.986667\n",
      "{'082.Ringed_Kingfisher', '148.Green_tailed_Towhee', '042.Vermilion_Flycatcher', '177.Prothonotary_Warbler', '010.Red_winged_Blackbird'}\n",
      "Episode 47: Test Accuracy: 0.9867\n",
      "\n",
      "Training Episode: 48...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.320000, avg. loss: 1.573931, test accuracy: 0.626667\n",
      "epoch: 1, accuracy: 0.760000, avg. loss: 0.582224, test accuracy: 0.893333\n",
      "epoch: 2, accuracy: 1.000000, avg. loss: 0.108573, test accuracy: 0.933333\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.081083, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.049244, test accuracy: 0.933333\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.035981, test accuracy: 0.933333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.031881, test accuracy: 0.946667\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.035951, test accuracy: 0.946667\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.031819, test accuracy: 0.946667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.025081, test accuracy: 0.946667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.030380, test accuracy: 0.946667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.031536, test accuracy: 0.946667\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.033228, test accuracy: 0.946667\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.024268, test accuracy: 0.946667\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.025379, test accuracy: 0.946667\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.029407, test accuracy: 0.946667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.023886, test accuracy: 0.946667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.031480, test accuracy: 0.946667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.022635, test accuracy: 0.946667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.030130, test accuracy: 0.946667\n",
      "{'088.Western_Meadowlark', '037.Acadian_Flycatcher', '072.Pomarine_Jaeger', '143.Caspian_Tern', '181.Worm_eating_Warbler'}\n",
      "Episode 48: Test Accuracy: 0.9467\n",
      "\n",
      "Training Episode: 49...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 1.852244, test accuracy: 0.426667\n",
      "epoch: 1, accuracy: 0.640000, avg. loss: 1.833521, test accuracy: 0.760000\n",
      "epoch: 2, accuracy: 0.680000, avg. loss: 0.809486, test accuracy: 0.840000\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.165196, test accuracy: 0.946667\n",
      "epoch: 4, accuracy: 0.960000, avg. loss: 0.120117, test accuracy: 0.960000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.020263, test accuracy: 0.960000\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.012816, test accuracy: 0.960000\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.015552, test accuracy: 0.960000\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.013790, test accuracy: 0.960000\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.018217, test accuracy: 0.960000\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.010151, test accuracy: 0.960000\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.011364, test accuracy: 0.960000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.014407, test accuracy: 0.960000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.012227, test accuracy: 0.960000\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.012978, test accuracy: 0.960000\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.012170, test accuracy: 0.960000\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.010142, test accuracy: 0.960000\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.013580, test accuracy: 0.973333\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.009179, test accuracy: 0.973333\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.008674, test accuracy: 0.973333\n",
      "{'084.Red_legged_Kittiwake', '150.Sage_Thrasher', '078.Gray_Kingbird', '157.Yellow_throated_Vireo', '200.Common_Yellowthroat'}\n",
      "Episode 49: Test Accuracy: 0.9733\n",
      "\n",
      "Training Episode: 50...\n",
      "{'train': 25, 'test': 75}\n",
      "epoch: 0, accuracy: 0.200000, avg. loss: 2.008113, test accuracy: 0.266667\n",
      "epoch: 1, accuracy: 0.680000, avg. loss: 1.002873, test accuracy: 0.466667\n",
      "epoch: 2, accuracy: 0.680000, avg. loss: 0.743475, test accuracy: 0.866667\n",
      "epoch: 3, accuracy: 1.000000, avg. loss: 0.139470, test accuracy: 0.826667\n",
      "epoch: 4, accuracy: 1.000000, avg. loss: 0.040448, test accuracy: 0.840000\n",
      "epoch: 5, accuracy: 1.000000, avg. loss: 0.036957, test accuracy: 0.853333\n",
      "epoch: 6, accuracy: 1.000000, avg. loss: 0.043143, test accuracy: 0.853333\n",
      "epoch: 7, accuracy: 1.000000, avg. loss: 0.023895, test accuracy: 0.853333\n",
      "epoch: 8, accuracy: 1.000000, avg. loss: 0.019426, test accuracy: 0.866667\n",
      "epoch: 9, accuracy: 1.000000, avg. loss: 0.022818, test accuracy: 0.866667\n",
      "epoch: 10, accuracy: 1.000000, avg. loss: 0.025562, test accuracy: 0.866667\n",
      "epoch: 11, accuracy: 1.000000, avg. loss: 0.019980, test accuracy: 0.880000\n",
      "epoch: 12, accuracy: 1.000000, avg. loss: 0.032202, test accuracy: 0.880000\n",
      "epoch: 13, accuracy: 1.000000, avg. loss: 0.079277, test accuracy: 0.893333\n",
      "epoch: 14, accuracy: 1.000000, avg. loss: 0.025480, test accuracy: 0.893333\n",
      "epoch: 15, accuracy: 1.000000, avg. loss: 0.019161, test accuracy: 0.906667\n",
      "epoch: 16, accuracy: 1.000000, avg. loss: 0.019804, test accuracy: 0.906667\n",
      "epoch: 17, accuracy: 1.000000, avg. loss: 0.019969, test accuracy: 0.906667\n",
      "epoch: 18, accuracy: 1.000000, avg. loss: 0.020112, test accuracy: 0.906667\n",
      "epoch: 19, accuracy: 1.000000, avg. loss: 0.023197, test accuracy: 0.906667\n",
      "{'041.Scissor_tailed_Flycatcher', '020.Yellow_breasted_Chat', '015.Lazuli_Bunting', '087.Mallard', '156.White_eyed_Vireo'}\n",
      "Episode 50: Test Accuracy: 0.9067\n",
      "\n",
      "Average Test Accuracy: 0.9333333333333332\n"
     ]
    }
   ],
   "source": [
    "print('Please wait patiently, it may take some seconds...')\n",
    "total_test_acc = 0.0\n",
    "\n",
    "for episode in range(EPISODES):\n",
    "    print(f\"\\nTraining Episode: {episode + 1}...\")\n",
    "    \n",
    "    loaded_model = torch.load(modelPath, map_location=device)\n",
    "    model = VisionTransformerTimm(64)\n",
    "    model.load_state_dict(loaded_model)\n",
    "    model.model.head = nn.Linear(model.model.head.in_features, 5).to(device)\n",
    "\n",
    "    for param in model.parameters():\n",
    "        param.requires_grad = True\n",
    "    \n",
    "    criterion = torch.nn.CrossEntropyLoss()\n",
    "    optimizer = torch.optim.SGD(params = model.parameters(), lr = LEARNING_RATE, momentum = MOMENTUM, weight_decay = WEIGHT_DECAY)\n",
    "    scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size = STEP_SIZE, gamma = GAMMA)\n",
    "    dataloaders, class_names, dataset_sizes = createEuroSatDataLoaders(data_transforms, dataDir, split=0.25, batch_size = BATCH_SIZE)\n",
    "    print(dataset_sizes)\n",
    "    \n",
    "    model, _, _ = train_fine_tuning(model, dataloaders, criterion, optimizer, scheduler, num_epochs = NUM_EPOCHS, learning_rate = LEARNING_RATE)\n",
    "    test_loader = dataloaders['test']\n",
    "    acc_test, test_loss = eval_method(net=model, data_loader=test_loader)\n",
    "    total_test_acc += acc_test\n",
    "    print(class_names)\n",
    "    print(f\"Episode {episode + 1}: Test Accuracy: {acc_test:.4f}\")\n",
    "\n",
    "average_test_acc = total_test_acc / EPISODES\n",
    "print(\"\\nAverage Test Accuracy:\", average_test_acc)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}