07.21_breath_sound.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "952e4415",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>filename</th>\n",
       "      <th>category</th>\n",
       "      <th>class</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0001-2.wav</td>\n",
       "      <td>healthy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0002-1.wav</td>\n",
       "      <td>healthy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0002-2.wav</td>\n",
       "      <td>healthy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0002-3.wav</td>\n",
       "      <td>healthy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0002-4.wav</td>\n",
       "      <td>healthy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>258</th>\n",
       "      <td>0614-2.wav</td>\n",
       "      <td>wheezing</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>259</th>\n",
       "      <td>0614-3.wav</td>\n",
       "      <td>wheezing</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>260</th>\n",
       "      <td>0614-4.wav</td>\n",
       "      <td>wheezing</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>261</th>\n",
       "      <td>0615-1.wav</td>\n",
       "      <td>wheezing</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>262</th>\n",
       "      <td>0615-2.wav</td>\n",
       "      <td>wheezing</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>263 rows × 3 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       filename  category  class\n",
       "0    0001-2.wav   healthy      0\n",
       "1    0002-1.wav   healthy      0\n",
       "2    0002-2.wav   healthy      0\n",
       "3    0002-3.wav   healthy      0\n",
       "4    0002-4.wav   healthy      0\n",
       "..          ...       ...    ...\n",
       "258  0614-2.wav  wheezing      1\n",
       "259  0614-3.wav  wheezing      1\n",
       "260  0614-4.wav  wheezing      1\n",
       "261  0615-1.wav  wheezing      1\n",
       "262  0615-2.wav  wheezing      1\n",
       "\n",
       "[263 rows x 3 columns]"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "from pathlib import Path\n",
    "\n",
    "Catholic_file = 'data/data2.csv'\n",
    "df = pd.read_csv(Catholic_file)\n",
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "f25ab30e",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>relative_path</th>\n",
       "      <th>class</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>/0001-2.wav</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>/0002-1.wav</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>/0002-2.wav</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>/0002-3.wav</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>/0002-4.wav</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  relative_path  class\n",
       "0   /0001-2.wav      0\n",
       "1   /0002-1.wav      0\n",
       "2   /0002-2.wav      0\n",
       "3   /0002-3.wav      0\n",
       "4   /0002-4.wav      0"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['relative_path'] = '/' + df['filename'].astype(str)\n",
    "df = df[['relative_path', 'class']]\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "926e2170",
   "metadata": {},
   "outputs": [],
   "source": [
    "import math, random\n",
    "import torch\n",
    "import torchaudio\n",
    "from torchaudio import transforms\n",
    "from IPython.display import Audio\n",
    "\n",
    "class Breath_sound_Util():\n",
    "  \n",
    "  def open(audio_file):\n",
    "    sig, sr = torchaudio.load(audio_file)\n",
    "    return (sig, sr)\n",
    "\n",
    "  def resample(aud, newsr):\n",
    "    sig, sr = aud\n",
    "\n",
    "    if (sr == newsr):\n",
    "     return aud\n",
    "\n",
    "    num_channels = sig.shape[0]\n",
    "    resig = torchaudio.transforms.Resample(sr, newsr)(sig[:1,:])\n",
    "    if (num_channels > 1):\n",
    "      retwo = torchaudio.transforms.Resample(sr, newsr)(sig[1:,:])\n",
    "      resig = torch.cat([resig, retwo])\n",
    "\n",
    "    return ((resig, newsr))\n",
    "\n",
    "  def pad(aud, max_ms):\n",
    "    sig, sr = aud\n",
    "    num_rows, sig_len = sig.shape\n",
    "    max_len = sr//1000 * max_ms\n",
    "   \n",
    "    if (sig_len > max_len):\n",
    "      sig = sig[:,:max_len]\n",
    " \n",
    "    elif (sig_len < max_len):\n",
    "\n",
    "      repeated = []\n",
    "      repeated.append(sig)\n",
    "      required_len = max_len - sig_len\n",
    "\n",
    "      while required_len > sig_len : \n",
    "        repeated.append(sig)\n",
    "        require_len -= sig_len\n",
    "      repeated.append(sig[:, :required_len])\n",
    " \n",
    "      sig = torch.cat(repeated, 1)\n",
    "\n",
    "    return (sig, sr)\n",
    "\n",
    "\n",
    "  def time_shift(aud, shift_limit):\n",
    "    sig,sr = aud\n",
    "    _, sig_len = sig.shape\n",
    "    shift_amt = int(random.random() * shift_limit * sig_len)\n",
    "    return (sig.roll(shift_amt), sr)\n",
    "\n",
    "  def spectro_gram(aud, n_mels=64, n_fft=1024, hop_len=None):\n",
    "    sig,sr = aud\n",
    "    aud = torchaudio.transforms.TimeStretch(hop_length=2048, fixed_rate=5000000)\n",
    "    top_db = 80\n",
    "    \n",
    "    spec = transforms.MelSpectrogram(sr, n_fft=n_fft, hop_length=hop_len, n_mels=n_mels)(sig)\n",
    "    spec = transforms.AmplitudeToDB(top_db=top_db)(spec)\n",
    "    return (spec)\n",
    "\n",
    "  def spectro_augment(spec, max_mask_pct=0.1, n_freq_masks=1, n_time_masks=1):\n",
    "    _, n_mels, n_steps = spec.shape\n",
    "    mask_value = spec.mean()\n",
    "    aug_spec = spec\n",
    "\n",
    "    freq_mask_param = max_mask_pct * n_mels\n",
    "    for _ in range(n_freq_masks):\n",
    "      aug_spec = transforms.FrequencyMasking(freq_mask_param)(aug_spec, mask_value)\n",
    "\n",
    "    time_mask_param = max_mask_pct * n_steps\n",
    "    for _ in range(n_time_masks):\n",
    "      aug_spec = transforms.TimeMasking(time_mask_param)(aug_spec, mask_value)\n",
    "\n",
    "    return aug_spec\n",
    "    print(aug_spec)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "07d02f06",
   "metadata": {},
   "source": [
    "#### (좌)최대길이에 맞춰 반복시켜 패딩 (우)최대길이에 맞춰 제로패딩"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "7f80eab9",
   "metadata": {},
   "source": [
    "![Screenshot from 2021-07-20 20-47-09](https://user-images.githubusercontent.com/74411831/126318715-e4d0ae11-38f9-42e2-9a2f-21282c7f674d.png)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "dd954efb",
   "metadata": {},
   "outputs": [],
   "source": [
    "data_path = 'data/all_data'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "162b2f4d",
   "metadata": {},
   "outputs": [],
   "source": [
    "from torch.utils.data import DataLoader, Dataset, random_split\n",
    "import torchaudio\n",
    "import torchvision\n",
    "\n",
    "class breathDS(Dataset):\n",
    "    \n",
    "  def __init__(self, df, data_path):\n",
    "    self.df = df\n",
    "    self.data_path = str(data_path)\n",
    "    self.duration = 4000\n",
    "    self.sr = 48000\n",
    "    self.shift_pct = 0.4\n",
    "            \n",
    "  def __len__(self):\n",
    "    return len(self.df)    \n",
    "    \n",
    "  def __getitem__(self, idx):\n",
    "    audio_file = self.data_path + self.df.loc[idx, 'relative_path']\n",
    "    class_id = self.df.loc[idx, 'class']\n",
    "    aud = Breath_sound_Util.open(audio_file)\n",
    "    reaud = Breath_sound_Util.resample(aud, self.sr)\n",
    "    dur_aud = Breath_sound_Util.pad(reaud, self.duration)\n",
    "    shift_aud = Breath_sound_Util.time_shift(dur_aud, self.shift_pct)\n",
    "    sgram = Breath_sound_Util.spectro_gram(shift_aud, n_mels=64, n_fft=1024, hop_len=None)\n",
    "    aug_sgram = Breath_sound_Util.spectro_augment(sgram, max_mask_pct=0.1, n_freq_masks=2, n_time_masks=2)\n",
    "\n",
    "    \n",
    "    return aug_sgram, class_id"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "fb573201",
   "metadata": {},
   "outputs": [],
   "source": [
    "from torch.utils.data import random_split\n",
    "\n",
    "brds = breathDS(df, data_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "a8f38daa",
   "metadata": {},
   "outputs": [],
   "source": [
    "num_items = len(brds)\n",
    "num_train = round(num_items * 0.8)\n",
    "num_val = num_items - num_train\n",
    "train_ds, val_ds = random_split(brds, [num_train, num_val])\n",
    "train_dl = torch.utils.data.DataLoader(train_ds, batch_size=16, shuffle=True)\n",
    "val_dl = torch.utils.data.DataLoader(val_ds, batch_size=16, shuffle=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "804962b3",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "device(type='cuda', index=0)"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import torch.nn as nn\n",
    "import torch.nn.functional as F\n",
    "from torch.nn import init\n",
    "\n",
    "# --------------------------------------------------------------\n",
    "# 호흡음의 Healthy, Wheezing을 판단하는 Binary Classification Model\n",
    "# --------------------------------------------------------------\n",
    "\n",
    "class WhoWheezing(nn.Module):\n",
    "\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "        conv_layers = []\n",
    "        self.conv1 = nn.Conv2d(1, 8, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2))\n",
    "        self.relu1 = nn.ReLU()\n",
    "        self.bn1 = nn.BatchNorm2d(8)\n",
    "        self.conv1.bias.data.zero_()\n",
    "        conv_layers += [self.conv1, self.relu1, self.bn1]\n",
    "\n",
    "        self.conv2 = nn.Conv2d(8, 16, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))\n",
    "        self.relu2 = nn.ReLU()\n",
    "        self.bn2 = nn.BatchNorm2d(16)\n",
    "        self.conv2.bias.data.zero_()\n",
    "        conv_layers += [self.conv2, self.relu2, self.bn2]\n",
    "\n",
    "        self.conv3 = nn.Conv2d(16, 32, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))\n",
    "        self.relu3 = nn.ReLU()\n",
    "        self.bn3 = nn.BatchNorm2d(32)\n",
    "        self.conv3.bias.data.zero_()\n",
    "        conv_layers += [self.conv3, self.relu3, self.bn3]\n",
    "\n",
    "        self.conv4 = nn.Conv2d(32, 64, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))\n",
    "        self.relu4 = nn.ReLU()\n",
    "        self.bn4 = nn.BatchNorm2d(64)\n",
    "        self.conv4.bias.data.zero_()\n",
    "        conv_layers += [self.conv4, self.relu4, self.bn4]\n",
    "\n",
    "        self.ap = nn.AdaptiveAvgPool2d(output_size=1)\n",
    "        self.lin = nn.Linear(in_features=64, out_features=2)\n",
    "        self.sigmoid = nn.Sigmoid()\n",
    "\n",
    "        self.conv = nn.Sequential(*conv_layers)\n",
    " \n",
    "    def forward(self, x):\n",
    "        x = self.conv(x)\n",
    "        x = self.ap(x)\n",
    "        x = x.view(x.shape[0], -1)\n",
    "        x = self.lin(x)\n",
    "        x = self.sigmoid(x)\n",
    "        return x\n",
    "\n",
    "Model1 = WhoWheezing()\n",
    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
    "Model1 = Model1.to(device)\n",
    "next(Model1.parameters()).device"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "3477e73a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch: 0, Loss: 0.6711, Accuracy: 0.7905\n",
      "Epoch: 1, Loss: 0.6222, Accuracy: 0.8143\n",
      "Epoch: 2, Loss: 0.5798, Accuracy: 0.8381\n",
      "Epoch: 3, Loss: 0.5534, Accuracy: 0.8333\n",
      "Epoch: 4, Loss: 0.5524, Accuracy: 0.7952\n",
      "Epoch: 5, Loss: 0.5210, Accuracy: 0.8429\n",
      "Epoch: 6, Loss: 0.5193, Accuracy: 0.8429\n",
      "Epoch: 7, Loss: 0.4949, Accuracy: 0.8286\n",
      "Epoch: 8, Loss: 0.4727, Accuracy: 0.8762\n",
      "Epoch: 9, Loss: 0.4933, Accuracy: 0.8810\n",
      "Epoch: 10, Loss: 0.4809, Accuracy: 0.8905\n",
      "Epoch: 11, Loss: 0.4645, Accuracy: 0.8571\n",
      "Epoch: 12, Loss: 0.4765, Accuracy: 0.8905\n",
      "Epoch: 13, Loss: 0.4916, Accuracy: 0.8762\n",
      "Epoch: 14, Loss: 0.4547, Accuracy: 0.8952\n",
      "Epoch: 15, Loss: 0.4409, Accuracy: 0.8810\n",
      "Epoch: 16, Loss: 0.4294, Accuracy: 0.9143\n",
      "Epoch: 17, Loss: 0.4389, Accuracy: 0.9095\n",
      "Epoch: 18, Loss: 0.4336, Accuracy: 0.9143\n",
      "Epoch: 19, Loss: 0.4232, Accuracy: 0.8952\n",
      "Epoch: 20, Loss: 0.4533, Accuracy: 0.9000\n",
      "Epoch: 21, Loss: 0.4436, Accuracy: 0.8952\n",
      "Epoch: 22, Loss: 0.4446, Accuracy: 0.9095\n",
      "Epoch: 23, Loss: 0.4335, Accuracy: 0.9000\n",
      "Epoch: 24, Loss: 0.4116, Accuracy: 0.9048\n",
      "Epoch: 25, Loss: 0.4085, Accuracy: 0.9048\n",
      "Epoch: 26, Loss: 0.4048, Accuracy: 0.9095\n",
      "Epoch: 27, Loss: 0.4482, Accuracy: 0.9000\n",
      "Epoch: 28, Loss: 0.4006, Accuracy: 0.9095\n",
      "Epoch: 29, Loss: 0.4278, Accuracy: 0.9000\n",
      "Epoch: 30, Loss: 0.4402, Accuracy: 0.8762\n",
      "Epoch: 31, Loss: 0.4294, Accuracy: 0.9048\n",
      "Epoch: 32, Loss: 0.4626, Accuracy: 0.8810\n",
      "Epoch: 33, Loss: 0.4313, Accuracy: 0.9000\n",
      "Epoch: 34, Loss: 0.4471, Accuracy: 0.9000\n",
      "Epoch: 35, Loss: 0.4039, Accuracy: 0.9048\n",
      "Epoch: 36, Loss: 0.4141, Accuracy: 0.9095\n",
      "Epoch: 37, Loss: 0.4307, Accuracy: 0.8952\n",
      "Epoch: 38, Loss: 0.4004, Accuracy: 0.9190\n",
      "Epoch: 39, Loss: 0.4135, Accuracy: 0.9095\n",
      "Epoch: 40, Loss: 0.4137, Accuracy: 0.9095\n",
      "Epoch: 41, Loss: 0.3968, Accuracy: 0.9143\n",
      "Epoch: 42, Loss: 0.4464, Accuracy: 0.8905\n",
      "Epoch: 43, Loss: 0.4460, Accuracy: 0.9000\n",
      "Epoch: 44, Loss: 0.4417, Accuracy: 0.9000\n",
      "Epoch: 45, Loss: 0.4415, Accuracy: 0.9048\n",
      "Epoch: 46, Loss: 0.4036, Accuracy: 0.9095\n",
      "Epoch: 47, Loss: 0.4022, Accuracy: 0.9095\n",
      "Epoch: 48, Loss: 0.4400, Accuracy: 0.9095\n",
      "Epoch: 49, Loss: 0.4288, Accuracy: 0.9095\n",
      "Epoch: 50, Loss: 0.4123, Accuracy: 0.9095\n",
      "Epoch: 51, Loss: 0.3969, Accuracy: 0.9190\n",
      "Epoch: 52, Loss: 0.4082, Accuracy: 0.9048\n",
      "Epoch: 53, Loss: 0.4129, Accuracy: 0.9095\n",
      "Epoch: 54, Loss: 0.4338, Accuracy: 0.9048\n",
      "Epoch: 55, Loss: 0.4238, Accuracy: 0.9095\n",
      "Epoch: 56, Loss: 0.4286, Accuracy: 0.9048\n",
      "Epoch: 57, Loss: 0.4086, Accuracy: 0.9095\n",
      "Epoch: 58, Loss: 0.4025, Accuracy: 0.9143\n",
      "Epoch: 59, Loss: 0.4384, Accuracy: 0.9143\n",
      "Epoch: 60, Loss: 0.4031, Accuracy: 0.9143\n",
      "Epoch: 61, Loss: 0.4275, Accuracy: 0.9190\n",
      "Epoch: 62, Loss: 0.4350, Accuracy: 0.9095\n",
      "Epoch: 63, Loss: 0.3960, Accuracy: 0.9095\n",
      "Epoch: 64, Loss: 0.3989, Accuracy: 0.9143\n",
      "Epoch: 65, Loss: 0.3948, Accuracy: 0.9190\n",
      "Epoch: 66, Loss: 0.4184, Accuracy: 0.9048\n",
      "Epoch: 67, Loss: 0.3918, Accuracy: 0.9143\n",
      "Epoch: 68, Loss: 0.4007, Accuracy: 0.9095\n",
      "Epoch: 69, Loss: 0.4265, Accuracy: 0.9143\n",
      "Epoch: 70, Loss: 0.3959, Accuracy: 0.9143\n",
      "Epoch: 71, Loss: 0.4466, Accuracy: 0.9048\n",
      "Epoch: 72, Loss: 0.3922, Accuracy: 0.9143\n",
      "Epoch: 73, Loss: 0.4036, Accuracy: 0.9143\n",
      "Epoch: 74, Loss: 0.3997, Accuracy: 0.9095\n",
      "Epoch: 75, Loss: 0.4094, Accuracy: 0.9143\n",
      "Epoch: 76, Loss: 0.4073, Accuracy: 0.9143\n",
      "Epoch: 77, Loss: 0.4371, Accuracy: 0.9143\n",
      "Epoch: 78, Loss: 0.4091, Accuracy: 0.9095\n",
      "Epoch: 79, Loss: 0.4151, Accuracy: 0.9048\n",
      "Epoch: 80, Loss: 0.4029, Accuracy: 0.9143\n",
      "Epoch: 81, Loss: 0.4277, Accuracy: 0.9095\n",
      "Epoch: 82, Loss: 0.4167, Accuracy: 0.9095\n",
      "Epoch: 83, Loss: 0.4033, Accuracy: 0.9143\n",
      "Epoch: 84, Loss: 0.4033, Accuracy: 0.9095\n",
      "Epoch: 85, Loss: 0.3921, Accuracy: 0.9190\n",
      "Epoch: 86, Loss: 0.4454, Accuracy: 0.9143\n",
      "Epoch: 87, Loss: 0.3930, Accuracy: 0.9190\n",
      "Epoch: 88, Loss: 0.3975, Accuracy: 0.9190\n",
      "Epoch: 89, Loss: 0.3891, Accuracy: 0.9190\n",
      "Epoch: 90, Loss: 0.4386, Accuracy: 0.9143\n",
      "Epoch: 91, Loss: 0.4494, Accuracy: 0.9048\n",
      "Epoch: 92, Loss: 0.3900, Accuracy: 0.9238\n",
      "Epoch: 93, Loss: 0.3977, Accuracy: 0.9190\n",
      "Epoch: 94, Loss: 0.3945, Accuracy: 0.9190\n",
      "Epoch: 95, Loss: 0.4005, Accuracy: 0.9143\n",
      "Epoch: 96, Loss: 0.3978, Accuracy: 0.9238\n",
      "Epoch: 97, Loss: 0.4249, Accuracy: 0.9190\n",
      "Epoch: 98, Loss: 0.4195, Accuracy: 0.9238\n",
      "Epoch: 99, Loss: 0.4348, Accuracy: 0.9190\n",
      "Finished Training\n"
     ]
    }
   ],
   "source": [
    "from sklearn.metrics import confusion_matrix\n",
    "import numpy as np\n",
    "import sklearn.metrics as metrics\n",
    "def training(model, train_dl, num_epochs):\n",
    "  criterion = nn.CrossEntropyLoss()\n",
    "  optimizer = torch.optim.Adam(model.parameters(),lr=0.001)\n",
    "  scheduler = torch.optim.lr_scheduler.OneCycleLR(optimizer, max_lr=0.001,\n",
    "                                                steps_per_epoch=int(len(train_dl)),\n",
    "                                                epochs=num_epochs,\n",
    "                                                anneal_strategy='linear')\n",
    "\n",
    "  for epoch in range(num_epochs):\n",
    "    running_loss = 0.0\n",
    "    correct_prediction = 0\n",
    "    total_prediction = 0\n",
    "\n",
    "    for i, data in enumerate(train_dl):\n",
    "        inputs, labels = data[0].to(device), data[1].to(device)\n",
    "        inputs_m, inputs_s = inputs.mean(), inputs.std()\n",
    "        inputs = (inputs - inputs_m) / inputs_s\n",
    "        optimizer.zero_grad()\n",
    "        outputs = model(inputs)\n",
    "        loss = criterion(outputs, labels)\n",
    "        loss.backward()\n",
    "        optimizer.step()\n",
    "        scheduler.step()\n",
    "        running_loss += loss.item()\n",
    "        _, prediction = torch.max(outputs,1)\n",
    "        correct_prediction += (prediction == labels).sum().item()\n",
    "        total_prediction += prediction.shape[0]\n",
    "\n",
    "    num_batches = len(train_dl)\n",
    "    avg_loss = running_loss / num_batches\n",
    "    acc = correct_prediction/total_prediction\n",
    "    print(f'Epoch: {epoch}, Loss: {avg_loss:.4f}, Accuracy: {acc:.4f}')\n",
    "\n",
    "  print('Finished Training')\n",
    "  \n",
    "num_epochs=100\n",
    "training(Model1, train_dl, num_epochs)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "74c94521",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 0 | Inputs [[[[ -7.3975425    5.007869     9.03249    ...  -2.9935834\n",
      "     -1.3400569  -15.080538  ]\n",
      "   [  0.6873282    1.7515504   10.396252   ...  -4.5290813\n",
      "      6.5511236   -3.7352905 ]\n",
      "   [ -0.5288943   -0.37542346   7.5962644  ...   0.65855396\n",
      "      7.384449    -7.735466  ]\n",
      "   ...\n",
      "   [-53.70318    -53.70318    -53.70318    ... -53.70318\n",
      "    -53.70318    -53.70318   ]\n",
      "   [-53.70318    -53.70318    -53.70318    ... -53.70318\n",
      "    -53.70318    -53.70318   ]\n",
      "   [-53.70318    -53.70318    -53.70318    ... -53.70318\n",
      "    -53.70318    -53.70318   ]]]\n",
      "\n",
      "\n",
      " [[[-12.698537   -16.158916   -21.610918   ... -39.639812\n",
      "    -33.077995   -21.931028  ]\n",
      "   [-12.99748    -22.309607   -27.015907   ... -35.536354\n",
      "    -36.839096   -27.541487  ]\n",
      "   [-15.517608   -34.834457   -36.397022   ... -33.983864\n",
      "    -32.449036   -26.76367   ]\n",
      "   ...\n",
      "   [-70.217224   -73.03935    -73.03935    ... -73.03935\n",
      "    -73.03935    -70.3933    ]\n",
      "   [-70.25016    -73.03935    -73.03935    ... -73.03935\n",
      "    -73.03935    -70.44217   ]\n",
      "   [-70.22058    -73.03935    -73.03935    ... -73.03935\n",
      "    -73.03935    -70.40161   ]]]\n",
      "\n",
      "\n",
      " [[[ -6.265468     2.4237454   -9.909275   ...   8.694908\n",
      "      9.125035     4.501435  ]\n",
      "   [-11.842156     2.6135828   -6.059274   ...  10.421913\n",
      "      6.6764545    3.627059  ]\n",
      "   [ -6.896884    -1.1081583   -7.900639   ...   6.982011\n",
      "      2.6443946   -1.6359001 ]\n",
      "   ...\n",
      "   [-43.12609    -43.12609    -43.12609    ... -43.12609\n",
      "    -43.12609    -43.12609   ]\n",
      "   [-43.12609    -43.12609    -43.12609    ... -43.12609\n",
      "    -43.12609    -43.12609   ]\n",
      "   [-43.12609    -43.12609    -43.12609    ... -43.12609\n",
      "    -43.12609    -43.12609   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  5.615943     6.6914554  -11.47643    ...   2.2374141\n",
      "      9.209687    -3.2239857 ]\n",
      "   [  0.56447786   0.53727716 -10.381094   ...  -3.6307049\n",
      "      7.014411    -2.1301656 ]\n",
      "   [ -5.7458076  -12.976804   -14.792371   ... -14.810526\n",
      "      4.16032     -3.1783628 ]\n",
      "   ...\n",
      "   [-66.74158    -66.74158    -66.74158    ... -66.74158\n",
      "    -66.74158    -66.74158   ]\n",
      "   [-66.74158    -66.74158    -66.74158    ... -66.74158\n",
      "    -66.74158    -66.74158   ]\n",
      "   [-66.74158    -66.74158    -66.74158    ... -66.74158\n",
      "    -66.74158    -66.74158   ]]]\n",
      "\n",
      "\n",
      " [[[-65.62885    -65.62885    -65.62885    ... -65.62885\n",
      "    -65.62885    -65.62885   ]\n",
      "   [-65.62885    -65.62885    -65.62885    ... -65.62885\n",
      "    -65.62885    -65.62885   ]\n",
      "   [-21.567707   -15.40509    -18.260426   ... -23.671822\n",
      "    -23.482332   -37.681908  ]\n",
      "   ...\n",
      "   [-81.21129    -81.21129    -81.21129    ... -81.21129\n",
      "    -81.21129    -81.21129   ]\n",
      "   [-81.21129    -81.21129    -81.21129    ... -81.21129\n",
      "    -81.21129    -81.21129   ]\n",
      "   [-81.21129    -81.21129    -81.21129    ... -81.21129\n",
      "    -81.21129    -81.21129   ]]]\n",
      "\n",
      "\n",
      " [[[-33.999146   -14.367752   -19.904707   ...  -7.625241\n",
      "    -17.12045    -39.052994  ]\n",
      "   [-37.417404   -16.83778    -24.80122    ... -12.69454\n",
      "    -23.417015   -24.19822   ]\n",
      "   [-23.538857   -20.743107   -22.609064   ... -21.716806\n",
      "    -37.61728    -20.537033  ]\n",
      "   ...\n",
      "   [-78.15124    -78.84859    -78.84859    ... -78.84859\n",
      "    -78.84859    -78.239334  ]\n",
      "   [-78.27666    -78.84859    -78.84859    ... -78.84859\n",
      "    -78.84859    -78.35646   ]\n",
      "   [-78.20459    -78.84859    -78.84859    ... -78.84859\n",
      "    -78.84859    -78.40573   ]]]] | Labels tensor([0, 0, 1, 0, 0, 0, 1, 0, 1, 1, 0, 1, 1, 0, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 1 | Inputs [[[[-12.702986   -7.2603726  -5.7886834 ... -18.132332  -33.576836\n",
      "     -9.197185 ]\n",
      "   [-14.215563  -10.415231   -8.743921  ... -22.586926  -27.882\n",
      "    -13.944378 ]\n",
      "   [-18.361431  -16.021645  -15.038525  ... -30.09679   -24.030378\n",
      "    -20.585726 ]\n",
      "   ...\n",
      "   [-69.93989   -69.93989   -69.93989   ... -69.93989   -69.93989\n",
      "    -69.93989  ]\n",
      "   [-69.93989   -69.93989   -69.93989   ... -69.93989   -69.93989\n",
      "    -69.93989  ]\n",
      "   [-69.93989   -69.93989   -69.93989   ... -69.93989   -69.93989\n",
      "    -69.93989  ]]]\n",
      "\n",
      "\n",
      " [[[-66.71853   -66.71853   -66.71853   ... -66.71853   -66.71853\n",
      "    -66.71853  ]\n",
      "   [-66.71853   -66.71853   -66.71853   ... -66.71853   -66.71853\n",
      "    -66.71853  ]\n",
      "   [-66.71853   -66.71853   -66.71853   ... -66.71853   -66.71853\n",
      "    -66.71853  ]\n",
      "   ...\n",
      "   [-82.79438   -82.79438   -82.79438   ... -82.79438   -82.79438\n",
      "    -82.79438  ]\n",
      "   [-82.79438   -82.79438   -82.79438   ... -82.79438   -82.79438\n",
      "    -82.79438  ]\n",
      "   [-82.79438   -82.79438   -82.79438   ... -82.79438   -82.79438\n",
      "    -82.79438  ]]]\n",
      "\n",
      "\n",
      " [[[  3.1550593  -0.7483888   2.66139   ...  -6.94746    -8.539922\n",
      "     -9.956426 ]\n",
      "   [ -2.0343056  -4.8818536   1.0204382 ...  -3.3062365  -7.250569\n",
      "     -6.8018727]\n",
      "   [-10.001774   -6.0050683  -2.4816773 ...  -1.2032208  -3.7103648\n",
      "     -3.0061598]\n",
      "   ...\n",
      "   [-47.402657  -47.096447  -46.741882  ... -47.355118  -48.39857\n",
      "    -48.241444 ]\n",
      "   [-47.468838  -48.58627   -45.625763  ... -46.61493   -48.6647\n",
      "    -45.528492 ]\n",
      "   [-45.773243  -50.245293  -49.122864  ... -48.2463    -46.91358\n",
      "    -47.730667 ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  1.3418316  -2.7876666  -5.3183246 ... -15.71349   -12.679968\n",
      "      3.5477355]\n",
      "   [  6.10896    -5.058526   -5.49282   ... -15.026197  -10.7560625\n",
      "      7.0730777]\n",
      "   [  5.1015368  -8.276676   -9.870783  ... -15.57879    -8.50783\n",
      "      6.7567725]\n",
      "   ...\n",
      "   [-48.448753  -47.160286  -46.544434  ... -48.21301   -47.44669\n",
      "    -47.707596 ]\n",
      "   [-46.98697   -44.46293   -46.355797  ... -46.634323  -45.91971\n",
      "    -49.268166 ]\n",
      "   [-46.93554   -44.410778  -47.517357  ... -46.51219   -46.743923\n",
      "    -48.048866 ]]]\n",
      "\n",
      "\n",
      " [[[-21.906473  -15.993167  -19.859241  ... -24.93367   -19.16999\n",
      "    -33.349007 ]\n",
      "   [-25.563238  -22.553024  -20.293018  ... -30.244654  -20.54508\n",
      "    -25.379875 ]\n",
      "   [-28.341698  -34.97253   -25.38378   ... -34.473217  -24.703875\n",
      "    -28.991692 ]\n",
      "   ...\n",
      "   [-81.38913   -81.38913   -81.38913   ... -81.38913   -81.38913\n",
      "    -81.38913  ]\n",
      "   [-81.38913   -81.38913   -81.38913   ... -81.38913   -81.38913\n",
      "    -81.38913  ]\n",
      "   [-81.38913   -81.38913   -81.38913   ... -81.38913   -81.38913\n",
      "    -81.38913  ]]]\n",
      "\n",
      "\n",
      " [[[-19.074636   -3.2038612  -6.9593196 ... -14.960331  -10.082049\n",
      "    -20.460941 ]\n",
      "   [-11.230565   -8.869671  -11.92954   ... -21.424368  -15.528227\n",
      "    -24.8646   ]\n",
      "   [-13.5589485 -19.738195  -20.703339  ... -28.957462  -25.61499\n",
      "    -32.485535 ]\n",
      "   ...\n",
      "   [-76.26867   -76.26867   -76.26867   ... -76.26867   -76.26867\n",
      "    -76.26867  ]\n",
      "   [-76.26867   -76.26867   -76.26867   ... -76.26867   -76.26867\n",
      "    -76.26867  ]\n",
      "   [-76.26867   -76.26867   -76.26867   ... -76.26867   -76.26867\n",
      "    -76.26867  ]]]] | Labels tensor([1, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 2 | Inputs [[[[ 3.7502214e-02 -3.3058257e+00 -5.4952675e-01 ... -3.0139956e-01\n",
      "    -4.9177580e+00 -3.3969278e+00]\n",
      "   [-5.0140839e+00 -8.9974755e-01  5.4205222e+00 ... -9.1363239e-01\n",
      "     5.0467622e-01  1.5951750e+00]\n",
      "   [-5.9472299e-03  6.7787628e+00  1.2789479e+01 ...  4.1491160e+00\n",
      "     1.0027552e+01  9.0135813e+00]\n",
      "   ...\n",
      "   [-4.4183990e+01 -4.4183990e+01 -4.4183990e+01 ... -4.4183990e+01\n",
      "    -4.4183990e+01 -4.4183990e+01]\n",
      "   [-4.4183990e+01 -4.4183990e+01 -4.4183990e+01 ... -4.4183990e+01\n",
      "    -4.4183990e+01 -4.4183990e+01]\n",
      "   [-4.4183990e+01 -4.4183990e+01 -4.4183990e+01 ... -4.4183990e+01\n",
      "    -4.4183990e+01 -4.4183990e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.7804552e+01 -1.6330896e+01 -3.1653687e+01 ... -1.1070454e+01\n",
      "    -1.2328651e+01 -1.5445896e+01]\n",
      "   [-2.4330990e+01 -2.0273460e+01 -2.5699471e+01 ... -1.4081632e+01\n",
      "    -1.8790316e+01 -1.9498646e+01]\n",
      "   [-4.1352715e+01 -2.7245499e+01 -2.8149405e+01 ... -2.0375008e+01\n",
      "    -3.0367008e+01 -2.5778887e+01]\n",
      "   ...\n",
      "   [-8.0805061e+01 -8.0805061e+01 -8.0805061e+01 ... -8.0805061e+01\n",
      "    -8.0805061e+01 -8.0805061e+01]\n",
      "   [-8.0805061e+01 -8.0805061e+01 -8.0805061e+01 ... -8.0805061e+01\n",
      "    -8.0805061e+01 -8.0805061e+01]\n",
      "   [-8.0805061e+01 -8.0805061e+01 -8.0805061e+01 ... -8.0805061e+01\n",
      "    -8.0805061e+01 -8.0805061e+01]]]\n",
      "\n",
      "\n",
      " [[[-4.3703980e+00 -1.2537691e+01 -1.0067202e+01 ... -7.1253176e+00\n",
      "    -6.8737226e+00 -4.9294201e+01]\n",
      "   [-7.6609879e+00 -1.8859051e+01 -1.5674958e+01 ... -1.2274318e+01\n",
      "    -1.1444996e+01 -1.1598474e+01]\n",
      "   [-9.1100769e+00 -3.1680470e+01 -2.6364773e+01 ... -2.1186640e+01\n",
      "    -1.9615051e+01 -8.9308090e+00]\n",
      "   ...\n",
      "   [-5.4928665e+01 -6.4063614e+01 -6.4063614e+01 ... -6.4063614e+01\n",
      "    -6.4063614e+01 -5.5419731e+01]\n",
      "   [-5.4976810e+01 -6.4063614e+01 -6.4063614e+01 ... -6.4063614e+01\n",
      "    -6.4063614e+01 -5.5460220e+01]\n",
      "   [-5.4919056e+01 -6.4063614e+01 -6.4063614e+01 ... -6.4063614e+01\n",
      "    -6.4063614e+01 -5.5391026e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-9.0260525e+00 -4.3846340e+00 -1.6762957e+01 ... -1.0313884e+01\n",
      "    -1.1123522e+01 -2.7383682e+01]\n",
      "   [-1.0922870e+01 -8.7924089e+00 -2.3381205e+01 ... -1.5225440e+01\n",
      "    -1.5061994e+01 -1.4421282e+01]\n",
      "   [-1.5450755e+01 -1.6456350e+01 -3.6396301e+01 ... -2.1734819e+01\n",
      "    -2.0720919e+01 -1.5041080e+01]\n",
      "   ...\n",
      "   [-7.2317490e+01 -7.2317490e+01 -7.2317490e+01 ... -7.2317490e+01\n",
      "    -7.2317490e+01 -7.2317490e+01]\n",
      "   [-7.2317490e+01 -7.2317490e+01 -7.2317490e+01 ... -7.2317490e+01\n",
      "    -7.2317490e+01 -7.2317490e+01]\n",
      "   [-7.2317490e+01 -7.2317490e+01 -7.2317490e+01 ... -7.2317490e+01\n",
      "    -7.2317490e+01 -7.2317490e+01]]]\n",
      "\n",
      "\n",
      " [[[ 1.9987925e+01  1.7795929e+01  3.4519982e+00 ...  2.2045226e+00\n",
      "     3.1256838e+00  5.5117216e+00]\n",
      "   [ 2.0832184e+01  1.8493824e+01  1.7361653e+00 ...  8.7438602e+00\n",
      "     8.5236330e+00  1.3711164e+01]\n",
      "   [ 1.6102245e+01  1.4508240e+01  5.1284738e+00 ...  8.3180704e+00\n",
      "     6.9943137e+00  1.1985387e+01]\n",
      "   ...\n",
      "   [-5.1669960e+01 -5.1669960e+01 -5.1669960e+01 ... -5.1669960e+01\n",
      "    -5.1669960e+01 -5.1669960e+01]\n",
      "   [-5.1669960e+01 -5.1669960e+01 -5.1669960e+01 ... -5.1669960e+01\n",
      "    -5.1669960e+01 -5.1669960e+01]\n",
      "   [-5.1669960e+01 -5.1669960e+01 -5.1669960e+01 ... -5.1669960e+01\n",
      "    -5.1669960e+01 -5.1669960e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.1663928e+01 -5.8151728e-01 -1.1977494e+00 ... -1.1318672e+01\n",
      "    -9.9570608e+00 -3.1098943e+00]\n",
      "   [-1.7512753e+01 -4.4133368e+00 -5.8004045e+00 ... -1.7645838e+01\n",
      "    -1.4573540e+01 -9.0543127e+00]\n",
      "   [-2.9253124e+01 -1.1376814e+01 -1.3659562e+01 ... -2.6997232e+01\n",
      "    -2.2050398e+01 -1.9779890e+01]\n",
      "   ...\n",
      "   [-7.4601395e+01 -7.4601395e+01 -7.4601395e+01 ... -7.4601395e+01\n",
      "    -7.4601395e+01 -7.4601395e+01]\n",
      "   [-7.4601395e+01 -7.4601395e+01 -7.4601395e+01 ... -7.4601395e+01\n",
      "    -7.4601395e+01 -7.4601395e+01]\n",
      "   [-7.4601395e+01 -7.4601395e+01 -7.4601395e+01 ... -7.4601395e+01\n",
      "    -7.4601395e+01 -7.4601395e+01]]]] | Labels tensor([1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 1, 1])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 3 | Inputs [[[[-14.144321   -35.493713   -21.159975   ... -17.696777\n",
      "    -20.136679    -9.423846  ]\n",
      "   [-12.86132    -36.59207    -27.278      ... -23.602736\n",
      "    -25.625366   -11.077969  ]\n",
      "   [-10.469681   -37.591473   -36.374516   ... -35.363533\n",
      "    -33.274708   -11.577783  ]\n",
      "   ...\n",
      "   [-70.042755   -70.042755   -70.042755   ... -70.042755\n",
      "    -70.042755   -70.042755  ]\n",
      "   [-70.042755   -70.042755   -70.042755   ... -70.042755\n",
      "    -70.042755   -70.042755  ]\n",
      "   [-70.042755   -70.042755   -70.042755   ... -70.042755\n",
      "    -70.042755   -70.042755  ]]]\n",
      "\n",
      "\n",
      " [[[ -9.950836   -11.390672   -21.806953   ...  -2.2714052\n",
      "     -2.5458221   -4.10437   ]\n",
      "   [-10.370586   -12.071228   -27.27462    ...  -5.8183537\n",
      "     -4.500327    -9.1861725 ]\n",
      "   [-14.199672   -16.11336    -27.031391   ... -12.506179\n",
      "    -10.028925   -16.99772   ]\n",
      "   ...\n",
      "   [-57.730156   -57.730156   -57.730156   ... -57.730156\n",
      "    -57.730156   -57.730156  ]\n",
      "   [-57.730156   -57.730156   -57.730156   ... -57.730156\n",
      "    -57.730156   -57.730156  ]\n",
      "   [-57.730156   -57.730156   -57.730156   ... -57.730156\n",
      "    -57.730156   -57.730156  ]]]\n",
      "\n",
      "\n",
      " [[[  3.2856555    9.22565      1.0611539  ...   1.6050242\n",
      "      9.937097     2.9729671 ]\n",
      "   [ -1.3749642   11.166786     3.2433896  ...   5.1837444\n",
      "     10.47666     -2.6230717 ]\n",
      "   [ -8.220156    12.692116     5.2895     ...   3.1424904\n",
      "      8.925601     6.401968  ]\n",
      "   ...\n",
      "   [-43.980305   -43.980305   -43.980305   ... -43.980305\n",
      "    -43.980305   -43.980305  ]\n",
      "   [-43.36243    -43.980305   -43.980305   ... -43.980305\n",
      "    -43.980305   -43.808624  ]\n",
      "   [-43.980305   -43.980305   -43.980305   ... -43.980305\n",
      "    -43.980305   -43.980305  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-10.678951    19.315508    27.578133   ...  26.157076\n",
      "     17.947748    11.631073  ]\n",
      "   [ 18.461899    14.971757    21.466772   ...  22.60292\n",
      "     19.974628    13.6016865 ]\n",
      "   [ 17.519674    11.094126    22.18613    ...  17.130888\n",
      "     16.269165    12.03143   ]\n",
      "   ...\n",
      "   [-30.57399    -30.57399    -30.57399    ... -30.57399\n",
      "    -30.57399    -30.57399   ]\n",
      "   [-30.57399    -30.57399    -30.57399    ... -30.57399\n",
      "    -30.57399    -30.57399   ]\n",
      "   [-30.57399    -30.57399    -30.57399    ... -30.57399\n",
      "    -30.57399    -30.57399   ]]]\n",
      "\n",
      "\n",
      " [[[  5.888687    -4.89018     -7.480956   ...  -2.3237588\n",
      "     -0.35269898   4.3498282 ]\n",
      "   [  4.1342835   -8.536467    -9.111145   ...  -0.5162857\n",
      "     -2.7963305    3.5774634 ]\n",
      "   [ -1.0007963   -4.866041    -8.059204   ...  -3.4436023\n",
      "     -4.1912827   -1.2407308 ]\n",
      "   ...\n",
      "   [-62.95185    -62.95185    -62.95185    ... -62.95185\n",
      "    -62.95185    -62.95185   ]\n",
      "   [-62.95185    -62.95185    -62.95185    ... -62.95185\n",
      "    -62.95185    -62.95185   ]\n",
      "   [-62.95185    -62.95185    -62.95185    ... -62.95185\n",
      "    -62.95185    -62.95185   ]]]\n",
      "\n",
      "\n",
      " [[[  4.043271     0.21465752   2.7932673  ...  -4.01606\n",
      "     -1.9812658   -3.295274  ]\n",
      "   [ -0.9892059   -5.862357     1.4577847  ...  -4.408725\n",
      "      1.9756355   -8.739704  ]\n",
      "   [-58.6386     -58.6386     -58.6386     ... -58.6386\n",
      "    -58.6386     -58.6386    ]\n",
      "   ...\n",
      "   [-75.956726   -75.956726   -75.956726   ... -75.956726\n",
      "    -75.956726   -75.956726  ]\n",
      "   [-75.956726   -75.956726   -75.956726   ... -75.956726\n",
      "    -75.956726   -75.956726  ]\n",
      "   [-75.956726   -75.956726   -75.956726   ... -75.956726\n",
      "    -75.956726   -75.956726  ]]]] | Labels tensor([0, 0, 1, 1, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 4 | Inputs [[[[  1.0590898   -3.9219186   -3.4696677  ...  -1.6799806\n",
      "      0.5282276   -1.1590769 ]\n",
      "   [ -1.9700022   -1.6427673   11.533743   ...  12.118874\n",
      "      8.921194     8.081469  ]\n",
      "   [  3.0911515    5.8904796   13.678684   ...  16.520555\n",
      "     12.516337    10.890476  ]\n",
      "   ...\n",
      "   [-47.992172   -49.79601    -49.475693   ... -48.759647\n",
      "    -49.15764    -46.096806  ]\n",
      "   [-45.257286   -47.455517   -47.895718   ... -49.353577\n",
      "    -47.96969    -45.691887  ]\n",
      "   [-47.986702   -48.179077   -48.160515   ... -49.49841\n",
      "    -47.985374   -47.33498   ]]]\n",
      "\n",
      "\n",
      " [[[-27.114088   -32.64339    -29.803865   ... -30.398981\n",
      "    -23.15131    -21.97775   ]\n",
      "   [-28.539305   -28.418858   -33.5151     ... -29.529118\n",
      "    -26.211277   -26.78528   ]\n",
      "   [-55.71938    -55.71938    -55.71938    ... -55.71938\n",
      "    -55.71938    -55.71938   ]\n",
      "   ...\n",
      "   [-63.165108   -63.165108   -63.165108   ... -63.165108\n",
      "    -63.165108   -63.165108  ]\n",
      "   [-63.165108   -63.165108   -63.165108   ... -63.165108\n",
      "    -63.165108   -63.165108  ]\n",
      "   [-63.165108   -63.165108   -63.165108   ... -63.165108\n",
      "    -63.165108   -63.165108  ]]]\n",
      "\n",
      "\n",
      " [[[-15.275363   -16.956875   -22.053185   ...  -6.9465237\n",
      "    -10.025319   -13.211418  ]\n",
      "   [-16.630833   -21.634132   -24.18641    ... -13.165581\n",
      "    -14.7233305  -19.623096  ]\n",
      "   [-21.789362   -28.860943   -25.429419   ... -26.894863\n",
      "    -22.699224   -28.503872  ]\n",
      "   ...\n",
      "   [-71.19869    -71.19869    -71.19869    ... -71.19869\n",
      "    -71.19869    -71.19869   ]\n",
      "   [-71.19869    -71.19869    -71.19869    ... -71.19869\n",
      "    -71.19869    -71.19869   ]\n",
      "   [-71.19869    -71.19869    -71.19869    ... -71.19869\n",
      "    -71.19869    -71.19869   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  2.2920368    6.8420167    0.40713575 ...   7.0278316\n",
      "      5.180749   -13.456289  ]\n",
      "   [  4.880439     9.191295    -2.0693464  ...   5.9019985\n",
      "      5.467076     3.4764419 ]\n",
      "   [  1.3764961    7.021986    -3.9736822  ...   0.84361696\n",
      "      2.8781333   -0.47336105]\n",
      "   ...\n",
      "   [-47.001926   -47.001926   -47.001926   ... -45.99817\n",
      "    -47.001926   -45.04047   ]\n",
      "   [-47.001926   -46.811554   -47.001926   ... -47.001926\n",
      "    -45.786785   -47.001926  ]\n",
      "   [-46.280056   -47.001926   -47.001926   ... -47.001926\n",
      "    -45.75155    -47.001926  ]]]\n",
      "\n",
      "\n",
      " [[[-13.0881605   -3.849354   -12.490469   ... -13.018997\n",
      "     -9.829923    -9.663649  ]\n",
      "   [-11.667689    -6.663497   -14.890293   ... -17.434767\n",
      "    -15.456973   -15.677572  ]\n",
      "   [-14.98642    -12.671176   -20.765175   ... -23.865105\n",
      "    -25.775047   -21.34777   ]\n",
      "   ...\n",
      "   [-67.2046     -67.2046     -67.2046     ... -67.2046\n",
      "    -67.2046     -67.2046    ]\n",
      "   [-67.2046     -67.2046     -67.2046     ... -67.2046\n",
      "    -67.2046     -67.2046    ]\n",
      "   [-67.2046     -67.2046     -67.2046     ... -67.2046\n",
      "    -67.2046     -67.2046    ]]]\n",
      "\n",
      "\n",
      " [[[-20.123575   -19.464024   -17.911047   ... -19.276985\n",
      "    -12.337366   -20.523594  ]\n",
      "   [-25.890398   -22.954262   -21.6385     ... -21.297375\n",
      "    -17.557484   -25.530386  ]\n",
      "   [-30.668161   -26.124924   -24.103985   ... -23.917196\n",
      "    -25.66224    -27.252138  ]\n",
      "   ...\n",
      "   [-80.557274   -80.557274   -80.557274   ... -80.557274\n",
      "    -80.557274   -80.557274  ]\n",
      "   [-80.557274   -80.557274   -80.557274   ... -80.557274\n",
      "    -80.557274   -80.557274  ]\n",
      "   [-80.557274   -80.557274   -80.557274   ... -80.557274\n",
      "    -80.557274   -80.557274  ]]]] | Labels tensor([1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 5 | Inputs [[[[-1.4884738e+01  1.1528646e+01  1.5727824e+01 ... -1.8265769e+00\n",
      "     1.0718242e+01  1.4686588e+01]\n",
      "   [-5.0268817e+00  1.5131358e+01  2.2203522e+01 ...  4.8840666e+00\n",
      "     6.9735055e+00  8.5156193e+00]\n",
      "   [-5.6853380e+00  1.3970729e+01  2.2580170e+01 ...  7.9579539e+00\n",
      "     4.8004432e+00  9.7183161e+00]\n",
      "   ...\n",
      "   [-4.6727505e+01 -4.9215565e+01 -4.6082973e+01 ... -4.8259659e+01\n",
      "    -4.6744839e+01 -4.7787781e+01]\n",
      "   [-4.6630951e+01 -4.7883789e+01 -4.6878696e+01 ... -4.7275215e+01\n",
      "    -4.5871529e+01 -4.6403740e+01]\n",
      "   [-4.7185379e+01 -4.7216705e+01 -4.8962978e+01 ... -4.7938217e+01\n",
      "    -4.7391354e+01 -4.7038502e+01]]]\n",
      "\n",
      "\n",
      " [[[-2.9000654e+01 -1.6595108e+01 -8.5672874e+00 ... -4.1432053e+01\n",
      "    -1.9708101e+01 -2.3144571e+01]\n",
      "   [-3.4323792e+01 -2.0008102e+01 -1.3694601e+01 ... -3.3970596e+01\n",
      "    -2.6064152e+01 -2.7917519e+01]\n",
      "   [-3.8115822e+01 -2.6540649e+01 -2.2963303e+01 ... -3.0463318e+01\n",
      "    -3.6554672e+01 -3.4377743e+01]\n",
      "   ...\n",
      "   [-8.7472122e+01 -8.7472122e+01 -8.7472122e+01 ... -8.7472122e+01\n",
      "    -8.7472122e+01 -8.7472122e+01]\n",
      "   [-8.7472122e+01 -8.7472122e+01 -8.7472122e+01 ... -8.7472122e+01\n",
      "    -8.7472122e+01 -8.7472122e+01]\n",
      "   [-8.7472122e+01 -8.7472122e+01 -8.7472122e+01 ... -8.7472122e+01\n",
      "    -8.7472122e+01 -8.7472122e+01]]]\n",
      "\n",
      "\n",
      " [[[ 1.9461551e+00  6.1605034e+00  2.3465607e-01 ... -7.1708336e+00\n",
      "    -6.6564858e-01  3.8867843e+00]\n",
      "   [-3.0157742e-01  4.3868618e+00  1.6077895e+00 ... -8.5303459e+00\n",
      "     1.1489867e+00  3.5224667e+00]\n",
      "   [ 3.6525550e+00  4.6898606e-01 -2.1903398e+00 ... -9.2257032e+00\n",
      "     1.1186678e+00 -8.9402497e-01]\n",
      "   ...\n",
      "   [-4.3277824e+01 -4.3277824e+01 -4.3277824e+01 ... -4.3277824e+01\n",
      "    -4.3277824e+01 -4.3277824e+01]\n",
      "   [-4.3277824e+01 -4.3277824e+01 -4.3277824e+01 ... -4.3277824e+01\n",
      "    -4.3277824e+01 -4.3277824e+01]\n",
      "   [-4.3277824e+01 -4.3277824e+01 -4.3277824e+01 ... -4.3277824e+01\n",
      "    -4.3277824e+01 -4.3277824e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ 9.7415657e+00  6.4681787e+00  1.6385319e+00 ...  3.4674883e+00\n",
      "     5.1825953e+00  9.2594357e+00]\n",
      "   [ 6.8195667e+00  1.0547724e+01  7.0315928e+00 ...  4.6811938e+00\n",
      "     3.3047414e+00  1.2444813e+01]\n",
      "   [ 2.8962638e+00  1.0563335e+01  4.2785559e+00 ...  6.2844362e+00\n",
      "     2.9324234e-01  1.0116559e+01]\n",
      "   ...\n",
      "   [-4.8365131e+01 -4.7716335e+01 -4.7448811e+01 ... -4.8585007e+01\n",
      "    -4.8764709e+01 -4.9817905e+01]\n",
      "   [-4.6760277e+01 -4.7007538e+01 -4.6160221e+01 ... -4.6669380e+01\n",
      "    -4.7620361e+01 -4.7955566e+01]\n",
      "   [-4.7627464e+01 -4.8257614e+01 -4.7856819e+01 ... -4.8534412e+01\n",
      "    -4.9684120e+01 -5.0645618e+01]]]\n",
      "\n",
      "\n",
      " [[[-7.5328941e+00 -7.7933173e+00 -3.1511591e+01 ...  2.9470327e+00\n",
      "    -1.0696254e+01 -1.7600504e+01]\n",
      "   [-1.3961994e+01 -8.8490191e+00 -2.0426085e+01 ...  6.6960163e-02\n",
      "    -9.1523409e+00 -2.3761442e+01]\n",
      "   [-2.2028687e+01 -1.2528342e+01 -2.2484585e+01 ... -5.2318401e+00\n",
      "    -1.1909189e+01 -2.4119831e+01]\n",
      "   ...\n",
      "   [-4.3829842e+01 -4.3829842e+01 -4.3829842e+01 ... -4.3829842e+01\n",
      "    -4.3829842e+01 -4.3829842e+01]\n",
      "   [-4.3829842e+01 -4.3829842e+01 -4.3829842e+01 ... -4.3829842e+01\n",
      "    -4.3829842e+01 -4.3829842e+01]\n",
      "   [-4.3829842e+01 -4.3829842e+01 -4.3829842e+01 ... -4.3829842e+01\n",
      "    -4.3829842e+01 -4.3829842e+01]]]\n",
      "\n",
      "\n",
      " [[[-3.5502243e+01 -3.6837231e+01 -4.1018433e+01 ... -3.5078056e+01\n",
      "    -4.5795563e+01 -4.7854256e+01]\n",
      "   [-3.7529888e+01 -2.7778439e+01 -2.8011320e+01 ... -2.9731474e+01\n",
      "    -2.9970623e+01 -4.2280636e+01]\n",
      "   [-2.6936739e+01 -2.8826170e+01 -2.7192757e+01 ... -3.0821209e+01\n",
      "    -3.2282936e+01 -2.9518082e+01]\n",
      "   ...\n",
      "   [-7.5355324e+01 -7.5355324e+01 -7.5355324e+01 ... -7.5355324e+01\n",
      "    -7.5355324e+01 -7.5355324e+01]\n",
      "   [-7.5355324e+01 -7.5355324e+01 -7.5355324e+01 ... -7.5355324e+01\n",
      "    -7.5355324e+01 -7.5355324e+01]\n",
      "   [-7.5355324e+01 -7.5355324e+01 -7.5355324e+01 ... -7.5355324e+01\n",
      "    -7.5355324e+01 -7.5355324e+01]]]] | Labels tensor([0, 0, 1, 1, 0, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1, 0])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 6 | Inputs [[[[-16.670235   -15.631141   -31.796772   ... -18.430656\n",
      "    -15.709898   -23.07064   ]\n",
      "   [-20.771238   -19.006235   -30.929188   ... -22.536129\n",
      "    -18.549042   -27.42025   ]\n",
      "   [-24.388437   -25.60008    -34.068375   ... -28.4525\n",
      "    -24.069956   -35.01221   ]\n",
      "   ...\n",
      "   [-67.260056   -67.260056   -67.260056   ... -67.260056\n",
      "    -67.260056   -67.260056  ]\n",
      "   [-67.260056   -67.260056   -67.260056   ... -67.260056\n",
      "    -67.260056   -67.260056  ]\n",
      "   [-67.260056   -67.260056   -67.260056   ... -67.260056\n",
      "    -67.260056   -67.260056  ]]]\n",
      "\n",
      "\n",
      " [[[-13.612261   -19.054752   -14.602523   ... -16.310251\n",
      "    -18.008183   -29.053978  ]\n",
      "   [-17.093657   -24.466162   -19.389954   ... -22.653017\n",
      "    -22.795202   -19.34468   ]\n",
      "   [-23.882675   -17.422663   -23.33836    ... -30.355713\n",
      "    -30.740227   -18.378847  ]\n",
      "   ...\n",
      "   [-81.01962    -82.36636    -82.36636    ... -82.36636\n",
      "    -82.36636    -76.29968   ]\n",
      "   [-81.08423    -82.36636    -82.36636    ... -82.36636\n",
      "    -82.36636    -76.378075  ]\n",
      "   [-80.89576    -82.36636    -82.36636    ... -82.36636\n",
      "    -82.36636    -76.30496   ]]]\n",
      "\n",
      "\n",
      " [[[ 25.195057    19.217184    -0.11527676 ...   1.29894\n",
      "     15.506459    22.067757  ]\n",
      "   [ 21.352066    18.691477    17.350466   ...  17.662838\n",
      "     22.344028    21.473646  ]\n",
      "   [-33.65107    -33.65107    -33.65107    ... -33.65107\n",
      "    -33.65107    -33.65107   ]\n",
      "   ...\n",
      "   [-43.201714   -46.1946     -46.1946     ... -46.1946\n",
      "    -46.1946     -42.822056  ]\n",
      "   [-41.260128   -45.976616   -45.839355   ... -46.1946\n",
      "    -46.1946     -41.941353  ]\n",
      "   [-43.07775    -46.1946     -46.1946     ... -46.1946\n",
      "    -46.1946     -42.32421   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ 11.073944    11.926144     8.508927   ...  15.49117\n",
      "      9.979454   -16.399017  ]\n",
      "   [  4.672908     6.933091     4.625796   ...   9.764328\n",
      "      5.8020215    0.6029346 ]\n",
      "   [  0.5042831   -0.23435129   3.2987282  ...   2.6901996\n",
      "     -1.4179633    2.3859718 ]\n",
      "   ...\n",
      "   [-31.874958   -31.874958   -31.874958   ... -31.874958\n",
      "    -31.874958   -31.874958  ]\n",
      "   [-31.874958   -31.874958   -31.874958   ... -31.874958\n",
      "    -31.874958   -31.874958  ]\n",
      "   [-31.874958   -31.874958   -31.874958   ... -31.874958\n",
      "    -31.874958   -31.874958  ]]]\n",
      "\n",
      "\n",
      " [[[ -6.4568396   -7.7318306  -21.083199   ... -12.999311\n",
      "     -9.524988    -3.8055553 ]\n",
      "   [ -9.715243   -12.501112   -20.449495   ... -14.076149\n",
      "    -12.368755   -10.219727  ]\n",
      "   [-16.329174   -18.484028   -23.23844    ... -18.212723\n",
      "    -18.050179   -20.43686   ]\n",
      "   ...\n",
      "   [-71.526276   -71.526276   -71.526276   ... -71.526276\n",
      "    -71.526276   -71.526276  ]\n",
      "   [-71.526276   -71.526276   -71.526276   ... -71.526276\n",
      "    -71.526276   -71.526276  ]\n",
      "   [-71.526276   -71.526276   -71.526276   ... -71.526276\n",
      "    -71.526276   -71.526276  ]]]\n",
      "\n",
      "\n",
      " [[[-58.52097    -58.52097    -58.52097    ... -58.52097\n",
      "    -58.52097    -58.52097   ]\n",
      "   [-58.52097    -58.52097    -58.52097    ... -58.52097\n",
      "    -58.52097    -58.52097   ]\n",
      "   [-58.52097    -58.52097    -58.52097    ... -58.52097\n",
      "    -58.52097    -58.52097   ]\n",
      "   ...\n",
      "   [-64.97543    -64.97543    -64.97543    ... -64.97543\n",
      "    -64.97543    -64.97543   ]\n",
      "   [-64.97543    -64.97543    -64.97543    ... -64.97543\n",
      "    -64.97543    -64.97543   ]\n",
      "   [-64.97543    -64.97543    -64.97543    ... -64.97543\n",
      "    -64.97543    -64.97543   ]]]] | Labels tensor([0, 0, 1, 0, 0, 0, 0, 1, 0, 1, 1, 1, 0, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 7 | Inputs [[[[-10.303779    -6.4830313   -3.432069   ...   3.3850155\n",
      "      0.86169773  -7.1828184 ]\n",
      "   [ -9.558815   -10.2128105   -8.729096   ...   0.8402281\n",
      "     -2.5637271   -6.4782624 ]\n",
      "   [-13.503817   -17.053913   -18.360611   ...  -5.1959467\n",
      "     -9.332502   -11.304834  ]\n",
      "   ...\n",
      "   [-74.60367    -74.60367    -74.60367    ... -74.60367\n",
      "    -74.60367    -74.60367   ]\n",
      "   [-74.60367    -74.60367    -74.60367    ... -74.60367\n",
      "    -74.60367    -74.60367   ]\n",
      "   [-74.60367    -74.60367    -74.60367    ... -74.60367\n",
      "    -74.60367    -74.60367   ]]]\n",
      "\n",
      "\n",
      " [[[ -6.014397    -9.624821     1.3117551  ... -13.780468\n",
      "     -5.1384454   -2.9425547 ]\n",
      "   [ -7.332919    -6.8189836   -0.76052475 ... -20.22477\n",
      "     -9.3641405   -9.576916  ]\n",
      "   [-12.654257   -10.5158615   -5.7038984  ... -14.904238\n",
      "    -16.350039   -22.920048  ]\n",
      "   ...\n",
      "   [-62.903442   -62.903442   -62.903442   ... -62.903442\n",
      "    -62.903442   -62.903442  ]\n",
      "   [-62.903442   -62.903442   -62.903442   ... -62.903442\n",
      "    -62.903442   -62.903442  ]\n",
      "   [-62.903442   -62.903442   -62.903442   ... -62.903442\n",
      "    -62.903442   -62.903442  ]]]\n",
      "\n",
      "\n",
      " [[[-27.449398   -28.756807   -30.052658   ... -25.075071\n",
      "    -26.357052   -27.549303  ]\n",
      "   [-34.080963   -35.390755   -36.67726    ... -31.711342\n",
      "    -32.9876     -34.166595  ]\n",
      "   [-57.66629    -57.66629    -57.66629    ... -57.66629\n",
      "    -57.66629    -57.66629   ]\n",
      "   ...\n",
      "   [-57.66629    -57.66629    -57.66629    ... -57.66629\n",
      "    -57.66629    -57.66629   ]\n",
      "   [-57.66629    -57.66629    -57.66629    ... -57.66629\n",
      "    -57.66629    -57.66629   ]\n",
      "   [-57.66629    -57.66629    -57.66629    ... -57.66629\n",
      "    -57.66629    -57.66629   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-22.64238    -19.703667   -20.505888   ... -15.018144\n",
      "    -18.71125    -17.659788  ]\n",
      "   [-29.157984   -23.927887   -24.275433   ... -19.851873\n",
      "    -22.717531   -23.590471  ]\n",
      "   [-37.199535   -31.4779     -29.694641   ... -27.282297\n",
      "    -29.157057   -32.46114   ]\n",
      "   ...\n",
      "   [-77.99632    -77.99632    -77.99632    ... -77.99632\n",
      "    -77.99632    -77.99632   ]\n",
      "   [-77.99632    -77.99632    -77.99632    ... -77.99632\n",
      "    -77.99632    -77.99632   ]\n",
      "   [-77.99632    -77.99632    -77.99632    ... -77.99632\n",
      "    -77.99632    -77.99632   ]]]\n",
      "\n",
      "\n",
      " [[[-23.194935   -17.84037    -22.424591   ...  -8.575464\n",
      "     -3.2737975  -10.9792185 ]\n",
      "   [-29.221733   -17.74589    -21.209087   ...  -7.703474\n",
      "     -5.8008046  -15.493349  ]\n",
      "   [-37.02221    -21.564438   -25.640781   ... -10.308395\n",
      "    -11.581336   -16.956556  ]\n",
      "   ...\n",
      "   [-82.07622    -82.07622    -82.07622    ... -82.07622\n",
      "    -82.07622    -82.07622   ]\n",
      "   [-82.07622    -82.07622    -82.07622    ... -82.07622\n",
      "    -82.07622    -82.07622   ]\n",
      "   [-82.07622    -82.07622    -82.07622    ... -82.07622\n",
      "    -82.07622    -82.07622   ]]]\n",
      "\n",
      "\n",
      " [[[-19.491049   -19.10981    -16.371288   ... -26.882162\n",
      "    -18.076393   -13.850548  ]\n",
      "   [-25.695946   -24.641685   -21.284218   ... -27.938377\n",
      "    -22.587374   -19.793758  ]\n",
      "   [-39.851242   -26.885044   -26.582815   ... -33.13813\n",
      "    -27.464142   -30.372856  ]\n",
      "   ...\n",
      "   [-71.829025   -71.829025   -71.829025   ... -71.829025\n",
      "    -71.829025   -71.829025  ]\n",
      "   [-71.829025   -71.829025   -71.829025   ... -71.829025\n",
      "    -71.829025   -71.829025  ]\n",
      "   [-71.829025   -71.829025   -71.829025   ... -71.829025\n",
      "    -71.829025   -71.829025  ]]]] | Labels tensor([0, 0, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 8 | Inputs [[[[-5.93724728e+00 -3.19382820e+01 -2.84446621e+01 ... -1.87272587e+01\n",
      "    -1.29670477e+01 -7.77460766e+00]\n",
      "   [-5.13539219e+00 -3.25628815e+01 -2.57295532e+01 ... -2.41699371e+01\n",
      "    -1.80005722e+01 -4.04702139e+00]\n",
      "   [-4.15896463e+00 -3.57859306e+01 -2.66112995e+01 ... -2.99251347e+01\n",
      "    -2.65256233e+01 -3.36561227e+00]\n",
      "   ...\n",
      "   [-5.40535316e+01 -6.38074722e+01 -6.38074722e+01 ... -6.38074722e+01\n",
      "    -6.38074722e+01 -5.19928894e+01]\n",
      "   [-5.40207710e+01 -6.38074722e+01 -6.38074722e+01 ... -6.38074722e+01\n",
      "    -6.38074722e+01 -5.20240402e+01]\n",
      "   [-5.39192772e+01 -6.38074722e+01 -6.38074722e+01 ... -6.38074722e+01\n",
      "    -6.38074722e+01 -5.19586105e+01]]]\n",
      "\n",
      "\n",
      " [[[-8.61477947e+00 -3.70351613e-01 -2.45803371e-02 ...  5.23485088e+00\n",
      "    -8.24832797e-01 -1.58605328e+01]\n",
      "   [-1.48879347e+01 -4.62825489e+00 -4.36210585e+00 ...  2.63725638e+00\n",
      "    -2.67940187e+00 -1.63277721e+01]\n",
      "   [-2.24325581e+01 -1.22723675e+01 -1.21369362e+01 ... -3.35392737e+00\n",
      "    -7.82219172e+00 -2.13372021e+01]\n",
      "   ...\n",
      "   [-6.96840286e+01 -6.96840286e+01 -6.96840286e+01 ... -6.96840286e+01\n",
      "    -6.96840286e+01 -6.96840286e+01]\n",
      "   [-6.96840286e+01 -6.96840286e+01 -6.96840286e+01 ... -6.96840286e+01\n",
      "    -6.96840286e+01 -6.96840286e+01]\n",
      "   [-6.96840286e+01 -6.96840286e+01 -6.96840286e+01 ... -6.96840286e+01\n",
      "    -6.96840286e+01 -6.96840286e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.21469679e+01 -2.08690224e+01 -2.34425201e+01 ... -1.99315987e+01\n",
      "    -2.38697205e+01 -8.91072845e+00]\n",
      "   [-9.40763092e+00 -2.69275951e+01 -2.94933529e+01 ... -2.37983799e+01\n",
      "    -2.93788052e+01 -9.28503895e+00]\n",
      "   [-8.75267410e+00 -3.29622421e+01 -3.79202614e+01 ... -3.05424728e+01\n",
      "    -3.56273270e+01 -9.04387760e+00]\n",
      "   ...\n",
      "   [-5.60093384e+01 -6.43656387e+01 -6.43656387e+01 ... -6.43656387e+01\n",
      "    -6.43656387e+01 -5.37157059e+01]\n",
      "   [-5.59352493e+01 -6.43656387e+01 -6.43656387e+01 ... -6.43656387e+01\n",
      "    -6.43656387e+01 -5.37359428e+01]\n",
      "   [-5.58110199e+01 -6.43656387e+01 -6.43656387e+01 ... -6.43656387e+01\n",
      "    -6.43656387e+01 -5.36680603e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ 3.34873581e+01  2.93856411e+01  2.94057293e+01 ...  4.07926941e+01\n",
      "     3.53640900e+01  4.36884651e+01]\n",
      "   [ 3.26919556e+01  3.15648098e+01  2.74877071e+01 ...  4.02523117e+01\n",
      "     3.50656166e+01  3.71661797e+01]\n",
      "   [ 3.42585030e+01  3.32283096e+01  2.57183647e+01 ...  3.73167953e+01\n",
      "     3.00954800e+01  3.28303108e+01]\n",
      "   ...\n",
      "   [-3.06753540e+01 -3.59231262e+01 -3.59231262e+01 ... -3.59231262e+01\n",
      "    -3.59231262e+01 -3.21112823e+01]\n",
      "   [-3.27361794e+01 -3.59231262e+01 -3.59231262e+01 ... -3.59231262e+01\n",
      "    -3.59231262e+01 -3.00725193e+01]\n",
      "   [-3.26178665e+01 -3.59231262e+01 -3.59231262e+01 ... -3.59231262e+01\n",
      "    -3.59231262e+01 -3.13166885e+01]]]\n",
      "\n",
      "\n",
      " [[[-2.33773327e+00  8.14503002e+00  1.66570683e+01 ... -3.38540375e-01\n",
      "     8.32655907e+00  1.28848562e+01]\n",
      "   [-7.57477665e+00  1.05792141e+01  1.57237444e+01 ...  6.59178352e+00\n",
      "     2.85620689e+00  6.36220837e+00]\n",
      "   [-1.24171124e+01  8.66968155e+00  1.11172523e+01 ...  4.98786497e+00\n",
      "     6.41091776e+00  6.60865116e+00]\n",
      "   ...\n",
      "   [-4.52106934e+01 -4.75004578e+01 -4.59696388e+01 ... -4.49130707e+01\n",
      "    -4.69098663e+01 -4.47083664e+01]\n",
      "   [-4.55804634e+01 -4.55315819e+01 -4.70626106e+01 ... -4.57927895e+01\n",
      "    -4.70299149e+01 -4.70780563e+01]\n",
      "   [-4.78598747e+01 -4.87503471e+01 -4.84194260e+01 ... -4.70972595e+01\n",
      "    -4.68342361e+01 -4.76755867e+01]]]\n",
      "\n",
      "\n",
      " [[[-2.04718661e+00  7.15980768e+00  1.73476830e+01 ...  6.51224995e+00\n",
      "    -4.46891975e+00 -1.64966431e+01]\n",
      "   [-8.29697990e+00  1.05773697e+01  2.32796192e+01 ...  4.21373987e+00\n",
      "    -2.33099985e+00 -2.98540807e+00]\n",
      "   [-1.28230906e+01  8.20930862e+00  2.40194550e+01 ... -6.68498218e-01\n",
      "    -2.11571074e+00 -6.77769470e+00]\n",
      "   ...\n",
      "   [-4.66547394e+01 -4.59889679e+01 -4.60507660e+01 ... -4.90340767e+01\n",
      "    -4.95646133e+01 -4.90003891e+01]\n",
      "   [-4.63242950e+01 -4.65750427e+01 -4.64836426e+01 ... -4.91870956e+01\n",
      "    -4.77504044e+01 -4.75052299e+01]\n",
      "   [-4.55264320e+01 -4.77954407e+01 -4.69464455e+01 ... -4.80780334e+01\n",
      "    -4.78073845e+01 -4.78253136e+01]]]] | Labels tensor([0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 1])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 9 | Inputs [[[[  3.0577989    5.771313     5.844733   ...  -2.538676\n",
      "     -3.9296389    3.6864328 ]\n",
      "   [  7.9492683    6.323645     5.5840626  ...  -4.4073124\n",
      "     -7.0850906    6.8918633 ]\n",
      "   [  6.1424947    1.6394247    1.2324908  ...  -5.413552\n",
      "     -2.2465081    5.0580716 ]\n",
      "   ...\n",
      "   [-46.30783    -47.75528    -48.68742    ... -45.87787\n",
      "    -47.95542    -48.091167  ]\n",
      "   [-47.123802   -46.099174   -46.52798    ... -47.009335\n",
      "    -48.39134    -48.027493  ]\n",
      "   [-47.03325    -47.713593   -47.738235   ... -47.1224\n",
      "    -47.567707   -48.88962   ]]]\n",
      "\n",
      "\n",
      " [[[  4.340914    -7.234434   -12.623024   ... -12.736322\n",
      "     -7.6534824   -5.931635  ]\n",
      "   [  3.875611    -4.54606     -9.316802   ... -13.3299265\n",
      "     -2.2720115  -11.227003  ]\n",
      "   [ -0.74055666  -4.2201815   -6.5604167  ... -13.381313\n",
      "     -2.8479252  -12.393072  ]\n",
      "   ...\n",
      "   [-47.237694   -46.25126    -45.359413   ... -47.059837\n",
      "    -45.150967   -46.510452  ]\n",
      "   [-44.379845   -44.478203   -46.196712   ... -46.55453\n",
      "    -45.84345    -46.767654  ]\n",
      "   [-44.796864   -44.36752    -45.671257   ... -46.7277\n",
      "    -47.66684    -46.118454  ]]]\n",
      "\n",
      "\n",
      " [[[ 13.92106     19.734793    13.0334835  ...   3.3101974\n",
      "      3.8470998   19.718306  ]\n",
      "   [ 17.861942    17.608965    18.522533   ...   3.3508298\n",
      "      6.9995403   21.683674  ]\n",
      "   [ 15.598372    14.94105     17.32392    ...   3.6451938\n",
      "      7.726638    19.20561   ]\n",
      "   ...\n",
      "   [-38.473614   -34.97682    -37.26847    ... -37.250267\n",
      "    -37.48387    -38.324554  ]\n",
      "   [-38.216404   -37.63363    -38.370888   ... -35.526535\n",
      "    -37.039314   -36.283237  ]\n",
      "   [-37.783974   -37.202435   -38.358723   ... -37.0996\n",
      "    -39.88315    -35.676247  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  1.5515275    7.890139     3.933702   ...   7.301209\n",
      "      0.74503946   1.3779745 ]\n",
      "   [  3.6602464    5.062521     0.14959143 ...   6.1039762\n",
      "     -4.916527    -3.6302924 ]\n",
      "   [  3.7983055   -1.2177991   -6.033062   ...   1.7555599\n",
      "     -9.652738    -8.040229  ]\n",
      "   ...\n",
      "   [-47.92164    -47.79698    -45.686234   ... -47.946304\n",
      "    -48.014473   -46.10662   ]\n",
      "   [-48.7936     -47.25762    -44.160828   ... -46.98768\n",
      "    -47.360603   -49.02016   ]\n",
      "   [-49.09813    -47.254005   -46.902042   ... -46.30979\n",
      "    -46.64799    -47.091106  ]]]\n",
      "\n",
      "\n",
      " [[[-34.242218   -25.47907    -19.082918   ... -13.881106\n",
      "    -17.57008    -22.078074  ]\n",
      "   [-29.479773   -25.251226   -20.36031    ... -20.13123\n",
      "    -22.017433   -22.483309  ]\n",
      "   [-32.824974   -26.317131   -24.128937   ... -27.807232\n",
      "    -25.41506    -25.659246  ]\n",
      "   ...\n",
      "   [-72.035446   -72.035446   -72.035446   ... -72.035446\n",
      "    -72.035446   -72.035446  ]\n",
      "   [-72.035446   -72.035446   -72.035446   ... -72.035446\n",
      "    -72.035446   -72.035446  ]\n",
      "   [-72.035446   -72.035446   -72.035446   ... -72.035446\n",
      "    -72.035446   -72.035446  ]]]\n",
      "\n",
      "\n",
      " [[[-58.150574   -58.150574   -58.150574   ... -58.150574\n",
      "    -58.150574   -58.150574  ]\n",
      "   [-58.150574   -58.150574   -58.150574   ... -58.150574\n",
      "    -58.150574   -58.150574  ]\n",
      "   [-58.150574   -58.150574   -58.150574   ... -58.150574\n",
      "    -58.150574   -58.150574  ]\n",
      "   ...\n",
      "   [-64.75932    -64.75932    -64.75932    ... -64.75932\n",
      "    -64.75932    -64.75932   ]\n",
      "   [-64.75932    -64.75932    -64.75932    ... -64.75932\n",
      "    -64.75932    -64.75932   ]\n",
      "   [-64.75932    -64.75932    -64.75932    ... -64.75932\n",
      "    -64.75932    -64.75932   ]]]] | Labels tensor([1, 1, 1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 10 | Inputs [[[[-13.26132      0.9289568   -5.492318   ...  -7.223467\n",
      "    -15.289558   -16.721891  ]\n",
      "   [ -9.220112     0.720064    -7.692477   ...  -5.0457664\n",
      "     -7.333337   -19.098146  ]\n",
      "   [-10.958185    -2.9220777  -12.756746   ...  -9.295563\n",
      "     -5.892823   -21.884157  ]\n",
      "   ...\n",
      "   [-48.95855    -46.60438    -47.687874   ... -47.387726\n",
      "    -47.87037    -47.05698   ]\n",
      "   [-47.577213   -45.153675   -46.16044    ... -46.344585\n",
      "    -45.961132   -45.129272  ]\n",
      "   [-48.63501    -47.26605    -46.180084   ... -46.494083\n",
      "    -46.973263   -47.704704  ]]]\n",
      "\n",
      "\n",
      " [[[-11.104235   -15.161087   -16.045122   ... -21.806232\n",
      "    -15.142607   -11.169708  ]\n",
      "   [-17.462921   -20.591087   -22.088886   ... -25.205095\n",
      "    -18.416988   -17.62716   ]\n",
      "   [-33.207893   -29.512453   -32.56018    ... -31.760761\n",
      "    -23.478525   -32.188778  ]\n",
      "   ...\n",
      "   [-83.26738    -83.26738    -83.26738    ... -83.26738\n",
      "    -83.26738    -83.26738   ]\n",
      "   [-83.26738    -83.26738    -83.26738    ... -83.26738\n",
      "    -83.26738    -83.26738   ]\n",
      "   [-83.26738    -83.26738    -83.26738    ... -83.26738\n",
      "    -83.26738    -83.26738   ]]]\n",
      "\n",
      "\n",
      " [[[-47.311703   -23.206078   -13.495189   ... -26.330223\n",
      "    -16.01901    -20.163813  ]\n",
      "   [-29.109623   -25.383102   -16.68832    ... -28.988197\n",
      "    -18.808752   -26.067858  ]\n",
      "   [-31.81865    -31.084787   -22.425083   ... -30.705997\n",
      "    -23.75355    -33.17611   ]\n",
      "   ...\n",
      "   [-68.98642    -68.98642    -68.98642    ... -68.98642\n",
      "    -68.98642    -68.98642   ]\n",
      "   [-68.98642    -68.98642    -68.98642    ... -68.98642\n",
      "    -68.98642    -68.98642   ]\n",
      "   [-68.98642    -68.98642    -68.98642    ... -68.98642\n",
      "    -68.98642    -68.98642   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-31.045147   -18.875502   -24.262445   ...  -5.340123\n",
      "     -5.9468794  -13.978575  ]\n",
      "   [-17.305548   -22.646267   -27.267517   ... -10.0563965\n",
      "    -12.093712   -16.99263   ]\n",
      "   [-19.197084   -29.33305    -33.19867    ... -17.983297\n",
      "    -25.546305   -22.66489   ]\n",
      "   ...\n",
      "   [-75.83156    -75.83156    -75.83156    ... -75.83156\n",
      "    -75.83156    -75.83156   ]\n",
      "   [-75.83156    -75.83156    -75.83156    ... -75.83156\n",
      "    -75.83156    -75.83156   ]\n",
      "   [-75.83156    -75.83156    -75.83156    ... -75.83156\n",
      "    -75.83156    -75.83156   ]]]\n",
      "\n",
      "\n",
      " [[[ -5.184238     6.6762834    0.57890797 ...   5.477682\n",
      "      5.379922   -17.390564  ]\n",
      "   [  7.227259     6.2382293   -0.12358848 ...   4.705262\n",
      "      7.0484514    6.547468  ]\n",
      "   [  3.442031     7.8581495    8.317643   ...   6.2446012\n",
      "     10.377459     9.911653  ]\n",
      "   ...\n",
      "   [-47.18798    -48.076992   -49.295635   ... -47.798077\n",
      "    -47.636757   -47.60755   ]\n",
      "   [-48.375362   -47.5922     -48.12371    ... -45.465004\n",
      "    -45.675198   -46.94326   ]\n",
      "   [-48.53994    -46.647446   -46.484154   ... -47.299313\n",
      "    -49.14867    -51.694992  ]]]\n",
      "\n",
      "\n",
      " [[[-22.631226   -11.060621    -8.28576    ... -18.646328\n",
      "    -13.82176    -21.688759  ]\n",
      "   [-28.102991   -13.733697   -12.199688   ... -22.139034\n",
      "    -19.248823   -26.551971  ]\n",
      "   [-38.036304   -18.608986   -18.691092   ... -28.596067\n",
      "    -28.420525   -35.130005  ]\n",
      "   ...\n",
      "   [-70.43198    -70.43198    -70.43198    ... -70.43198\n",
      "    -70.43198    -70.43198   ]\n",
      "   [-70.43198    -70.43198    -70.43198    ... -70.43198\n",
      "    -70.43198    -70.43198   ]\n",
      "   [-70.43198    -70.43198    -70.43198    ... -70.43198\n",
      "    -70.43198    -70.43198   ]]]] | Labels tensor([1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1, 0, 0, 1, 1])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 11 | Inputs [[[[-14.765573   -16.374762   -19.722654   ... -26.066364\n",
      "    -29.494232   -11.971731  ]\n",
      "   [-21.338612   -20.911158   -23.709564   ... -29.62775\n",
      "    -34.278625   -13.175989  ]\n",
      "   [-24.854713   -27.925304   -30.922413   ... -34.52689\n",
      "    -41.529755   -15.77153   ]\n",
      "   ...\n",
      "   [-72.62237    -72.62237    -72.62237    ... -72.62237\n",
      "    -72.62237    -72.62237   ]\n",
      "   [-72.62237    -72.62237    -72.62237    ... -72.62237\n",
      "    -72.62237    -72.62237   ]\n",
      "   [-72.62237    -72.62237    -72.62237    ... -72.62237\n",
      "    -72.62237    -72.62237   ]]]\n",
      "\n",
      "\n",
      " [[[-60.143944   -60.143944   -60.143944   ... -60.143944\n",
      "    -60.143944   -60.143944  ]\n",
      "   [-60.143944   -60.143944   -60.143944   ... -60.143944\n",
      "    -60.143944   -60.143944  ]\n",
      "   [-60.143944   -60.143944   -60.143944   ... -60.143944\n",
      "    -60.143944   -60.143944  ]\n",
      "   ...\n",
      "   [-73.37503    -73.37503    -73.37503    ... -73.37503\n",
      "    -73.37503    -73.37503   ]\n",
      "   [-73.37503    -73.37503    -73.37503    ... -73.37503\n",
      "    -73.37503    -73.37503   ]\n",
      "   [-73.37503    -73.37503    -73.37503    ... -73.37503\n",
      "    -73.37503    -73.37503   ]]]\n",
      "\n",
      "\n",
      " [[[-34.720886   -34.720886   -34.720886   ... -34.720886\n",
      "    -34.720886   -34.720886  ]\n",
      "   [-34.720886   -34.720886   -34.720886   ... -34.720886\n",
      "    -34.720886   -34.720886  ]\n",
      "   [-34.720886   -34.720886   -34.720886   ... -34.720886\n",
      "    -34.720886   -34.720886  ]\n",
      "   ...\n",
      "   [-43.230274   -43.230274   -43.230274   ... -43.230274\n",
      "    -43.230274   -43.230274  ]\n",
      "   [-40.5332     -43.230274   -43.230274   ... -43.230274\n",
      "    -43.230274   -43.230274  ]\n",
      "   [-42.64116    -43.230274   -43.230274   ... -43.230274\n",
      "    -43.230274   -43.230274  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  0.68409556  -2.161857    -6.074774   ...   0.61279494\n",
      "      0.9144837   -1.812686  ]\n",
      "   [ -5.8575625   -8.035126    -9.717485   ...  -4.196142\n",
      "     -4.5201097   -5.7467866 ]\n",
      "   [-19.082645   -19.82455    -15.562459   ... -12.663666\n",
      "    -13.195963   -12.69311   ]\n",
      "   ...\n",
      "   [-63.2509     -63.2509     -63.2509     ... -63.2509\n",
      "    -63.2509     -63.2509    ]\n",
      "   [-63.2509     -63.2509     -63.2509     ... -63.2509\n",
      "    -63.2509     -63.2509    ]\n",
      "   [-63.2509     -63.2509     -63.2509     ... -63.2509\n",
      "    -63.2509     -63.2509    ]]]\n",
      "\n",
      "\n",
      " [[[-42.379463   -16.988766   -20.892971   ... -12.120586\n",
      "    -11.432756   -22.80624   ]\n",
      "   [-19.635164   -17.706884   -25.042854   ...  -9.857101\n",
      "    -14.968739   -22.64466   ]\n",
      "   [-21.080853   -22.047308   -30.247276   ... -13.187318\n",
      "    -21.716965   -24.95773   ]\n",
      "   ...\n",
      "   [-64.96833    -64.96833    -64.96833    ... -64.96833\n",
      "    -64.96833    -64.96833   ]\n",
      "   [-64.96833    -64.96833    -64.96833    ... -64.96833\n",
      "    -64.96833    -64.96833   ]\n",
      "   [-64.96833    -64.96833    -64.96833    ... -64.96833\n",
      "    -64.96833    -64.96833   ]]]\n",
      "\n",
      "\n",
      " [[[ -5.106756     0.69069093  -2.4270658  ...  -4.9254184\n",
      "    -30.969343    -3.2498395 ]\n",
      "   [ -3.2567172    3.8959024   -0.9574021  ...  -2.031836\n",
      "     -4.042136    -5.721697  ]\n",
      "   [ -6.498476     2.5583622   -1.4161596  ...   1.1160773\n",
      "     -2.0738263   -2.0605593 ]\n",
      "   ...\n",
      "   [-45.193207   -45.362602   -47.786633   ... -48.61939\n",
      "    -47.012733   -45.158176  ]\n",
      "   [-45.62829    -46.35643    -46.356773   ... -47.522255\n",
      "    -48.120903   -45.9695    ]\n",
      "   [-46.985733   -45.99067    -47.185783   ... -46.347664\n",
      "    -48.088478   -44.8646    ]]]] | Labels tensor([0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 1, 0, 1, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 12 | Inputs [[[[-27.327154   -4.7211766  -9.1133585 ... -22.389282   -9.396254\n",
      "    -12.174841 ]\n",
      "   [-20.100958   -9.681279  -15.259932  ... -26.910515   -9.732088\n",
      "    -12.381139 ]\n",
      "   [-24.14539   -16.516518  -22.119257  ... -33.311775  -13.876\n",
      "    -15.589354 ]\n",
      "   ...\n",
      "   [-73.805084  -73.805084  -73.805084  ... -73.805084  -73.805084\n",
      "    -73.805084 ]\n",
      "   [-73.805084  -73.805084  -73.805084  ... -73.805084  -73.805084\n",
      "    -73.805084 ]\n",
      "   [-73.805084  -73.805084  -73.805084  ... -73.805084  -73.805084\n",
      "    -73.805084 ]]]\n",
      "\n",
      "\n",
      " [[[ -0.1944035  -5.80287   -18.334896  ... -35.01104    -9.780904\n",
      "     -6.6788025]\n",
      "   [ -5.5577717  -9.621907  -20.73015   ... -33.56196   -11.802233\n",
      "     -4.6016817]\n",
      "   [-14.115339  -16.603046  -25.619995  ... -37.359703  -15.259584\n",
      "     -8.724104 ]\n",
      "   ...\n",
      "   [-77.61718   -77.61718   -77.61718   ... -77.61718   -77.61718\n",
      "    -77.61718  ]\n",
      "   [-77.61718   -77.61718   -77.61718   ... -77.61718   -77.61718\n",
      "    -77.61718  ]\n",
      "   [-77.61718   -77.61718   -77.61718   ... -77.61718   -77.61718\n",
      "    -77.61718  ]]]\n",
      "\n",
      "\n",
      " [[[ -5.0289445  10.036064   11.082949  ...  -3.1198273  18.8064\n",
      "    -12.13213  ]\n",
      "   [ 10.699989    4.386405    8.703727  ...  -2.044757   21.774323\n",
      "     19.702957 ]\n",
      "   [  7.37452     6.0431533   3.6508179 ...   2.0688171  21.11758\n",
      "     16.833242 ]\n",
      "   ...\n",
      "   [-47.671947  -46.456085  -45.893524  ... -46.137054  -46.61936\n",
      "    -48.430477 ]\n",
      "   [-44.523006  -46.16627   -46.80531   ... -43.97533   -45.835384\n",
      "    -49.44282  ]\n",
      "   [-46.63078   -47.628204  -48.318695  ... -45.89474   -47.74536\n",
      "    -48.260902 ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ -5.840507   -2.9582846  -2.0692704 ... -10.018598   -9.031252\n",
      "    -18.350609 ]\n",
      "   [ -7.954606   -7.068549   -4.604563  ... -13.789426   -9.926262\n",
      "    -13.087755 ]\n",
      "   [ -3.187635   -4.596416   -8.462282  ... -12.0579605  -8.701266\n",
      "    -10.27696  ]\n",
      "   ...\n",
      "   [-47.15872   -47.138348  -48.396946  ... -47.039116  -49.512783\n",
      "    -45.201336 ]\n",
      "   [-46.41714   -46.714966  -47.386345  ... -47.0001    -47.104435\n",
      "    -47.91168  ]\n",
      "   [-46.153072  -49.249363  -47.12269   ... -48.58025   -46.7552\n",
      "    -49.304012 ]]]\n",
      "\n",
      "\n",
      " [[[-28.19599   -15.094232  -15.787285  ... -18.852463  -18.538689\n",
      "    -36.16219  ]\n",
      "   [-17.08459   -15.913077  -15.318376  ... -22.132874  -17.126907\n",
      "    -26.465137 ]\n",
      "   [-17.639442  -20.942131  -18.589031  ... -28.466528  -19.903658\n",
      "    -22.225113 ]\n",
      "   ...\n",
      "   [-64.29617   -64.29617   -64.29617   ... -64.29617   -64.29617\n",
      "    -64.29617  ]\n",
      "   [-64.29617   -64.29617   -64.29617   ... -64.29617   -64.29617\n",
      "    -64.29617  ]\n",
      "   [-64.29617   -64.29617   -64.29617   ... -64.29617   -64.29617\n",
      "    -64.29617  ]]]\n",
      "\n",
      "\n",
      " [[[-64.798805  -64.798805  -64.798805  ... -64.798805  -64.798805\n",
      "    -64.798805 ]\n",
      "   [-64.798805  -64.798805  -64.798805  ... -64.798805  -64.798805\n",
      "    -64.798805 ]\n",
      "   [-64.798805  -64.798805  -64.798805  ... -64.798805  -64.798805\n",
      "    -64.798805 ]\n",
      "   ...\n",
      "   [-64.798805  -64.798805  -64.798805  ... -64.798805  -64.798805\n",
      "    -64.798805 ]\n",
      "   [-76.38321   -76.38321   -76.38321   ... -76.38321   -76.38321\n",
      "    -76.38321  ]\n",
      "   [-76.38321   -76.38321   -76.38321   ... -76.38321   -76.38321\n",
      "    -76.38321  ]]]] | Labels tensor([0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0])\n",
      "(2, 1, 64, 376)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch: 13 | Inputs [[[[-34.30631  -20.573061 -21.049976 ... -19.40988  -27.037968\n",
      "    -44.902428]\n",
      "   [-32.16592  -22.852657 -25.248505 ... -26.015635 -32.552296\n",
      "    -34.497185]\n",
      "   [-31.349766 -27.123592 -31.608265 ... -40.482178 -33.739838\n",
      "    -38.534405]\n",
      "   ...\n",
      "   [-66.27602  -66.27602  -66.27602  ... -66.27602  -66.27602\n",
      "    -66.27602 ]\n",
      "   [-66.27602  -66.27602  -66.27602  ... -66.27602  -66.27602\n",
      "    -66.27602 ]\n",
      "   [-66.27602  -66.27602  -66.27602  ... -66.27602  -66.27602\n",
      "    -66.27602 ]]]\n",
      "\n",
      "\n",
      " [[[-29.134066 -20.572622 -18.241837 ... -15.657551 -34.16446\n",
      "    -28.595808]\n",
      "   [-28.759789 -25.226738 -23.509056 ... -20.056492 -27.275475\n",
      "    -29.671211]\n",
      "   [-31.142653 -28.49413  -31.715841 ... -27.812794 -30.729563\n",
      "    -29.277805]\n",
      "   ...\n",
      "   [-68.41237  -68.41237  -68.41237  ... -68.41237  -68.41237\n",
      "    -68.41237 ]\n",
      "   [-68.41237  -68.41237  -68.41237  ... -68.41237  -68.41237\n",
      "    -68.41237 ]\n",
      "   [-68.41237  -68.41237  -68.41237  ... -68.41237  -68.41237\n",
      "    -68.41237 ]]]] | Labels tensor([0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 0 | Inputs [[[[  0.41032118   4.781559     8.340166   ...  -4.0793586\n",
      "      3.50028      7.210383  ]\n",
      "   [  5.3150887    3.0164983    8.887243   ...   1.8664138\n",
      "      2.1792192    1.287861  ]\n",
      "   [-37.032734   -37.032734   -37.032734   ... -37.032734\n",
      "    -37.032734   -37.032734  ]\n",
      "   ...\n",
      "   [-47.122902   -46.888615   -47.122902   ... -47.122902\n",
      "    -47.122902   -47.122902  ]\n",
      "   [-47.122902   -46.90278    -46.04437    ... -45.053616\n",
      "    -46.98778    -45.71712   ]\n",
      "   [-47.122902   -47.122902   -47.122902   ... -46.151546\n",
      "    -47.122902   -47.122902  ]]]\n",
      "\n",
      "\n",
      " [[[-24.69037     -8.065089    -8.182378   ... -15.974154\n",
      "    -27.631208   -20.340542  ]\n",
      "   [-30.282402   -11.291064   -12.503238   ... -22.325844\n",
      "    -28.341768   -22.237768  ]\n",
      "   [-31.022036   -17.51056    -20.139616   ... -34.325233\n",
      "    -30.801945   -27.85467   ]\n",
      "   ...\n",
      "   [-75.04247    -75.04247    -75.04247    ... -75.04247\n",
      "    -75.04247    -75.04247   ]\n",
      "   [-75.04247    -75.04247    -75.04247    ... -75.04247\n",
      "    -75.04247    -75.04247   ]\n",
      "   [-75.04247    -75.04247    -75.04247    ... -75.04247\n",
      "    -75.04247    -75.04247   ]]]\n",
      "\n",
      "\n",
      " [[[-26.92687    -33.008583   -38.20967    ... -15.964157\n",
      "    -25.230171   -46.843018  ]\n",
      "   [-33.554195   -30.516577   -37.911686   ... -21.486692\n",
      "    -26.852707   -27.539185  ]\n",
      "   [-35.446327   -30.568775   -36.7227     ... -31.720121\n",
      "    -31.885187   -30.406313  ]\n",
      "   ...\n",
      "   [-66.113846   -66.113846   -66.113846   ... -66.113846\n",
      "    -66.113846   -66.113846  ]\n",
      "   [-66.113846   -66.113846   -66.113846   ... -66.113846\n",
      "    -66.113846   -66.113846  ]\n",
      "   [-66.113846   -66.113846   -66.113846   ... -66.113846\n",
      "    -66.113846   -66.113846  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-18.349375     1.2956116   -7.05256    ...   3.6711502\n",
      "      8.174406    10.704684  ]\n",
      "   [ 11.529072     8.269139     7.0508265  ...   2.3783374\n",
      "      8.189163     8.85226   ]\n",
      "   [ 14.899612     4.9129615   12.624744   ...   2.6919494\n",
      "      8.343298     9.046627  ]\n",
      "   ...\n",
      "   [-43.19055    -45.763577   -45.763577   ... -45.763577\n",
      "    -45.763577   -44.91573   ]\n",
      "   [-43.058113   -45.763577   -45.763577   ... -45.099262\n",
      "    -45.763577   -45.763577  ]\n",
      "   [-45.48181    -45.763577   -45.763577   ... -45.763577\n",
      "    -45.763577   -45.212532  ]]]\n",
      "\n",
      "\n",
      " [[[ -1.9091291   -3.2825303  -19.76896    ...  -4.141076\n",
      "     -8.45309     -3.7728164 ]\n",
      "   [ -4.5251184   -4.81578    -14.355955   ...  -2.7192187\n",
      "    -13.309931    -1.9184096 ]\n",
      "   [ -7.945485    -9.444429    -9.629542   ...  -1.7973442\n",
      "    -10.980095    -6.246574  ]\n",
      "   ...\n",
      "   [-44.03625    -46.61651    -47.223034   ... -46.383255\n",
      "    -46.68145    -48.392647  ]\n",
      "   [-45.260544   -46.420303   -48.382675   ... -47.20138\n",
      "    -46.411892   -48.55231   ]\n",
      "   [-48.55231    -48.107407   -47.777206   ... -48.11486\n",
      "    -47.725822   -45.936466  ]]]\n",
      "\n",
      "\n",
      " [[[-28.962095   -15.636224    -8.687841   ... -12.713223\n",
      "     -7.3893147  -27.24728   ]\n",
      "   [-25.006107   -19.376904   -14.161254   ... -19.205456\n",
      "    -13.923422   -15.464319  ]\n",
      "   [-25.69703    -25.395937   -21.822056   ... -28.76548\n",
      "    -27.29376    -18.90835   ]\n",
      "   ...\n",
      "   [-74.58846    -74.58846    -74.58846    ... -74.58846\n",
      "    -74.58846    -74.58846   ]\n",
      "   [-74.58846    -74.58846    -74.58846    ... -74.58846\n",
      "    -74.58846    -74.58846   ]\n",
      "   [-74.58846    -74.58846    -74.58846    ... -74.58846\n",
      "    -74.58846    -74.58846   ]]]] | Labels tensor([1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1, 1, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 1 | Inputs [[[[  5.2334876    0.9317207   -1.5780711  ...  15.178986\n",
      "      8.042108     6.1841397 ]\n",
      "   [  5.36614      1.3743252   -6.5658293  ...  20.696123\n",
      "      8.016622     0.32556352]\n",
      "   [  0.6607075    4.189826    -2.1135066  ...  20.809958\n",
      "      9.863896    -7.2515564 ]\n",
      "   ...\n",
      "   [-48.42224    -46.477806   -46.284542   ... -46.3586\n",
      "    -47.58449    -45.613598  ]\n",
      "   [-48.00659    -46.93695    -46.56565    ... -46.25198\n",
      "    -45.738758   -46.132107  ]\n",
      "   [-47.544083   -46.825104   -48.53743    ... -48.9826\n",
      "    -47.949806   -47.064816  ]]]\n",
      "\n",
      "\n",
      " [[[-18.622149   -18.197863   -25.35206    ... -20.708467\n",
      "    -20.43189    -17.91819   ]\n",
      "   [-24.434544   -22.863533   -29.536999   ... -25.622631\n",
      "    -24.527235   -23.990524  ]\n",
      "   [-34.56698    -30.865385   -36.598392   ... -31.43524\n",
      "    -31.589096   -36.906105  ]\n",
      "   ...\n",
      "   [-70.750656   -70.750656   -70.750656   ... -70.750656\n",
      "    -70.750656   -70.750656  ]\n",
      "   [-70.750656   -70.750656   -70.750656   ... -70.750656\n",
      "    -70.750656   -70.750656  ]\n",
      "   [-70.750656   -70.750656   -70.750656   ... -70.750656\n",
      "    -70.750656   -70.750656  ]]]\n",
      "\n",
      "\n",
      " [[[-30.798338    -8.015583   -11.389251   ...  -3.3635104\n",
      "     -4.7149034  -20.541676  ]\n",
      "   [-21.954058   -11.303224   -14.520477   ...  -7.8738303\n",
      "     -6.8593526  -26.758835  ]\n",
      "   [-25.526947   -17.231285   -16.978933   ... -14.720058\n",
      "    -10.850639   -19.283009  ]\n",
      "   ...\n",
      "   [-69.970894   -69.970894   -69.970894   ... -69.970894\n",
      "    -69.970894   -69.970894  ]\n",
      "   [-69.970894   -69.970894   -69.970894   ... -69.970894\n",
      "    -69.970894   -69.970894  ]\n",
      "   [-69.970894   -69.970894   -69.970894   ... -69.970894\n",
      "    -69.970894   -69.970894  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ 15.896267    -6.271986    -2.4600115  ...   1.7985518\n",
      "     -0.17275253  11.717229  ]\n",
      "   [ 15.83809      4.1423984    4.1530395  ...   5.1586275\n",
      "      0.61282414  10.596029  ]\n",
      "   [ 17.000534     3.5609925    3.3725774  ...   1.6708343\n",
      "      0.32034206  15.924704  ]\n",
      "   ...\n",
      "   [-43.49407    -44.231075   -44.231075   ... -44.231075\n",
      "    -44.231075   -42.043285  ]\n",
      "   [-44.07217    -44.231075   -44.231075   ... -44.231075\n",
      "    -44.231075   -41.990376  ]\n",
      "   [-44.231075   -44.231075   -44.231075   ... -44.231075\n",
      "    -44.231075   -42.189453  ]]]\n",
      "\n",
      "\n",
      " [[[-22.097202    -6.3789053   -5.8254576  ...  -8.270039\n",
      "    -12.303472   -28.955423  ]\n",
      "   [-21.785776   -11.794411   -11.049515   ... -11.485413\n",
      "    -18.328438   -33.164604  ]\n",
      "   [-21.465017   -21.314667   -20.112186   ... -17.448204\n",
      "    -29.937086   -29.776928  ]\n",
      "   ...\n",
      "   [-69.23054    -69.23054    -69.23054    ... -69.23054\n",
      "    -69.23054    -69.23054   ]\n",
      "   [-69.23054    -69.23054    -69.23054    ... -69.23054\n",
      "    -69.23054    -69.23054   ]\n",
      "   [-69.23054    -69.23054    -69.23054    ... -69.23054\n",
      "    -69.23054    -69.23054   ]]]\n",
      "\n",
      "\n",
      " [[[-71.26381    -43.540653   -47.903008   ... -47.49107\n",
      "    -61.594463   -52.670593  ]\n",
      "   [-50.002213   -47.654068   -53.845215   ... -51.228832\n",
      "    -59.019188   -59.043022  ]\n",
      "   [-52.226177   -48.958073   -51.828285   ... -58.21618\n",
      "    -60.347084   -57.159733  ]\n",
      "   ...\n",
      "   [-80.27312    -80.27312    -80.27312    ... -80.27312\n",
      "    -80.27312    -80.27312   ]\n",
      "   [-80.27312    -80.27312    -80.27312    ... -80.27312\n",
      "    -80.27312    -80.27312   ]\n",
      "   [-80.27312    -80.27312    -80.27312    ... -80.27312\n",
      "    -80.27312    -80.27312   ]]]] | Labels tensor([0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 2 | Inputs [[[[ -4.6625333   -4.380852   -24.395817   ... -10.988678\n",
      "    -14.599016    -2.8093295 ]\n",
      "   [ -1.4627664   -2.5399365  -14.627079   ...  -3.4924412\n",
      "     -8.179388    -4.398237  ]\n",
      "   [ -4.65347     -2.4231324   -7.983728   ...  -3.4815028\n",
      "     -2.8968265   -9.331184  ]\n",
      "   ...\n",
      "   [-49.26776    -47.53632    -45.73622    ... -45.04609\n",
      "    -46.953796   -48.41724   ]\n",
      "   [-48.477398   -48.407616   -45.97947    ... -45.41643\n",
      "    -46.160984   -45.549393  ]\n",
      "   [-45.28898    -47.103886   -47.38215    ... -47.771698\n",
      "    -47.119198   -46.77791   ]]]\n",
      "\n",
      "\n",
      " [[[-25.314817   -18.5146     -19.200523   ... -17.208878\n",
      "    -15.411518   -29.585289  ]\n",
      "   [-31.559261   -24.84106    -23.959812   ... -21.700518\n",
      "    -21.795378   -18.173868  ]\n",
      "   [-41.949463   -29.02487    -30.990202   ... -29.579676\n",
      "    -31.789547   -20.532293  ]\n",
      "   ...\n",
      "   [-68.528404   -56.58636    -68.528404   ... -68.528404\n",
      "    -68.528404   -68.528404  ]\n",
      "   [-68.528404   -56.386482   -68.528404   ... -68.528404\n",
      "    -68.528404   -68.528404  ]\n",
      "   [-68.528404   -56.132935   -68.528404   ... -68.528404\n",
      "    -68.528404   -68.528404  ]]]\n",
      "\n",
      "\n",
      " [[[-11.309383    -1.9659256   -3.405495   ...  -8.179011\n",
      "    -10.012708    -9.631797  ]\n",
      "   [ -6.9905195    0.16068748  -5.628298   ...  -6.7059603\n",
      "    -13.466675    -5.66455   ]\n",
      "   [ -7.7569766   -1.6435497   -9.219465   ...  -6.1256723\n",
      "    -18.75719     -8.909993  ]\n",
      "   ...\n",
      "   [-47.958412   -46.597847   -44.869186   ... -46.419273\n",
      "    -44.38079    -45.82356   ]\n",
      "   [-46.84826    -46.323006   -46.446003   ... -44.748055\n",
      "    -44.23323    -44.36771   ]\n",
      "   [-48.19588    -47.5421     -46.9496     ... -46.122192\n",
      "    -46.450787   -46.008263  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-11.717073    10.858782    12.277456   ...  -0.5451919\n",
      "     -5.524941     2.064904  ]\n",
      "   [ -8.737746     7.005324     9.763553   ...   2.9160235\n",
      "     -2.9883265    1.7627594 ]\n",
      "   [ -5.197661     0.19508077   9.438871   ...   1.6006237\n",
      "     -6.0276403    2.7702217 ]\n",
      "   ...\n",
      "   [-46.756016   -20.664028   -43.31582    ... -49.62637\n",
      "    -48.34945    -48.031006  ]\n",
      "   [-48.74418    -20.539324   -42.89002    ... -47.169018\n",
      "    -47.96669    -49.63503   ]\n",
      "   [-46.51407    -20.470825   -42.992363   ... -48.020416\n",
      "    -49.910984   -49.136654  ]]]\n",
      "\n",
      "\n",
      " [[[-39.97021    -10.892794   -13.954783   ... -20.053942\n",
      "    -15.201141   -16.45705   ]\n",
      "   [-17.90459    -15.852069   -19.32761    ... -23.442564\n",
      "    -19.155231   -19.52564   ]\n",
      "   [-21.576962   -24.170055   -29.041203   ... -28.338074\n",
      "    -25.602531   -24.191933  ]\n",
      "   ...\n",
      "   [-69.54002    -69.54002    -69.54002    ... -69.54002\n",
      "    -69.54002    -69.54002   ]\n",
      "   [-69.54002    -69.54002    -69.54002    ... -69.54002\n",
      "    -69.54002    -69.54002   ]\n",
      "   [-69.54002    -69.54002    -69.54002    ... -69.54002\n",
      "    -69.54002    -69.54002   ]]]\n",
      "\n",
      "\n",
      " [[[-11.223902   -12.106621   -10.191086   ...  -8.780102\n",
      "     -5.540407   -11.883083  ]\n",
      "   [-17.795795   -14.843279   -11.674862   ... -13.800976\n",
      "    -11.471671   -12.313438  ]\n",
      "   [-32.39157    -16.189703   -16.69526    ... -19.943121\n",
      "    -22.234621   -17.322237  ]\n",
      "   ...\n",
      "   [-69.2756     -69.2756     -69.2756     ... -69.2756\n",
      "    -69.2756     -69.2756    ]\n",
      "   [-69.2756     -69.2756     -69.2756     ... -69.2756\n",
      "    -69.2756     -69.2756    ]\n",
      "   [-69.2756     -69.2756     -69.2756     ... -69.2756\n",
      "    -69.2756     -69.2756    ]]]] | Labels tensor([1, 0, 0, 1, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 3 | Inputs [[[[ -4.0409465   -4.45592     -0.9975818  ... -32.390106\n",
      "    -13.67161     -8.878049  ]\n",
      "   [-10.135574   -10.388079    -6.121585   ... -21.394064\n",
      "    -17.37991    -11.737024  ]\n",
      "   [-20.135883   -21.329212   -15.380654   ... -23.457253\n",
      "    -22.640478   -17.361626  ]\n",
      "   ...\n",
      "   [-75.23507    -75.23507    -75.23507    ... -75.23507\n",
      "    -75.23507    -75.23507   ]\n",
      "   [-75.23507    -75.23507    -75.23507    ... -75.23507\n",
      "    -75.23507    -75.23507   ]\n",
      "   [-75.23507    -75.23507    -75.23507    ... -75.23507\n",
      "    -75.23507    -75.23507   ]]]\n",
      "\n",
      "\n",
      " [[[-42.043785   -42.043785   -42.043785   ... -42.043785\n",
      "    -42.043785   -42.043785  ]\n",
      "   [-42.043785   -42.043785   -42.043785   ... -42.043785\n",
      "    -42.043785   -42.043785  ]\n",
      "   [ -3.9837728    3.5614624   -9.339413   ...  -8.679208\n",
      "      2.7769232    5.4799266 ]\n",
      "   ...\n",
      "   [-57.325077   -57.325077   -57.325077   ... -57.325077\n",
      "    -57.325077   -57.325077  ]\n",
      "   [-57.325077   -57.325077   -57.325077   ... -57.325077\n",
      "    -57.325077   -57.325077  ]\n",
      "   [-57.325077   -57.325077   -57.325077   ... -57.325077\n",
      "    -57.325077   -57.325077  ]]]\n",
      "\n",
      "\n",
      " [[[-16.132587   -20.901493   -27.27608    ... -22.957733\n",
      "    -19.518908   -22.86769   ]\n",
      "   [-22.19549    -24.602673   -28.7677     ... -28.558594\n",
      "    -20.604647   -27.156475  ]\n",
      "   [-29.754122   -29.149092   -33.81757    ... -34.41406\n",
      "    -24.39618    -28.35333   ]\n",
      "   ...\n",
      "   [-73.31582    -73.31582    -73.31582    ... -73.31582\n",
      "    -73.31582    -73.31582   ]\n",
      "   [-73.31582    -73.31582    -73.31582    ... -73.31582\n",
      "    -73.31582    -73.31582   ]\n",
      "   [-73.31582    -73.31582    -73.31582    ... -73.31582\n",
      "    -73.31582    -73.31582   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-51.342796   -45.67381    -40.74123    ... -41.84066\n",
      "    -42.37294    -42.794014  ]\n",
      "   [-50.41899    -51.08641    -42.190956   ... -41.446068\n",
      "    -42.718006   -44.522736  ]\n",
      "   [-57.146282   -57.146282   -57.146282   ... -57.146282\n",
      "    -57.146282   -57.146282  ]\n",
      "   ...\n",
      "   [-72.44962    -72.44962    -72.44962    ... -72.44962\n",
      "    -72.44962    -72.44962   ]\n",
      "   [-72.44962    -72.44962    -72.44962    ... -72.44962\n",
      "    -72.44962    -72.44962   ]\n",
      "   [-72.44962    -72.44962    -72.44962    ... -72.44962\n",
      "    -72.44962    -72.44962   ]]]\n",
      "\n",
      "\n",
      " [[[ -4.654767    -4.962629    -7.0133505  ...  -1.763519\n",
      "     -8.6889925   -2.5100873 ]\n",
      "   [ -9.353065    -9.886026   -12.955223   ...  -6.4279003\n",
      "    -14.57893     -9.121324  ]\n",
      "   [-17.70642    -18.517405   -21.486254   ... -14.749321\n",
      "    -26.186676   -28.718393  ]\n",
      "   ...\n",
      "   [-69.44404    -69.44404    -69.44404    ... -69.44404\n",
      "    -69.44404    -69.44404   ]\n",
      "   [-69.44404    -69.44404    -69.44404    ... -69.44404\n",
      "    -69.44404    -69.44404   ]\n",
      "   [-69.44404    -69.44404    -69.44404    ... -69.44404\n",
      "    -69.44404    -69.44404   ]]]\n",
      "\n",
      "\n",
      " [[[  2.6130462   10.469007    12.311224   ...  10.351434\n",
      "     11.277752     6.2170944 ]\n",
      "   [  2.865279     7.9236865   10.596906   ...   8.825492\n",
      "     11.247993     7.432342  ]\n",
      "   [ -0.18831462   1.8509353    6.8090115  ...   3.6020052\n",
      "      8.739904     7.35075   ]\n",
      "   ...\n",
      "   [-43.44773    -43.44773    -43.44773    ... -43.44773\n",
      "    -43.44773    -43.44773   ]\n",
      "   [-43.44773    -43.44773    -43.44773    ... -43.44773\n",
      "    -43.44773    -43.44773   ]\n",
      "   [-43.44773    -43.44773    -43.44773    ... -43.44773\n",
      "    -43.44773    -43.44773   ]]]] | Labels tensor([1, 1, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 4 | Inputs [[[[-1.80459900e+01 -3.01970711e+01 -1.39680195e+01 ... -1.49962282e+01\n",
      "    -2.21991749e+01 -1.63955326e+01]\n",
      "   [-2.02325439e+01 -2.36491852e+01 -1.36873722e+01 ... -1.94530449e+01\n",
      "    -2.19120884e+01 -1.93467178e+01]\n",
      "   [-2.58899860e+01 -1.90844765e+01 -1.70404625e+01 ... -2.41091805e+01\n",
      "    -2.40781822e+01 -2.53812370e+01]\n",
      "   ...\n",
      "   [-8.69680634e+01 -8.69680634e+01 -8.69680634e+01 ... -8.69680634e+01\n",
      "    -8.69680634e+01 -8.69680634e+01]\n",
      "   [-8.69680634e+01 -8.69680634e+01 -8.69680634e+01 ... -8.69680634e+01\n",
      "    -8.69680634e+01 -8.69680634e+01]\n",
      "   [-8.69680634e+01 -8.69680634e+01 -8.69680634e+01 ... -8.69680634e+01\n",
      "    -8.69680634e+01 -8.69680634e+01]]]\n",
      "\n",
      "\n",
      " [[[ 3.13554311e+00 -9.72759533e+00 -2.40227723e+00 ... -1.82189655e+01\n",
      "    -3.61874998e-02 -3.70626116e+00]\n",
      "   [-1.07987630e+00 -5.06735134e+00  1.09387410e+00 ... -2.67665768e+00\n",
      "     2.98247963e-01 -5.11999035e+00]\n",
      "   [-8.67095089e+00 -2.56696653e+00 -8.73712003e-01 ... -8.45943540e-02\n",
      "    -2.21430123e-01 -7.50794363e+00]\n",
      "   ...\n",
      "   [-4.79234352e+01 -4.58919182e+01 -4.49535980e+01 ... -4.70876007e+01\n",
      "    -4.61536179e+01 -4.50234032e+01]\n",
      "   [-4.60481339e+01 -4.48332138e+01 -4.47684441e+01 ... -4.49606895e+01\n",
      "    -4.35935249e+01 -4.57472534e+01]\n",
      "   [-4.52957382e+01 -4.58871231e+01 -4.72708092e+01 ... -4.57361450e+01\n",
      "    -4.51021194e+01 -4.57447586e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.57052097e+01 -1.31806393e+01 -3.29472208e+00 ... -1.42097836e+01\n",
      "    -1.26194000e+01 -2.16038246e+01]\n",
      "   [-1.41376019e+01 -1.59129372e+01 -9.74041080e+00 ... -1.49133339e+01\n",
      "    -1.67416172e+01 -1.08249378e+01]\n",
      "   [-1.44654131e+01 -1.77708645e+01 -2.33376465e+01 ... -1.99114761e+01\n",
      "    -2.40203838e+01 -1.00415401e+01]\n",
      "   ...\n",
      "   [-5.59909019e+01 -6.67289429e+01 -6.67289429e+01 ... -6.67289429e+01\n",
      "    -6.67289429e+01 -5.56498260e+01]\n",
      "   [-5.60381279e+01 -6.67289429e+01 -6.67289429e+01 ... -6.67289429e+01\n",
      "    -6.67289429e+01 -5.56985855e+01]\n",
      "   [-5.59790764e+01 -6.67289429e+01 -6.67289429e+01 ... -6.67289429e+01\n",
      "    -6.67289429e+01 -5.56419449e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ 8.36928177e+00  1.17081022e+01  4.72282219e+00 ...  1.38408489e+01\n",
      "     1.60203018e+01  1.49913502e+01]\n",
      "   [ 2.13410640e+00  1.35280809e+01  9.29350471e+00 ...  1.67222290e+01\n",
      "     1.79413948e+01  1.97284241e+01]\n",
      "   [-9.29014564e-01  1.41914139e+01  1.06065998e+01 ...  1.24517832e+01\n",
      "     1.51224480e+01  1.80700989e+01]\n",
      "   ...\n",
      "   [-4.58995361e+01 -4.68509445e+01 -4.63519211e+01 ... -4.71365242e+01\n",
      "    -4.71365242e+01 -4.71365242e+01]\n",
      "   [-4.52007256e+01 -4.55954590e+01 -4.56116676e+01 ... -4.53863983e+01\n",
      "    -4.71365242e+01 -4.71365242e+01]\n",
      "   [-4.70830841e+01 -4.67623177e+01 -4.71365242e+01 ... -4.64114075e+01\n",
      "    -4.68725433e+01 -4.60866127e+01]]]\n",
      "\n",
      "\n",
      " [[[-9.97999763e+00 -1.46608868e+01 -2.46513596e+01 ...  1.80575228e+00\n",
      "    -2.05470676e+01 -9.92971992e+00]\n",
      "   [-1.65451660e+01 -1.90631447e+01 -1.50465765e+01 ... -2.26652718e+00\n",
      "    -1.26646814e+01 -1.24446211e+01]\n",
      "   [-3.45501366e+01 -1.98395882e+01 -1.47930126e+01 ... -9.48909092e+00\n",
      "    -1.52028313e+01 -1.50327349e+01]\n",
      "   ...\n",
      "   [-6.66841812e+01 -6.66841812e+01 -6.66841812e+01 ... -6.66841812e+01\n",
      "    -6.66841812e+01 -6.66841812e+01]\n",
      "   [-6.66841812e+01 -6.66841812e+01 -6.66841812e+01 ... -6.66841812e+01\n",
      "    -6.66841812e+01 -6.66841812e+01]\n",
      "   [-6.66841812e+01 -6.66841812e+01 -6.66841812e+01 ... -6.66841812e+01\n",
      "    -6.66841812e+01 -6.66841812e+01]]]\n",
      "\n",
      "\n",
      " [[[-2.09007282e+01 -6.70010519e+00 -6.31990147e+00 ... -8.23965263e+00\n",
      "    -1.49541368e+01 -1.40449753e+01]\n",
      "   [-1.29641533e+01 -6.18607807e+00 -6.75093460e+00 ... -1.00919924e+01\n",
      "    -1.03801079e+01 -1.98927193e+01]\n",
      "   [-1.63874130e+01 -1.07551966e+01 -8.13972855e+00 ... -1.35648336e+01\n",
      "    -1.26263371e+01 -1.81460915e+01]\n",
      "   ...\n",
      "   [-4.71139107e+01 -4.67015038e+01 -4.65220833e+01 ... -4.63931999e+01\n",
      "    -4.58038406e+01 -4.44200096e+01]\n",
      "   [-4.58651009e+01 -4.32656326e+01 -4.58733711e+01 ... -4.68487740e+01\n",
      "    -4.53007584e+01 -4.78118210e+01]\n",
      "   [-4.81475372e+01 -4.66580238e+01 -4.77476044e+01 ... -4.62494049e+01\n",
      "    -4.70726929e+01 -4.65768852e+01]]]] | Labels tensor([0, 1, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 5 | Inputs [[[[-27.54424    -24.605072   -34.582756   ... -16.005043\n",
      "    -21.994663   -21.63599   ]\n",
      "   [-33.539368   -29.403435   -32.145615   ... -20.37822\n",
      "    -26.80826    -28.093489  ]\n",
      "   [-37.79052    -36.989895   -36.16951    ... -28.094406\n",
      "    -33.71594    -36.38772   ]\n",
      "   ...\n",
      "   [-79.42512    -79.42512    -79.42512    ... -79.42512\n",
      "    -79.42512    -79.42512   ]\n",
      "   [-79.42512    -79.42512    -79.42512    ... -79.42512\n",
      "    -79.42512    -79.42512   ]\n",
      "   [-79.42512    -79.42512    -79.42512    ... -79.42512\n",
      "    -79.42512    -79.42512   ]]]\n",
      "\n",
      "\n",
      " [[[  9.809509     6.975687     9.380623   ...   2.431201\n",
      "      3.9753625   10.436455  ]\n",
      "   [  9.827856     4.3298326   18.050278   ...   6.4578824\n",
      "      6.462203    12.427824  ]\n",
      "   [  8.774419     3.5109277   20.42508    ...   9.274587\n",
      "     12.636309    10.342349  ]\n",
      "   ...\n",
      "   [-41.94439    -46.203995   -47.04798    ... -47.811493\n",
      "    -48.209587   -42.744072  ]\n",
      "   [-42.80606    -47.282898   -48.66771    ... -45.22876\n",
      "    -47.22601    -42.536674  ]\n",
      "   [-41.375774   -47.763306   -49.221607   ... -46.247032\n",
      "    -46.464252   -42.585228  ]]]\n",
      "\n",
      "\n",
      " [[[-23.8249     -19.543188   -15.101288   ... -12.799562\n",
      "    -19.188688   -23.43964   ]\n",
      "   [-29.090536   -25.11511    -19.804798   ... -19.05766\n",
      "    -21.451275   -24.936686  ]\n",
      "   [-37.178288   -33.282658   -27.79874    ... -31.834452\n",
      "    -27.209484   -30.432398  ]\n",
      "   ...\n",
      "   [-78.682594   -78.682594   -78.682594   ... -78.682594\n",
      "    -78.682594   -78.682594  ]\n",
      "   [-78.682594   -78.682594   -78.682594   ... -78.682594\n",
      "    -78.682594   -78.682594  ]\n",
      "   [-78.682594   -78.682594   -78.682594   ... -78.682594\n",
      "    -78.682594   -78.682594  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[ -5.8016877   -1.6657071   -3.0403826  ... -10.976233\n",
      "     -4.3299837  -15.192184  ]\n",
      "   [-12.346364    -6.953375    -7.26863    ... -16.397106\n",
      "     -5.8308926   -5.869437  ]\n",
      "   [-18.268854   -16.38556    -14.827014   ... -20.05886\n",
      "    -10.853451    -8.920257  ]\n",
      "   ...\n",
      "   [-71.07507    -71.07507    -71.07507    ... -71.07507\n",
      "    -71.07507    -71.07507   ]\n",
      "   [-71.07507    -71.07507    -71.07507    ... -71.07507\n",
      "    -71.07507    -71.07507   ]\n",
      "   [-71.07507    -71.07507    -71.07507    ... -71.07507\n",
      "    -71.07507    -71.07507   ]]]\n",
      "\n",
      "\n",
      " [[[-27.178286   -25.197325   -24.914585   ... -13.726601\n",
      "    -23.366817   -16.29914   ]\n",
      "   [-22.938328   -22.481451   -30.006748   ... -18.556282\n",
      "    -28.345564   -21.731901  ]\n",
      "   [-25.203264   -21.523815   -25.16755    ... -26.212414\n",
      "    -24.981749   -27.80616   ]\n",
      "   ...\n",
      "   [-69.092285   -69.092285   -69.092285   ... -69.092285\n",
      "    -69.092285   -69.092285  ]\n",
      "   [-69.092285   -69.092285   -69.092285   ... -69.092285\n",
      "    -69.092285   -69.092285  ]\n",
      "   [-69.092285   -69.092285   -69.092285   ... -69.092285\n",
      "    -69.092285   -69.092285  ]]]\n",
      "\n",
      "\n",
      " [[[ -0.10461204  -3.7795267   -4.004055   ...  11.207954\n",
      "      7.581079     0.33991885]\n",
      "   [ -4.328993    -4.594058    -7.756248   ...  10.599055\n",
      "     11.857302     3.2294364 ]\n",
      "   [ -7.029683    -6.6422296  -14.423718   ...   6.330914\n",
      "      9.765363    -1.2355205 ]\n",
      "   ...\n",
      "   [-44.65849    -46.459526   -44.740604   ... -45.67673\n",
      "    -46.684296   -50.38529   ]\n",
      "   [-47.186028   -45.27265    -44.833862   ... -46.923393\n",
      "    -47.263416   -48.033848  ]\n",
      "   [-46.51312    -46.13143    -47.49137    ... -47.524414\n",
      "    -47.817085   -48.46847   ]]]] | Labels tensor([0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 1])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 6 | Inputs [[[[-1.41518984e+01  4.95426893e+00  4.80478668e+00 ...  7.81596947e+00\n",
      "     1.09842739e+01 -7.13849592e+00]\n",
      "   [-1.57379255e+01  3.74321103e+00  1.05102406e+01 ...  1.34583616e+01\n",
      "     8.89789200e+00 -8.43206584e-01]\n",
      "   [-7.63242531e+00  1.16930914e+00  8.96189976e+00 ...  1.37964897e+01\n",
      "     4.30644417e+00 -1.82163805e-01]\n",
      "   ...\n",
      "   [-4.65570908e+01 -4.61459274e+01 -4.74201965e+01 ... -4.56787262e+01\n",
      "    -4.79802742e+01 -4.77227325e+01]\n",
      "   [-4.88012276e+01 -4.66139069e+01 -4.59312592e+01 ... -4.67800713e+01\n",
      "    -4.74176788e+01 -4.68389359e+01]\n",
      "   [-4.78952942e+01 -4.91812897e+01 -4.84919701e+01 ... -4.78861847e+01\n",
      "    -4.77086601e+01 -4.82785378e+01]]]\n",
      "\n",
      "\n",
      " [[[-3.35543900e+01 -1.63172951e+01 -2.64988213e+01 ... -1.76245499e+01\n",
      "    -1.54440689e+01 -1.65719013e+01]\n",
      "   [-3.69678726e+01 -1.91493130e+01 -2.50421429e+01 ... -1.82430458e+01\n",
      "    -1.97343330e+01 -1.78144398e+01]\n",
      "   [-3.07272301e+01 -2.00208569e+01 -2.45881195e+01 ... -2.30515327e+01\n",
      "    -2.29941559e+01 -1.63141232e+01]\n",
      "   ...\n",
      "   [-7.15863113e+01 -7.15863113e+01 -7.15863113e+01 ... -7.15863113e+01\n",
      "    -7.15863113e+01 -7.15863113e+01]\n",
      "   [-7.15863113e+01 -7.15863113e+01 -7.15863113e+01 ... -7.15863113e+01\n",
      "    -7.15863113e+01 -7.15863113e+01]\n",
      "   [-7.15863113e+01 -7.15863113e+01 -7.15863113e+01 ... -7.15863113e+01\n",
      "    -7.15863113e+01 -7.15863113e+01]]]\n",
      "\n",
      "\n",
      " [[[ 7.09078693e+00 -3.41210634e-01  1.35179985e+00 ...  7.24582672e+00\n",
      "     1.20188093e+01  1.15368004e+01]\n",
      "   [ 3.41816640e+00 -1.14909041e+00  1.18085945e+00 ...  1.02899942e+01\n",
      "     1.34615345e+01  6.99837494e+00]\n",
      "   [-1.93837953e+00  2.92617059e+00  3.74189997e+00 ...  1.40306587e+01\n",
      "     1.08090820e+01  8.90430832e+00]\n",
      "   ...\n",
      "   [-4.64259148e+01 -4.67124634e+01 -4.80063019e+01 ... -4.64517365e+01\n",
      "    -4.67079239e+01 -4.73254395e+01]\n",
      "   [-4.48456917e+01 -4.60891418e+01 -4.78461761e+01 ... -4.73796654e+01\n",
      "    -4.82653999e+01 -4.91330490e+01]\n",
      "   [-4.80642548e+01 -4.79553947e+01 -4.81343384e+01 ... -4.75076866e+01\n",
      "    -4.85228195e+01 -5.09826050e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-3.23519516e+01  2.00965476e+00  1.51830502e+01 ...  1.04361401e+01\n",
      "     1.02186203e+01  5.51745701e+00]\n",
      "   [-8.86461163e+00  2.10958838e+00  1.79440231e+01 ...  8.07947731e+00\n",
      "     6.68323898e+00  1.80079830e+00]\n",
      "   [-1.16717300e+01  4.54551554e+00  1.66958656e+01 ...  2.10709882e+00\n",
      "     1.91459674e-02  2.66201162e+00]\n",
      "   ...\n",
      "   [-3.61828079e+01 -3.73551636e+01 -3.85980606e+01 ... -3.67965698e+01\n",
      "    -3.50371857e+01 -3.46982727e+01]\n",
      "   [-3.61661072e+01 -3.55822220e+01 -3.60559158e+01 ... -3.57130585e+01\n",
      "    -3.66989479e+01 -3.47449913e+01]\n",
      "   [-3.62221947e+01 -3.63192062e+01 -3.63871651e+01 ... -3.55191994e+01\n",
      "    -3.56409340e+01 -3.68219490e+01]]]\n",
      "\n",
      "\n",
      " [[[-6.08300161e+00  1.48315287e+00  2.57210040e+00 ... -4.66134930e+00\n",
      "    -7.72239780e+00 -8.87889194e+00]\n",
      "   [-8.44823647e+00 -1.30016863e+00 -1.34252441e+00 ... -9.30272388e+00\n",
      "    -1.19550896e+01 -1.53334484e+01]\n",
      "   [-1.35900974e+01 -7.37220001e+00 -8.53899384e+00 ... -1.72031460e+01\n",
      "    -1.89760456e+01 -3.29114304e+01]\n",
      "   ...\n",
      "   [-6.59068832e+01 -6.59068832e+01 -6.59068832e+01 ... -6.59068832e+01\n",
      "    -6.59068832e+01 -6.59068832e+01]\n",
      "   [-6.59068832e+01 -6.59068832e+01 -6.59068832e+01 ... -6.59068832e+01\n",
      "    -6.59068832e+01 -6.59068832e+01]\n",
      "   [-6.59068832e+01 -6.59068832e+01 -6.59068832e+01 ... -6.59068832e+01\n",
      "    -6.59068832e+01 -6.59068832e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.19975529e+01 -9.69309711e+00 -2.27929592e+01 ... -1.58403702e+01\n",
      "    -9.91460037e+00 -1.34956303e+01]\n",
      "   [-1.75711231e+01 -1.52326765e+01 -1.91928825e+01 ... -2.13736420e+01\n",
      "    -1.47200241e+01 -1.75171967e+01]\n",
      "   [-2.81391640e+01 -2.50768852e+01 -2.02915993e+01 ... -3.18101139e+01\n",
      "    -2.22788715e+01 -2.49137516e+01]\n",
      "   ...\n",
      "   [-6.54279785e+01 -6.54279785e+01 -6.54279785e+01 ... -6.54279785e+01\n",
      "    -6.54279785e+01 -6.54279785e+01]\n",
      "   [-6.54279785e+01 -6.54279785e+01 -6.54279785e+01 ... -6.54279785e+01\n",
      "    -6.54279785e+01 -6.54279785e+01]\n",
      "   [-6.54279785e+01 -6.54279785e+01 -6.54279785e+01 ... -6.54279785e+01\n",
      "    -6.54279785e+01 -6.54279785e+01]]]] | Labels tensor([0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 7 | Inputs [[[[-11.064204   -6.9089193 -12.498863  ...  -9.585625  -11.454724\n",
      "    -11.8392105]\n",
      "   [-14.403443   -8.708781  -11.936371  ...  -8.830539   -9.780848\n",
      "     -5.533289 ]\n",
      "   [ -5.800486   -6.6770573  -6.1931515 ...  -1.9151145  -6.9470053\n",
      "     -5.6161556]\n",
      "   ...\n",
      "   [-66.90645   -74.965706  -74.965706  ... -74.965706  -74.965706\n",
      "    -66.67623  ]\n",
      "   [-66.93411   -74.965706  -74.965706  ... -74.965706  -74.965706\n",
      "    -66.70793  ]\n",
      "   [-66.87668   -74.965706  -74.965706  ... -74.965706  -74.965706\n",
      "    -66.63349  ]]]\n",
      "\n",
      "\n",
      " [[[ -7.4315715 -20.046822  -13.723978  ... -16.541803  -16.470682\n",
      "    -36.61301  ]\n",
      "   [ -4.4398694 -13.489398  -16.468227  ... -19.314013  -14.692207\n",
      "    -17.512667 ]\n",
      "   [ -3.4994042 -15.403958  -14.983234  ... -18.01762   -17.112291\n",
      "    -17.812738 ]\n",
      "   ...\n",
      "   [-69.14514   -70.258224  -70.258224  ... -70.258224  -70.258224\n",
      "    -67.564224 ]\n",
      "   [-69.18891   -70.258224  -70.258224  ... -70.258224  -70.258224\n",
      "    -67.57083  ]\n",
      "   [-69.132866  -70.258224  -70.258224  ... -70.258224  -70.258224\n",
      "    -67.467186 ]]]\n",
      "\n",
      "\n",
      " [[[-42.9265    -42.9265    -42.9265    ... -31.912535  -32.429855\n",
      "     -9.247425 ]\n",
      "   [-42.9265    -42.9265    -42.9265    ... -36.05057   -35.131535\n",
      "     -9.390917 ]\n",
      "   [-42.9265    -42.9265    -42.9265    ... -43.139233  -38.96675\n",
      "     -8.305247 ]\n",
      "   ...\n",
      "   [-42.9265    -42.9265    -42.9265    ... -70.89151   -70.89151\n",
      "    -67.463486 ]\n",
      "   [-42.9265    -42.9265    -42.9265    ... -70.89151   -70.89151\n",
      "    -68.01687  ]\n",
      "   [-42.9265    -42.9265    -42.9265    ... -70.89151   -70.89151\n",
      "    -68.23981  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-23.189503  -12.458358   -6.7279506 ... -10.70175   -17.121805\n",
      "    -10.481558 ]\n",
      "   [-23.426275  -14.003174  -12.283264  ... -11.232362  -17.607449\n",
      "    -17.0889   ]\n",
      "   [-21.570883  -17.737047  -11.832243  ... -10.068784  -20.777458\n",
      "    -28.5151   ]\n",
      "   ...\n",
      "   [-57.66839   -57.66839   -57.66839   ... -57.66839   -57.66839\n",
      "    -57.66839  ]\n",
      "   [-57.66839   -57.66839   -57.66839   ... -57.66839   -57.66839\n",
      "    -57.66839  ]\n",
      "   [-57.66839   -57.66839   -57.66839   ... -57.66839   -57.66839\n",
      "    -57.66839  ]]]\n",
      "\n",
      "\n",
      " [[[-15.008345  -12.679447  -15.725718  ... -11.342906  -22.151297\n",
      "    -20.54619  ]\n",
      "   [-15.830584  -13.827162  -20.88015   ... -17.159618  -26.25523\n",
      "    -23.960331 ]\n",
      "   [-20.908886  -18.833803  -29.452078  ... -27.975697  -25.934437\n",
      "    -29.664803 ]\n",
      "   ...\n",
      "   [-64.23094   -64.23094   -64.23094   ... -64.23094   -64.23094\n",
      "    -64.23094  ]\n",
      "   [-64.23094   -64.23094   -64.23094   ... -64.23094   -64.23094\n",
      "    -64.23094  ]\n",
      "   [-64.23094   -64.23094   -64.23094   ... -64.23094   -64.23094\n",
      "    -64.23094  ]]]\n",
      "\n",
      "\n",
      " [[[  5.9995885  -2.8656206   3.8991044 ...  10.410179    6.387372\n",
      "      3.2207227]\n",
      "   [  0.6032081  -8.289767   -1.2539171 ...  10.246154    1.4861096\n",
      "     -2.4666855]\n",
      "   [ -9.181265  -17.822935   -9.635764  ...   7.701995   -6.783186\n",
      "     -5.593619 ]\n",
      "   ...\n",
      "   [-67.06136   -67.06136   -67.06136   ... -67.06136   -67.06136\n",
      "    -67.06136  ]\n",
      "   [-67.06136   -67.06136   -67.06136   ... -67.06136   -67.06136\n",
      "    -67.06136  ]\n",
      "   [-67.06136   -67.06136   -67.06136   ... -67.06136   -67.06136\n",
      "    -67.06136  ]]]] | Labels tensor([0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 8 | Inputs [[[[-17.504105  -20.548058  -17.290062  ...  -9.315282  -14.952741\n",
      "    -14.709768 ]\n",
      "   [-61.794983  -61.794983  -61.794983  ... -61.794983  -61.794983\n",
      "    -61.794983 ]\n",
      "   [-18.596138  -23.028046  -25.506363  ... -18.43958   -29.169073\n",
      "    -22.635647 ]\n",
      "   ...\n",
      "   [-79.08216   -79.08216   -79.08216   ... -79.08216   -79.08216\n",
      "    -79.08216  ]\n",
      "   [-79.08216   -79.08216   -79.08216   ... -79.08216   -79.08216\n",
      "    -79.08216  ]\n",
      "   [-79.08216   -79.08216   -79.08216   ... -79.08216   -79.08216\n",
      "    -79.08216  ]]]\n",
      "\n",
      "\n",
      " [[[-13.104696  -20.071508  -22.981474  ... -13.564444  -16.044085\n",
      "    -11.3521385]\n",
      "   [-16.243992  -20.295692  -26.09127   ... -17.063894  -18.95419\n",
      "    -15.97361  ]\n",
      "   [-22.515675  -25.160563  -31.592463  ... -23.873365  -23.758648\n",
      "    -23.887003 ]\n",
      "   ...\n",
      "   [-79.35167   -79.35167   -79.35167   ... -79.35167   -79.35167\n",
      "    -79.35167  ]\n",
      "   [-79.35167   -79.35167   -79.35167   ... -79.35167   -79.35167\n",
      "    -79.35167  ]\n",
      "   [-79.35167   -79.35167   -79.35167   ... -79.35167   -79.35167\n",
      "    -79.35167  ]]]\n",
      "\n",
      "\n",
      " [[[-11.024256  -12.583972   -9.827834  ...  -7.5651145 -14.605196\n",
      "    -10.606113 ]\n",
      "   [-17.660965  -18.668177  -13.333182  ... -12.201159  -20.534536\n",
      "    -12.168641 ]\n",
      "   [-27.221981  -27.8005    -19.666342  ... -18.509958  -30.196842\n",
      "    -16.79128  ]\n",
      "   ...\n",
      "   [-65.37124   -65.37124   -65.37124   ... -65.37124   -65.37124\n",
      "    -65.37124  ]\n",
      "   [-65.37124   -65.37124   -65.37124   ... -65.37124   -65.37124\n",
      "    -65.37124  ]\n",
      "   [-65.37124   -65.37124   -65.37124   ... -65.37124   -65.37124\n",
      "    -65.37124  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  7.735075    8.919165    9.056179  ...  11.881262   11.645399\n",
      "      4.095229 ]\n",
      "   [  3.3367424   4.718794    6.943588  ...   6.781798    8.707505\n",
      "     -1.7656254]\n",
      "   [ -4.5175414  -2.3899128   2.818464  ...   1.7499233   3.043147\n",
      "     -2.8070512]\n",
      "   ...\n",
      "   [-31.867447  -31.867447  -31.867447  ... -31.867447  -31.867447\n",
      "    -31.867447 ]\n",
      "   [-31.867447  -31.867447  -31.867447  ... -31.867447  -31.867447\n",
      "    -31.867447 ]\n",
      "   [-31.867447  -31.867447  -31.867447  ... -31.867447  -31.867447\n",
      "    -31.867447 ]]]\n",
      "\n",
      "\n",
      " [[[-41.28028   -16.55359   -13.250781  ... -33.225662  -25.84341\n",
      "    -32.17346  ]\n",
      "   [-22.298555  -21.611444  -18.01546   ... -29.52789   -28.04572\n",
      "    -29.657848 ]\n",
      "   [-24.492828  -29.062523  -26.472853  ... -32.554146  -33.284363\n",
      "    -26.862867 ]\n",
      "   ...\n",
      "   [-74.14062   -74.14062   -74.14062   ... -74.14062   -74.14062\n",
      "    -74.14062  ]\n",
      "   [-74.14062   -74.14062   -74.14062   ... -74.14062   -74.14062\n",
      "    -74.14062  ]\n",
      "   [-74.14062   -74.14062   -74.14062   ... -74.14062   -74.14062\n",
      "    -74.14062  ]]]\n",
      "\n",
      "\n",
      " [[[-13.45648   -14.858533  -27.257986  ... -36.088913  -15.794028\n",
      "    -40.35477  ]\n",
      "   [-18.463848  -18.815067  -17.664328  ... -17.59076   -16.61683\n",
      "    -25.513527 ]\n",
      "   [-26.663378  -20.117895  -16.59312   ... -19.129866  -20.512867\n",
      "    -20.760298 ]\n",
      "   ...\n",
      "   [-59.98359   -59.98359   -59.98359   ... -59.98359   -59.98359\n",
      "    -59.98359  ]\n",
      "   [-59.98359   -59.98359   -59.98359   ... -59.98359   -59.98359\n",
      "    -59.98359  ]\n",
      "   [-69.90257   -69.90257   -69.90257   ... -69.90257   -69.90257\n",
      "    -69.90257  ]]]] | Labels tensor([0, 0, 0, 1, 1, 1, 1, 0, 1, 0, 1, 1, 0, 1, 0, 0])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 9 | Inputs [[[[-38.180717   -36.352425   -14.120342   ... -42.260628\n",
      "    -48.822113   -36.40067   ]\n",
      "   [-39.756012   -38.005806   -13.641741   ... -44.18917\n",
      "    -50.999413   -39.026417  ]\n",
      "   [-43.314377   -40.809944   -15.1034565  ... -44.76559\n",
      "    -53.665203   -43.789482  ]\n",
      "   ...\n",
      "   [-64.50119    -64.50119    -64.50119    ... -64.50119\n",
      "    -64.50119    -64.50119   ]\n",
      "   [-64.50119    -64.50119    -64.50119    ... -64.50119\n",
      "    -64.50119    -64.50119   ]\n",
      "   [-64.50119    -64.50119    -64.50119    ... -64.50119\n",
      "    -64.50119    -64.50119   ]]]\n",
      "\n",
      "\n",
      " [[[  1.8208705    9.779903    11.024435   ...  -1.1421776\n",
      "      4.036704     7.5649805 ]\n",
      "   [  1.5925136    7.9945974    8.574881   ...   3.289139\n",
      "      4.9710426    3.307352  ]\n",
      "   [  7.3042607    5.5298204    4.755139   ...   3.3470283\n",
      "      6.5671854   -2.122073  ]\n",
      "   ...\n",
      "   [-46.085907   -47.094757   -45.189247   ... -47.841755\n",
      "    -45.17544    -45.131756  ]\n",
      "   [-44.331646   -44.666435   -46.31736    ... -44.72445\n",
      "    -45.98137    -47.230988  ]\n",
      "   [-45.921516   -46.17485    -46.42166    ... -45.89589\n",
      "    -48.837387   -49.618492  ]]]\n",
      "\n",
      "\n",
      " [[[-13.968797   -11.538733   -26.748716   ... -19.707684\n",
      "    -14.174205    -7.986837  ]\n",
      "   [-18.500504   -17.76074    -21.886375   ... -23.471989\n",
      "    -20.52862    -14.614372  ]\n",
      "   [-24.35523    -23.37436    -24.904375   ... -29.6998\n",
      "    -26.683006   -23.55708   ]\n",
      "   ...\n",
      "   [-70.84253    -70.84253    -70.84253    ... -70.84253\n",
      "    -70.84253    -70.84253   ]\n",
      "   [-70.84253    -70.84253    -70.84253    ... -70.84253\n",
      "    -70.84253    -70.84253   ]\n",
      "   [-70.84253    -70.84253    -70.84253    ... -70.84253\n",
      "    -70.84253    -70.84253   ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[  0.9527928    6.4280677    5.1756196  ... -16.078705\n",
      "      4.832307    -1.5835478 ]\n",
      "   [  6.8558598    8.74172      4.2113304  ...   0.27370486\n",
      "      8.937391     0.36822927]\n",
      "   [  3.1974802    6.9581633    4.5985847  ...  -1.0927835\n",
      "      5.779138     7.6635003 ]\n",
      "   ...\n",
      "   [-48.150806   -44.859474   -47.379173   ... -47.17328\n",
      "    -49.102898   -45.77179   ]\n",
      "   [-48.10927    -47.389027   -47.83458    ... -47.27752\n",
      "    -46.607513   -48.162495  ]\n",
      "   [-50.53423    -48.12339    -47.07596    ... -45.428547\n",
      "    -46.348804   -46.21996   ]]]\n",
      "\n",
      "\n",
      " [[[-18.981716    -9.172977    -4.441716   ... -13.9523945\n",
      "    -19.992596   -42.037983  ]\n",
      "   [-19.949764   -12.193603   -10.442442   ... -20.098812\n",
      "    -20.938797   -26.815311  ]\n",
      "   [-24.748552   -18.62059    -22.577627   ... -31.150562\n",
      "    -23.19005    -22.409971  ]\n",
      "   ...\n",
      "   [-67.56446    -67.56446    -67.56446    ... -67.56446\n",
      "    -67.56446    -67.56446   ]\n",
      "   [-67.56446    -67.56446    -67.56446    ... -67.56446\n",
      "    -67.56446    -67.56446   ]\n",
      "   [-67.56446    -67.56446    -67.56446    ... -67.56446\n",
      "    -67.56446    -67.56446   ]]]\n",
      "\n",
      "\n",
      " [[[  0.65607846   1.4959639   -6.0878186  ... -14.942288\n",
      "     -7.4953766  -10.233379  ]\n",
      "   [ -4.52126      4.534871   -11.478287   ...  -1.4523829\n",
      "      1.4595352  -16.120659  ]\n",
      "   [ -5.306469     4.415786    -0.3731467  ...   2.2411754\n",
      "      5.456705   -27.248373  ]\n",
      "   ...\n",
      "   [-48.702644   -46.599777   -45.938908   ... -47.002796\n",
      "    -47.92903    -48.869175  ]\n",
      "   [-44.54413    -46.659245   -45.130646   ... -46.856796\n",
      "    -46.72825    -48.016647  ]\n",
      "   [-47.481407   -48.90923    -47.86444    ... -45.59837\n",
      "    -48.104164   -45.701134  ]]]] | Labels tensor([0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 1, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 10 | Inputs [[[[-22.472494  -11.98537   -11.156849  ... -14.150076  -11.57426\n",
      "     -4.209364 ]\n",
      "   [ -9.37768   -14.393028   -6.8530173 ... -11.872759  -14.802191\n",
      "    -10.395957 ]\n",
      "   [-11.227013  -10.929445   -8.304149  ... -14.15846   -20.187265\n",
      "    -18.297611 ]\n",
      "   ...\n",
      "   [-44.871174  -47.598846  -47.66321   ... -48.67893   -44.955116\n",
      "    -50.61631  ]\n",
      "   [-46.65435   -47.229385  -45.421127  ... -46.803154  -45.08282\n",
      "    -45.906624 ]\n",
      "   [-48.02865   -46.1419    -47.78482   ... -46.613003  -47.99066\n",
      "    -45.111317 ]]]\n",
      "\n",
      "\n",
      " [[[-28.362164  -18.521084  -16.242191  ... -24.111916  -18.745235\n",
      "    -23.749317 ]\n",
      "   [-23.749056  -22.084564  -21.509716  ... -30.680717  -22.83709\n",
      "    -28.603779 ]\n",
      "   [-27.747448  -27.225044  -30.796755  ... -35.253136  -30.258102\n",
      "    -31.056473 ]\n",
      "   ...\n",
      "   [-87.136986  -87.136986  -87.136986  ... -87.136986  -87.136986\n",
      "    -87.136986 ]\n",
      "   [-87.136986  -87.136986  -87.136986  ... -87.136986  -87.136986\n",
      "    -87.136986 ]\n",
      "   [-87.136986  -87.136986  -87.136986  ... -87.136986  -87.136986\n",
      "    -87.136986 ]]]\n",
      "\n",
      "\n",
      " [[[-13.868613  -23.294134  -34.738335  ... -41.398983  -19.442795\n",
      "    -27.258814 ]\n",
      "   [-15.584302  -28.101568  -35.258858  ... -41.398983  -25.685642\n",
      "    -18.835918 ]\n",
      "   [-17.348343  -35.271988  -33.23394   ... -41.398983  -32.95912\n",
      "    -17.909206 ]\n",
      "   ...\n",
      "   [-52.515205  -63.64754   -63.64754   ... -41.398983  -63.64754\n",
      "    -50.366512 ]\n",
      "   [-52.51165   -63.64754   -63.64754   ... -41.398983  -63.64754\n",
      "    -50.465874 ]\n",
      "   [-52.431015  -63.64754   -63.64754   ... -41.398983  -63.64754\n",
      "    -50.437046 ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-17.00245   -22.060505  -22.746674  ... -28.182789  -27.618818\n",
      "    -11.481103 ]\n",
      "   [-19.897552  -23.287622  -24.635666  ... -21.387154  -23.843506\n",
      "    -16.946262 ]\n",
      "   [-25.03469   -26.331207  -28.466248  ... -22.003     -25.747818\n",
      "    -26.257957 ]\n",
      "   ...\n",
      "   [-70.81916   -70.81916   -70.81916   ... -70.81916   -70.81916\n",
      "    -70.81916  ]\n",
      "   [-70.81916   -70.81916   -70.81916   ... -70.81916   -70.81916\n",
      "    -70.81916  ]\n",
      "   [-70.81916   -70.81916   -70.81916   ... -70.81916   -70.81916\n",
      "    -70.81916  ]]]\n",
      "\n",
      "\n",
      " [[[-39.085194  -23.57153   -26.844847  ... -18.426416  -29.879421\n",
      "    -27.49831  ]\n",
      "   [-45.13684   -28.650034  -28.301258  ... -23.02253   -31.415958\n",
      "    -34.13221  ]\n",
      "   [-41.36908   -32.5762    -29.770725  ... -29.718058  -36.66877\n",
      "    -43.69673  ]\n",
      "   ...\n",
      "   [-66.91493   -66.91493   -66.91493   ... -66.91493   -66.91493\n",
      "    -66.91493  ]\n",
      "   [-66.91493   -66.91493   -66.91493   ... -66.91493   -66.91493\n",
      "    -66.91493  ]\n",
      "   [-66.91493   -66.91493   -66.91493   ... -66.91493   -66.91493\n",
      "    -66.91493  ]]]\n",
      "\n",
      "\n",
      " [[[-30.415878  -19.419441  -13.531236  ... -22.776491  -19.786016\n",
      "    -21.203108 ]\n",
      "   [-17.155582  -24.83562   -19.152618  ... -25.38723   -22.921593\n",
      "    -22.041996 ]\n",
      "   [-21.025097  -22.329693  -27.718153  ... -29.997719  -28.58561\n",
      "    -24.51911  ]\n",
      "   ...\n",
      "   [-73.68895   -73.68895   -73.68895   ... -73.68895   -73.68895\n",
      "    -73.68895  ]\n",
      "   [-73.68895   -73.68895   -73.68895   ... -73.68895   -73.68895\n",
      "    -73.68895  ]\n",
      "   [-73.68895   -73.68895   -73.68895   ... -73.68895   -73.68895\n",
      "    -73.68895  ]]]] | Labels tensor([1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 1, 0, 0, 0, 0])\n",
      "(16, 1, 64, 376)\n",
      "Epoch: 11 | Inputs [[[[-1.29726410e+01 -1.71797981e+01 -1.29474535e+01 ... -4.46639347e+00\n",
      "    -1.02110004e+01 -8.49606705e+00]\n",
      "   [-1.75711975e+01 -8.72747612e+00 -8.57494831e+00 ... -7.75650311e+00\n",
      "    -1.49746981e+01 -1.10343342e+01]\n",
      "   [-1.37699022e+01 -8.10549164e+00 -9.31813812e+00 ... -1.43385849e+01\n",
      "    -1.58404016e+01 -9.13316727e+00]\n",
      "   ...\n",
      "   [-6.47424698e+01 -6.47424698e+01 -6.47424698e+01 ... -6.47424698e+01\n",
      "    -6.47424698e+01 -6.47424698e+01]\n",
      "   [-6.47424698e+01 -6.47424698e+01 -6.47424698e+01 ... -6.47424698e+01\n",
      "    -6.47424698e+01 -6.47424698e+01]\n",
      "   [-6.47424698e+01 -6.47424698e+01 -6.47424698e+01 ... -6.47424698e+01\n",
      "    -6.47424698e+01 -6.47424698e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.48493681e+01 -1.02393932e+01  6.52967644e+00 ...  8.22240412e-01\n",
      "    -5.40210342e+00 -7.15413690e-01]\n",
      "   [-1.38891625e+00  3.44959784e+00  8.94361401e+00 ...  5.90128899e+00\n",
      "     2.56283879e+00  5.83130074e+00]\n",
      "   [-4.82711124e+00  5.34113789e+00  7.19390392e+00 ...  7.43341255e+00\n",
      "     6.28764868e+00  7.93631601e+00]\n",
      "   ...\n",
      "   [-4.62624512e+01 -4.76925354e+01 -4.89021378e+01 ... -4.81501198e+01\n",
      "    -4.83443146e+01 -4.90249901e+01]\n",
      "   [-4.57874489e+01 -4.72388878e+01 -4.90249901e+01 ... -4.69256783e+01\n",
      "    -4.90249901e+01 -4.52044373e+01]\n",
      "   [-4.63382530e+01 -4.78347435e+01 -4.87888870e+01 ... -4.72999954e+01\n",
      "    -4.90109100e+01 -4.73794479e+01]]]\n",
      "\n",
      "\n",
      " [[[-1.52878113e+01 -1.70048122e+01 -1.80511932e+01 ... -1.67979488e+01\n",
      "    -1.67594795e+01 -3.37624588e+01]\n",
      "   [-2.06865864e+01 -2.36238747e+01 -2.40132236e+01 ... -2.13725014e+01\n",
      "    -2.16898384e+01 -2.26147442e+01]\n",
      "   [-1.81202164e+01 -2.53308754e+01 -2.84374008e+01 ... -2.72146511e+01\n",
      "    -2.76914444e+01 -2.17231770e+01]\n",
      "   ...\n",
      "   [-8.24124527e+01 -8.29845810e+01 -8.29845810e+01 ... -8.29845810e+01\n",
      "    -8.29845810e+01 -8.29845810e+01]\n",
      "   [-8.25814896e+01 -8.29845810e+01 -8.29845810e+01 ... -8.29845810e+01\n",
      "    -8.29845810e+01 -8.29845810e+01]\n",
      "   [-8.26200409e+01 -8.29845810e+01 -8.29845810e+01 ... -8.29845810e+01\n",
      "    -8.29845810e+01 -8.29845810e+01]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-2.26152077e+01 -5.90113144e+01 -5.90113144e+01 ... -1.67565365e+01\n",
      "    -3.38916473e+01 -1.83554134e+01]\n",
      "   [-1.55865812e+01 -5.90113144e+01 -5.90113144e+01 ... -2.11666183e+01\n",
      "    -3.31909904e+01 -2.30078888e+01]\n",
      "   [-1.96954346e+01 -5.90113144e+01 -5.90113144e+01 ... -2.83222218e+01\n",
      "    -3.41545753e+01 -3.07431316e+01]\n",
      "   ...\n",
      "   [-7.01635056e+01 -5.90113144e+01 -5.90113144e+01 ... -7.01635056e+01\n",
      "    -7.01635056e+01 -7.01635056e+01]\n",
      "   [-7.01635056e+01 -5.90113144e+01 -5.90113144e+01 ... -7.01635056e+01\n",
      "    -7.01635056e+01 -7.01635056e+01]\n",
      "   [-7.01635056e+01 -5.90113144e+01 -5.90113144e+01 ... -7.01635056e+01\n",
      "    -7.01635056e+01 -7.01635056e+01]]]\n",
      "\n",
      "\n",
      " [[[-8.38525295e+00 -3.86147976e-01 -4.00707901e-01 ... -6.99776125e+00\n",
      "    -7.47153234e+00 -1.51610394e+01]\n",
      "   [-7.61671925e+00  8.60858858e-02 -1.00884938e+00 ... -5.79113913e+00\n",
      "    -9.25018883e+00 -8.96459103e+00]\n",
      "   [-5.58846664e+00 -4.51145411e+00 -5.61284542e+00 ... -8.33106327e+00\n",
      "    -1.21957932e+01 -1.26470623e+01]\n",
      "   ...\n",
      "   [-3.67749214e+01 -3.67749214e+01 -3.67749214e+01 ... -3.67749214e+01\n",
      "    -3.67749214e+01 -3.67749214e+01]\n",
      "   [-4.72989311e+01 -4.72989311e+01 -4.72989311e+01 ... -4.72989311e+01\n",
      "    -4.72989311e+01 -4.72989311e+01]\n",
      "   [-4.72989311e+01 -4.72989311e+01 -4.72989311e+01 ... -4.72989311e+01\n",
      "    -4.72989311e+01 -4.72989311e+01]]]\n",
      "\n",
      "\n",
      " [[[ 8.96491432e+00 -1.27113876e+01 -6.53498125e+00 ... -8.37479210e+00\n",
      "    -8.71419716e+00  8.24138069e+00]\n",
      "   [ 1.14720964e+01 -1.51193608e-02 -1.16084719e+00 ... -2.15565491e+00\n",
      "    -2.90797925e+00  1.44543228e+01]\n",
      "   [ 7.56885958e+00  1.60956478e+01  1.51453409e+01 ...  1.50483589e+01\n",
      "     1.59221878e+01  1.71012993e+01]\n",
      "   ...\n",
      "   [-4.65163841e+01 -4.63485680e+01 -4.65767632e+01 ... -4.99678001e+01\n",
      "    -4.71816711e+01 -4.54992256e+01]\n",
      "   [-4.69727287e+01 -4.64720230e+01 -4.61796989e+01 ... -4.57185059e+01\n",
      "    -4.62818069e+01 -4.51637726e+01]\n",
      "   [-4.57064400e+01 -4.85268822e+01 -4.81708298e+01 ... -4.62291145e+01\n",
      "    -4.67929611e+01 -4.76083679e+01]]]] | Labels tensor([0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0, 1, 1])\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(16, 1, 64, 376)\n",
      "Epoch: 12 | Inputs [[[[-44.391136  -28.35315   -27.923471  ... -47.778275  -65.05765\n",
      "    -41.27161  ]\n",
      "   [-39.074863  -30.5932    -32.445072  ... -49.34575   -65.05765\n",
      "    -41.66435  ]\n",
      "   [-37.96091   -35.059933  -40.43709   ... -50.87969   -63.645203\n",
      "    -43.286987 ]\n",
      "   ...\n",
      "   [-65.05765   -65.05765   -65.05765   ... -65.05765   -65.05765\n",
      "    -65.05765  ]\n",
      "   [-65.05765   -65.05765   -65.05765   ... -65.05765   -65.05765\n",
      "    -65.05765  ]\n",
      "   [-65.05765   -65.05765   -65.05765   ... -65.05765   -65.05765\n",
      "    -65.05765  ]]]\n",
      "\n",
      "\n",
      " [[[-14.240812  -14.82009    -3.060028  ...  -9.834965  -19.06961\n",
      "    -21.733292 ]\n",
      "   [ -5.8013673  -9.536169   -3.8414865 ... -10.361556  -15.5327015\n",
      "     -9.319998 ]\n",
      "   [ -6.0389524  -8.619074   -8.995945  ... -12.627011  -18.437597\n",
      "     -9.428795 ]\n",
      "   ...\n",
      "   [-50.0438    -47.848034  -47.25229   ... -46.935368  -46.186024\n",
      "    -45.775856 ]\n",
      "   [-47.37566   -45.12159   -44.050552  ... -48.39378   -46.12255\n",
      "    -45.719585 ]\n",
      "   [-48.60083   -45.80414   -46.465878  ... -46.73803   -47.001476\n",
      "    -47.955593 ]]]\n",
      "\n",
      "\n",
      " [[[ -1.5364621  -4.8102527 -10.298048  ...  -4.042132  -17.712454\n",
      "     -5.1935434]\n",
      "   [ -5.350003   -4.1857924 -14.497107  ...  -4.7783604 -11.192726\n",
      "    -10.852765 ]\n",
      "   [ -9.1297245  -7.9734225 -10.022191  ...  -6.829151  -14.2909155\n",
      "     -8.978205 ]\n",
      "   ...\n",
      "   [-46.539875  -47.220314  -47.44434   ... -46.669296  -46.62023\n",
      "    -46.43513  ]\n",
      "   [-45.644188  -46.350365  -45.592075  ... -47.62355   -46.742977\n",
      "    -47.025986 ]\n",
      "   [-47.687653  -47.7371    -46.902164  ... -48.853218  -46.551952\n",
      "    -45.97908  ]]]\n",
      "\n",
      "\n",
      " ...\n",
      "\n",
      "\n",
      " [[[-34.142212  -30.866055  -17.679888  ... -37.764442  -36.927856\n",
      "    -29.178951 ]\n",
      "   [-30.842579  -30.769066  -14.967593  ... -37.87191   -31.095083\n",
      "    -34.895485 ]\n",
      "   [-35.232662  -35.475334  -15.4988    ... -38.869644  -31.71125\n",
      "    -40.801243 ]\n",
      "   ...\n",
      "   [-62.93333   -62.93333   -62.93333   ... -62.93333   -62.93333\n",
      "    -62.93333  ]\n",
      "   [-62.93333   -62.93333   -62.93333   ... -62.93333   -62.93333\n",
      "    -62.93333  ]\n",
      "   [-62.93333   -62.93333   -62.93333   ... -62.93333   -62.93333\n",
      "    -62.93333  ]]]\n",
      "\n",
      "\n",
      " [[[-17.955688  -25.422737  -30.08348   ... -18.91456   -15.960645\n",
      "    -19.879623 ]\n",
      "   [-23.021915  -31.285099  -31.967289  ... -24.62347   -21.513432\n",
      "    -25.955563 ]\n",
      "   [-30.801857  -34.66955   -30.997198  ... -27.297003  -30.46988\n",
      "    -34.63643  ]\n",
      "   ...\n",
      "   [-81.76873   -81.76873   -81.76873   ... -81.76873   -81.76873\n",
      "    -81.76873  ]\n",
      "   [-81.76873   -81.76873   -81.76873   ... -81.76873   -81.76873\n",
      "    -81.76873  ]\n",
      "   [-81.76873   -81.76873   -81.76873   ... -81.76873   -81.76873\n",
      "    -81.76873  ]]]\n",
      "\n",
      "\n",
      " [[[ -5.5420036  -6.8835645  -4.9211683 ...  -9.11941    -6.46945\n",
      "     -3.483342 ]\n",
      "   [ -9.039669  -11.8515835 -10.452862  ... -15.396358  -11.423761\n",
      "     -9.197176 ]\n",
      "   [-59.7679    -59.7679    -59.7679    ... -59.7679    -59.7679\n",
      "    -59.7679   ]\n",
      "   ...\n",
      "   [-73.15157   -73.15157   -73.15157   ... -73.15157   -73.15157\n",
      "    -73.15157  ]\n",
      "   [-73.15157   -73.15157   -73.15157   ... -73.15157   -73.15157\n",
      "    -73.15157  ]\n",
      "   [-73.15157   -73.15157   -73.15157   ... -73.15157   -73.15157\n",
      "    -73.15157  ]]]] | Labels tensor([0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0])\n",
      "(2, 1, 64, 376)\n",
      "Epoch: 13 | Inputs [[[[-25.110043 -18.212618 -23.541803 ... -14.511891 -18.80581\n",
      "    -17.181162]\n",
      "   [-21.979342 -21.537552 -25.217682 ... -16.597769 -21.201494\n",
      "    -23.536552]\n",
      "   [-24.912006 -27.712364 -30.207918 ... -22.403584 -25.49794\n",
      "    -33.250248]\n",
      "   ...\n",
      "   [-62.80931  -62.80931  -62.80931  ... -62.80931  -62.80931\n",
      "    -62.80931 ]\n",
      "   [-62.80931  -62.80931  -62.80931  ... -62.80931  -62.80931\n",
      "    -62.80931 ]\n",
      "   [-62.80931  -62.80931  -62.80931  ... -62.80931  -62.80931\n",
      "    -62.80931 ]]]\n",
      "\n",
      "\n",
      " [[[-30.557173 -33.69393  -36.486336 ... -24.906853 -30.036001\n",
      "    -26.447334]\n",
      "   [-35.42009  -35.924755 -39.825893 ... -29.547188 -33.886936\n",
      "    -25.32927 ]\n",
      "   [-39.49123  -29.960506 -31.524637 ... -28.215267 -30.192886\n",
      "    -27.4817  ]\n",
      "   ...\n",
      "   [-66.01755  -66.01755  -66.01755  ... -66.01755  -66.01755\n",
      "    -66.01755 ]\n",
      "   [-66.01755  -66.01755  -66.01755  ... -66.01755  -66.01755\n",
      "    -66.01755 ]\n",
      "   [-66.01755  -66.01755  -66.01755  ... -66.01755  -66.01755\n",
      "    -66.01755 ]]]] | Labels tensor([0, 0])\n"
     ]
    }
   ],
   "source": [
    "import numpy as np\n",
    "for epoch in range(2):\n",
    "    for i, data in enumerate(train_dl, 0):\n",
    "        inputs, labels = data\n",
    "        inputs = np.array(inputs)\n",
    "        print(inputs.shape)\n",
    "        print(f'Epoch: {i} | Inputs {inputs} | Labels {labels}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "551cac03",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "----------------------------------------------------------------\n",
      "        Layer (type)               Output Shape         Param #\n",
      "================================================================\n",
      "            Conv2d-1           [-1, 8, 32, 188]             208\n",
      "            Conv2d-2           [-1, 8, 32, 188]             208\n",
      "              ReLU-3           [-1, 8, 32, 188]               0\n",
      "              ReLU-4           [-1, 8, 32, 188]               0\n",
      "       BatchNorm2d-5           [-1, 8, 32, 188]              16\n",
      "       BatchNorm2d-6           [-1, 8, 32, 188]              16\n",
      "            Conv2d-7           [-1, 16, 16, 94]           1,168\n",
      "            Conv2d-8           [-1, 16, 16, 94]           1,168\n",
      "              ReLU-9           [-1, 16, 16, 94]               0\n",
      "             ReLU-10           [-1, 16, 16, 94]               0\n",
      "      BatchNorm2d-11           [-1, 16, 16, 94]              32\n",
      "      BatchNorm2d-12           [-1, 16, 16, 94]              32\n",
      "           Conv2d-13            [-1, 32, 8, 47]           4,640\n",
      "           Conv2d-14            [-1, 32, 8, 47]           4,640\n",
      "             ReLU-15            [-1, 32, 8, 47]               0\n",
      "             ReLU-16            [-1, 32, 8, 47]               0\n",
      "      BatchNorm2d-17            [-1, 32, 8, 47]              64\n",
      "      BatchNorm2d-18            [-1, 32, 8, 47]              64\n",
      "           Conv2d-19            [-1, 64, 4, 24]          18,496\n",
      "           Conv2d-20            [-1, 64, 4, 24]          18,496\n",
      "             ReLU-21            [-1, 64, 4, 24]               0\n",
      "             ReLU-22            [-1, 64, 4, 24]               0\n",
      "      BatchNorm2d-23            [-1, 64, 4, 24]             128\n",
      "      BatchNorm2d-24            [-1, 64, 4, 24]             128\n",
      "AdaptiveAvgPool2d-25             [-1, 64, 1, 1]               0\n",
      "           Linear-26                    [-1, 2]             130\n",
      "          Sigmoid-27                    [-1, 2]               0\n",
      "================================================================\n",
      "Total params: 49,634\n",
      "Trainable params: 49,634\n",
      "Non-trainable params: 0\n",
      "----------------------------------------------------------------\n",
      "Input size (MB): 0.09\n",
      "Forward/backward pass size (MB): 4.14\n",
      "Params size (MB): 0.19\n",
      "Estimated Total Size (MB): 4.42\n",
      "----------------------------------------------------------------\n"
     ]
    }
   ],
   "source": [
    "import torchsummary\n",
    "torchsummary.summary(Model1, (1,64,376))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "849dc4a2",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Accuracy: 0.87, Total items: 53\n",
      "Accuracy 0.8\n",
      "precision 1.0\n",
      "recall 0.6666666666666666\n",
      "f1 0.8\n",
      "accuracy 0.8\n",
      "precision 1.0\n",
      "recall 0.6666666666666666\n",
      "f1 0.8\n",
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.67      1.00      0.80         2\n",
      "           1       1.00      0.67      0.80         3\n",
      "\n",
      "    accuracy                           0.80         5\n",
      "   macro avg       0.83      0.83      0.80         5\n",
      "weighted avg       0.87      0.80      0.80         5\n",
      "\n"
     ]
    }
   ],
   "source": [
    "def inference (model, val_dl):\n",
    "  correct_prediction = 0\n",
    "  total_prediction = 0\n",
    "\n",
    "  with torch.no_grad():\n",
    "    for data in val_dl:\n",
    "      \n",
    "      inputs, labels = data[0].to(device), data[1].to(device)\n",
    "      inputs_m, inputs_s = inputs.mean(), inputs.std()\n",
    "      inputs = (inputs - inputs_m) / inputs_s\n",
    "      outputs = model(inputs)\n",
    " \n",
    "      _, prediction = torch.max(outputs,1)\n",
    "      \n",
    "      correct_prediction += (prediction == labels).sum().item()\n",
    "      total_prediction += prediction.shape[0]\n",
    "    \n",
    "  acc = correct_prediction/total_prediction\n",
    "  print(f'Accuracy: {acc:.2f}, Total items: {total_prediction}')\n",
    "  \n",
    "  labels = labels.cpu().numpy()\n",
    "  prediction = prediction.cpu().numpy()\n",
    "  accuracy = np.mean(np.equal(labels,prediction))\n",
    "  right = np.sum(labels * prediction == 1)\n",
    "  precision = right / np.sum(prediction)\n",
    "  recall = right / np.sum(labels)\n",
    "  f1 = 2 * precision*recall/(precision+recall)\n",
    "  print('Accuracy', accuracy)\n",
    "  print('precision', precision)\n",
    "  print('recall', recall)\n",
    "  print('f1', f1)\n",
    "  print('accuracy', metrics.accuracy_score(labels,prediction) )\n",
    "  print('precision', metrics.precision_score(labels,prediction) )\n",
    "  print('recall', metrics.recall_score(labels,prediction) )\n",
    "  print('f1', metrics.f1_score(labels,prediction) )\n",
    "  print(metrics.classification_report(labels,prediction))\n",
    "\n",
    "inference(Model1, val_dl)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}