diff --git "a/test.ipynb" "b/test.ipynb"
deleted file mode 100644--- "a/test.ipynb"
+++ /dev/null
@@ -1,35985 +0,0 @@
-{
- "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 1,
-   "id": "18e11bfc",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import numpy as np\n",
-    "import matplotlib.pyplot as plt\n",
-    "import torch as t\n",
-    "from datasets import load_dataset"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "52ee59a9",
-   "metadata": {},
-   "source": [
-    "## Loading the hf data to pytorch dataset\n",
-    "* pytorch dataset is a wrapper of  tuples\n",
-    "* load_dataset is a wrapper around dict"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "id": "119adc87",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "9dceefe92dfd4163b390dc85e2718c8b",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4b5d6915a87342729f011cf37abe6f3b",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Loading dataset shards:   0%|          | 0/64 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "1b526a3efcc44bdab5eec592fd80cc49",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "cd901d6815154e0bade114a4bf69f803",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "train_hf = load_dataset(\"chainyo/rvl-cdip\", split=\"train\")\n",
-    "val_hf = load_dataset(\"chainyo/rvl-cdip\", split=\"val\")\n",
-    "test_hf = load_dataset(\"chainyo/rvl-cdip\", split=\"test\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "id": "411ba4c8",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "8c363a5b417a4af09eb117d9c2f731a4",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "61f0f71a4717455d9ec8956d16c1eff6",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Loading dataset shards:   0%|          | 0/64 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "f8e737e85ad34057893667f32b827f10",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "fb9aed3d9b4245ba9fd7c6cce97694ec",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Resolving data files:   0%|          | 0/119 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "train = load_dataset(\"chainyo/rvl-cdip\", split=\"train\").with_format(type=\"torch\", columns=[\"image\", \"label\"])\n",
-    "val = load_dataset(\"chainyo/rvl-cdip\", split=\"val\").with_format(type=\"torch\", columns=[\"image\", \"label\"])\n",
-    "test = load_dataset(\"chainyo/rvl-cdip\", split=\"test\").with_format(type=\"torch\", columns=[\"image\", \"label\"])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 46,
-   "id": "27e7977b",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Index    | Shape                | Label\n",
-      "----------------------------------------\n",
-      "0        | torch.Size([1, 1000, 754]) | 0\n",
-      "1        | torch.Size([1, 1000, 781]) | 0\n",
-      "2        | torch.Size([1, 1000, 754]) | 0\n",
-      "3        | torch.Size([1, 1000, 754]) | 0\n",
-      "4        | torch.Size([1, 1000, 762]) | 0\n",
-      "5        | torch.Size([1, 1000, 762]) | 0\n",
-      "6        | torch.Size([1, 1000, 783]) | 0\n",
-      "7        | torch.Size([1, 1000, 777]) | 0\n",
-      "8        | torch.Size([1, 1000, 762]) | 0\n",
-      "9        | torch.Size([1, 1000, 754]) | 0\n",
-      "10       | torch.Size([1, 1000, 762]) | 0\n",
-      "11       | torch.Size([1, 1000, 754]) | 0\n",
-      "12       | torch.Size([1, 1000, 762]) | 0\n",
-      "13       | torch.Size([1, 1000, 754]) | 0\n",
-      "14       | torch.Size([1, 1000, 754]) | 0\n",
-      "15       | torch.Size([1, 1000, 771]) | 0\n",
-      "16       | torch.Size([1, 1000, 754]) | 0\n",
-      "17       | torch.Size([1, 1000, 754]) | 0\n",
-      "18       | torch.Size([1, 1000, 784]) | 0\n",
-      "19       | torch.Size([1, 1000, 808]) | 0\n",
-      "20       | torch.Size([1, 1000, 754]) | 0\n",
-      "21       | torch.Size([1, 1000, 757]) | 0\n",
-      "22       | torch.Size([1, 1000, 789]) | 0\n",
-      "23       | torch.Size([1, 1000, 754]) | 0\n",
-      "24       | torch.Size([1, 1000, 762]) | 0\n",
-      "25       | torch.Size([1, 1000, 781]) | 0\n",
-      "26       | torch.Size([1, 1000, 762]) | 0\n",
-      "27       | torch.Size([1, 1000, 754]) | 0\n",
-      "28       | torch.Size([1, 1000, 754]) | 0\n",
-      "29       | torch.Size([1, 1000, 780]) | 0\n",
-      "30       | torch.Size([1, 1000, 777]) | 0\n",
-      "31       | torch.Size([1, 1000, 754]) | 0\n",
-      "32       | torch.Size([1, 1000, 762]) | 0\n",
-      "33       | torch.Size([1, 1000, 754]) | 0\n",
-      "34       | torch.Size([1, 1000, 797]) | 0\n",
-      "35       | torch.Size([1, 1000, 754]) | 0\n",
-      "36       | torch.Size([1, 1000, 780]) | 0\n",
-      "37       | torch.Size([1, 1000, 802]) | 0\n",
-      "38       | torch.Size([1, 1000, 754]) | 0\n",
-      "39       | torch.Size([1, 1000, 796]) | 0\n",
-      "40       | torch.Size([1, 1000, 754]) | 0\n",
-      "41       | torch.Size([1, 1000, 754]) | 0\n",
-      "42       | torch.Size([1, 1000, 754]) | 0\n",
-      "43       | torch.Size([1, 1000, 802]) | 0\n",
-      "44       | torch.Size([1, 1000, 757]) | 0\n",
-      "45       | torch.Size([1, 1000, 776]) | 0\n",
-      "46       | torch.Size([1, 1000, 754]) | 0\n",
-      "47       | torch.Size([1, 1000, 754]) | 0\n",
-      "48       | torch.Size([1, 1000, 777]) | 0\n",
-      "49       | torch.Size([1, 1000, 771]) | 0\n",
-      "50       | torch.Size([1, 1000, 754]) | 0\n",
-      "51       | torch.Size([1, 1000, 754]) | 0\n",
-      "52       | torch.Size([1, 1000, 757]) | 0\n",
-      "53       | torch.Size([1, 1000, 762]) | 0\n",
-      "54       | torch.Size([1, 1000, 757]) | 0\n",
-      "55       | torch.Size([1, 1000, 754]) | 0\n",
-      "56       | torch.Size([1, 1000, 780]) | 0\n",
-      "57       | torch.Size([1, 1000, 754]) | 0\n",
-      "58       | torch.Size([1, 1000, 777]) | 0\n",
-      "59       | torch.Size([1, 1000, 762]) | 0\n",
-      "60       | torch.Size([1, 1000, 757]) | 0\n",
-      "61       | torch.Size([1, 1000, 757]) | 0\n",
-      "62       | torch.Size([1, 1000, 757]) | 0\n",
-      "63       | torch.Size([1, 1000, 762]) | 0\n",
-      "64       | torch.Size([1, 1000, 762]) | 0\n",
-      "65       | torch.Size([1, 1000, 754]) | 0\n",
-      "66       | torch.Size([1, 1000, 754]) | 0\n",
-      "67       | torch.Size([1, 1000, 754]) | 0\n",
-      "68       | torch.Size([1, 1000, 754]) | 0\n",
-      "69       | torch.Size([1, 1000, 754]) | 0\n",
-      "70       | torch.Size([1, 1000, 757]) | 0\n",
-      "71       | torch.Size([1, 1000, 754]) | 0\n",
-      "72       | torch.Size([1, 1000, 782]) | 0\n",
-      "73       | torch.Size([1, 1000, 777]) | 0\n",
-      "74       | torch.Size([1, 1000, 762]) | 0\n",
-      "75       | torch.Size([1, 1000, 754]) | 0\n",
-      "76       | torch.Size([1, 1000, 754]) | 0\n",
-      "77       | torch.Size([1, 1000, 772]) | 0\n",
-      "78       | torch.Size([1, 1000, 754]) | 0\n",
-      "79       | torch.Size([1, 1000, 754]) | 0\n",
-      "80       | torch.Size([1, 1000, 754]) | 0\n",
-      "81       | torch.Size([1, 1000, 757]) | 0\n",
-      "82       | torch.Size([1, 1000, 754]) | 0\n",
-      "83       | torch.Size([1, 1000, 754]) | 0\n",
-      "84       | torch.Size([1, 1000, 757]) | 0\n",
-      "85       | torch.Size([1, 1000, 780]) | 0\n",
-      "86       | torch.Size([1, 1000, 951]) | 0\n",
-      "87       | torch.Size([1, 1000, 754]) | 0\n",
-      "88       | torch.Size([1, 1000, 751]) | 0\n",
-      "89       | torch.Size([1, 1000, 757]) | 0\n",
-      "90       | torch.Size([1, 1000, 754]) | 0\n",
-      "91       | torch.Size([1, 1000, 771]) | 0\n",
-      "92       | torch.Size([1, 1000, 754]) | 0\n",
-      "93       | torch.Size([1, 1000, 754]) | 0\n",
-      "94       | torch.Size([1, 1000, 754]) | 0\n",
-      "95       | torch.Size([1, 1000, 762]) | 0\n",
-      "96       | torch.Size([1, 1000, 771]) | 0\n",
-      "97       | torch.Size([1, 1000, 754]) | 0\n",
-      "98       | torch.Size([1, 1000, 611]) | 0\n",
-      "99       | torch.Size([1, 1000, 754]) | 0\n",
-      "100      | torch.Size([1, 1000, 780]) | 0\n",
-      "101      | torch.Size([1, 1000, 813]) | 0\n",
-      "102      | torch.Size([1, 1000, 754]) | 0\n",
-      "103      | torch.Size([1, 1000, 773]) | 0\n",
-      "104      | torch.Size([1, 1000, 754]) | 0\n",
-      "105      | torch.Size([1, 1000, 774]) | 0\n",
-      "106      | torch.Size([1, 1000, 762]) | 0\n",
-      "107      | torch.Size([1, 1000, 762]) | 0\n",
-      "108      | torch.Size([1, 1000, 762]) | 0\n",
-      "109      | torch.Size([1, 1000, 773]) | 0\n",
-      "110      | torch.Size([1, 1000, 754]) | 0\n",
-      "111      | torch.Size([1, 1000, 754]) | 0\n",
-      "112      | torch.Size([1, 1000, 754]) | 0\n",
-      "113      | torch.Size([1, 1000, 751]) | 0\n",
-      "114      | torch.Size([1, 1000, 741]) | 0\n",
-      "115      | torch.Size([1, 1000, 754]) | 0\n",
-      "116      | torch.Size([1, 1000, 754]) | 0\n",
-      "117      | torch.Size([1, 1000, 771]) | 0\n",
-      "118      | torch.Size([1, 1000, 754]) | 0\n",
-      "119      | torch.Size([1, 1000, 775]) | 0\n",
-      "120      | torch.Size([1, 1000, 777]) | 0\n",
-      "121      | torch.Size([1, 1000, 754]) | 0\n",
-      "122      | torch.Size([1, 1000, 754]) | 0\n",
-      "123      | torch.Size([1, 1000, 772]) | 0\n",
-      "124      | torch.Size([1, 1000, 762]) | 0\n",
-      "125      | torch.Size([1, 1000, 762]) | 0\n",
-      "126      | torch.Size([1, 1000, 772]) | 0\n",
-      "127      | torch.Size([1, 1000, 778]) | 0\n",
-      "128      | torch.Size([1, 1000, 786]) | 0\n",
-      "129      | torch.Size([1, 1000, 754]) | 0\n",
-      "130      | torch.Size([1, 1000, 754]) | 0\n",
-      "131      | torch.Size([1, 1000, 777]) | 0\n",
-      "132      | torch.Size([1, 1000, 769]) | 0\n",
-      "133      | torch.Size([1, 1000, 774]) | 0\n",
-      "134      | torch.Size([1, 1000, 754]) | 0\n",
-      "135      | torch.Size([1, 1000, 762]) | 0\n",
-      "136      | torch.Size([1, 1000, 762]) | 0\n",
-      "137      | torch.Size([1, 1000, 754]) | 0\n",
-      "138      | torch.Size([1, 1000, 754]) | 0\n",
-      "139      | torch.Size([1, 1000, 781]) | 0\n",
-      "140      | torch.Size([1, 1000, 754]) | 0\n",
-      "141      | torch.Size([1, 1000, 787]) | 0\n",
-      "142      | torch.Size([1, 1000, 754]) | 0\n",
-      "143      | torch.Size([1, 1000, 754]) | 0\n",
-      "144      | torch.Size([1, 1000, 762]) | 0\n",
-      "145      | torch.Size([1, 1000, 767]) | 0\n",
-      "146      | torch.Size([1, 1000, 770]) | 0\n",
-      "147      | torch.Size([1, 1000, 754]) | 0\n",
-      "148      | torch.Size([1, 1000, 754]) | 0\n",
-      "149      | torch.Size([1, 1000, 754]) | 0\n",
-      "150      | torch.Size([1, 1000, 754]) | 0\n",
-      "151      | torch.Size([1, 1000, 777]) | 0\n",
-      "152      | torch.Size([1, 1000, 967]) | 0\n",
-      "153      | torch.Size([1, 1000, 802]) | 0\n",
-      "154      | torch.Size([1, 1000, 754]) | 0\n",
-      "155      | torch.Size([1, 1000, 740]) | 0\n",
-      "156      | torch.Size([1, 1000, 754]) | 0\n",
-      "157      | torch.Size([1, 1000, 757]) | 0\n",
-      "158      | torch.Size([1, 1000, 754]) | 0\n",
-      "159      | torch.Size([1, 1000, 611]) | 0\n",
-      "160      | torch.Size([1, 1000, 754]) | 0\n",
-      "161      | torch.Size([1, 1000, 762]) | 0\n",
-      "162      | torch.Size([1, 1000, 757]) | 0\n",
-      "163      | torch.Size([1, 1000, 754]) | 0\n",
-      "164      | torch.Size([1, 1000, 777]) | 0\n",
-      "165      | torch.Size([1, 1000, 782]) | 0\n",
-      "166      | torch.Size([1, 1000, 781]) | 0\n",
-      "167      | torch.Size([1, 1000, 754]) | 0\n",
-      "168      | torch.Size([1, 1000, 754]) | 0\n",
-      "169      | torch.Size([1, 1000, 771]) | 0\n",
-      "170      | torch.Size([1, 1000, 754]) | 0\n",
-      "171      | torch.Size([1, 1000, 762]) | 0\n",
-      "172      | torch.Size([1, 1000, 783]) | 0\n",
-      "173      | torch.Size([1, 1000, 751]) | 0\n",
-      "174      | torch.Size([1, 1000, 802]) | 0\n",
-      "175      | torch.Size([1, 1000, 754]) | 0\n",
-      "176      | torch.Size([1, 1000, 754]) | 0\n",
-      "177      | torch.Size([1, 1000, 754]) | 0\n",
-      "178      | torch.Size([1, 1000, 757]) | 0\n",
-      "179      | torch.Size([1, 1000, 754]) | 0\n",
-      "180      | torch.Size([1, 1000, 754]) | 0\n",
-      "181      | torch.Size([1, 1000, 783]) | 0\n",
-      "182      | torch.Size([1, 1000, 754]) | 0\n",
-      "183      | torch.Size([1, 1000, 778]) | 0\n",
-      "184      | torch.Size([1, 1000, 777]) | 0\n",
-      "185      | torch.Size([1, 1000, 754]) | 0\n",
-      "186      | torch.Size([1, 1000, 824]) | 0\n",
-      "187      | torch.Size([1, 1000, 754]) | 0\n",
-      "188      | torch.Size([1, 1000, 757]) | 0\n",
-      "189      | torch.Size([1, 1000, 754]) | 0\n",
-      "190      | torch.Size([1, 1000, 777]) | 0\n",
-      "191      | torch.Size([1, 1000, 751]) | 0\n",
-      "192      | torch.Size([1, 1000, 754]) | 0\n",
-      "193      | torch.Size([1, 1000, 754]) | 0\n",
-      "194      | torch.Size([1, 1000, 757]) | 0\n",
-      "195      | torch.Size([1, 1000, 754]) | 0\n",
-      "196      | torch.Size([1, 1000, 757]) | 0\n",
-      "197      | torch.Size([1, 1000, 762]) | 0\n",
-      "198      | torch.Size([1, 1000, 776]) | 0\n",
-      "199      | torch.Size([1, 1000, 777]) | 0\n",
-      "200      | torch.Size([1, 1000, 754]) | 0\n",
-      "201      | torch.Size([1, 1000, 754]) | 0\n",
-      "202      | torch.Size([1, 1000, 754]) | 0\n",
-      "203      | torch.Size([1, 1000, 790]) | 0\n",
-      "204      | torch.Size([1, 1000, 754]) | 0\n",
-      "205      | torch.Size([1, 1000, 754]) | 0\n",
-      "206      | torch.Size([1, 1000, 772]) | 0\n",
-      "207      | torch.Size([1, 1000, 761]) | 0\n",
-      "208      | torch.Size([1, 1000, 754]) | 0\n",
-      "209      | torch.Size([1, 1000, 771]) | 0\n",
-      "210      | torch.Size([1, 1000, 754]) | 0\n",
-      "211      | torch.Size([1, 1000, 754]) | 0\n",
-      "212      | torch.Size([1, 1000, 771]) | 0\n",
-      "213      | torch.Size([1, 1000, 754]) | 0\n",
-      "214      | torch.Size([1, 1000, 786]) | 0\n",
-      "215      | torch.Size([1, 1000, 754]) | 0\n",
-      "216      | torch.Size([1, 1000, 1473]) | 0\n",
-      "217      | torch.Size([1, 1000, 754]) | 0\n",
-      "218      | torch.Size([1, 1000, 754]) | 0\n",
-      "219      | torch.Size([1, 1000, 781]) | 0\n",
-      "220      | torch.Size([1, 1000, 771]) | 0\n",
-      "221      | torch.Size([1, 1000, 754]) | 0\n",
-      "222      | torch.Size([1, 1000, 773]) | 0\n",
-      "223      | torch.Size([1, 1000, 754]) | 0\n",
-      "224      | torch.Size([1, 1000, 754]) | 0\n",
-      "225      | torch.Size([1, 1000, 777]) | 0\n",
-      "226      | torch.Size([1, 1000, 786]) | 0\n",
-      "227      | torch.Size([1, 1000, 754]) | 0\n",
-      "228      | torch.Size([1, 1000, 754]) | 0\n",
-      "229      | torch.Size([1, 1000, 754]) | 0\n",
-      "230      | torch.Size([1, 1000, 754]) | 0\n",
-      "231      | torch.Size([1, 1000, 754]) | 0\n",
-      "232      | torch.Size([1, 1000, 754]) | 0\n",
-      "233      | torch.Size([1, 1000, 754]) | 0\n",
-      "234      | torch.Size([1, 1000, 754]) | 0\n",
-      "235      | torch.Size([1, 1000, 757]) | 0\n",
-      "236      | torch.Size([1, 1000, 777]) | 0\n",
-      "237      | torch.Size([1, 1000, 779]) | 0\n",
-      "238      | torch.Size([1, 1000, 787]) | 0\n",
-      "239      | torch.Size([1, 1000, 772]) | 0\n",
-      "240      | torch.Size([1, 1000, 754]) | 0\n",
-      "241      | torch.Size([1, 1000, 754]) | 0\n",
-      "242      | torch.Size([1, 1000, 1473]) | 0\n",
-      "243      | torch.Size([1, 1000, 763]) | 0\n",
-      "244      | torch.Size([1, 1000, 754]) | 0\n",
-      "245      | torch.Size([1, 1000, 773]) | 0\n",
-      "246      | torch.Size([1, 1000, 777]) | 0\n",
-      "247      | torch.Size([1, 1000, 757]) | 0\n",
-      "248      | torch.Size([1, 1000, 754]) | 0\n",
-      "249      | torch.Size([1, 1000, 761]) | 0\n",
-      "250      | torch.Size([1, 1000, 800]) | 0\n",
-      "251      | torch.Size([1, 1000, 754]) | 0\n",
-      "252      | torch.Size([1, 1000, 754]) | 0\n",
-      "253      | torch.Size([1, 1000, 865]) | 0\n",
-      "254      | torch.Size([1, 1000, 762]) | 0\n",
-      "255      | torch.Size([1, 1000, 762]) | 0\n",
-      "256      | torch.Size([1, 1000, 762]) | 0\n",
-      "257      | torch.Size([1, 1000, 763]) | 0\n",
-      "258      | torch.Size([1, 1000, 754]) | 0\n",
-      "259      | torch.Size([1, 1000, 757]) | 0\n",
-      "260      | torch.Size([1, 1000, 754]) | 0\n",
-      "261      | torch.Size([1, 1000, 780]) | 0\n",
-      "262      | torch.Size([1, 1000, 754]) | 0\n",
-      "263      | torch.Size([1, 1000, 762]) | 0\n",
-      "264      | torch.Size([1, 1000, 754]) | 0\n",
-      "265      | torch.Size([1, 1000, 787]) | 0\n",
-      "266      | torch.Size([1, 1000, 777]) | 0\n",
-      "267      | torch.Size([1, 1000, 751]) | 0\n",
-      "268      | torch.Size([1, 1000, 754]) | 0\n",
-      "269      | torch.Size([1, 1000, 782]) | 0\n",
-      "270      | torch.Size([1, 1000, 762]) | 0\n",
-      "271      | torch.Size([1, 1000, 810]) | 0\n",
-      "272      | torch.Size([1, 1000, 782]) | 0\n",
-      "273      | torch.Size([1, 1000, 762]) | 0\n",
-      "274      | torch.Size([1, 1000, 777]) | 0\n",
-      "275      | torch.Size([1, 1000, 813]) | 0\n",
-      "276      | torch.Size([1, 1000, 770]) | 0\n",
-      "277      | torch.Size([1, 1000, 774]) | 0\n",
-      "278      | torch.Size([1, 1000, 780]) | 0\n",
-      "279      | torch.Size([1, 1000, 762]) | 0\n",
-      "280      | torch.Size([1, 1000, 783]) | 0\n",
-      "281      | torch.Size([1, 1000, 754]) | 0\n",
-      "282      | torch.Size([1, 1000, 754]) | 0\n",
-      "283      | torch.Size([1, 1000, 754]) | 0\n",
-      "284      | torch.Size([1, 1000, 757]) | 0\n",
-      "285      | torch.Size([1, 1000, 778]) | 0\n",
-      "286      | torch.Size([1, 1000, 754]) | 0\n",
-      "287      | torch.Size([1, 1000, 772]) | 0\n",
-      "288      | torch.Size([1, 1000, 762]) | 0\n",
-      "289      | torch.Size([1, 1000, 754]) | 0\n",
-      "290      | torch.Size([1, 1000, 751]) | 0\n",
-      "291      | torch.Size([1, 1000, 762]) | 0\n",
-      "292      | torch.Size([1, 1000, 754]) | 0\n",
-      "293      | torch.Size([1, 1000, 781]) | 0\n",
-      "294      | torch.Size([1, 1000, 754]) | 0\n",
-      "295      | torch.Size([1, 1000, 798]) | 0\n",
-      "296      | torch.Size([1, 1000, 754]) | 0\n",
-      "297      | torch.Size([1, 1000, 757]) | 0\n",
-      "298      | torch.Size([1, 1000, 754]) | 0\n",
-      "299      | torch.Size([1, 1000, 967]) | 0\n",
-      "300      | torch.Size([1, 1000, 771]) | 0\n",
-      "301      | torch.Size([1, 1000, 754]) | 0\n",
-      "302      | torch.Size([1, 1000, 772]) | 0\n",
-      "303      | torch.Size([1, 1000, 782]) | 0\n",
-      "304      | torch.Size([1, 1000, 771]) | 0\n",
-      "305      | torch.Size([1, 1000, 754]) | 0\n",
-      "306      | torch.Size([1, 1000, 788]) | 0\n",
-      "307      | torch.Size([1, 1000, 754]) | 0\n",
-      "308      | torch.Size([1, 1000, 754]) | 0\n",
-      "309      | torch.Size([1, 1000, 743]) | 0\n",
-      "310      | torch.Size([1, 1000, 754]) | 0\n",
-      "311      | torch.Size([1, 1000, 751]) | 0\n",
-      "312      | torch.Size([1, 1000, 611]) | 0\n",
-      "313      | torch.Size([1, 1000, 774]) | 0\n",
-      "314      | torch.Size([1, 1000, 762]) | 0\n",
-      "315      | torch.Size([1, 1000, 754]) | 0\n",
-      "316      | torch.Size([1, 1000, 653]) | 0\n",
-      "317      | torch.Size([1, 1000, 611]) | 0\n",
-      "318      | torch.Size([1, 1000, 777]) | 0\n",
-      "319      | torch.Size([1, 1000, 813]) | 0\n",
-      "320      | torch.Size([1, 1000, 754]) | 0\n",
-      "321      | torch.Size([1, 1000, 754]) | 0\n",
-      "322      | torch.Size([1, 1000, 782]) | 0\n",
-      "323      | torch.Size([1, 1000, 790]) | 0\n",
-      "324      | torch.Size([1, 1000, 672]) | 0\n",
-      "325      | torch.Size([1, 1000, 774]) | 0\n",
-      "326      | torch.Size([1, 1000, 754]) | 0\n",
-      "327      | torch.Size([1, 1000, 754]) | 0\n",
-      "328      | torch.Size([1, 1000, 754]) | 0\n",
-      "329      | torch.Size([1, 1000, 771]) | 0\n",
-      "330      | torch.Size([1, 1000, 754]) | 0\n",
-      "331      | torch.Size([1, 1000, 754]) | 0\n",
-      "332      | torch.Size([1, 1000, 844]) | 0\n",
-      "333      | torch.Size([1, 1000, 754]) | 0\n",
-      "334      | torch.Size([1, 1000, 772]) | 0\n",
-      "335      | torch.Size([1, 1000, 754]) | 0\n",
-      "336      | torch.Size([1, 1000, 754]) | 0\n",
-      "337      | torch.Size([1, 1000, 754]) | 0\n",
-      "338      | torch.Size([1, 1000, 771]) | 0\n",
-      "339      | torch.Size([1, 1000, 762]) | 0\n",
-      "340      | torch.Size([1, 1000, 754]) | 0\n",
-      "341      | torch.Size([1, 1000, 754]) | 0\n",
-      "342      | torch.Size([1, 1000, 754]) | 0\n",
-      "343      | torch.Size([1, 1000, 762]) | 0\n",
-      "344      | torch.Size([1, 1000, 762]) | 0\n",
-      "345      | torch.Size([1, 1000, 809]) | 0\n",
-      "346      | torch.Size([1, 1000, 762]) | 0\n",
-      "347      | torch.Size([1, 1000, 781]) | 0\n",
-      "348      | torch.Size([1, 1000, 775]) | 0\n",
-      "349      | torch.Size([1, 1000, 802]) | 0\n",
-      "350      | torch.Size([1, 1000, 754]) | 0\n",
-      "351      | torch.Size([1, 1000, 786]) | 0\n",
-      "352      | torch.Size([1, 1000, 771]) | 0\n",
-      "353      | torch.Size([1, 1000, 806]) | 0\n",
-      "354      | torch.Size([1, 1000, 771]) | 0\n",
-      "355      | torch.Size([1, 1000, 784]) | 0\n",
-      "356      | torch.Size([1, 1000, 754]) | 0\n",
-      "357      | torch.Size([1, 1000, 754]) | 0\n",
-      "358      | torch.Size([1, 1000, 772]) | 0\n",
-      "359      | torch.Size([1, 1000, 770]) | 0\n",
-      "360      | torch.Size([1, 1000, 788]) | 0\n",
-      "361      | torch.Size([1, 1000, 794]) | 0\n",
-      "362      | torch.Size([1, 1000, 762]) | 0\n",
-      "363      | torch.Size([1, 1000, 762]) | 0\n",
-      "364      | torch.Size([1, 1000, 754]) | 0\n",
-      "365      | torch.Size([1, 1000, 773]) | 0\n",
-      "366      | torch.Size([1, 1000, 754]) | 0\n",
-      "367      | torch.Size([1, 1000, 611]) | 0\n",
-      "368      | torch.Size([1, 1000, 754]) | 0\n",
-      "369      | torch.Size([1, 1000, 784]) | 0\n",
-      "370      | torch.Size([1, 1000, 754]) | 0\n",
-      "371      | torch.Size([1, 1000, 762]) | 0\n",
-      "372      | torch.Size([1, 1000, 762]) | 0\n",
-      "373      | torch.Size([1, 1000, 754]) | 0\n",
-      "374      | torch.Size([1, 1000, 762]) | 0\n",
-      "375      | torch.Size([1, 1000, 777]) | 0\n",
-      "376      | torch.Size([1, 1000, 606]) | 0\n",
-      "377      | torch.Size([1, 1000, 762]) | 0\n",
-      "378      | torch.Size([1, 1000, 778]) | 0\n",
-      "379      | torch.Size([1, 1000, 754]) | 0\n",
-      "380      | torch.Size([1, 1000, 762]) | 0\n",
-      "381      | torch.Size([1, 1000, 754]) | 0\n",
-      "382      | torch.Size([1, 1000, 762]) | 0\n",
-      "383      | torch.Size([1, 1000, 754]) | 0\n",
-      "384      | torch.Size([1, 1000, 754]) | 0\n",
-      "385      | torch.Size([1, 1000, 762]) | 0\n",
-      "386      | torch.Size([1, 1000, 762]) | 0\n",
-      "387      | torch.Size([1, 1000, 782]) | 0\n",
-      "388      | torch.Size([1, 1000, 754]) | 0\n",
-      "389      | torch.Size([1, 1000, 781]) | 0\n",
-      "390      | torch.Size([1, 1000, 762]) | 0\n",
-      "391      | torch.Size([1, 1000, 754]) | 0\n",
-      "392      | torch.Size([1, 1000, 778]) | 0\n",
-      "393      | torch.Size([1, 1000, 801]) | 0\n",
-      "394      | torch.Size([1, 1000, 762]) | 0\n",
-      "395      | torch.Size([1, 1000, 754]) | 0\n",
-      "396      | torch.Size([1, 1000, 782]) | 0\n",
-      "397      | torch.Size([1, 1000, 754]) | 0\n",
-      "398      | torch.Size([1, 1000, 781]) | 0\n",
-      "399      | torch.Size([1, 1000, 779]) | 0\n",
-      "400      | torch.Size([1, 1000, 754]) | 0\n",
-      "401      | torch.Size([1, 1000, 762]) | 0\n",
-      "402      | torch.Size([1, 1000, 778]) | 0\n",
-      "403      | torch.Size([1, 1000, 808]) | 0\n",
-      "404      | torch.Size([1, 1000, 754]) | 0\n",
-      "405      | torch.Size([1, 1000, 762]) | 0\n",
-      "406      | torch.Size([1, 1000, 754]) | 0\n",
-      "407      | torch.Size([1, 1000, 754]) | 0\n",
-      "408      | torch.Size([1, 1000, 754]) | 0\n",
-      "409      | torch.Size([1, 1000, 776]) | 0\n",
-      "410      | torch.Size([1, 1000, 762]) | 0\n",
-      "411      | torch.Size([1, 1000, 802]) | 0\n",
-      "412      | torch.Size([1, 1000, 754]) | 0\n",
-      "413      | torch.Size([1, 1000, 772]) | 0\n",
-      "414      | torch.Size([1, 1000, 762]) | 0\n",
-      "415      | torch.Size([1, 1000, 751]) | 0\n",
-      "416      | torch.Size([1, 1000, 771]) | 0\n",
-      "417      | torch.Size([1, 1000, 762]) | 0\n",
-      "418      | torch.Size([1, 1000, 762]) | 0\n",
-      "419      | torch.Size([1, 1000, 754]) | 0\n",
-      "420      | torch.Size([1, 1000, 780]) | 0\n",
-      "421      | torch.Size([1, 1000, 808]) | 0\n",
-      "422      | torch.Size([1, 1000, 754]) | 0\n",
-      "423      | torch.Size([1, 1000, 781]) | 0\n",
-      "424      | torch.Size([1, 1000, 766]) | 0\n",
-      "425      | torch.Size([1, 1000, 777]) | 0\n",
-      "426      | torch.Size([1, 1000, 762]) | 0\n",
-      "427      | torch.Size([1, 1000, 754]) | 0\n",
-      "428      | torch.Size([1, 1000, 754]) | 0\n",
-      "429      | torch.Size([1, 1000, 754]) | 0\n",
-      "430      | torch.Size([1, 1000, 766]) | 0\n",
-      "431      | torch.Size([1, 1000, 777]) | 0\n",
-      "432      | torch.Size([1, 1000, 762]) | 0\n",
-      "433      | torch.Size([1, 1000, 798]) | 0\n",
-      "434      | torch.Size([1, 1000, 831]) | 0\n",
-      "435      | torch.Size([1, 1000, 757]) | 0\n",
-      "436      | torch.Size([1, 1000, 786]) | 0\n",
-      "437      | torch.Size([1, 1000, 777]) | 0\n",
-      "438      | torch.Size([1, 1000, 777]) | 0\n",
-      "439      | torch.Size([1, 1000, 757]) | 0\n",
-      "440      | torch.Size([1, 1000, 754]) | 0\n",
-      "441      | torch.Size([1, 1000, 754]) | 0\n",
-      "442      | torch.Size([1, 1000, 751]) | 0\n",
-      "443      | torch.Size([1, 1000, 805]) | 0\n",
-      "444      | torch.Size([1, 1000, 762]) | 0\n",
-      "445      | torch.Size([1, 1000, 762]) | 0\n",
-      "446      | torch.Size([1, 1000, 754]) | 0\n",
-      "447      | torch.Size([1, 1000, 754]) | 0\n",
-      "448      | torch.Size([1, 1000, 781]) | 0\n",
-      "449      | torch.Size([1, 1000, 754]) | 0\n",
-      "450      | torch.Size([1, 1000, 754]) | 0\n",
-      "451      | torch.Size([1, 1000, 771]) | 0\n",
-      "452      | torch.Size([1, 1000, 754]) | 0\n",
-      "453      | torch.Size([1, 1000, 784]) | 0\n",
-      "454      | torch.Size([1, 1000, 751]) | 0\n",
-      "455      | torch.Size([1, 1000, 754]) | 0\n",
-      "456      | torch.Size([1, 1000, 754]) | 0\n",
-      "457      | torch.Size([1, 1000, 754]) | 0\n",
-      "458      | torch.Size([1, 1000, 812]) | 0\n",
-      "459      | torch.Size([1, 1000, 776]) | 0\n",
-      "460      | torch.Size([1, 1000, 754]) | 0\n",
-      "461      | torch.Size([1, 1000, 829]) | 0\n",
-      "462      | torch.Size([1, 1000, 774]) | 0\n",
-      "463      | torch.Size([1, 1000, 762]) | 0\n",
-      "464      | torch.Size([1, 1000, 754]) | 0\n",
-      "465      | torch.Size([1, 1000, 781]) | 0\n",
-      "466      | torch.Size([1, 1000, 754]) | 0\n",
-      "467      | torch.Size([1, 1000, 777]) | 0\n",
-      "468      | torch.Size([1, 1000, 786]) | 0\n",
-      "469      | torch.Size([1, 1000, 757]) | 0\n",
-      "470      | torch.Size([1, 1000, 754]) | 0\n",
-      "471      | torch.Size([1, 1000, 782]) | 0\n",
-      "472      | torch.Size([1, 1000, 754]) | 0\n",
-      "473      | torch.Size([1, 1000, 754]) | 0\n",
-      "474      | torch.Size([1, 1000, 754]) | 0\n",
-      "475      | torch.Size([1, 1000, 765]) | 0\n",
-      "476      | torch.Size([1, 1000, 754]) | 0\n",
-      "477      | torch.Size([1, 1000, 754]) | 0\n",
-      "478      | torch.Size([1, 1000, 762]) | 0\n",
-      "479      | torch.Size([1, 1000, 751]) | 0\n",
-      "480      | torch.Size([1, 1000, 754]) | 0\n",
-      "481      | torch.Size([1, 1000, 754]) | 0\n",
-      "482      | torch.Size([1, 1000, 762]) | 0\n",
-      "483      | torch.Size([1, 1000, 792]) | 0\n",
-      "484      | torch.Size([1, 1000, 754]) | 0\n",
-      "485      | torch.Size([1, 1000, 754]) | 0\n",
-      "486      | torch.Size([1, 1000, 754]) | 0\n",
-      "487      | torch.Size([1, 1000, 754]) | 0\n",
-      "488      | torch.Size([1, 1000, 777]) | 0\n",
-      "489      | torch.Size([1, 1000, 757]) | 0\n",
-      "490      | torch.Size([1, 1000, 774]) | 0\n",
-      "491      | torch.Size([1, 1000, 762]) | 0\n",
-      "492      | torch.Size([1, 1000, 754]) | 0\n",
-      "493      | torch.Size([1, 1000, 771]) | 0\n",
-      "494      | torch.Size([1, 1000, 772]) | 0\n",
-      "495      | torch.Size([1, 1000, 762]) | 0\n",
-      "496      | torch.Size([1, 1000, 754]) | 0\n",
-      "497      | torch.Size([1, 1000, 754]) | 0\n",
-      "498      | torch.Size([1, 1000, 788]) | 0\n",
-      "499      | torch.Size([1, 1000, 754]) | 0\n",
-      "500      | torch.Size([1, 1000, 751]) | 0\n",
-      "501      | torch.Size([1, 1000, 740]) | 0\n",
-      "502      | torch.Size([1, 1000, 762]) | 0\n",
-      "503      | torch.Size([1, 1000, 781]) | 0\n",
-      "504      | torch.Size([1, 1000, 762]) | 0\n",
-      "505      | torch.Size([1, 1000, 731]) | 0\n",
-      "506      | torch.Size([1, 1000, 805]) | 0\n",
-      "507      | torch.Size([1, 1000, 750]) | 0\n",
-      "508      | torch.Size([1, 1000, 771]) | 0\n",
-      "509      | torch.Size([1, 1000, 754]) | 0\n",
-      "510      | torch.Size([1, 1000, 777]) | 0\n",
-      "511      | torch.Size([1, 1000, 754]) | 0\n",
-      "512      | torch.Size([1, 1000, 791]) | 0\n",
-      "513      | torch.Size([1, 1000, 762]) | 0\n",
-      "514      | torch.Size([1, 1000, 754]) | 0\n",
-      "515      | torch.Size([1, 1000, 762]) | 0\n",
-      "516      | torch.Size([1, 1000, 777]) | 0\n",
-      "517      | torch.Size([1, 1000, 776]) | 0\n",
-      "518      | torch.Size([1, 1000, 777]) | 0\n",
-      "519      | torch.Size([1, 1000, 843]) | 0\n",
-      "520      | torch.Size([1, 1000, 754]) | 0\n",
-      "521      | torch.Size([1, 1000, 754]) | 0\n",
-      "522      | torch.Size([1, 1000, 754]) | 0\n",
-      "523      | torch.Size([1, 1000, 754]) | 0\n",
-      "524      | torch.Size([1, 1000, 805]) | 0\n",
-      "525      | torch.Size([1, 1000, 757]) | 0\n",
-      "526      | torch.Size([1, 1000, 776]) | 0\n",
-      "527      | torch.Size([1, 1000, 754]) | 0\n",
-      "528      | torch.Size([1, 1000, 771]) | 0\n",
-      "529      | torch.Size([1, 1000, 776]) | 0\n",
-      "530      | torch.Size([1, 1000, 611]) | 0\n",
-      "531      | torch.Size([1, 1000, 777]) | 0\n",
-      "532      | torch.Size([1, 1000, 777]) | 0\n",
-      "533      | torch.Size([1, 1000, 771]) | 0\n",
-      "534      | torch.Size([1, 1000, 620]) | 0\n",
-      "535      | torch.Size([1, 1000, 754]) | 0\n",
-      "536      | torch.Size([1, 1000, 768]) | 0\n",
-      "537      | torch.Size([1, 1000, 754]) | 0\n",
-      "538      | torch.Size([1, 1000, 754]) | 0\n",
-      "539      | torch.Size([1, 1000, 754]) | 0\n",
-      "540      | torch.Size([1, 1000, 774]) | 0\n",
-      "541      | torch.Size([1, 1000, 754]) | 0\n",
-      "542      | torch.Size([1, 1000, 757]) | 0\n",
-      "543      | torch.Size([1, 1000, 754]) | 0\n",
-      "544      | torch.Size([1, 1000, 754]) | 0\n",
-      "545      | torch.Size([1, 1000, 770]) | 0\n",
-      "546      | torch.Size([1, 1000, 754]) | 0\n",
-      "547      | torch.Size([1, 1000, 754]) | 0\n",
-      "548      | torch.Size([1, 1000, 754]) | 0\n",
-      "549      | torch.Size([1, 1000, 754]) | 0\n",
-      "550      | torch.Size([1, 1000, 754]) | 0\n",
-      "551      | torch.Size([1, 1000, 754]) | 0\n",
-      "552      | torch.Size([1, 1000, 762]) | 0\n",
-      "553      | torch.Size([1, 1000, 757]) | 0\n",
-      "554      | torch.Size([1, 1000, 752]) | 0\n",
-      "555      | torch.Size([1, 1000, 778]) | 0\n",
-      "556      | torch.Size([1, 1000, 754]) | 0\n",
-      "557      | torch.Size([1, 1000, 754]) | 0\n",
-      "558      | torch.Size([1, 1000, 777]) | 0\n",
-      "559      | torch.Size([1, 1000, 754]) | 0\n",
-      "560      | torch.Size([1, 1000, 757]) | 0\n",
-      "561      | torch.Size([1, 1000, 851]) | 0\n",
-      "562      | torch.Size([1, 1000, 771]) | 0\n",
-      "563      | torch.Size([1, 1000, 744]) | 0\n",
-      "564      | torch.Size([1, 1000, 751]) | 0\n",
-      "565      | torch.Size([1, 1000, 777]) | 0\n",
-      "566      | torch.Size([1, 1000, 780]) | 0\n",
-      "567      | torch.Size([1, 1000, 771]) | 0\n",
-      "568      | torch.Size([1, 1000, 754]) | 0\n",
-      "569      | torch.Size([1, 1000, 754]) | 0\n",
-      "570      | torch.Size([1, 1000, 754]) | 0\n",
-      "571      | torch.Size([1, 1000, 757]) | 0\n",
-      "572      | torch.Size([1, 1000, 823]) | 0\n",
-      "573      | torch.Size([1, 1000, 779]) | 0\n",
-      "574      | torch.Size([1, 1000, 757]) | 0\n",
-      "575      | torch.Size([1, 1000, 757]) | 0\n",
-      "576      | torch.Size([1, 1000, 762]) | 0\n",
-      "577      | torch.Size([1, 1000, 754]) | 0\n",
-      "578      | torch.Size([1, 1000, 780]) | 0\n",
-      "579      | torch.Size([1, 1000, 754]) | 0\n",
-      "580      | torch.Size([1, 1000, 789]) | 0\n",
-      "581      | torch.Size([1, 1000, 762]) | 0\n",
-      "582      | torch.Size([1, 1000, 754]) | 0\n",
-      "583      | torch.Size([1, 1000, 754]) | 0\n",
-      "584      | torch.Size([1, 1000, 754]) | 0\n",
-      "585      | torch.Size([1, 1000, 762]) | 0\n",
-      "586      | torch.Size([1, 1000, 757]) | 0\n",
-      "587      | torch.Size([1, 1000, 754]) | 0\n",
-      "588      | torch.Size([1, 1000, 754]) | 0\n",
-      "589      | torch.Size([1, 1000, 754]) | 0\n",
-      "590      | torch.Size([1, 1000, 777]) | 0\n",
-      "591      | torch.Size([1, 1000, 802]) | 0\n",
-      "592      | torch.Size([1, 1000, 754]) | 0\n",
-      "593      | torch.Size([1, 1000, 754]) | 0\n",
-      "594      | torch.Size([1, 1000, 754]) | 0\n",
-      "595      | torch.Size([1, 1000, 738]) | 0\n",
-      "596      | torch.Size([1, 1000, 786]) | 0\n",
-      "597      | torch.Size([1, 1000, 754]) | 0\n",
-      "598      | torch.Size([1, 1000, 762]) | 0\n",
-      "599      | torch.Size([1, 1000, 754]) | 0\n",
-      "600      | torch.Size([1, 1000, 757]) | 0\n",
-      "601      | torch.Size([1, 1000, 754]) | 0\n",
-      "602      | torch.Size([1, 1000, 754]) | 0\n",
-      "603      | torch.Size([1, 1000, 754]) | 0\n",
-      "604      | torch.Size([1, 1000, 784]) | 0\n",
-      "605      | torch.Size([1, 1000, 754]) | 0\n",
-      "606      | torch.Size([1, 1000, 777]) | 0\n",
-      "607      | torch.Size([1, 1000, 777]) | 0\n",
-      "608      | torch.Size([1, 1000, 823]) | 0\n",
-      "609      | torch.Size([1, 1000, 754]) | 0\n",
-      "610      | torch.Size([1, 1000, 754]) | 0\n",
-      "611      | torch.Size([1, 1000, 754]) | 0\n",
-      "612      | torch.Size([1, 1000, 754]) | 0\n",
-      "613      | torch.Size([1, 1000, 757]) | 0\n",
-      "614      | torch.Size([1, 1000, 754]) | 0\n",
-      "615      | torch.Size([1, 1000, 754]) | 0\n",
-      "616      | torch.Size([1, 1000, 771]) | 0\n",
-      "617      | torch.Size([1, 1000, 777]) | 0\n",
-      "618      | torch.Size([1, 1000, 786]) | 0\n",
-      "619      | torch.Size([1, 1000, 772]) | 0\n",
-      "620      | torch.Size([1, 1000, 780]) | 0\n",
-      "621      | torch.Size([1, 1000, 754]) | 0\n",
-      "622      | torch.Size([1, 1000, 790]) | 0\n",
-      "623      | torch.Size([1, 1000, 802]) | 0\n",
-      "624      | torch.Size([1, 1000, 762]) | 0\n",
-      "625      | torch.Size([1, 1000, 762]) | 0\n",
-      "626      | torch.Size([1, 1000, 725]) | 0\n",
-      "627      | torch.Size([1, 1000, 782]) | 0\n",
-      "628      | torch.Size([1, 1000, 771]) | 0\n",
-      "629      | torch.Size([1, 1000, 788]) | 0\n",
-      "630      | torch.Size([1, 1000, 785]) | 0\n",
-      "631      | torch.Size([1, 1000, 754]) | 0\n",
-      "632      | torch.Size([1, 1000, 819]) | 0\n",
-      "633      | torch.Size([1, 1000, 754]) | 0\n",
-      "634      | torch.Size([1, 1000, 784]) | 0\n",
-      "635      | torch.Size([1, 1000, 754]) | 0\n",
-      "636      | torch.Size([1, 1000, 790]) | 0\n",
-      "637      | torch.Size([1, 1000, 777]) | 0\n",
-      "638      | torch.Size([1, 1000, 754]) | 0\n",
-      "639      | torch.Size([1, 1000, 611]) | 0\n",
-      "640      | torch.Size([1, 1000, 772]) | 0\n",
-      "641      | torch.Size([1, 1000, 764]) | 0\n",
-      "642      | torch.Size([1, 1000, 754]) | 0\n",
-      "643      | torch.Size([1, 1000, 754]) | 0\n",
-      "644      | torch.Size([1, 1000, 773]) | 0\n",
-      "645      | torch.Size([1, 1000, 754]) | 0\n",
-      "646      | torch.Size([1, 1000, 771]) | 0\n",
-      "647      | torch.Size([1, 1000, 751]) | 0\n",
-      "648      | torch.Size([1, 1000, 867]) | 0\n",
-      "649      | torch.Size([1, 1000, 766]) | 0\n",
-      "650      | torch.Size([1, 1000, 754]) | 0\n",
-      "651      | torch.Size([1, 1000, 754]) | 0\n",
-      "652      | torch.Size([1, 1000, 773]) | 0\n",
-      "653      | torch.Size([1, 1000, 754]) | 0\n",
-      "654      | torch.Size([1, 1000, 754]) | 0\n",
-      "655      | torch.Size([1, 1000, 754]) | 0\n",
-      "656      | torch.Size([1, 1000, 754]) | 0\n",
-      "657      | torch.Size([1, 1000, 773]) | 0\n",
-      "658      | torch.Size([1, 1000, 754]) | 0\n",
-      "659      | torch.Size([1, 1000, 771]) | 0\n",
-      "660      | torch.Size([1, 1000, 781]) | 0\n",
-      "661      | torch.Size([1, 1000, 754]) | 0\n",
-      "662      | torch.Size([1, 1000, 754]) | 0\n",
-      "663      | torch.Size([1, 1000, 754]) | 0\n",
-      "664      | torch.Size([1, 1000, 785]) | 0\n",
-      "665      | torch.Size([1, 1000, 791]) | 0\n",
-      "666      | torch.Size([1, 1000, 611]) | 0\n",
-      "667      | torch.Size([1, 1000, 776]) | 0\n",
-      "668      | torch.Size([1, 1000, 762]) | 0\n",
-      "669      | torch.Size([1, 1000, 769]) | 0\n",
-      "670      | torch.Size([1, 1000, 757]) | 0\n",
-      "671      | torch.Size([1, 1000, 754]) | 0\n",
-      "672      | torch.Size([1, 1000, 763]) | 0\n",
-      "673      | torch.Size([1, 1000, 762]) | 0\n",
-      "674      | torch.Size([1, 1000, 757]) | 0\n",
-      "675      | torch.Size([1, 1000, 754]) | 0\n",
-      "676      | torch.Size([1, 1000, 780]) | 0\n",
-      "677      | torch.Size([1, 1000, 802]) | 0\n",
-      "678      | torch.Size([1, 1000, 754]) | 0\n",
-      "679      | torch.Size([1, 1000, 754]) | 0\n",
-      "680      | torch.Size([1, 1000, 754]) | 0\n",
-      "681      | torch.Size([1, 1000, 771]) | 0\n",
-      "682      | torch.Size([1, 1000, 754]) | 0\n",
-      "683      | torch.Size([1, 1000, 754]) | 0\n",
-      "684      | torch.Size([1, 1000, 773]) | 0\n",
-      "685      | torch.Size([1, 1000, 757]) | 0\n",
-      "686      | torch.Size([1, 1000, 774]) | 0\n",
-      "687      | torch.Size([1, 1000, 754]) | 0\n",
-      "688      | torch.Size([1, 1000, 754]) | 0\n",
-      "689      | torch.Size([1, 1000, 771]) | 0\n",
-      "690      | torch.Size([1, 1000, 762]) | 0\n",
-      "691      | torch.Size([1, 1000, 754]) | 0\n",
-      "692      | torch.Size([1, 1000, 754]) | 0\n",
-      "693      | torch.Size([1, 1000, 762]) | 0\n",
-      "694      | torch.Size([1, 1000, 787]) | 0\n",
-      "695      | torch.Size([1, 1000, 754]) | 0\n",
-      "696      | torch.Size([1, 1000, 843]) | 0\n",
-      "697      | torch.Size([1, 1000, 754]) | 0\n",
-      "698      | torch.Size([1, 1000, 754]) | 0\n",
-      "699      | torch.Size([1, 1000, 762]) | 0\n",
-      "700      | torch.Size([1, 1000, 754]) | 0\n",
-      "701      | torch.Size([1, 1000, 754]) | 0\n",
-      "702      | torch.Size([1, 1000, 762]) | 0\n",
-      "703      | torch.Size([1, 1000, 762]) | 0\n",
-      "704      | torch.Size([1, 1000, 611]) | 0\n",
-      "705      | torch.Size([1, 1000, 754]) | 0\n",
-      "706      | torch.Size([1, 1000, 802]) | 0\n",
-      "707      | torch.Size([1, 1000, 757]) | 0\n",
-      "708      | torch.Size([1, 1000, 757]) | 0\n",
-      "709      | torch.Size([1, 1000, 754]) | 0\n",
-      "710      | torch.Size([1, 1000, 754]) | 0\n",
-      "711      | torch.Size([1, 1000, 754]) | 0\n",
-      "712      | torch.Size([1, 1000, 843]) | 0\n",
-      "713      | torch.Size([1, 1000, 790]) | 0\n",
-      "714      | torch.Size([1, 1000, 779]) | 0\n",
-      "715      | torch.Size([1, 1000, 754]) | 0\n",
-      "716      | torch.Size([1, 1000, 762]) | 0\n",
-      "717      | torch.Size([1, 1000, 754]) | 0\n",
-      "718      | torch.Size([1, 1000, 809]) | 0\n",
-      "719      | torch.Size([1, 1000, 834]) | 0\n",
-      "720      | torch.Size([1, 1000, 781]) | 0\n",
-      "721      | torch.Size([1, 1000, 802]) | 0\n",
-      "722      | torch.Size([1, 1000, 781]) | 0\n",
-      "723      | torch.Size([1, 1000, 754]) | 0\n",
-      "724      | torch.Size([1, 1000, 754]) | 0\n",
-      "725      | torch.Size([1, 1000, 754]) | 0\n",
-      "726      | torch.Size([1, 1000, 754]) | 0\n",
-      "727      | torch.Size([1, 1000, 650]) | 0\n",
-      "728      | torch.Size([1, 1000, 754]) | 0\n",
-      "729      | torch.Size([1, 1000, 754]) | 0\n",
-      "730      | torch.Size([1, 1000, 754]) | 0\n",
-      "731      | torch.Size([1, 1000, 762]) | 0\n",
-      "732      | torch.Size([1, 1000, 757]) | 0\n",
-      "733      | torch.Size([1, 1000, 754]) | 0\n",
-      "734      | torch.Size([1, 1000, 757]) | 0\n",
-      "735      | torch.Size([1, 1000, 754]) | 0\n",
-      "736      | torch.Size([1, 1000, 785]) | 0\n",
-      "737      | torch.Size([1, 1000, 778]) | 0\n",
-      "738      | torch.Size([1, 1000, 754]) | 0\n",
-      "739      | torch.Size([1, 1000, 777]) | 0\n",
-      "740      | torch.Size([1, 1000, 754]) | 0\n",
-      "741      | torch.Size([1, 1000, 754]) | 0\n",
-      "742      | torch.Size([1, 1000, 762]) | 0\n",
-      "743      | torch.Size([1, 1000, 762]) | 0\n",
-      "744      | torch.Size([1, 1000, 778]) | 0\n",
-      "745      | torch.Size([1, 1000, 762]) | 0\n",
-      "746      | torch.Size([1, 1000, 754]) | 0\n",
-      "747      | torch.Size([1, 1000, 762]) | 0\n",
-      "748      | torch.Size([1, 1000, 762]) | 0\n",
-      "749      | torch.Size([1, 1000, 808]) | 0\n",
-      "750      | torch.Size([1, 1000, 764]) | 0\n",
-      "751      | torch.Size([1, 1000, 754]) | 0\n",
-      "752      | torch.Size([1, 1000, 754]) | 0\n",
-      "753      | torch.Size([1, 1000, 775]) | 0\n",
-      "754      | torch.Size([1, 1000, 789]) | 0\n",
-      "755      | torch.Size([1, 1000, 795]) | 0\n",
-      "756      | torch.Size([1, 1000, 754]) | 0\n",
-      "757      | torch.Size([1, 1000, 754]) | 0\n",
-      "758      | torch.Size([1, 1000, 754]) | 0\n",
-      "759      | torch.Size([1, 1000, 780]) | 0\n",
-      "760      | torch.Size([1, 1000, 754]) | 0\n",
-      "761      | torch.Size([1, 1000, 783]) | 0\n",
-      "762      | torch.Size([1, 1000, 754]) | 0\n",
-      "763      | torch.Size([1, 1000, 754]) | 0\n",
-      "764      | torch.Size([1, 1000, 785]) | 0\n",
-      "765      | torch.Size([1, 1000, 778]) | 0\n",
-      "766      | torch.Size([1, 1000, 777]) | 0\n",
-      "767      | torch.Size([1, 1000, 775]) | 0\n",
-      "768      | torch.Size([1, 1000, 774]) | 0\n",
-      "769      | torch.Size([1, 1000, 754]) | 0\n",
-      "770      | torch.Size([1, 1000, 754]) | 0\n",
-      "771      | torch.Size([1, 1000, 774]) | 0\n",
-      "772      | torch.Size([1, 1000, 754]) | 0\n",
-      "773      | torch.Size([1, 1000, 771]) | 0\n",
-      "774      | torch.Size([1, 1000, 757]) | 0\n",
-      "775      | torch.Size([1, 1000, 786]) | 0\n",
-      "776      | torch.Size([1, 1000, 762]) | 0\n",
-      "777      | torch.Size([1, 1000, 771]) | 0\n",
-      "778      | torch.Size([1, 1000, 774]) | 0\n",
-      "779      | torch.Size([1, 1000, 754]) | 0\n",
-      "780      | torch.Size([1, 1000, 863]) | 0\n",
-      "781      | torch.Size([1, 1000, 783]) | 0\n",
-      "782      | torch.Size([1, 1000, 754]) | 0\n",
-      "783      | torch.Size([1, 1000, 754]) | 0\n",
-      "784      | torch.Size([1, 1000, 789]) | 0\n",
-      "785      | torch.Size([1, 1000, 754]) | 0\n",
-      "786      | torch.Size([1, 1000, 757]) | 0\n",
-      "787      | torch.Size([1, 1000, 754]) | 0\n",
-      "788      | torch.Size([1, 1000, 754]) | 0\n",
-      "789      | torch.Size([1, 1000, 754]) | 0\n",
-      "790      | torch.Size([1, 1000, 748]) | 0\n",
-      "791      | torch.Size([1, 1000, 754]) | 0\n",
-      "792      | torch.Size([1, 1000, 754]) | 0\n",
-      "793      | torch.Size([1, 1000, 754]) | 0\n",
-      "794      | torch.Size([1, 1000, 777]) | 0\n",
-      "795      | torch.Size([1, 1000, 773]) | 0\n",
-      "796      | torch.Size([1, 1000, 754]) | 0\n",
-      "797      | torch.Size([1, 1000, 781]) | 0\n",
-      "798      | torch.Size([1, 1000, 765]) | 0\n",
-      "799      | torch.Size([1, 1000, 757]) | 0\n",
-      "800      | torch.Size([1, 1000, 773]) | 0\n",
-      "801      | torch.Size([1, 1000, 762]) | 0\n",
-      "802      | torch.Size([1, 1000, 754]) | 0\n",
-      "803      | torch.Size([1, 1000, 781]) | 0\n",
-      "804      | torch.Size([1, 1000, 1473]) | 0\n",
-      "805      | torch.Size([1, 1000, 757]) | 0\n",
-      "806      | torch.Size([1, 1000, 783]) | 0\n",
-      "807      | torch.Size([1, 1000, 769]) | 0\n",
-      "808      | torch.Size([1, 1000, 771]) | 0\n",
-      "809      | torch.Size([1, 1000, 754]) | 0\n",
-      "810      | torch.Size([1, 1000, 777]) | 0\n",
-      "811      | torch.Size([1, 1000, 754]) | 0\n",
-      "812      | torch.Size([1, 1000, 767]) | 0\n",
-      "813      | torch.Size([1, 1000, 754]) | 0\n",
-      "814      | torch.Size([1, 1000, 779]) | 0\n",
-      "815      | torch.Size([1, 1000, 754]) | 0\n",
-      "816      | torch.Size([1, 1000, 754]) | 0\n",
-      "817      | torch.Size([1, 1000, 778]) | 0\n",
-      "818      | torch.Size([1, 1000, 762]) | 0\n",
-      "819      | torch.Size([1, 1000, 754]) | 0\n",
-      "820      | torch.Size([1, 1000, 754]) | 0\n",
-      "821      | torch.Size([1, 1000, 948]) | 0\n",
-      "822      | torch.Size([1, 1000, 782]) | 0\n",
-      "823      | torch.Size([1, 1000, 757]) | 0\n",
-      "824      | torch.Size([1, 1000, 754]) | 0\n",
-      "825      | torch.Size([1, 1000, 754]) | 0\n",
-      "826      | torch.Size([1, 1000, 754]) | 0\n",
-      "827      | torch.Size([1, 1000, 762]) | 0\n",
-      "828      | torch.Size([1, 1000, 762]) | 0\n",
-      "829      | torch.Size([1, 1000, 804]) | 0\n",
-      "830      | torch.Size([1, 1000, 754]) | 0\n",
-      "831      | torch.Size([1, 1000, 762]) | 0\n",
-      "832      | torch.Size([1, 1000, 762]) | 0\n",
-      "833      | torch.Size([1, 1000, 754]) | 0\n",
-      "834      | torch.Size([1, 1000, 754]) | 0\n",
-      "835      | torch.Size([1, 1000, 771]) | 0\n",
-      "836      | torch.Size([1, 1000, 754]) | 0\n",
-      "837      | torch.Size([1, 1000, 754]) | 0\n",
-      "838      | torch.Size([1, 1000, 763]) | 0\n",
-      "839      | torch.Size([1, 1000, 754]) | 0\n",
-      "840      | torch.Size([1, 1000, 762]) | 0\n",
-      "841      | torch.Size([1, 1000, 732]) | 0\n",
-      "842      | torch.Size([1, 1000, 754]) | 0\n",
-      "843      | torch.Size([1, 1000, 754]) | 0\n",
-      "844      | torch.Size([1, 1000, 802]) | 0\n",
-      "845      | torch.Size([1, 1000, 754]) | 0\n",
-      "846      | torch.Size([1, 1000, 773]) | 0\n",
-      "847      | torch.Size([1, 1000, 754]) | 0\n",
-      "848      | torch.Size([1, 1000, 777]) | 0\n",
-      "849      | torch.Size([1, 1000, 771]) | 0\n",
-      "850      | torch.Size([1, 1000, 762]) | 0\n",
-      "851      | torch.Size([1, 1000, 754]) | 0\n",
-      "852      | torch.Size([1, 1000, 771]) | 0\n",
-      "853      | torch.Size([1, 1000, 754]) | 0\n",
-      "854      | torch.Size([1, 1000, 762]) | 0\n",
-      "855      | torch.Size([1, 1000, 762]) | 0\n",
-      "856      | torch.Size([1, 1000, 757]) | 0\n",
-      "857      | torch.Size([1, 1000, 777]) | 0\n",
-      "858      | torch.Size([1, 1000, 754]) | 0\n",
-      "859      | torch.Size([1, 1000, 779]) | 0\n",
-      "860      | torch.Size([1, 1000, 771]) | 0\n",
-      "861      | torch.Size([1, 1000, 762]) | 0\n",
-      "862      | torch.Size([1, 1000, 754]) | 0\n",
-      "863      | torch.Size([1, 1000, 754]) | 0\n",
-      "864      | torch.Size([1, 1000, 773]) | 0\n",
-      "865      | torch.Size([1, 1000, 796]) | 0\n",
-      "866      | torch.Size([1, 1000, 785]) | 0\n",
-      "867      | torch.Size([1, 1000, 754]) | 0\n",
-      "868      | torch.Size([1, 1000, 754]) | 0\n",
-      "869      | torch.Size([1, 1000, 803]) | 0\n",
-      "870      | torch.Size([1, 1000, 845]) | 0\n",
-      "871      | torch.Size([1, 1000, 754]) | 0\n",
-      "872      | torch.Size([1, 1000, 783]) | 0\n",
-      "873      | torch.Size([1, 1000, 777]) | 0\n",
-      "874      | torch.Size([1, 1000, 754]) | 0\n",
-      "875      | torch.Size([1, 1000, 762]) | 0\n",
-      "876      | torch.Size([1, 1000, 754]) | 0\n",
-      "877      | torch.Size([1, 1000, 754]) | 0\n",
-      "878      | torch.Size([1, 1000, 782]) | 0\n",
-      "879      | torch.Size([1, 1000, 762]) | 0\n",
-      "880      | torch.Size([1, 1000, 754]) | 0\n",
-      "881      | torch.Size([1, 1000, 771]) | 0\n",
-      "882      | torch.Size([1, 1000, 773]) | 0\n",
-      "883      | torch.Size([1, 1000, 754]) | 0\n",
-      "884      | torch.Size([1, 1000, 754]) | 0\n",
-      "885      | torch.Size([1, 1000, 754]) | 0\n",
-      "886      | torch.Size([1, 1000, 791]) | 0\n",
-      "887      | torch.Size([1, 1000, 757]) | 0\n",
-      "888      | torch.Size([1, 1000, 762]) | 0\n",
-      "889      | torch.Size([1, 1000, 754]) | 0\n",
-      "890      | torch.Size([1, 1000, 757]) | 0\n",
-      "891      | torch.Size([1, 1000, 825]) | 0\n",
-      "892      | torch.Size([1, 1000, 777]) | 0\n",
-      "893      | torch.Size([1, 1000, 754]) | 0\n",
-      "894      | torch.Size([1, 1000, 771]) | 0\n",
-      "895      | torch.Size([1, 1000, 754]) | 0\n",
-      "896      | torch.Size([1, 1000, 754]) | 0\n",
-      "897      | torch.Size([1, 1000, 754]) | 0\n",
-      "898      | torch.Size([1, 1000, 754]) | 0\n",
-      "899      | torch.Size([1, 1000, 776]) | 0\n",
-      "900      | torch.Size([1, 1000, 754]) | 0\n",
-      "901      | torch.Size([1, 1000, 771]) | 0\n",
-      "902      | torch.Size([1, 1000, 754]) | 0\n",
-      "903      | torch.Size([1, 1000, 762]) | 0\n",
-      "904      | torch.Size([1, 1000, 771]) | 0\n",
-      "905      | torch.Size([1, 1000, 754]) | 0\n",
-      "906      | torch.Size([1, 1000, 762]) | 0\n",
-      "907      | torch.Size([1, 1000, 777]) | 0\n",
-      "908      | torch.Size([1, 1000, 803]) | 0\n",
-      "909      | torch.Size([1, 1000, 754]) | 0\n",
-      "910      | torch.Size([1, 1000, 791]) | 0\n",
-      "911      | torch.Size([1, 1000, 754]) | 0\n",
-      "912      | torch.Size([1, 1000, 762]) | 0\n",
-      "913      | torch.Size([1, 1000, 757]) | 0\n",
-      "914      | torch.Size([1, 1000, 762]) | 0\n",
-      "915      | torch.Size([1, 1000, 757]) | 0\n",
-      "916      | torch.Size([1, 1000, 754]) | 0\n",
-      "917      | torch.Size([1, 1000, 754]) | 0\n",
-      "918      | torch.Size([1, 1000, 754]) | 0\n",
-      "919      | torch.Size([1, 1000, 754]) | 0\n",
-      "920      | torch.Size([1, 1000, 789]) | 0\n",
-      "921      | torch.Size([1, 1000, 754]) | 0\n",
-      "922      | torch.Size([1, 1000, 794]) | 0\n",
-      "923      | torch.Size([1, 1000, 754]) | 0\n",
-      "924      | torch.Size([1, 1000, 780]) | 0\n",
-      "925      | torch.Size([1, 1000, 772]) | 0\n",
-      "926      | torch.Size([1, 1000, 761]) | 0\n",
-      "927      | torch.Size([1, 1000, 754]) | 0\n",
-      "928      | torch.Size([1, 1000, 762]) | 0\n",
-      "929      | torch.Size([1, 1000, 969]) | 0\n",
-      "930      | torch.Size([1, 1000, 804]) | 0\n",
-      "931      | torch.Size([1, 1000, 754]) | 0\n",
-      "932      | torch.Size([1, 1000, 754]) | 0\n",
-      "933      | torch.Size([1, 1000, 762]) | 0\n",
-      "934      | torch.Size([1, 1000, 754]) | 0\n",
-      "935      | torch.Size([1, 1000, 757]) | 0\n",
-      "936      | torch.Size([1, 1000, 770]) | 0\n",
-      "937      | torch.Size([1, 1000, 762]) | 0\n",
-      "938      | torch.Size([1, 1000, 754]) | 0\n",
-      "939      | torch.Size([1, 1000, 754]) | 0\n",
-      "940      | torch.Size([1, 1000, 754]) | 0\n",
-      "941      | torch.Size([1, 1000, 754]) | 0\n",
-      "942      | torch.Size([1, 1000, 785]) | 0\n",
-      "943      | torch.Size([1, 1000, 754]) | 0\n",
-      "944      | torch.Size([1, 1000, 754]) | 0\n",
-      "945      | torch.Size([1, 1000, 754]) | 0\n",
-      "946      | torch.Size([1, 1000, 754]) | 0\n",
-      "947      | torch.Size([1, 1000, 754]) | 0\n",
-      "948      | torch.Size([1, 1000, 754]) | 0\n",
-      "949      | torch.Size([1, 1000, 812]) | 0\n",
-      "950      | torch.Size([1, 1000, 754]) | 0\n",
-      "951      | torch.Size([1, 1000, 754]) | 0\n",
-      "952      | torch.Size([1, 1000, 790]) | 0\n",
-      "953      | torch.Size([1, 1000, 754]) | 0\n",
-      "954      | torch.Size([1, 1000, 754]) | 0\n",
-      "955      | torch.Size([1, 1000, 762]) | 0\n",
-      "956      | torch.Size([1, 1000, 754]) | 0\n",
-      "957      | torch.Size([1, 1000, 762]) | 0\n",
-      "958      | torch.Size([1, 1000, 754]) | 0\n",
-      "959      | torch.Size([1, 1000, 754]) | 0\n",
-      "960      | torch.Size([1, 1000, 754]) | 0\n",
-      "961      | torch.Size([1, 1000, 771]) | 0\n",
-      "962      | torch.Size([1, 1000, 782]) | 0\n",
-      "963      | torch.Size([1, 1000, 754]) | 0\n",
-      "964      | torch.Size([1, 1000, 611]) | 0\n",
-      "965      | torch.Size([1, 1000, 754]) | 0\n",
-      "966      | torch.Size([1, 1000, 757]) | 0\n",
-      "967      | torch.Size([1, 1000, 797]) | 0\n",
-      "968      | torch.Size([1, 1000, 754]) | 0\n",
-      "969      | torch.Size([1, 1000, 784]) | 0\n",
-      "970      | torch.Size([1, 1000, 802]) | 0\n",
-      "971      | torch.Size([1, 1000, 611]) | 0\n",
-      "972      | torch.Size([1, 1000, 762]) | 0\n",
-      "973      | torch.Size([1, 1000, 754]) | 0\n",
-      "974      | torch.Size([1, 1000, 782]) | 0\n",
-      "975      | torch.Size([1, 1000, 777]) | 0\n",
-      "976      | torch.Size([1, 1000, 762]) | 0\n",
-      "977      | torch.Size([1, 1000, 771]) | 0\n",
-      "978      | torch.Size([1, 1000, 754]) | 0\n",
-      "979      | torch.Size([1, 1000, 754]) | 0\n",
-      "980      | torch.Size([1, 1000, 771]) | 0\n",
-      "981      | torch.Size([1, 1000, 771]) | 0\n",
-      "982      | torch.Size([1, 1000, 754]) | 0\n",
-      "983      | torch.Size([1, 1000, 762]) | 0\n",
-      "984      | torch.Size([1, 1000, 754]) | 0\n",
-      "985      | torch.Size([1, 1000, 762]) | 0\n",
-      "986      | torch.Size([1, 1000, 754]) | 0\n",
-      "987      | torch.Size([1, 1000, 754]) | 0\n",
-      "988      | torch.Size([1, 1000, 771]) | 0\n",
-      "989      | torch.Size([1, 1000, 762]) | 0\n",
-      "990      | torch.Size([1, 1000, 762]) | 0\n",
-      "991      | torch.Size([1, 1000, 611]) | 0\n",
-      "992      | torch.Size([1, 1000, 782]) | 0\n",
-      "993      | torch.Size([1, 1000, 778]) | 0\n",
-      "994      | torch.Size([1, 1000, 777]) | 0\n",
-      "995      | torch.Size([1, 1000, 757]) | 0\n",
-      "996      | torch.Size([1, 1000, 762]) | 0\n",
-      "997      | torch.Size([1, 1000, 771]) | 0\n",
-      "998      | torch.Size([1, 1000, 782]) | 0\n",
-      "999      | torch.Size([1, 1000, 754]) | 0\n",
-      "1000     | torch.Size([1, 1000, 780]) | 0\n",
-      "1001     | torch.Size([1, 1000, 762]) | 0\n",
-      "1002     | torch.Size([1, 1000, 754]) | 0\n",
-      "1003     | torch.Size([1, 1000, 751]) | 0\n",
-      "1004     | torch.Size([1, 1000, 762]) | 0\n",
-      "1005     | torch.Size([1, 1000, 754]) | 0\n",
-      "1006     | torch.Size([1, 1000, 762]) | 0\n",
-      "1007     | torch.Size([1, 1000, 754]) | 0\n",
-      "1008     | torch.Size([1, 1000, 754]) | 0\n",
-      "1009     | torch.Size([1, 1000, 777]) | 0\n",
-      "1010     | torch.Size([1, 1000, 867]) | 0\n",
-      "1011     | torch.Size([1, 1000, 762]) | 0\n",
-      "1012     | torch.Size([1, 1000, 860]) | 0\n",
-      "1013     | torch.Size([1, 1000, 757]) | 0\n",
-      "1014     | torch.Size([1, 1000, 777]) | 0\n",
-      "1015     | torch.Size([1, 1000, 754]) | 0\n",
-      "1016     | torch.Size([1, 1000, 754]) | 0\n",
-      "1017     | torch.Size([1, 1000, 782]) | 0\n",
-      "1018     | torch.Size([1, 1000, 782]) | 0\n",
-      "1019     | torch.Size([1, 1000, 762]) | 0\n",
-      "1020     | torch.Size([1, 1000, 782]) | 0\n",
-      "1021     | torch.Size([1, 1000, 777]) | 0\n",
-      "1022     | torch.Size([1, 1000, 762]) | 0\n",
-      "1023     | torch.Size([1, 1000, 648]) | 0\n",
-      "1024     | torch.Size([1, 1000, 757]) | 0\n",
-      "1025     | torch.Size([1, 1000, 754]) | 0\n",
-      "1026     | torch.Size([1, 1000, 757]) | 0\n",
-      "1027     | torch.Size([1, 1000, 777]) | 0\n",
-      "1028     | torch.Size([1, 1000, 789]) | 0\n",
-      "1029     | torch.Size([1, 1000, 850]) | 0\n",
-      "1030     | torch.Size([1, 1000, 791]) | 0\n",
-      "1031     | torch.Size([1, 1000, 754]) | 0\n",
-      "1032     | torch.Size([1, 1000, 754]) | 0\n",
-      "1033     | torch.Size([1, 1000, 754]) | 0\n",
-      "1034     | torch.Size([1, 1000, 776]) | 0\n",
-      "1035     | torch.Size([1, 1000, 751]) | 0\n",
-      "1036     | torch.Size([1, 1000, 754]) | 0\n",
-      "1037     | torch.Size([1, 1000, 757]) | 0\n",
-      "1038     | torch.Size([1, 1000, 771]) | 0\n",
-      "1039     | torch.Size([1, 1000, 754]) | 0\n",
-      "1040     | torch.Size([1, 1000, 782]) | 0\n",
-      "1041     | torch.Size([1, 1000, 757]) | 0\n",
-      "1042     | torch.Size([1, 1000, 754]) | 0\n",
-      "1043     | torch.Size([1, 1000, 754]) | 0\n",
-      "1044     | torch.Size([1, 1000, 757]) | 0\n",
-      "1045     | torch.Size([1, 1000, 811]) | 0\n",
-      "1046     | torch.Size([1, 1000, 754]) | 0\n",
-      "1047     | torch.Size([1, 1000, 781]) | 0\n",
-      "1048     | torch.Size([1, 1000, 754]) | 0\n",
-      "1049     | torch.Size([1, 1000, 762]) | 0\n",
-      "1050     | torch.Size([1, 1000, 762]) | 0\n",
-      "1051     | torch.Size([1, 1000, 754]) | 0\n",
-      "1052     | torch.Size([1, 1000, 754]) | 0\n",
-      "1053     | torch.Size([1, 1000, 787]) | 0\n",
-      "1054     | torch.Size([1, 1000, 781]) | 0\n",
-      "1055     | torch.Size([1, 1000, 757]) | 0\n",
-      "1056     | torch.Size([1, 1000, 762]) | 0\n",
-      "1057     | torch.Size([1, 1000, 762]) | 0\n",
-      "1058     | torch.Size([1, 1000, 771]) | 0\n",
-      "1059     | torch.Size([1, 1000, 762]) | 0\n",
-      "1060     | torch.Size([1, 1000, 754]) | 0\n",
-      "1061     | torch.Size([1, 1000, 777]) | 0\n",
-      "1062     | torch.Size([1, 1000, 754]) | 0\n",
-      "1063     | torch.Size([1, 1000, 772]) | 0\n",
-      "1064     | torch.Size([1, 1000, 754]) | 0\n",
-      "1065     | torch.Size([1, 1000, 754]) | 0\n",
-      "1066     | torch.Size([1, 1000, 784]) | 0\n",
-      "1067     | torch.Size([1, 1000, 754]) | 0\n",
-      "1068     | torch.Size([1, 1000, 754]) | 0\n",
-      "1069     | torch.Size([1, 1000, 754]) | 0\n",
-      "1070     | torch.Size([1, 1000, 794]) | 0\n",
-      "1071     | torch.Size([1, 1000, 754]) | 0\n",
-      "1072     | torch.Size([1, 1000, 762]) | 0\n",
-      "1073     | torch.Size([1, 1000, 762]) | 0\n",
-      "1074     | torch.Size([1, 1000, 787]) | 0\n",
-      "1075     | torch.Size([1, 1000, 754]) | 0\n",
-      "1076     | torch.Size([1, 1000, 771]) | 0\n",
-      "1077     | torch.Size([1, 1000, 862]) | 0\n",
-      "1078     | torch.Size([1, 1000, 1473]) | 0\n",
-      "1079     | torch.Size([1, 1000, 791]) | 0\n",
-      "1080     | torch.Size([1, 1000, 762]) | 0\n",
-      "1081     | torch.Size([1, 1000, 754]) | 0\n",
-      "1082     | torch.Size([1, 1000, 754]) | 0\n",
-      "1083     | torch.Size([1, 1000, 754]) | 0\n",
-      "1084     | torch.Size([1, 1000, 754]) | 0\n",
-      "1085     | torch.Size([1, 1000, 754]) | 0\n",
-      "1086     | torch.Size([1, 1000, 770]) | 0\n",
-      "1087     | torch.Size([1, 1000, 754]) | 0\n",
-      "1088     | torch.Size([1, 1000, 785]) | 0\n",
-      "1089     | torch.Size([1, 1000, 771]) | 0\n",
-      "1090     | torch.Size([1, 1000, 754]) | 0\n",
-      "1091     | torch.Size([1, 1000, 754]) | 0\n",
-      "1092     | torch.Size([1, 1000, 786]) | 0\n",
-      "1093     | torch.Size([1, 1000, 754]) | 0\n",
-      "1094     | torch.Size([1, 1000, 754]) | 0\n",
-      "1095     | torch.Size([1, 1000, 754]) | 0\n",
-      "1096     | torch.Size([1, 1000, 754]) | 0\n",
-      "1097     | torch.Size([1, 1000, 802]) | 0\n",
-      "1098     | torch.Size([1, 1000, 760]) | 0\n",
-      "1099     | torch.Size([1, 1000, 762]) | 0\n",
-      "1100     | torch.Size([1, 1000, 777]) | 0\n",
-      "1101     | torch.Size([1, 1000, 771]) | 0\n",
-      "1102     | torch.Size([1, 1000, 754]) | 0\n",
-      "1103     | torch.Size([1, 1000, 754]) | 0\n",
-      "1104     | torch.Size([1, 1000, 754]) | 0\n",
-      "1105     | torch.Size([1, 1000, 777]) | 0\n",
-      "1106     | torch.Size([1, 1000, 754]) | 0\n",
-      "1107     | torch.Size([1, 1000, 776]) | 0\n",
-      "1108     | torch.Size([1, 1000, 774]) | 0\n",
-      "1109     | torch.Size([1, 1000, 802]) | 0\n",
-      "1110     | torch.Size([1, 1000, 762]) | 0\n",
-      "1111     | torch.Size([1, 1000, 754]) | 0\n",
-      "1112     | torch.Size([1, 1000, 762]) | 0\n",
-      "1113     | torch.Size([1, 1000, 779]) | 0\n",
-      "1114     | torch.Size([1, 1000, 763]) | 0\n",
-      "1115     | torch.Size([1, 1000, 754]) | 0\n",
-      "1116     | torch.Size([1, 1000, 754]) | 0\n",
-      "1117     | torch.Size([1, 1000, 754]) | 0\n",
-      "1118     | torch.Size([1, 1000, 754]) | 0\n",
-      "1119     | torch.Size([1, 1000, 754]) | 0\n",
-      "1120     | torch.Size([1, 1000, 773]) | 0\n",
-      "1121     | torch.Size([1, 1000, 754]) | 0\n",
-      "1122     | torch.Size([1, 1000, 762]) | 0\n",
-      "1123     | torch.Size([1, 1000, 762]) | 0\n",
-      "1124     | torch.Size([1, 1000, 762]) | 0\n",
-      "1125     | torch.Size([1, 1000, 721]) | 0\n",
-      "1126     | torch.Size([1, 1000, 611]) | 0\n",
-      "1127     | torch.Size([1, 1000, 757]) | 0\n",
-      "1128     | torch.Size([1, 1000, 754]) | 0\n",
-      "1129     | torch.Size([1, 1000, 754]) | 0\n",
-      "1130     | torch.Size([1, 1000, 754]) | 0\n",
-      "1131     | torch.Size([1, 1000, 754]) | 0\n",
-      "1132     | torch.Size([1, 1000, 754]) | 0\n",
-      "1133     | torch.Size([1, 1000, 754]) | 0\n",
-      "1134     | torch.Size([1, 1000, 826]) | 0\n",
-      "1135     | torch.Size([1, 1000, 802]) | 0\n",
-      "1136     | torch.Size([1, 1000, 754]) | 0\n",
-      "1137     | torch.Size([1, 1000, 773]) | 0\n",
-      "1138     | torch.Size([1, 1000, 754]) | 0\n",
-      "1139     | torch.Size([1, 1000, 778]) | 0\n",
-      "1140     | torch.Size([1, 1000, 771]) | 0\n",
-      "1141     | torch.Size([1, 1000, 777]) | 0\n",
-      "1142     | torch.Size([1, 1000, 754]) | 0\n",
-      "1143     | torch.Size([1, 1000, 810]) | 0\n",
-      "1144     | torch.Size([1, 1000, 787]) | 0\n",
-      "1145     | torch.Size([1, 1000, 754]) | 0\n",
-      "1146     | torch.Size([1, 1000, 754]) | 0\n",
-      "1147     | torch.Size([1, 1000, 773]) | 0\n",
-      "1148     | torch.Size([1, 1000, 771]) | 0\n",
-      "1149     | torch.Size([1, 1000, 777]) | 0\n",
-      "1150     | torch.Size([1, 1000, 754]) | 0\n",
-      "1151     | torch.Size([1, 1000, 756]) | 0\n",
-      "1152     | torch.Size([1, 1000, 754]) | 0\n",
-      "1153     | torch.Size([1, 1000, 777]) | 0\n",
-      "1154     | torch.Size([1, 1000, 754]) | 0\n",
-      "1155     | torch.Size([1, 1000, 782]) | 0\n",
-      "1156     | torch.Size([1, 1000, 754]) | 0\n",
-      "1157     | torch.Size([1, 1000, 754]) | 0\n",
-      "1158     | torch.Size([1, 1000, 754]) | 0\n",
-      "1159     | torch.Size([1, 1000, 762]) | 0\n",
-      "1160     | torch.Size([1, 1000, 670]) | 0\n",
-      "1161     | torch.Size([1, 1000, 754]) | 0\n",
-      "1162     | torch.Size([1, 1000, 754]) | 0\n",
-      "1163     | torch.Size([1, 1000, 757]) | 0\n",
-      "1164     | torch.Size([1, 1000, 773]) | 0\n",
-      "1165     | torch.Size([1, 1000, 812]) | 0\n",
-      "1166     | torch.Size([1, 1000, 754]) | 0\n",
-      "1167     | torch.Size([1, 1000, 754]) | 0\n",
-      "1168     | torch.Size([1, 1000, 762]) | 0\n",
-      "1169     | torch.Size([1, 1000, 788]) | 0\n",
-      "1170     | torch.Size([1, 1000, 754]) | 0\n",
-      "1171     | torch.Size([1, 1000, 754]) | 0\n",
-      "1172     | torch.Size([1, 1000, 754]) | 0\n",
-      "1173     | torch.Size([1, 1000, 773]) | 0\n",
-      "1174     | torch.Size([1, 1000, 754]) | 0\n",
-      "1175     | torch.Size([1, 1000, 754]) | 0\n",
-      "1176     | torch.Size([1, 1000, 864]) | 0\n",
-      "1177     | torch.Size([1, 1000, 754]) | 0\n",
-      "1178     | torch.Size([1, 1000, 611]) | 0\n",
-      "1179     | torch.Size([1, 1000, 648]) | 0\n",
-      "1180     | torch.Size([1, 1000, 754]) | 0\n",
-      "1181     | torch.Size([1, 1000, 757]) | 0\n",
-      "1182     | torch.Size([1, 1000, 754]) | 0\n",
-      "1183     | torch.Size([1, 1000, 762]) | 0\n",
-      "1184     | torch.Size([1, 1000, 754]) | 0\n",
-      "1185     | torch.Size([1, 1000, 754]) | 0\n",
-      "1186     | torch.Size([1, 1000, 762]) | 0\n",
-      "1187     | torch.Size([1, 1000, 754]) | 0\n",
-      "1188     | torch.Size([1, 1000, 754]) | 0\n",
-      "1189     | torch.Size([1, 1000, 754]) | 0\n",
-      "1190     | torch.Size([1, 1000, 781]) | 0\n",
-      "1191     | torch.Size([1, 1000, 767]) | 0\n",
-      "1192     | torch.Size([1, 1000, 762]) | 0\n",
-      "1193     | torch.Size([1, 1000, 754]) | 0\n",
-      "1194     | torch.Size([1, 1000, 783]) | 0\n",
-      "1195     | torch.Size([1, 1000, 757]) | 0\n",
-      "1196     | torch.Size([1, 1000, 777]) | 0\n",
-      "1197     | torch.Size([1, 1000, 751]) | 0\n",
-      "1198     | torch.Size([1, 1000, 762]) | 0\n",
-      "1199     | torch.Size([1, 1000, 762]) | 0\n",
-      "1200     | torch.Size([1, 1000, 754]) | 0\n",
-      "1201     | torch.Size([1, 1000, 762]) | 0\n",
-      "1202     | torch.Size([1, 1000, 754]) | 0\n",
-      "1203     | torch.Size([1, 1000, 784]) | 0\n",
-      "1204     | torch.Size([1, 1000, 762]) | 0\n",
-      "1205     | torch.Size([1, 1000, 777]) | 0\n",
-      "1206     | torch.Size([1, 1000, 754]) | 0\n",
-      "1207     | torch.Size([1, 1000, 754]) | 0\n",
-      "1208     | torch.Size([1, 1000, 754]) | 0\n",
-      "1209     | torch.Size([1, 1000, 754]) | 0\n",
-      "1210     | torch.Size([1, 1000, 754]) | 0\n",
-      "1211     | torch.Size([1, 1000, 762]) | 0\n",
-      "1212     | torch.Size([1, 1000, 754]) | 0\n",
-      "1213     | torch.Size([1, 1000, 809]) | 0\n",
-      "1214     | torch.Size([1, 1000, 780]) | 0\n",
-      "1215     | torch.Size([1, 1000, 754]) | 0\n",
-      "1216     | torch.Size([1, 1000, 754]) | 0\n",
-      "1217     | torch.Size([1, 1000, 754]) | 0\n",
-      "1218     | torch.Size([1, 1000, 754]) | 0\n",
-      "1219     | torch.Size([1, 1000, 611]) | 0\n",
-      "1220     | torch.Size([1, 1000, 754]) | 0\n",
-      "1221     | torch.Size([1, 1000, 782]) | 0\n",
-      "1222     | torch.Size([1, 1000, 777]) | 0\n",
-      "1223     | torch.Size([1, 1000, 771]) | 0\n",
-      "1224     | torch.Size([1, 1000, 762]) | 0\n",
-      "1225     | torch.Size([1, 1000, 754]) | 0\n",
-      "1226     | torch.Size([1, 1000, 789]) | 0\n",
-      "1227     | torch.Size([1, 1000, 783]) | 0\n",
-      "1228     | torch.Size([1, 1000, 754]) | 0\n",
-      "1229     | torch.Size([1, 1000, 786]) | 0\n",
-      "1230     | torch.Size([1, 1000, 754]) | 0\n",
-      "1231     | torch.Size([1, 1000, 754]) | 0\n",
-      "1232     | torch.Size([1, 1000, 757]) | 0\n",
-      "1233     | torch.Size([1, 1000, 754]) | 0\n",
-      "1234     | torch.Size([1, 1000, 779]) | 0\n",
-      "1235     | torch.Size([1, 1000, 754]) | 0\n",
-      "1236     | torch.Size([1, 1000, 754]) | 0\n",
-      "1237     | torch.Size([1, 1000, 754]) | 0\n",
-      "1238     | torch.Size([1, 1000, 754]) | 0\n",
-      "1239     | torch.Size([1, 1000, 785]) | 0\n",
-      "1240     | torch.Size([1, 1000, 757]) | 0\n",
-      "1241     | torch.Size([1, 1000, 809]) | 0\n",
-      "1242     | torch.Size([1, 1000, 1013]) | 0\n",
-      "1243     | torch.Size([1, 1000, 780]) | 0\n",
-      "1244     | torch.Size([1, 1000, 754]) | 0\n",
-      "1245     | torch.Size([1, 1000, 754]) | 0\n",
-      "1246     | torch.Size([1, 1000, 754]) | 0\n",
-      "1247     | torch.Size([1, 1000, 762]) | 0\n",
-      "1248     | torch.Size([1, 1000, 754]) | 0\n",
-      "1249     | torch.Size([1, 1000, 786]) | 0\n",
-      "1250     | torch.Size([1, 1000, 780]) | 0\n",
-      "1251     | torch.Size([1, 1000, 762]) | 0\n",
-      "1252     | torch.Size([1, 1000, 754]) | 0\n",
-      "1253     | torch.Size([1, 1000, 754]) | 0\n",
-      "1254     | torch.Size([1, 1000, 762]) | 0\n",
-      "1255     | torch.Size([1, 1000, 754]) | 0\n",
-      "1256     | torch.Size([1, 1000, 771]) | 0\n",
-      "1257     | torch.Size([1, 1000, 754]) | 0\n",
-      "1258     | torch.Size([1, 1000, 754]) | 0\n",
-      "1259     | torch.Size([1, 1000, 762]) | 0\n",
-      "1260     | torch.Size([1, 1000, 754]) | 0\n",
-      "1261     | torch.Size([1, 1000, 611]) | 0\n",
-      "1262     | torch.Size([1, 1000, 786]) | 0\n",
-      "1263     | torch.Size([1, 1000, 757]) | 0\n",
-      "1264     | torch.Size([1, 1000, 754]) | 0\n",
-      "1265     | torch.Size([1, 1000, 762]) | 0\n",
-      "1266     | torch.Size([1, 1000, 751]) | 0\n",
-      "1267     | torch.Size([1, 1000, 754]) | 0\n",
-      "1268     | torch.Size([1, 1000, 778]) | 0\n",
-      "1269     | torch.Size([1, 1000, 754]) | 0\n",
-      "1270     | torch.Size([1, 1000, 757]) | 0\n",
-      "1271     | torch.Size([1, 1000, 754]) | 0\n",
-      "1272     | torch.Size([1, 1000, 754]) | 0\n",
-      "1273     | torch.Size([1, 1000, 754]) | 0\n",
-      "1274     | torch.Size([1, 1000, 785]) | 0\n",
-      "1275     | torch.Size([1, 1000, 762]) | 0\n",
-      "1276     | torch.Size([1, 1000, 777]) | 0\n",
-      "1277     | torch.Size([1, 1000, 786]) | 0\n",
-      "1278     | torch.Size([1, 1000, 754]) | 0\n",
-      "1279     | torch.Size([1, 1000, 754]) | 0\n",
-      "1280     | torch.Size([1, 1000, 754]) | 0\n",
-      "1281     | torch.Size([1, 1000, 777]) | 0\n",
-      "1282     | torch.Size([1, 1000, 754]) | 0\n",
-      "1283     | torch.Size([1, 1000, 754]) | 0\n",
-      "1284     | torch.Size([1, 1000, 754]) | 0\n",
-      "1285     | torch.Size([1, 1000, 777]) | 0\n",
-      "1286     | torch.Size([1, 1000, 777]) | 0\n",
-      "1287     | torch.Size([1, 1000, 778]) | 0\n",
-      "1288     | torch.Size([1, 1000, 762]) | 0\n",
-      "1289     | torch.Size([1, 1000, 754]) | 0\n",
-      "1290     | torch.Size([1, 1000, 757]) | 0\n",
-      "1291     | torch.Size([1, 1000, 754]) | 0\n",
-      "1292     | torch.Size([1, 1000, 762]) | 0\n",
-      "1293     | torch.Size([1, 1000, 611]) | 0\n",
-      "1294     | torch.Size([1, 1000, 780]) | 0\n",
-      "1295     | torch.Size([1, 1000, 813]) | 0\n",
-      "1296     | torch.Size([1, 1000, 754]) | 0\n",
-      "1297     | torch.Size([1, 1000, 754]) | 0\n",
-      "1298     | torch.Size([1, 1000, 768]) | 0\n",
-      "1299     | torch.Size([1, 1000, 784]) | 0\n",
-      "1300     | torch.Size([1, 1000, 771]) | 0\n",
-      "1301     | torch.Size([1, 1000, 770]) | 0\n",
-      "1302     | torch.Size([1, 1000, 743]) | 0\n",
-      "1303     | torch.Size([1, 1000, 754]) | 0\n",
-      "1304     | torch.Size([1, 1000, 754]) | 0\n",
-      "1305     | torch.Size([1, 1000, 778]) | 0\n",
-      "1306     | torch.Size([1, 1000, 754]) | 0\n",
-      "1307     | torch.Size([1, 1000, 754]) | 0\n",
-      "1308     | torch.Size([1, 1000, 754]) | 0\n",
-      "1309     | torch.Size([1, 1000, 754]) | 0\n",
-      "1310     | torch.Size([1, 1000, 754]) | 0\n",
-      "1311     | torch.Size([1, 1000, 754]) | 0\n",
-      "1312     | torch.Size([1, 1000, 611]) | 0\n",
-      "1313     | torch.Size([1, 1000, 754]) | 0\n",
-      "1314     | torch.Size([1, 1000, 779]) | 0\n",
-      "1315     | torch.Size([1, 1000, 754]) | 0\n",
-      "1316     | torch.Size([1, 1000, 754]) | 0\n",
-      "1317     | torch.Size([1, 1000, 754]) | 0\n",
-      "1318     | torch.Size([1, 1000, 754]) | 0\n",
-      "1319     | torch.Size([1, 1000, 754]) | 0\n",
-      "1320     | torch.Size([1, 1000, 777]) | 0\n",
-      "1321     | torch.Size([1, 1000, 754]) | 0\n",
-      "1322     | torch.Size([1, 1000, 757]) | 0\n",
-      "1323     | torch.Size([1, 1000, 757]) | 0\n",
-      "1324     | torch.Size([1, 1000, 754]) | 0\n",
-      "1325     | torch.Size([1, 1000, 754]) | 0\n",
-      "1326     | torch.Size([1, 1000, 754]) | 0\n",
-      "1327     | torch.Size([1, 1000, 777]) | 0\n",
-      "1328     | torch.Size([1, 1000, 762]) | 0\n",
-      "1329     | torch.Size([1, 1000, 762]) | 0\n",
-      "1330     | torch.Size([1, 1000, 754]) | 0\n",
-      "1331     | torch.Size([1, 1000, 757]) | 0\n",
-      "1332     | torch.Size([1, 1000, 754]) | 0\n",
-      "1333     | torch.Size([1, 1000, 789]) | 0\n",
-      "1334     | torch.Size([1, 1000, 783]) | 0\n",
-      "1335     | torch.Size([1, 1000, 754]) | 0\n",
-      "1336     | torch.Size([1, 1000, 784]) | 0\n",
-      "1337     | torch.Size([1, 1000, 775]) | 0\n",
-      "1338     | torch.Size([1, 1000, 780]) | 0\n",
-      "1339     | torch.Size([1, 1000, 762]) | 0\n",
-      "1340     | torch.Size([1, 1000, 754]) | 0\n",
-      "1341     | torch.Size([1, 1000, 762]) | 0\n",
-      "1342     | torch.Size([1, 1000, 754]) | 0\n",
-      "1343     | torch.Size([1, 1000, 757]) | 0\n",
-      "1344     | torch.Size([1, 1000, 754]) | 0\n",
-      "1345     | torch.Size([1, 1000, 751]) | 0\n",
-      "1346     | torch.Size([1, 1000, 762]) | 0\n",
-      "1347     | torch.Size([1, 1000, 802]) | 0\n",
-      "1348     | torch.Size([1, 1000, 762]) | 0\n",
-      "1349     | torch.Size([1, 1000, 762]) | 0\n",
-      "1350     | torch.Size([1, 1000, 754]) | 0\n",
-      "1351     | torch.Size([1, 1000, 762]) | 0\n",
-      "1352     | torch.Size([1, 1000, 754]) | 0\n",
-      "1353     | torch.Size([1, 1000, 754]) | 0\n",
-      "1354     | torch.Size([1, 1000, 754]) | 0\n",
-      "1355     | torch.Size([1, 1000, 754]) | 0\n",
-      "1356     | torch.Size([1, 1000, 779]) | 0\n",
-      "1357     | torch.Size([1, 1000, 754]) | 0\n",
-      "1358     | torch.Size([1, 1000, 787]) | 0\n",
-      "1359     | torch.Size([1, 1000, 784]) | 0\n",
-      "1360     | torch.Size([1, 1000, 762]) | 0\n",
-      "1361     | torch.Size([1, 1000, 754]) | 0\n",
-      "1362     | torch.Size([1, 1000, 777]) | 0\n",
-      "1363     | torch.Size([1, 1000, 762]) | 0\n",
-      "1364     | torch.Size([1, 1000, 757]) | 0\n",
-      "1365     | torch.Size([1, 1000, 757]) | 0\n",
-      "1366     | torch.Size([1, 1000, 781]) | 0\n",
-      "1367     | torch.Size([1, 1000, 762]) | 0\n",
-      "1368     | torch.Size([1, 1000, 764]) | 0\n",
-      "1369     | torch.Size([1, 1000, 777]) | 0\n",
-      "1370     | torch.Size([1, 1000, 757]) | 0\n",
-      "1371     | torch.Size([1, 1000, 780]) | 0\n",
-      "1372     | torch.Size([1, 1000, 754]) | 0\n",
-      "1373     | torch.Size([1, 1000, 754]) | 0\n",
-      "1374     | torch.Size([1, 1000, 843]) | 0\n",
-      "1375     | torch.Size([1, 1000, 754]) | 0\n",
-      "1376     | torch.Size([1, 1000, 754]) | 0\n",
-      "1377     | torch.Size([1, 1000, 783]) | 0\n",
-      "1378     | torch.Size([1, 1000, 754]) | 0\n",
-      "1379     | torch.Size([1, 1000, 762]) | 0\n",
-      "1380     | torch.Size([1, 1000, 751]) | 0\n",
-      "1381     | torch.Size([1, 1000, 781]) | 0\n",
-      "1382     | torch.Size([1, 1000, 754]) | 0\n",
-      "1383     | torch.Size([1, 1000, 784]) | 0\n",
-      "1384     | torch.Size([1, 1000, 767]) | 0\n",
-      "1385     | torch.Size([1, 1000, 611]) | 0\n",
-      "1386     | torch.Size([1, 1000, 754]) | 0\n",
-      "1387     | torch.Size([1, 1000, 754]) | 0\n",
-      "1388     | torch.Size([1, 1000, 754]) | 0\n",
-      "1389     | torch.Size([1, 1000, 754]) | 0\n",
-      "1390     | torch.Size([1, 1000, 771]) | 0\n",
-      "1391     | torch.Size([1, 1000, 762]) | 0\n",
-      "1392     | torch.Size([1, 1000, 771]) | 0\n",
-      "1393     | torch.Size([1, 1000, 611]) | 0\n",
-      "1394     | torch.Size([1, 1000, 762]) | 0\n",
-      "1395     | torch.Size([1, 1000, 611]) | 0\n",
-      "1396     | torch.Size([1, 1000, 754]) | 0\n",
-      "1397     | torch.Size([1, 1000, 780]) | 0\n",
-      "1398     | torch.Size([1, 1000, 771]) | 0\n",
-      "1399     | torch.Size([1, 1000, 779]) | 0\n",
-      "1400     | torch.Size([1, 1000, 754]) | 0\n",
-      "1401     | torch.Size([1, 1000, 754]) | 0\n",
-      "1402     | torch.Size([1, 1000, 754]) | 0\n",
-      "1403     | torch.Size([1, 1000, 799]) | 0\n",
-      "1404     | torch.Size([1, 1000, 802]) | 0\n",
-      "1405     | torch.Size([1, 1000, 793]) | 0\n",
-      "1406     | torch.Size([1, 1000, 784]) | 0\n",
-      "1407     | torch.Size([1, 1000, 754]) | 0\n",
-      "1408     | torch.Size([1, 1000, 778]) | 0\n",
-      "1409     | torch.Size([1, 1000, 778]) | 0\n",
-      "1410     | torch.Size([1, 1000, 762]) | 0\n",
-      "1411     | torch.Size([1, 1000, 780]) | 0\n",
-      "1412     | torch.Size([1, 1000, 771]) | 0\n",
-      "1413     | torch.Size([1, 1000, 754]) | 0\n",
-      "1414     | torch.Size([1, 1000, 757]) | 0\n",
-      "1415     | torch.Size([1, 1000, 769]) | 0\n",
-      "1416     | torch.Size([1, 1000, 754]) | 0\n",
-      "1417     | torch.Size([1, 1000, 779]) | 0\n",
-      "1418     | torch.Size([1, 1000, 754]) | 0\n",
-      "1419     | torch.Size([1, 1000, 788]) | 0\n",
-      "1420     | torch.Size([1, 1000, 867]) | 0\n",
-      "1421     | torch.Size([1, 1000, 762]) | 0\n",
-      "1422     | torch.Size([1, 1000, 754]) | 0\n",
-      "1423     | torch.Size([1, 1000, 754]) | 0\n",
-      "1424     | torch.Size([1, 1000, 777]) | 0\n",
-      "1425     | torch.Size([1, 1000, 754]) | 0\n",
-      "1426     | torch.Size([1, 1000, 1473]) | 0\n",
-      "1427     | torch.Size([1, 1000, 762]) | 0\n",
-      "1428     | torch.Size([1, 1000, 771]) | 0\n",
-      "1429     | torch.Size([1, 1000, 766]) | 0\n",
-      "1430     | torch.Size([1, 1000, 757]) | 0\n",
-      "1431     | torch.Size([1, 1000, 754]) | 0\n",
-      "1432     | torch.Size([1, 1000, 754]) | 0\n",
-      "1433     | torch.Size([1, 1000, 754]) | 0\n",
-      "1434     | torch.Size([1, 1000, 757]) | 0\n",
-      "1435     | torch.Size([1, 1000, 754]) | 0\n",
-      "1436     | torch.Size([1, 1000, 796]) | 0\n",
-      "1437     | torch.Size([1, 1000, 757]) | 0\n",
-      "1438     | torch.Size([1, 1000, 611]) | 0\n",
-      "1439     | torch.Size([1, 1000, 754]) | 0\n",
-      "1440     | torch.Size([1, 1000, 808]) | 0\n",
-      "1441     | torch.Size([1, 1000, 764]) | 0\n",
-      "1442     | torch.Size([1, 1000, 754]) | 0\n",
-      "1443     | torch.Size([1, 1000, 754]) | 0\n",
-      "1444     | torch.Size([1, 1000, 754]) | 0\n",
-      "1445     | torch.Size([1, 1000, 771]) | 0\n",
-      "1446     | torch.Size([1, 1000, 771]) | 0\n",
-      "1447     | torch.Size([1, 1000, 754]) | 0\n",
-      "1448     | torch.Size([1, 1000, 792]) | 0\n",
-      "1449     | torch.Size([1, 1000, 792]) | 0\n",
-      "1450     | torch.Size([1, 1000, 762]) | 0\n",
-      "1451     | torch.Size([1, 1000, 771]) | 0\n",
-      "1452     | torch.Size([1, 1000, 754]) | 0\n",
-      "1453     | torch.Size([1, 1000, 754]) | 0\n",
-      "1454     | torch.Size([1, 1000, 757]) | 0\n",
-      "1455     | torch.Size([1, 1000, 778]) | 0\n",
-      "1456     | torch.Size([1, 1000, 754]) | 0\n",
-      "1457     | torch.Size([1, 1000, 793]) | 0\n",
-      "1458     | torch.Size([1, 1000, 1473]) | 0\n",
-      "1459     | torch.Size([1, 1000, 754]) | 0\n",
-      "1460     | torch.Size([1, 1000, 757]) | 0\n",
-      "1461     | torch.Size([1, 1000, 778]) | 0\n",
-      "1462     | torch.Size([1, 1000, 754]) | 0\n",
-      "1463     | torch.Size([1, 1000, 762]) | 0\n",
-      "1464     | torch.Size([1, 1000, 754]) | 0\n",
-      "1465     | torch.Size([1, 1000, 783]) | 0\n",
-      "1466     | torch.Size([1, 1000, 762]) | 0\n",
-      "1467     | torch.Size([1, 1000, 762]) | 0\n",
-      "1468     | torch.Size([1, 1000, 754]) | 0\n",
-      "1469     | torch.Size([1, 1000, 771]) | 0\n",
-      "1470     | torch.Size([1, 1000, 810]) | 0\n",
-      "1471     | torch.Size([1, 1000, 754]) | 0\n",
-      "1472     | torch.Size([1, 1000, 754]) | 0\n",
-      "1473     | torch.Size([1, 1000, 777]) | 0\n",
-      "1474     | torch.Size([1, 1000, 782]) | 0\n",
-      "1475     | torch.Size([1, 1000, 757]) | 0\n",
-      "1476     | torch.Size([1, 1000, 754]) | 0\n",
-      "1477     | torch.Size([1, 1000, 754]) | 0\n",
-      "1478     | torch.Size([1, 1000, 762]) | 0\n",
-      "1479     | torch.Size([1, 1000, 762]) | 0\n",
-      "1480     | torch.Size([1, 1000, 754]) | 0\n",
-      "1481     | torch.Size([1, 1000, 744]) | 0\n",
-      "1482     | torch.Size([1, 1000, 802]) | 0\n",
-      "1483     | torch.Size([1, 1000, 754]) | 0\n",
-      "1484     | torch.Size([1, 1000, 754]) | 0\n",
-      "1485     | torch.Size([1, 1000, 774]) | 0\n",
-      "1486     | torch.Size([1, 1000, 648]) | 0\n",
-      "1487     | torch.Size([1, 1000, 754]) | 0\n",
-      "1488     | torch.Size([1, 1000, 757]) | 0\n",
-      "1489     | torch.Size([1, 1000, 975]) | 0\n",
-      "1490     | torch.Size([1, 1000, 611]) | 0\n",
-      "1491     | torch.Size([1, 1000, 754]) | 0\n",
-      "1492     | torch.Size([1, 1000, 771]) | 0\n",
-      "1493     | torch.Size([1, 1000, 762]) | 0\n",
-      "1494     | torch.Size([1, 1000, 778]) | 0\n",
-      "1495     | torch.Size([1, 1000, 754]) | 0\n",
-      "1496     | torch.Size([1, 1000, 773]) | 0\n",
-      "1497     | torch.Size([1, 1000, 777]) | 0\n",
-      "1498     | torch.Size([1, 1000, 777]) | 0\n",
-      "1499     | torch.Size([1, 1000, 775]) | 0\n",
-      "1500     | torch.Size([1, 1000, 754]) | 0\n",
-      "1501     | torch.Size([1, 1000, 754]) | 0\n",
-      "1502     | torch.Size([1, 1000, 779]) | 0\n",
-      "1503     | torch.Size([1, 1000, 754]) | 0\n",
-      "1504     | torch.Size([1, 1000, 785]) | 0\n",
-      "1505     | torch.Size([1, 1000, 773]) | 0\n",
-      "1506     | torch.Size([1, 1000, 754]) | 0\n",
-      "1507     | torch.Size([1, 1000, 757]) | 0\n",
-      "1508     | torch.Size([1, 1000, 762]) | 0\n",
-      "1509     | torch.Size([1, 1000, 773]) | 0\n",
-      "1510     | torch.Size([1, 1000, 754]) | 0\n",
-      "1511     | torch.Size([1, 1000, 786]) | 0\n",
-      "1512     | torch.Size([1, 1000, 754]) | 0\n",
-      "1513     | torch.Size([1, 1000, 777]) | 0\n",
-      "1514     | torch.Size([1, 1000, 754]) | 0\n",
-      "1515     | torch.Size([1, 1000, 785]) | 0\n",
-      "1516     | torch.Size([1, 1000, 754]) | 0\n",
-      "1517     | torch.Size([1, 1000, 774]) | 0\n",
-      "1518     | torch.Size([1, 1000, 777]) | 0\n",
-      "1519     | torch.Size([1, 1000, 754]) | 0\n",
-      "1520     | torch.Size([1, 1000, 782]) | 0\n",
-      "1521     | torch.Size([1, 1000, 771]) | 0\n",
-      "1522     | torch.Size([1, 1000, 792]) | 0\n",
-      "1523     | torch.Size([1, 1000, 762]) | 0\n",
-      "1524     | torch.Size([1, 1000, 771]) | 0\n",
-      "1525     | torch.Size([1, 1000, 754]) | 0\n",
-      "1526     | torch.Size([1, 1000, 754]) | 0\n",
-      "1527     | torch.Size([1, 1000, 771]) | 0\n",
-      "1528     | torch.Size([1, 1000, 754]) | 0\n",
-      "1529     | torch.Size([1, 1000, 773]) | 0\n",
-      "1530     | torch.Size([1, 1000, 762]) | 0\n",
-      "1531     | torch.Size([1, 1000, 754]) | 0\n",
-      "1532     | torch.Size([1, 1000, 754]) | 0\n",
-      "1533     | torch.Size([1, 1000, 771]) | 0\n",
-      "1534     | torch.Size([1, 1000, 777]) | 0\n",
-      "1535     | torch.Size([1, 1000, 754]) | 0\n",
-      "1536     | torch.Size([1, 1000, 777]) | 0\n",
-      "1537     | torch.Size([1, 1000, 754]) | 0\n",
-      "1538     | torch.Size([1, 1000, 754]) | 0\n",
-      "1539     | torch.Size([1, 1000, 649]) | 0\n",
-      "1540     | torch.Size([1, 1000, 754]) | 0\n",
-      "1541     | torch.Size([1, 1000, 754]) | 0\n",
-      "1542     | torch.Size([1, 1000, 868]) | 0\n",
-      "1543     | torch.Size([1, 1000, 757]) | 0\n",
-      "1544     | torch.Size([1, 1000, 757]) | 0\n",
-      "1545     | torch.Size([1, 1000, 771]) | 0\n",
-      "1546     | torch.Size([1, 1000, 754]) | 0\n",
-      "1547     | torch.Size([1, 1000, 791]) | 0\n",
-      "1548     | torch.Size([1, 1000, 757]) | 0\n",
-      "1549     | torch.Size([1, 1000, 757]) | 0\n",
-      "1550     | torch.Size([1, 1000, 764]) | 0\n",
-      "1551     | torch.Size([1, 1000, 762]) | 0\n",
-      "1552     | torch.Size([1, 1000, 754]) | 0\n",
-      "1553     | torch.Size([1, 1000, 771]) | 0\n",
-      "1554     | torch.Size([1, 1000, 762]) | 0\n",
-      "1555     | torch.Size([1, 1000, 797]) | 0\n",
-      "1556     | torch.Size([1, 1000, 754]) | 0\n",
-      "1557     | torch.Size([1, 1000, 762]) | 0\n",
-      "1558     | torch.Size([1, 1000, 754]) | 0\n",
-      "1559     | torch.Size([1, 1000, 754]) | 0\n",
-      "1560     | torch.Size([1, 1000, 802]) | 0\n",
-      "1561     | torch.Size([1, 1000, 762]) | 0\n",
-      "1562     | torch.Size([1, 1000, 771]) | 0\n",
-      "1563     | torch.Size([1, 1000, 820]) | 0\n",
-      "1564     | torch.Size([1, 1000, 754]) | 0\n",
-      "1565     | torch.Size([1, 1000, 754]) | 0\n",
-      "1566     | torch.Size([1, 1000, 731]) | 0\n",
-      "1567     | torch.Size([1, 1000, 754]) | 0\n",
-      "1568     | torch.Size([1, 1000, 774]) | 0\n",
-      "1569     | torch.Size([1, 1000, 611]) | 0\n",
-      "1570     | torch.Size([1, 1000, 762]) | 0\n",
-      "1571     | torch.Size([1, 1000, 754]) | 0\n",
-      "1572     | torch.Size([1, 1000, 777]) | 0\n",
-      "1573     | torch.Size([1, 1000, 754]) | 0\n",
-      "1574     | torch.Size([1, 1000, 754]) | 0\n",
-      "1575     | torch.Size([1, 1000, 754]) | 0\n",
-      "1576     | torch.Size([1, 1000, 754]) | 0\n",
-      "1577     | torch.Size([1, 1000, 757]) | 0\n",
-      "1578     | torch.Size([1, 1000, 754]) | 0\n",
-      "1579     | torch.Size([1, 1000, 754]) | 0\n",
-      "1580     | torch.Size([1, 1000, 754]) | 0\n",
-      "1581     | torch.Size([1, 1000, 787]) | 0\n",
-      "1582     | torch.Size([1, 1000, 754]) | 0\n",
-      "1583     | torch.Size([1, 1000, 762]) | 0\n",
-      "1584     | torch.Size([1, 1000, 754]) | 0\n",
-      "1585     | torch.Size([1, 1000, 754]) | 0\n",
-      "1586     | torch.Size([1, 1000, 754]) | 0\n",
-      "1587     | torch.Size([1, 1000, 812]) | 0\n",
-      "1588     | torch.Size([1, 1000, 779]) | 0\n",
-      "1589     | torch.Size([1, 1000, 757]) | 0\n",
-      "1590     | torch.Size([1, 1000, 754]) | 0\n",
-      "1591     | torch.Size([1, 1000, 754]) | 0\n",
-      "1592     | torch.Size([1, 1000, 757]) | 0\n",
-      "1593     | torch.Size([1, 1000, 777]) | 0\n",
-      "1594     | torch.Size([1, 1000, 754]) | 0\n",
-      "1595     | torch.Size([1, 1000, 776]) | 0\n",
-      "1596     | torch.Size([1, 1000, 821]) | 0\n",
-      "1597     | torch.Size([1, 1000, 757]) | 0\n",
-      "1598     | torch.Size([1, 1000, 789]) | 0\n",
-      "1599     | torch.Size([1, 1000, 754]) | 0\n",
-      "1600     | torch.Size([1, 1000, 754]) | 0\n",
-      "1601     | torch.Size([1, 1000, 754]) | 0\n",
-      "1602     | torch.Size([1, 1000, 799]) | 0\n",
-      "1603     | torch.Size([1, 1000, 754]) | 0\n",
-      "1604     | torch.Size([1, 1000, 754]) | 0\n",
-      "1605     | torch.Size([1, 1000, 754]) | 0\n",
-      "1606     | torch.Size([1, 1000, 754]) | 0\n",
-      "1607     | torch.Size([1, 1000, 751]) | 0\n",
-      "1608     | torch.Size([1, 1000, 754]) | 0\n",
-      "1609     | torch.Size([1, 1000, 777]) | 0\n",
-      "1610     | torch.Size([1, 1000, 769]) | 0\n",
-      "1611     | torch.Size([1, 1000, 764]) | 0\n",
-      "1612     | torch.Size([1, 1000, 768]) | 0\n",
-      "1613     | torch.Size([1, 1000, 754]) | 0\n",
-      "1614     | torch.Size([1, 1000, 762]) | 0\n",
-      "1615     | torch.Size([1, 1000, 802]) | 0\n",
-      "1616     | torch.Size([1, 1000, 754]) | 0\n",
-      "1617     | torch.Size([1, 1000, 777]) | 0\n",
-      "1618     | torch.Size([1, 1000, 777]) | 0\n",
-      "1619     | torch.Size([1, 1000, 754]) | 0\n",
-      "1620     | torch.Size([1, 1000, 754]) | 0\n",
-      "1621     | torch.Size([1, 1000, 802]) | 0\n",
-      "1622     | torch.Size([1, 1000, 773]) | 0\n",
-      "1623     | torch.Size([1, 1000, 762]) | 0\n",
-      "1624     | torch.Size([1, 1000, 754]) | 0\n",
-      "1625     | torch.Size([1, 1000, 762]) | 0\n",
-      "1626     | torch.Size([1, 1000, 754]) | 0\n",
-      "1627     | torch.Size([1, 1000, 785]) | 0\n",
-      "1628     | torch.Size([1, 1000, 754]) | 0\n",
-      "1629     | torch.Size([1, 1000, 611]) | 0\n",
-      "1630     | torch.Size([1, 1000, 762]) | 0\n",
-      "1631     | torch.Size([1, 1000, 754]) | 0\n",
-      "1632     | torch.Size([1, 1000, 754]) | 0\n",
-      "1633     | torch.Size([1, 1000, 754]) | 0\n",
-      "1634     | torch.Size([1, 1000, 1358]) | 0\n",
-      "1635     | torch.Size([1, 1000, 754]) | 0\n",
-      "1636     | torch.Size([1, 1000, 762]) | 0\n",
-      "1637     | torch.Size([1, 1000, 784]) | 0\n",
-      "1638     | torch.Size([1, 1000, 777]) | 0\n",
-      "1639     | torch.Size([1, 1000, 762]) | 0\n",
-      "1640     | torch.Size([1, 1000, 778]) | 0\n",
-      "1641     | torch.Size([1, 1000, 771]) | 0\n",
-      "1642     | torch.Size([1, 1000, 793]) | 0\n",
-      "1643     | torch.Size([1, 1000, 769]) | 0\n",
-      "1644     | torch.Size([1, 1000, 751]) | 0\n",
-      "1645     | torch.Size([1, 1000, 754]) | 0\n",
-      "1646     | torch.Size([1, 1000, 796]) | 0\n",
-      "1647     | torch.Size([1, 1000, 785]) | 0\n",
-      "1648     | torch.Size([1, 1000, 772]) | 0\n",
-      "1649     | torch.Size([1, 1000, 777]) | 0\n",
-      "1650     | torch.Size([1, 1000, 757]) | 0\n",
-      "1651     | torch.Size([1, 1000, 754]) | 0\n",
-      "1652     | torch.Size([1, 1000, 754]) | 0\n",
-      "1653     | torch.Size([1, 1000, 754]) | 0\n",
-      "1654     | torch.Size([1, 1000, 781]) | 0\n",
-      "1655     | torch.Size([1, 1000, 757]) | 0\n",
-      "1656     | torch.Size([1, 1000, 754]) | 0\n",
-      "1657     | torch.Size([1, 1000, 754]) | 0\n",
-      "1658     | torch.Size([1, 1000, 754]) | 0\n",
-      "1659     | torch.Size([1, 1000, 754]) | 0\n",
-      "1660     | torch.Size([1, 1000, 754]) | 0\n",
-      "1661     | torch.Size([1, 1000, 754]) | 0\n",
-      "1662     | torch.Size([1, 1000, 757]) | 0\n",
-      "1663     | torch.Size([1, 1000, 762]) | 0\n",
-      "1664     | torch.Size([1, 1000, 771]) | 0\n",
-      "1665     | torch.Size([1, 1000, 762]) | 0\n",
-      "1666     | torch.Size([1, 1000, 787]) | 0\n",
-      "1667     | torch.Size([1, 1000, 780]) | 0\n",
-      "1668     | torch.Size([1, 1000, 771]) | 0\n",
-      "1669     | torch.Size([1, 1000, 754]) | 0\n",
-      "1670     | torch.Size([1, 1000, 762]) | 0\n",
-      "1671     | torch.Size([1, 1000, 757]) | 0\n",
-      "1672     | torch.Size([1, 1000, 754]) | 0\n",
-      "1673     | torch.Size([1, 1000, 762]) | 0\n",
-      "1674     | torch.Size([1, 1000, 753]) | 0\n",
-      "1675     | torch.Size([1, 1000, 782]) | 0\n",
-      "1676     | torch.Size([1, 1000, 780]) | 0\n",
-      "1677     | torch.Size([1, 1000, 783]) | 0\n",
-      "1678     | torch.Size([1, 1000, 754]) | 0\n",
-      "1679     | torch.Size([1, 1000, 786]) | 0\n",
-      "1680     | torch.Size([1, 1000, 782]) | 0\n",
-      "1681     | torch.Size([1, 1000, 781]) | 0\n",
-      "1682     | torch.Size([1, 1000, 754]) | 0\n",
-      "1683     | torch.Size([1, 1000, 772]) | 0\n",
-      "1684     | torch.Size([1, 1000, 754]) | 0\n",
-      "1685     | torch.Size([1, 1000, 754]) | 0\n",
-      "1686     | torch.Size([1, 1000, 777]) | 0\n",
-      "1687     | torch.Size([1, 1000, 754]) | 0\n",
-      "1688     | torch.Size([1, 1000, 762]) | 0\n",
-      "1689     | torch.Size([1, 1000, 809]) | 0\n",
-      "1690     | torch.Size([1, 1000, 754]) | 0\n",
-      "1691     | torch.Size([1, 1000, 754]) | 0\n",
-      "1692     | torch.Size([1, 1000, 754]) | 0\n",
-      "1693     | torch.Size([1, 1000, 777]) | 0\n",
-      "1694     | torch.Size([1, 1000, 754]) | 0\n",
-      "1695     | torch.Size([1, 1000, 780]) | 0\n",
-      "1696     | torch.Size([1, 1000, 754]) | 0\n",
-      "1697     | torch.Size([1, 1000, 754]) | 0\n",
-      "1698     | torch.Size([1, 1000, 771]) | 0\n",
-      "1699     | torch.Size([1, 1000, 751]) | 0\n",
-      "1700     | torch.Size([1, 1000, 771]) | 0\n",
-      "1701     | torch.Size([1, 1000, 754]) | 0\n",
-      "1702     | torch.Size([1, 1000, 754]) | 0\n",
-      "1703     | torch.Size([1, 1000, 773]) | 0\n",
-      "1704     | torch.Size([1, 1000, 754]) | 0\n",
-      "1705     | torch.Size([1, 1000, 754]) | 0\n",
-      "1706     | torch.Size([1, 1000, 780]) | 0\n",
-      "1707     | torch.Size([1, 1000, 754]) | 0\n",
-      "1708     | torch.Size([1, 1000, 771]) | 0\n",
-      "1709     | torch.Size([1, 1000, 841]) | 0\n",
-      "1710     | torch.Size([1, 1000, 754]) | 0\n",
-      "1711     | torch.Size([1, 1000, 733]) | 0\n",
-      "1712     | torch.Size([1, 1000, 762]) | 0\n",
-      "1713     | torch.Size([1, 1000, 762]) | 0\n",
-      "1714     | torch.Size([1, 1000, 754]) | 0\n",
-      "1715     | torch.Size([1, 1000, 724]) | 0\n",
-      "1716     | torch.Size([1, 1000, 754]) | 0\n",
-      "1717     | torch.Size([1, 1000, 762]) | 0\n",
-      "1718     | torch.Size([1, 1000, 757]) | 0\n",
-      "1719     | torch.Size([1, 1000, 762]) | 0\n",
-      "1720     | torch.Size([1, 1000, 793]) | 0\n",
-      "1721     | torch.Size([1, 1000, 762]) | 0\n",
-      "1722     | torch.Size([1, 1000, 762]) | 0\n",
-      "1723     | torch.Size([1, 1000, 754]) | 0\n",
-      "1724     | torch.Size([1, 1000, 791]) | 0\n",
-      "1725     | torch.Size([1, 1000, 754]) | 0\n",
-      "1726     | torch.Size([1, 1000, 613]) | 0\n",
-      "1727     | torch.Size([1, 1000, 785]) | 0\n",
-      "1728     | torch.Size([1, 1000, 754]) | 0\n",
-      "1729     | torch.Size([1, 1000, 754]) | 0\n",
-      "1730     | torch.Size([1, 1000, 771]) | 0\n",
-      "1731     | torch.Size([1, 1000, 754]) | 0\n",
-      "1732     | torch.Size([1, 1000, 802]) | 0\n",
-      "1733     | torch.Size([1, 1000, 762]) | 0\n",
-      "1734     | torch.Size([1, 1000, 762]) | 0\n",
-      "1735     | torch.Size([1, 1000, 786]) | 0\n",
-      "1736     | torch.Size([1, 1000, 754]) | 0\n",
-      "1737     | torch.Size([1, 1000, 762]) | 0\n",
-      "1738     | torch.Size([1, 1000, 754]) | 0\n",
-      "1739     | torch.Size([1, 1000, 778]) | 0\n",
-      "1740     | torch.Size([1, 1000, 774]) | 0\n",
-      "1741     | torch.Size([1, 1000, 754]) | 0\n",
-      "1742     | torch.Size([1, 1000, 781]) | 0\n",
-      "1743     | torch.Size([1, 1000, 754]) | 0\n",
-      "1744     | torch.Size([1, 1000, 754]) | 0\n",
-      "1745     | torch.Size([1, 1000, 772]) | 0\n",
-      "1746     | torch.Size([1, 1000, 754]) | 0\n",
-      "1747     | torch.Size([1, 1000, 754]) | 0\n",
-      "1748     | torch.Size([1, 1000, 754]) | 0\n",
-      "1749     | torch.Size([1, 1000, 754]) | 0\n",
-      "1750     | torch.Size([1, 1000, 762]) | 0\n",
-      "1751     | torch.Size([1, 1000, 791]) | 0\n",
-      "1752     | torch.Size([1, 1000, 754]) | 0\n",
-      "1753     | torch.Size([1, 1000, 802]) | 0\n",
-      "1754     | torch.Size([1, 1000, 754]) | 0\n",
-      "1755     | torch.Size([1, 1000, 768]) | 0\n",
-      "1756     | torch.Size([1, 1000, 820]) | 0\n",
-      "1757     | torch.Size([1, 1000, 754]) | 0\n",
-      "1758     | torch.Size([1, 1000, 785]) | 0\n",
-      "1759     | torch.Size([1, 1000, 754]) | 0\n",
-      "1760     | torch.Size([1, 1000, 757]) | 0\n",
-      "1761     | torch.Size([1, 1000, 771]) | 0\n",
-      "1762     | torch.Size([1, 1000, 754]) | 0\n",
-      "1763     | torch.Size([1, 1000, 762]) | 0\n",
-      "1764     | torch.Size([1, 1000, 771]) | 0\n",
-      "1765     | torch.Size([1, 1000, 762]) | 0\n",
-      "1766     | torch.Size([1, 1000, 754]) | 0\n",
-      "1767     | torch.Size([1, 1000, 783]) | 0\n",
-      "1768     | torch.Size([1, 1000, 773]) | 0\n",
-      "1769     | torch.Size([1, 1000, 762]) | 0\n",
-      "1770     | torch.Size([1, 1000, 762]) | 0\n",
-      "1771     | torch.Size([1, 1000, 611]) | 0\n",
-      "1772     | torch.Size([1, 1000, 751]) | 0\n",
-      "1773     | torch.Size([1, 1000, 761]) | 0\n",
-      "1774     | torch.Size([1, 1000, 754]) | 0\n",
-      "1775     | torch.Size([1, 1000, 788]) | 0\n",
-      "1776     | torch.Size([1, 1000, 611]) | 0\n",
-      "1777     | torch.Size([1, 1000, 754]) | 0\n",
-      "1778     | torch.Size([1, 1000, 778]) | 0\n",
-      "1779     | torch.Size([1, 1000, 762]) | 0\n",
-      "1780     | torch.Size([1, 1000, 762]) | 0\n",
-      "1781     | torch.Size([1, 1000, 762]) | 0\n",
-      "1782     | torch.Size([1, 1000, 959]) | 0\n",
-      "1783     | torch.Size([1, 1000, 762]) | 0\n",
-      "1784     | torch.Size([1, 1000, 762]) | 0\n",
-      "1785     | torch.Size([1, 1000, 772]) | 0\n",
-      "1786     | torch.Size([1, 1000, 754]) | 0\n",
-      "1787     | torch.Size([1, 1000, 798]) | 0\n",
-      "1788     | torch.Size([1, 1000, 754]) | 0\n",
-      "1789     | torch.Size([1, 1000, 777]) | 0\n",
-      "1790     | torch.Size([1, 1000, 754]) | 0\n",
-      "1791     | torch.Size([1, 1000, 774]) | 0\n",
-      "1792     | torch.Size([1, 1000, 757]) | 0\n",
-      "1793     | torch.Size([1, 1000, 754]) | 0\n",
-      "1794     | torch.Size([1, 1000, 782]) | 0\n",
-      "1795     | torch.Size([1, 1000, 762]) | 0\n",
-      "1796     | torch.Size([1, 1000, 754]) | 0\n",
-      "1797     | torch.Size([1, 1000, 754]) | 0\n",
-      "1798     | torch.Size([1, 1000, 762]) | 0\n",
-      "1799     | torch.Size([1, 1000, 651]) | 0\n",
-      "1800     | torch.Size([1, 1000, 762]) | 0\n",
-      "1801     | torch.Size([1, 1000, 754]) | 0\n",
-      "1802     | torch.Size([1, 1000, 771]) | 0\n",
-      "1803     | torch.Size([1, 1000, 757]) | 0\n",
-      "1804     | torch.Size([1, 1000, 754]) | 0\n",
-      "1805     | torch.Size([1, 1000, 762]) | 0\n",
-      "1806     | torch.Size([1, 1000, 777]) | 0\n",
-      "1807     | torch.Size([1, 1000, 757]) | 0\n",
-      "1808     | torch.Size([1, 1000, 754]) | 0\n",
-      "1809     | torch.Size([1, 1000, 754]) | 0\n",
-      "1810     | torch.Size([1, 1000, 843]) | 0\n",
-      "1811     | torch.Size([1, 1000, 754]) | 0\n",
-      "1812     | torch.Size([1, 1000, 754]) | 0\n",
-      "1813     | torch.Size([1, 1000, 757]) | 0\n",
-      "1814     | torch.Size([1, 1000, 754]) | 0\n",
-      "1815     | torch.Size([1, 1000, 757]) | 0\n",
-      "1816     | torch.Size([1, 1000, 762]) | 0\n",
-      "1817     | torch.Size([1, 1000, 757]) | 0\n",
-      "1818     | torch.Size([1, 1000, 769]) | 0\n",
-      "1819     | torch.Size([1, 1000, 754]) | 0\n",
-      "1820     | torch.Size([1, 1000, 754]) | 0\n",
-      "1821     | torch.Size([1, 1000, 754]) | 0\n",
-      "1822     | torch.Size([1, 1000, 754]) | 0\n",
-      "1823     | torch.Size([1, 1000, 754]) | 0\n",
-      "1824     | torch.Size([1, 1000, 762]) | 0\n",
-      "1825     | torch.Size([1, 1000, 754]) | 0\n",
-      "1826     | torch.Size([1, 1000, 762]) | 0\n",
-      "1827     | torch.Size([1, 1000, 754]) | 0\n",
-      "1828     | torch.Size([1, 1000, 754]) | 0\n",
-      "1829     | torch.Size([1, 1000, 772]) | 0\n",
-      "1830     | torch.Size([1, 1000, 741]) | 0\n",
-      "1831     | torch.Size([1, 1000, 754]) | 0\n",
-      "1832     | torch.Size([1, 1000, 754]) | 0\n",
-      "1833     | torch.Size([1, 1000, 611]) | 0\n",
-      "1834     | torch.Size([1, 1000, 754]) | 0\n",
-      "1835     | torch.Size([1, 1000, 754]) | 0\n",
-      "1836     | torch.Size([1, 1000, 777]) | 0\n",
-      "1837     | torch.Size([1, 1000, 778]) | 0\n",
-      "1838     | torch.Size([1, 1000, 798]) | 0\n",
-      "1839     | torch.Size([1, 1000, 754]) | 0\n",
-      "1840     | torch.Size([1, 1000, 754]) | 0\n",
-      "1841     | torch.Size([1, 1000, 754]) | 0\n",
-      "1842     | torch.Size([1, 1000, 757]) | 0\n",
-      "1843     | torch.Size([1, 1000, 754]) | 0\n",
-      "1844     | torch.Size([1, 1000, 762]) | 0\n",
-      "1845     | torch.Size([1, 1000, 781]) | 0\n",
-      "1846     | torch.Size([1, 1000, 754]) | 0\n",
-      "1847     | torch.Size([1, 1000, 754]) | 0\n",
-      "1848     | torch.Size([1, 1000, 754]) | 0\n",
-      "1849     | torch.Size([1, 1000, 787]) | 0\n",
-      "1850     | torch.Size([1, 1000, 760]) | 0\n",
-      "1851     | torch.Size([1, 1000, 757]) | 0\n",
-      "1852     | torch.Size([1, 1000, 754]) | 0\n",
-      "1853     | torch.Size([1, 1000, 786]) | 0\n",
-      "1854     | torch.Size([1, 1000, 754]) | 0\n",
-      "1855     | torch.Size([1, 1000, 797]) | 0\n",
-      "1856     | torch.Size([1, 1000, 762]) | 0\n",
-      "1857     | torch.Size([1, 1000, 783]) | 0\n",
-      "1858     | torch.Size([1, 1000, 762]) | 0\n",
-      "1859     | torch.Size([1, 1000, 815]) | 0\n",
-      "1860     | torch.Size([1, 1000, 798]) | 0\n",
-      "1861     | torch.Size([1, 1000, 776]) | 0\n",
-      "1862     | torch.Size([1, 1000, 754]) | 0\n",
-      "1863     | torch.Size([1, 1000, 751]) | 0\n",
-      "1864     | torch.Size([1, 1000, 797]) | 0\n",
-      "1865     | torch.Size([1, 1000, 757]) | 0\n",
-      "1866     | torch.Size([1, 1000, 754]) | 0\n",
-      "1867     | torch.Size([1, 1000, 771]) | 0\n",
-      "1868     | torch.Size([1, 1000, 777]) | 0\n",
-      "1869     | torch.Size([1, 1000, 757]) | 0\n",
-      "1870     | torch.Size([1, 1000, 777]) | 0\n",
-      "1871     | torch.Size([1, 1000, 842]) | 0\n",
-      "1872     | torch.Size([1, 1000, 754]) | 0\n",
-      "1873     | torch.Size([1, 1000, 754]) | 0\n",
-      "1874     | torch.Size([1, 1000, 611]) | 0\n",
-      "1875     | torch.Size([1, 1000, 754]) | 0\n",
-      "1876     | torch.Size([1, 1000, 754]) | 0\n",
-      "1877     | torch.Size([1, 1000, 754]) | 0\n",
-      "1878     | torch.Size([1, 1000, 754]) | 0\n",
-      "1879     | torch.Size([1, 1000, 779]) | 0\n",
-      "1880     | torch.Size([1, 1000, 754]) | 0\n",
-      "1881     | torch.Size([1, 1000, 762]) | 0\n",
-      "1882     | torch.Size([1, 1000, 771]) | 0\n",
-      "1883     | torch.Size([1, 1000, 754]) | 0\n",
-      "1884     | torch.Size([1, 1000, 762]) | 0\n",
-      "1885     | torch.Size([1, 1000, 754]) | 0\n",
-      "1886     | torch.Size([1, 1000, 777]) | 0\n",
-      "1887     | torch.Size([1, 1000, 754]) | 0\n",
-      "1888     | torch.Size([1, 1000, 754]) | 0\n",
-      "1889     | torch.Size([1, 1000, 777]) | 0\n",
-      "1890     | torch.Size([1, 1000, 779]) | 0\n",
-      "1891     | torch.Size([1, 1000, 754]) | 0\n",
-      "1892     | torch.Size([1, 1000, 762]) | 0\n",
-      "1893     | torch.Size([1, 1000, 762]) | 0\n",
-      "1894     | torch.Size([1, 1000, 782]) | 0\n",
-      "1895     | torch.Size([1, 1000, 754]) | 0\n",
-      "1896     | torch.Size([1, 1000, 762]) | 0\n",
-      "1897     | torch.Size([1, 1000, 754]) | 0\n",
-      "1898     | torch.Size([1, 1000, 754]) | 0\n",
-      "1899     | torch.Size([1, 1000, 798]) | 0\n",
-      "1900     | torch.Size([1, 1000, 762]) | 0\n",
-      "1901     | torch.Size([1, 1000, 757]) | 0\n",
-      "1902     | torch.Size([1, 1000, 754]) | 0\n",
-      "1903     | torch.Size([1, 1000, 754]) | 0\n",
-      "1904     | torch.Size([1, 1000, 781]) | 0\n",
-      "1905     | torch.Size([1, 1000, 772]) | 0\n",
-      "1906     | torch.Size([1, 1000, 754]) | 0\n",
-      "1907     | torch.Size([1, 1000, 754]) | 0\n",
-      "1908     | torch.Size([1, 1000, 790]) | 0\n",
-      "1909     | torch.Size([1, 1000, 611]) | 0\n",
-      "1910     | torch.Size([1, 1000, 754]) | 0\n",
-      "1911     | torch.Size([1, 1000, 787]) | 0\n",
-      "1912     | torch.Size([1, 1000, 757]) | 0\n",
-      "1913     | torch.Size([1, 1000, 767]) | 0\n",
-      "1914     | torch.Size([1, 1000, 754]) | 0\n",
-      "1915     | torch.Size([1, 1000, 770]) | 0\n",
-      "1916     | torch.Size([1, 1000, 790]) | 0\n",
-      "1917     | torch.Size([1, 1000, 762]) | 0\n",
-      "1918     | torch.Size([1, 1000, 754]) | 0\n",
-      "1919     | torch.Size([1, 1000, 754]) | 0\n",
-      "1920     | torch.Size([1, 1000, 754]) | 0\n",
-      "1921     | torch.Size([1, 1000, 754]) | 0\n",
-      "1922     | torch.Size([1, 1000, 754]) | 0\n",
-      "1923     | torch.Size([1, 1000, 754]) | 0\n",
-      "1924     | torch.Size([1, 1000, 757]) | 0\n",
-      "1925     | torch.Size([1, 1000, 754]) | 0\n",
-      "1926     | torch.Size([1, 1000, 863]) | 0\n",
-      "1927     | torch.Size([1, 1000, 777]) | 0\n",
-      "1928     | torch.Size([1, 1000, 789]) | 0\n",
-      "1929     | torch.Size([1, 1000, 754]) | 0\n",
-      "1930     | torch.Size([1, 1000, 754]) | 0\n",
-      "1931     | torch.Size([1, 1000, 771]) | 0\n",
-      "1932     | torch.Size([1, 1000, 762]) | 0\n",
-      "1933     | torch.Size([1, 1000, 769]) | 0\n",
-      "1934     | torch.Size([1, 1000, 842]) | 0\n",
-      "1935     | torch.Size([1, 1000, 951]) | 0\n",
-      "1936     | torch.Size([1, 1000, 780]) | 0\n",
-      "1937     | torch.Size([1, 1000, 754]) | 0\n",
-      "1938     | torch.Size([1, 1000, 762]) | 0\n",
-      "1939     | torch.Size([1, 1000, 762]) | 0\n",
-      "1940     | torch.Size([1, 1000, 771]) | 0\n",
-      "1941     | torch.Size([1, 1000, 956]) | 0\n",
-      "1942     | torch.Size([1, 1000, 762]) | 0\n",
-      "1943     | torch.Size([1, 1000, 777]) | 0\n",
-      "1944     | torch.Size([1, 1000, 777]) | 0\n",
-      "1945     | torch.Size([1, 1000, 779]) | 0\n",
-      "1946     | torch.Size([1, 1000, 754]) | 0\n",
-      "1947     | torch.Size([1, 1000, 754]) | 0\n",
-      "1948     | torch.Size([1, 1000, 777]) | 0\n",
-      "1949     | torch.Size([1, 1000, 754]) | 0\n",
-      "1950     | torch.Size([1, 1000, 803]) | 0\n",
-      "1951     | torch.Size([1, 1000, 762]) | 0\n",
-      "1952     | torch.Size([1, 1000, 754]) | 0\n",
-      "1953     | torch.Size([1, 1000, 754]) | 0\n",
-      "1954     | torch.Size([1, 1000, 762]) | 0\n",
-      "1955     | torch.Size([1, 1000, 762]) | 0\n",
-      "1956     | torch.Size([1, 1000, 754]) | 0\n",
-      "1957     | torch.Size([1, 1000, 754]) | 0\n",
-      "1958     | torch.Size([1, 1000, 781]) | 0\n",
-      "1959     | torch.Size([1, 1000, 757]) | 0\n",
-      "1960     | torch.Size([1, 1000, 774]) | 0\n",
-      "1961     | torch.Size([1, 1000, 754]) | 0\n",
-      "1962     | torch.Size([1, 1000, 808]) | 0\n",
-      "1963     | torch.Size([1, 1000, 781]) | 0\n",
-      "1964     | torch.Size([1, 1000, 721]) | 0\n",
-      "1965     | torch.Size([1, 1000, 754]) | 0\n",
-      "1966     | torch.Size([1, 1000, 789]) | 0\n",
-      "1967     | torch.Size([1, 1000, 745]) | 0\n",
-      "1968     | torch.Size([1, 1000, 754]) | 0\n",
-      "1969     | torch.Size([1, 1000, 754]) | 0\n",
-      "1970     | torch.Size([1, 1000, 777]) | 0\n",
-      "1971     | torch.Size([1, 1000, 751]) | 0\n",
-      "1972     | torch.Size([1, 1000, 762]) | 0\n",
-      "1973     | torch.Size([1, 1000, 788]) | 0\n",
-      "1974     | torch.Size([1, 1000, 754]) | 0\n",
-      "1975     | torch.Size([1, 1000, 773]) | 0\n",
-      "1976     | torch.Size([1, 1000, 762]) | 0\n",
-      "1977     | torch.Size([1, 1000, 762]) | 0\n",
-      "1978     | torch.Size([1, 1000, 762]) | 0\n",
-      "1979     | torch.Size([1, 1000, 754]) | 0\n",
-      "1980     | torch.Size([1, 1000, 757]) | 0\n",
-      "1981     | torch.Size([1, 1000, 774]) | 0\n",
-      "1982     | torch.Size([1, 1000, 777]) | 0\n",
-      "1983     | torch.Size([1, 1000, 777]) | 0\n",
-      "1984     | torch.Size([1, 1000, 762]) | 0\n",
-      "1985     | torch.Size([1, 1000, 754]) | 0\n",
-      "1986     | torch.Size([1, 1000, 754]) | 0\n",
-      "1987     | torch.Size([1, 1000, 754]) | 0\n",
-      "1988     | torch.Size([1, 1000, 754]) | 0\n",
-      "1989     | torch.Size([1, 1000, 794]) | 0\n",
-      "1990     | torch.Size([1, 1000, 762]) | 0\n",
-      "1991     | torch.Size([1, 1000, 754]) | 0\n",
-      "1992     | torch.Size([1, 1000, 777]) | 0\n",
-      "1993     | torch.Size([1, 1000, 757]) | 0\n",
-      "1994     | torch.Size([1, 1000, 771]) | 0\n",
-      "1995     | torch.Size([1, 1000, 754]) | 0\n",
-      "1996     | torch.Size([1, 1000, 754]) | 0\n",
-      "1997     | torch.Size([1, 1000, 754]) | 0\n",
-      "1998     | torch.Size([1, 1000, 762]) | 0\n",
-      "1999     | torch.Size([1, 1000, 754]) | 0\n",
-      "2000     | torch.Size([1, 1000, 754]) | 0\n",
-      "2001     | torch.Size([1, 1000, 754]) | 0\n",
-      "2002     | torch.Size([1, 1000, 777]) | 0\n",
-      "2003     | torch.Size([1, 1000, 781]) | 0\n",
-      "2004     | torch.Size([1, 1000, 777]) | 0\n",
-      "2005     | torch.Size([1, 1000, 765]) | 0\n",
-      "2006     | torch.Size([1, 1000, 778]) | 0\n",
-      "2007     | torch.Size([1, 1000, 785]) | 0\n",
-      "2008     | torch.Size([1, 1000, 772]) | 0\n",
-      "2009     | torch.Size([1, 1000, 777]) | 0\n",
-      "2010     | torch.Size([1, 1000, 808]) | 0\n",
-      "2011     | torch.Size([1, 1000, 754]) | 0\n",
-      "2012     | torch.Size([1, 1000, 762]) | 0\n",
-      "2013     | torch.Size([1, 1000, 754]) | 0\n",
-      "2014     | torch.Size([1, 1000, 754]) | 0\n",
-      "2015     | torch.Size([1, 1000, 762]) | 0\n",
-      "2016     | torch.Size([1, 1000, 754]) | 0\n",
-      "2017     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2018     | torch.Size([1, 1000, 754]) | 0\n",
-      "2019     | torch.Size([1, 1000, 754]) | 0\n",
-      "2020     | torch.Size([1, 1000, 772]) | 0\n",
-      "2021     | torch.Size([1, 1000, 762]) | 0\n",
-      "2022     | torch.Size([1, 1000, 754]) | 0\n",
-      "2023     | torch.Size([1, 1000, 772]) | 0\n",
-      "2024     | torch.Size([1, 1000, 762]) | 0\n",
-      "2025     | torch.Size([1, 1000, 762]) | 0\n",
-      "2026     | torch.Size([1, 1000, 762]) | 0\n",
-      "2027     | torch.Size([1, 1000, 754]) | 0\n",
-      "2028     | torch.Size([1, 1000, 762]) | 0\n",
-      "2029     | torch.Size([1, 1000, 754]) | 0\n",
-      "2030     | torch.Size([1, 1000, 798]) | 0\n",
-      "2031     | torch.Size([1, 1000, 771]) | 0\n",
-      "2032     | torch.Size([1, 1000, 757]) | 0\n",
-      "2033     | torch.Size([1, 1000, 762]) | 0\n",
-      "2034     | torch.Size([1, 1000, 771]) | 0\n",
-      "2035     | torch.Size([1, 1000, 777]) | 0\n",
-      "2036     | torch.Size([1, 1000, 960]) | 0\n",
-      "2037     | torch.Size([1, 1000, 754]) | 0\n",
-      "2038     | torch.Size([1, 1000, 757]) | 0\n",
-      "2039     | torch.Size([1, 1000, 754]) | 0\n",
-      "2040     | torch.Size([1, 1000, 771]) | 0\n",
-      "2041     | torch.Size([1, 1000, 768]) | 0\n",
-      "2042     | torch.Size([1, 1000, 754]) | 0\n",
-      "2043     | torch.Size([1, 1000, 781]) | 0\n",
-      "2044     | torch.Size([1, 1000, 754]) | 0\n",
-      "2045     | torch.Size([1, 1000, 754]) | 0\n",
-      "2046     | torch.Size([1, 1000, 745]) | 0\n",
-      "2047     | torch.Size([1, 1000, 754]) | 0\n",
-      "2048     | torch.Size([1, 1000, 765]) | 0\n",
-      "2049     | torch.Size([1, 1000, 788]) | 0\n",
-      "2050     | torch.Size([1, 1000, 781]) | 0\n",
-      "2051     | torch.Size([1, 1000, 781]) | 0\n",
-      "2052     | torch.Size([1, 1000, 754]) | 0\n",
-      "2053     | torch.Size([1, 1000, 757]) | 0\n",
-      "2054     | torch.Size([1, 1000, 754]) | 0\n",
-      "2055     | torch.Size([1, 1000, 777]) | 0\n",
-      "2056     | torch.Size([1, 1000, 777]) | 0\n",
-      "2057     | torch.Size([1, 1000, 754]) | 0\n",
-      "2058     | torch.Size([1, 1000, 762]) | 0\n",
-      "2059     | torch.Size([1, 1000, 789]) | 0\n",
-      "2060     | torch.Size([1, 1000, 754]) | 0\n",
-      "2061     | torch.Size([1, 1000, 773]) | 0\n",
-      "2062     | torch.Size([1, 1000, 762]) | 0\n",
-      "2063     | torch.Size([1, 1000, 762]) | 0\n",
-      "2064     | torch.Size([1, 1000, 782]) | 0\n",
-      "2065     | torch.Size([1, 1000, 754]) | 0\n",
-      "2066     | torch.Size([1, 1000, 754]) | 0\n",
-      "2067     | torch.Size([1, 1000, 779]) | 0\n",
-      "2068     | torch.Size([1, 1000, 762]) | 0\n",
-      "2069     | torch.Size([1, 1000, 771]) | 0\n",
-      "2070     | torch.Size([1, 1000, 769]) | 0\n",
-      "2071     | torch.Size([1, 1000, 777]) | 0\n",
-      "2072     | torch.Size([1, 1000, 754]) | 0\n",
-      "2073     | torch.Size([1, 1000, 754]) | 0\n",
-      "2074     | torch.Size([1, 1000, 754]) | 0\n",
-      "2075     | torch.Size([1, 1000, 762]) | 0\n",
-      "2076     | torch.Size([1, 1000, 762]) | 0\n",
-      "2077     | torch.Size([1, 1000, 754]) | 0\n",
-      "2078     | torch.Size([1, 1000, 754]) | 0\n",
-      "2079     | torch.Size([1, 1000, 754]) | 0\n",
-      "2080     | torch.Size([1, 1000, 764]) | 0\n",
-      "2081     | torch.Size([1, 1000, 762]) | 0\n",
-      "2082     | torch.Size([1, 1000, 754]) | 0\n",
-      "2083     | torch.Size([1, 1000, 754]) | 0\n",
-      "2084     | torch.Size([1, 1000, 754]) | 0\n",
-      "2085     | torch.Size([1, 1000, 762]) | 0\n",
-      "2086     | torch.Size([1, 1000, 754]) | 0\n",
-      "2087     | torch.Size([1, 1000, 754]) | 0\n",
-      "2088     | torch.Size([1, 1000, 757]) | 0\n",
-      "2089     | torch.Size([1, 1000, 777]) | 0\n",
-      "2090     | torch.Size([1, 1000, 762]) | 0\n",
-      "2091     | torch.Size([1, 1000, 777]) | 0\n",
-      "2092     | torch.Size([1, 1000, 776]) | 0\n",
-      "2093     | torch.Size([1, 1000, 754]) | 0\n",
-      "2094     | torch.Size([1, 1000, 754]) | 0\n",
-      "2095     | torch.Size([1, 1000, 771]) | 0\n",
-      "2096     | torch.Size([1, 1000, 762]) | 0\n",
-      "2097     | torch.Size([1, 1000, 754]) | 0\n",
-      "2098     | torch.Size([1, 1000, 754]) | 0\n",
-      "2099     | torch.Size([1, 1000, 762]) | 0\n",
-      "2100     | torch.Size([1, 1000, 777]) | 0\n",
-      "2101     | torch.Size([1, 1000, 770]) | 0\n",
-      "2102     | torch.Size([1, 1000, 754]) | 0\n",
-      "2103     | torch.Size([1, 1000, 754]) | 0\n",
-      "2104     | torch.Size([1, 1000, 794]) | 0\n",
-      "2105     | torch.Size([1, 1000, 762]) | 0\n",
-      "2106     | torch.Size([1, 1000, 754]) | 0\n",
-      "2107     | torch.Size([1, 1000, 757]) | 0\n",
-      "2108     | torch.Size([1, 1000, 803]) | 0\n",
-      "2109     | torch.Size([1, 1000, 766]) | 0\n",
-      "2110     | torch.Size([1, 1000, 771]) | 0\n",
-      "2111     | torch.Size([1, 1000, 754]) | 0\n",
-      "2112     | torch.Size([1, 1000, 754]) | 0\n",
-      "2113     | torch.Size([1, 1000, 754]) | 0\n",
-      "2114     | torch.Size([1, 1000, 754]) | 0\n",
-      "2115     | torch.Size([1, 1000, 754]) | 0\n",
-      "2116     | torch.Size([1, 1000, 771]) | 0\n",
-      "2117     | torch.Size([1, 1000, 772]) | 0\n",
-      "2118     | torch.Size([1, 1000, 669]) | 0\n",
-      "2119     | torch.Size([1, 1000, 757]) | 0\n",
-      "2120     | torch.Size([1, 1000, 762]) | 0\n",
-      "2121     | torch.Size([1, 1000, 762]) | 0\n",
-      "2122     | torch.Size([1, 1000, 762]) | 0\n",
-      "2123     | torch.Size([1, 1000, 772]) | 0\n",
-      "2124     | torch.Size([1, 1000, 757]) | 0\n",
-      "2125     | torch.Size([1, 1000, 762]) | 0\n",
-      "2126     | torch.Size([1, 1000, 754]) | 0\n",
-      "2127     | torch.Size([1, 1000, 754]) | 0\n",
-      "2128     | torch.Size([1, 1000, 762]) | 0\n",
-      "2129     | torch.Size([1, 1000, 762]) | 0\n",
-      "2130     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2131     | torch.Size([1, 1000, 762]) | 0\n",
-      "2132     | torch.Size([1, 1000, 762]) | 0\n",
-      "2133     | torch.Size([1, 1000, 802]) | 0\n",
-      "2134     | torch.Size([1, 1000, 754]) | 0\n",
-      "2135     | torch.Size([1, 1000, 754]) | 0\n",
-      "2136     | torch.Size([1, 1000, 771]) | 0\n",
-      "2137     | torch.Size([1, 1000, 780]) | 0\n",
-      "2138     | torch.Size([1, 1000, 754]) | 0\n",
-      "2139     | torch.Size([1, 1000, 754]) | 0\n",
-      "2140     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2141     | torch.Size([1, 1000, 751]) | 0\n",
-      "2142     | torch.Size([1, 1000, 754]) | 0\n",
-      "2143     | torch.Size([1, 1000, 802]) | 0\n",
-      "2144     | torch.Size([1, 1000, 757]) | 0\n",
-      "2145     | torch.Size([1, 1000, 754]) | 0\n",
-      "2146     | torch.Size([1, 1000, 759]) | 0\n",
-      "2147     | torch.Size([1, 1000, 754]) | 0\n",
-      "2148     | torch.Size([1, 1000, 754]) | 0\n",
-      "2149     | torch.Size([1, 1000, 754]) | 0\n",
-      "2150     | torch.Size([1, 1000, 779]) | 0\n",
-      "2151     | torch.Size([1, 1000, 762]) | 0\n",
-      "2152     | torch.Size([1, 1000, 784]) | 0\n",
-      "2153     | torch.Size([1, 1000, 754]) | 0\n",
-      "2154     | torch.Size([1, 1000, 762]) | 0\n",
-      "2155     | torch.Size([1, 1000, 771]) | 0\n",
-      "2156     | torch.Size([1, 1000, 754]) | 0\n",
-      "2157     | torch.Size([1, 1000, 754]) | 0\n",
-      "2158     | torch.Size([1, 1000, 757]) | 0\n",
-      "2159     | torch.Size([1, 1000, 762]) | 0\n",
-      "2160     | torch.Size([1, 1000, 780]) | 0\n",
-      "2161     | torch.Size([1, 1000, 754]) | 0\n",
-      "2162     | torch.Size([1, 1000, 790]) | 0\n",
-      "2163     | torch.Size([1, 1000, 754]) | 0\n",
-      "2164     | torch.Size([1, 1000, 754]) | 0\n",
-      "2165     | torch.Size([1, 1000, 787]) | 0\n",
-      "2166     | torch.Size([1, 1000, 809]) | 0\n",
-      "2167     | torch.Size([1, 1000, 775]) | 0\n",
-      "2168     | torch.Size([1, 1000, 844]) | 0\n",
-      "2169     | torch.Size([1, 1000, 754]) | 0\n",
-      "2170     | torch.Size([1, 1000, 796]) | 0\n",
-      "2171     | torch.Size([1, 1000, 762]) | 0\n",
-      "2172     | torch.Size([1, 1000, 754]) | 0\n",
-      "2173     | torch.Size([1, 1000, 754]) | 0\n",
-      "2174     | torch.Size([1, 1000, 754]) | 0\n",
-      "2175     | torch.Size([1, 1000, 757]) | 0\n",
-      "2176     | torch.Size([1, 1000, 754]) | 0\n",
-      "2177     | torch.Size([1, 1000, 611]) | 0\n",
-      "2178     | torch.Size([1, 1000, 783]) | 0\n",
-      "2179     | torch.Size([1, 1000, 754]) | 0\n",
-      "2180     | torch.Size([1, 1000, 762]) | 0\n",
-      "2181     | torch.Size([1, 1000, 777]) | 0\n",
-      "2182     | torch.Size([1, 1000, 778]) | 0\n",
-      "2183     | torch.Size([1, 1000, 754]) | 0\n",
-      "2184     | torch.Size([1, 1000, 764]) | 0\n",
-      "2185     | torch.Size([1, 1000, 774]) | 0\n",
-      "2186     | torch.Size([1, 1000, 757]) | 0\n",
-      "2187     | torch.Size([1, 1000, 754]) | 0\n",
-      "2188     | torch.Size([1, 1000, 771]) | 0\n",
-      "2189     | torch.Size([1, 1000, 762]) | 0\n",
-      "2190     | torch.Size([1, 1000, 754]) | 0\n",
-      "2191     | torch.Size([1, 1000, 782]) | 0\n",
-      "2192     | torch.Size([1, 1000, 758]) | 0\n",
-      "2193     | torch.Size([1, 1000, 762]) | 0\n",
-      "2194     | torch.Size([1, 1000, 754]) | 0\n",
-      "2195     | torch.Size([1, 1000, 772]) | 0\n",
-      "2196     | torch.Size([1, 1000, 777]) | 0\n",
-      "2197     | torch.Size([1, 1000, 754]) | 0\n",
-      "2198     | torch.Size([1, 1000, 611]) | 0\n",
-      "2199     | torch.Size([1, 1000, 785]) | 0\n",
-      "2200     | torch.Size([1, 1000, 754]) | 0\n",
-      "2201     | torch.Size([1, 1000, 754]) | 0\n",
-      "2202     | torch.Size([1, 1000, 754]) | 0\n",
-      "2203     | torch.Size([1, 1000, 754]) | 0\n",
-      "2204     | torch.Size([1, 1000, 754]) | 0\n",
-      "2205     | torch.Size([1, 1000, 754]) | 0\n",
-      "2206     | torch.Size([1, 1000, 611]) | 0\n",
-      "2207     | torch.Size([1, 1000, 754]) | 0\n",
-      "2208     | torch.Size([1, 1000, 794]) | 0\n",
-      "2209     | torch.Size([1, 1000, 611]) | 0\n",
-      "2210     | torch.Size([1, 1000, 762]) | 0\n",
-      "2211     | torch.Size([1, 1000, 808]) | 0\n",
-      "2212     | torch.Size([1, 1000, 787]) | 0\n",
-      "2213     | torch.Size([1, 1000, 762]) | 0\n",
-      "2214     | torch.Size([1, 1000, 771]) | 0\n",
-      "2215     | torch.Size([1, 1000, 771]) | 0\n",
-      "2216     | torch.Size([1, 1000, 754]) | 0\n",
-      "2217     | torch.Size([1, 1000, 788]) | 0\n",
-      "2218     | torch.Size([1, 1000, 842]) | 0\n",
-      "2219     | torch.Size([1, 1000, 757]) | 0\n",
-      "2220     | torch.Size([1, 1000, 810]) | 0\n",
-      "2221     | torch.Size([1, 1000, 774]) | 0\n",
-      "2222     | torch.Size([1, 1000, 786]) | 0\n",
-      "2223     | torch.Size([1, 1000, 611]) | 0\n",
-      "2224     | torch.Size([1, 1000, 611]) | 0\n",
-      "2225     | torch.Size([1, 1000, 754]) | 0\n",
-      "2226     | torch.Size([1, 1000, 788]) | 0\n",
-      "2227     | torch.Size([1, 1000, 785]) | 0\n",
-      "2228     | torch.Size([1, 1000, 784]) | 0\n",
-      "2229     | torch.Size([1, 1000, 754]) | 0\n",
-      "2230     | torch.Size([1, 1000, 841]) | 0\n",
-      "2231     | torch.Size([1, 1000, 754]) | 0\n",
-      "2232     | torch.Size([1, 1000, 754]) | 0\n",
-      "2233     | torch.Size([1, 1000, 762]) | 0\n",
-      "2234     | torch.Size([1, 1000, 762]) | 0\n",
-      "2235     | torch.Size([1, 1000, 783]) | 0\n",
-      "2236     | torch.Size([1, 1000, 751]) | 0\n",
-      "2237     | torch.Size([1, 1000, 780]) | 0\n",
-      "2238     | torch.Size([1, 1000, 851]) | 0\n",
-      "2239     | torch.Size([1, 1000, 757]) | 0\n",
-      "2240     | torch.Size([1, 1000, 762]) | 0\n",
-      "2241     | torch.Size([1, 1000, 754]) | 0\n",
-      "2242     | torch.Size([1, 1000, 754]) | 0\n",
-      "2243     | torch.Size([1, 1000, 754]) | 0\n",
-      "2244     | torch.Size([1, 1000, 611]) | 0\n",
-      "2245     | torch.Size([1, 1000, 754]) | 0\n",
-      "2246     | torch.Size([1, 1000, 787]) | 0\n",
-      "2247     | torch.Size([1, 1000, 781]) | 0\n",
-      "2248     | torch.Size([1, 1000, 777]) | 0\n",
-      "2249     | torch.Size([1, 1000, 778]) | 0\n",
-      "2250     | torch.Size([1, 1000, 754]) | 0\n",
-      "2251     | torch.Size([1, 1000, 754]) | 0\n",
-      "2252     | torch.Size([1, 1000, 771]) | 0\n",
-      "2253     | torch.Size([1, 1000, 1015]) | 0\n",
-      "2254     | torch.Size([1, 1000, 781]) | 0\n",
-      "2255     | torch.Size([1, 1000, 754]) | 0\n",
-      "2256     | torch.Size([1, 1000, 768]) | 0\n",
-      "2257     | torch.Size([1, 1000, 779]) | 0\n",
-      "2258     | torch.Size([1, 1000, 754]) | 0\n",
-      "2259     | torch.Size([1, 1000, 754]) | 0\n",
-      "2260     | torch.Size([1, 1000, 754]) | 0\n",
-      "2261     | torch.Size([1, 1000, 762]) | 0\n",
-      "2262     | torch.Size([1, 1000, 754]) | 0\n",
-      "2263     | torch.Size([1, 1000, 754]) | 0\n",
-      "2264     | torch.Size([1, 1000, 762]) | 0\n",
-      "2265     | torch.Size([1, 1000, 754]) | 0\n",
-      "2266     | torch.Size([1, 1000, 754]) | 0\n",
-      "2267     | torch.Size([1, 1000, 762]) | 0\n",
-      "2268     | torch.Size([1, 1000, 773]) | 0\n",
-      "2269     | torch.Size([1, 1000, 757]) | 0\n",
-      "2270     | torch.Size([1, 1000, 754]) | 0\n",
-      "2271     | torch.Size([1, 1000, 787]) | 0\n",
-      "2272     | torch.Size([1, 1000, 794]) | 0\n",
-      "2273     | torch.Size([1, 1000, 754]) | 0\n",
-      "2274     | torch.Size([1, 1000, 774]) | 0\n",
-      "2275     | torch.Size([1, 1000, 777]) | 0\n",
-      "2276     | torch.Size([1, 1000, 762]) | 0\n",
-      "2277     | torch.Size([1, 1000, 754]) | 0\n",
-      "2278     | torch.Size([1, 1000, 754]) | 0\n",
-      "2279     | torch.Size([1, 1000, 762]) | 0\n",
-      "2280     | torch.Size([1, 1000, 762]) | 0\n",
-      "2281     | torch.Size([1, 1000, 754]) | 0\n",
-      "2282     | torch.Size([1, 1000, 754]) | 0\n",
-      "2283     | torch.Size([1, 1000, 784]) | 0\n",
-      "2284     | torch.Size([1, 1000, 759]) | 0\n",
-      "2285     | torch.Size([1, 1000, 804]) | 0\n",
-      "2286     | torch.Size([1, 1000, 754]) | 0\n",
-      "2287     | torch.Size([1, 1000, 754]) | 0\n",
-      "2288     | torch.Size([1, 1000, 754]) | 0\n",
-      "2289     | torch.Size([1, 1000, 754]) | 0\n",
-      "2290     | torch.Size([1, 1000, 772]) | 0\n",
-      "2291     | torch.Size([1, 1000, 776]) | 0\n",
-      "2292     | torch.Size([1, 1000, 776]) | 0\n",
-      "2293     | torch.Size([1, 1000, 757]) | 0\n",
-      "2294     | torch.Size([1, 1000, 771]) | 0\n",
-      "2295     | torch.Size([1, 1000, 754]) | 0\n",
-      "2296     | torch.Size([1, 1000, 763]) | 0\n",
-      "2297     | torch.Size([1, 1000, 787]) | 0\n",
-      "2298     | torch.Size([1, 1000, 791]) | 0\n",
-      "2299     | torch.Size([1, 1000, 762]) | 0\n",
-      "2300     | torch.Size([1, 1000, 782]) | 0\n",
-      "2301     | torch.Size([1, 1000, 754]) | 0\n",
-      "2302     | torch.Size([1, 1000, 754]) | 0\n",
-      "2303     | torch.Size([1, 1000, 754]) | 0\n",
-      "2304     | torch.Size([1, 1000, 754]) | 0\n",
-      "2305     | torch.Size([1, 1000, 783]) | 0\n",
-      "2306     | torch.Size([1, 1000, 754]) | 0\n",
-      "2307     | torch.Size([1, 1000, 787]) | 0\n",
-      "2308     | torch.Size([1, 1000, 762]) | 0\n",
-      "2309     | torch.Size([1, 1000, 754]) | 0\n",
-      "2310     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2311     | torch.Size([1, 1000, 787]) | 0\n",
-      "2312     | torch.Size([1, 1000, 762]) | 0\n",
-      "2313     | torch.Size([1, 1000, 754]) | 0\n",
-      "2314     | torch.Size([1, 1000, 785]) | 0\n",
-      "2315     | torch.Size([1, 1000, 787]) | 0\n",
-      "2316     | torch.Size([1, 1000, 754]) | 0\n",
-      "2317     | torch.Size([1, 1000, 777]) | 0\n",
-      "2318     | torch.Size([1, 1000, 754]) | 0\n",
-      "2319     | torch.Size([1, 1000, 754]) | 0\n",
-      "2320     | torch.Size([1, 1000, 754]) | 0\n",
-      "2321     | torch.Size([1, 1000, 739]) | 0\n",
-      "2322     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2323     | torch.Size([1, 1000, 781]) | 0\n",
-      "2324     | torch.Size([1, 1000, 812]) | 0\n",
-      "2325     | torch.Size([1, 1000, 791]) | 0\n",
-      "2326     | torch.Size([1, 1000, 762]) | 0\n",
-      "2327     | torch.Size([1, 1000, 754]) | 0\n",
-      "2328     | torch.Size([1, 1000, 779]) | 0\n",
-      "2329     | torch.Size([1, 1000, 1015]) | 0\n",
-      "2330     | torch.Size([1, 1000, 754]) | 0\n",
-      "2331     | torch.Size([1, 1000, 757]) | 0\n",
-      "2332     | torch.Size([1, 1000, 754]) | 0\n",
-      "2333     | torch.Size([1, 1000, 765]) | 0\n",
-      "2334     | torch.Size([1, 1000, 757]) | 0\n",
-      "2335     | torch.Size([1, 1000, 762]) | 0\n",
-      "2336     | torch.Size([1, 1000, 754]) | 0\n",
-      "2337     | torch.Size([1, 1000, 777]) | 0\n",
-      "2338     | torch.Size([1, 1000, 751]) | 0\n",
-      "2339     | torch.Size([1, 1000, 757]) | 0\n",
-      "2340     | torch.Size([1, 1000, 780]) | 0\n",
-      "2341     | torch.Size([1, 1000, 785]) | 0\n",
-      "2342     | torch.Size([1, 1000, 754]) | 0\n",
-      "2343     | torch.Size([1, 1000, 779]) | 0\n",
-      "2344     | torch.Size([1, 1000, 754]) | 0\n",
-      "2345     | torch.Size([1, 1000, 782]) | 0\n",
-      "2346     | torch.Size([1, 1000, 754]) | 0\n",
-      "2347     | torch.Size([1, 1000, 754]) | 0\n",
-      "2348     | torch.Size([1, 1000, 850]) | 0\n",
-      "2349     | torch.Size([1, 1000, 754]) | 0\n",
-      "2350     | torch.Size([1, 1000, 754]) | 0\n",
-      "2351     | torch.Size([1, 1000, 762]) | 0\n",
-      "2352     | torch.Size([1, 1000, 1473]) | 0\n",
-      "2353     | torch.Size([1, 1000, 754]) | 0\n",
-      "2354     | torch.Size([1, 1000, 754]) | 0\n",
-      "2355     | torch.Size([1, 1000, 777]) | 0\n",
-      "2356     | torch.Size([1, 1000, 762]) | 0\n",
-      "2357     | torch.Size([1, 1000, 754]) | 0\n",
-      "2358     | torch.Size([1, 1000, 784]) | 0\n",
-      "2359     | torch.Size([1, 1000, 757]) | 0\n",
-      "2360     | torch.Size([1, 1000, 762]) | 0\n",
-      "2361     | torch.Size([1, 1000, 754]) | 0\n",
-      "2362     | torch.Size([1, 1000, 779]) | 0\n",
-      "2363     | torch.Size([1, 1000, 754]) | 0\n",
-      "2364     | torch.Size([1, 1000, 762]) | 0\n",
-      "2365     | torch.Size([1, 1000, 765]) | 0\n",
-      "2366     | torch.Size([1, 1000, 754]) | 0\n",
-      "2367     | torch.Size([1, 1000, 754]) | 0\n",
-      "2368     | torch.Size([1, 1000, 754]) | 0\n",
-      "2369     | torch.Size([1, 1000, 754]) | 0\n",
-      "2370     | torch.Size([1, 1000, 762]) | 0\n",
-      "2371     | torch.Size([1, 1000, 779]) | 0\n",
-      "2372     | torch.Size([1, 1000, 754]) | 0\n",
-      "2373     | torch.Size([1, 1000, 780]) | 0\n",
-      "2374     | torch.Size([1, 1000, 788]) | 0\n",
-      "2375     | torch.Size([1, 1000, 754]) | 0\n",
-      "2376     | torch.Size([1, 1000, 754]) | 0\n",
-      "2377     | torch.Size([1, 1000, 783]) | 0\n",
-      "2378     | torch.Size([1, 1000, 754]) | 0\n",
-      "2379     | torch.Size([1, 1000, 754]) | 0\n",
-      "2380     | torch.Size([1, 1000, 754]) | 0\n",
-      "2381     | torch.Size([1, 1000, 611]) | 0\n",
-      "2382     | torch.Size([1, 1000, 754]) | 0\n",
-      "2383     | torch.Size([1, 1000, 754]) | 0\n",
-      "2384     | torch.Size([1, 1000, 762]) | 0\n",
-      "2385     | torch.Size([1, 1000, 754]) | 0\n",
-      "2386     | torch.Size([1, 1000, 754]) | 0\n",
-      "2387     | torch.Size([1, 1000, 754]) | 0\n",
-      "2388     | torch.Size([1, 1000, 757]) | 0\n",
-      "2389     | torch.Size([1, 1000, 762]) | 0\n",
-      "2390     | torch.Size([1, 1000, 811]) | 0\n",
-      "2391     | torch.Size([1, 1000, 741]) | 0\n",
-      "2392     | torch.Size([1, 1000, 770]) | 0\n",
-      "2393     | torch.Size([1, 1000, 754]) | 0\n",
-      "2394     | torch.Size([1, 1000, 754]) | 0\n",
-      "2395     | torch.Size([1, 1000, 785]) | 0\n",
-      "2396     | torch.Size([1, 1000, 802]) | 0\n",
-      "2397     | torch.Size([1, 1000, 754]) | 0\n",
-      "2398     | torch.Size([1, 1000, 781]) | 0\n",
-      "2399     | torch.Size([1, 1000, 777]) | 0\n",
-      "2400     | torch.Size([1, 1000, 786]) | 0\n",
-      "2401     | torch.Size([1, 1000, 754]) | 0\n",
-      "2402     | torch.Size([1, 1000, 802]) | 0\n",
-      "2403     | torch.Size([1, 1000, 754]) | 0\n",
-      "2404     | torch.Size([1, 1000, 754]) | 0\n",
-      "2405     | torch.Size([1, 1000, 754]) | 0\n",
-      "2406     | torch.Size([1, 1000, 754]) | 0\n",
-      "2407     | torch.Size([1, 1000, 777]) | 0\n",
-      "2408     | torch.Size([1, 1000, 754]) | 0\n",
-      "2409     | torch.Size([1, 1000, 762]) | 0\n",
-      "2410     | torch.Size([1, 1000, 778]) | 0\n",
-      "2411     | torch.Size([1, 1000, 754]) | 0\n",
-      "2412     | torch.Size([1, 1000, 754]) | 0\n",
-      "2413     | torch.Size([1, 1000, 757]) | 0\n",
-      "2414     | torch.Size([1, 1000, 754]) | 0\n",
-      "2415     | torch.Size([1, 1000, 754]) | 0\n",
-      "2416     | torch.Size([1, 1000, 771]) | 0\n",
-      "2417     | torch.Size([1, 1000, 790]) | 0\n",
-      "2418     | torch.Size([1, 1000, 757]) | 0\n",
-      "2419     | torch.Size([1, 1000, 771]) | 0\n",
-      "2420     | torch.Size([1, 1000, 757]) | 0\n",
-      "2421     | torch.Size([1, 1000, 754]) | 0\n",
-      "2422     | torch.Size([1, 1000, 757]) | 0\n",
-      "2423     | torch.Size([1, 1000, 779]) | 0\n",
-      "2424     | torch.Size([1, 1000, 762]) | 0\n",
-      "2425     | torch.Size([1, 1000, 754]) | 0\n",
-      "2426     | torch.Size([1, 1000, 754]) | 0\n",
-      "2427     | torch.Size([1, 1000, 771]) | 0\n",
-      "2428     | torch.Size([1, 1000, 762]) | 0\n",
-      "2429     | torch.Size([1, 1000, 845]) | 0\n",
-      "2430     | torch.Size([1, 1000, 754]) | 0\n",
-      "2431     | torch.Size([1, 1000, 754]) | 0\n",
-      "2432     | torch.Size([1, 1000, 754]) | 0\n",
-      "2433     | torch.Size([1, 1000, 754]) | 0\n",
-      "2434     | torch.Size([1, 1000, 827]) | 0\n",
-      "2435     | torch.Size([1, 1000, 783]) | 0\n",
-      "2436     | torch.Size([1, 1000, 771]) | 0\n",
-      "2437     | torch.Size([1, 1000, 754]) | 0\n",
-      "2438     | torch.Size([1, 1000, 775]) | 0\n",
-      "2439     | torch.Size([1, 1000, 754]) | 0\n",
-      "2440     | torch.Size([1, 1000, 754]) | 0\n",
-      "2441     | torch.Size([1, 1000, 774]) | 0\n",
-      "2442     | torch.Size([1, 1000, 783]) | 0\n",
-      "2443     | torch.Size([1, 1000, 754]) | 0\n",
-      "2444     | torch.Size([1, 1000, 762]) | 0\n",
-      "2445     | torch.Size([1, 1000, 754]) | 0\n",
-      "2446     | torch.Size([1, 1000, 785]) | 0\n",
-      "2447     | torch.Size([1, 1000, 757]) | 0\n",
-      "2448     | torch.Size([1, 1000, 777]) | 0\n",
-      "2449     | torch.Size([1, 1000, 754]) | 0\n",
-      "2450     | torch.Size([1, 1000, 811]) | 0\n",
-      "2451     | torch.Size([1, 1000, 762]) | 0\n",
-      "2452     | torch.Size([1, 1000, 807]) | 0\n",
-      "2453     | torch.Size([1, 1000, 802]) | 0\n",
-      "2454     | torch.Size([1, 1000, 754]) | 0\n",
-      "2455     | torch.Size([1, 1000, 777]) | 0\n",
-      "2456     | torch.Size([1, 1000, 762]) | 0\n",
-      "2457     | torch.Size([1, 1000, 754]) | 0\n",
-      "2458     | torch.Size([1, 1000, 762]) | 0\n",
-      "2459     | torch.Size([1, 1000, 754]) | 0\n",
-      "2460     | torch.Size([1, 1000, 754]) | 0\n",
-      "2461     | torch.Size([1, 1000, 754]) | 0\n",
-      "2462     | torch.Size([1, 1000, 811]) | 0\n",
-      "2463     | torch.Size([1, 1000, 762]) | 0\n",
-      "2464     | torch.Size([1, 1000, 777]) | 0\n",
-      "2465     | torch.Size([1, 1000, 754]) | 0\n",
-      "2466     | torch.Size([1, 1000, 754]) | 0\n",
-      "2467     | torch.Size([1, 1000, 792]) | 0\n",
-      "2468     | torch.Size([1, 1000, 762]) | 0\n",
-      "2469     | torch.Size([1, 1000, 802]) | 0\n",
-      "2470     | torch.Size([1, 1000, 754]) | 0\n",
-      "2471     | torch.Size([1, 1000, 754]) | 0\n",
-      "2472     | torch.Size([1, 1000, 754]) | 0\n",
-      "2473     | torch.Size([1, 1000, 762]) | 0\n",
-      "2474     | torch.Size([1, 1000, 754]) | 0\n",
-      "2475     | torch.Size([1, 1000, 801]) | 0\n",
-      "2476     | torch.Size([1, 1000, 782]) | 0\n",
-      "2477     | torch.Size([1, 1000, 754]) | 0\n",
-      "2478     | torch.Size([1, 1000, 754]) | 0\n",
-      "2479     | torch.Size([1, 1000, 611]) | 0\n",
-      "2480     | torch.Size([1, 1000, 754]) | 0\n",
-      "2481     | torch.Size([1, 1000, 754]) | 0\n",
-      "2482     | torch.Size([1, 1000, 785]) | 0\n",
-      "2483     | torch.Size([1, 1000, 783]) | 0\n",
-      "2484     | torch.Size([1, 1000, 754]) | 0\n",
-      "2485     | torch.Size([1, 1000, 754]) | 0\n",
-      "2486     | torch.Size([1, 1000, 757]) | 0\n",
-      "2487     | torch.Size([1, 1000, 777]) | 0\n",
-      "2488     | torch.Size([1, 1000, 771]) | 0\n",
-      "2489     | torch.Size([1, 1000, 754]) | 0\n",
-      "2490     | torch.Size([1, 1000, 772]) | 0\n",
-      "2491     | torch.Size([1, 1000, 773]) | 0\n",
-      "2492     | torch.Size([1, 1000, 754]) | 0\n",
-      "2493     | torch.Size([1, 1000, 754]) | 0\n",
-      "2494     | torch.Size([1, 1000, 780]) | 0\n",
-      "2495     | torch.Size([1, 1000, 762]) | 0\n",
-      "2496     | torch.Size([1, 1000, 754]) | 0\n",
-      "2497     | torch.Size([1, 1000, 785]) | 0\n",
-      "2498     | torch.Size([1, 1000, 757]) | 0\n",
-      "2499     | torch.Size([1, 1000, 762]) | 0\n",
-      "2500     | torch.Size([1, 1000, 754]) | 0\n",
-      "2501     | torch.Size([1, 1000, 762]) | 0\n",
-      "2502     | torch.Size([1, 1000, 754]) | 0\n",
-      "2503     | torch.Size([1, 1000, 754]) | 0\n",
-      "2504     | torch.Size([1, 1000, 754]) | 0\n",
-      "2505     | torch.Size([1, 1000, 757]) | 0\n",
-      "2506     | torch.Size([1, 1000, 772]) | 0\n",
-      "2507     | torch.Size([1, 1000, 779]) | 0\n",
-      "2508     | torch.Size([1, 1000, 769]) | 0\n",
-      "2509     | torch.Size([1, 1000, 784]) | 0\n",
-      "2510     | torch.Size([1, 1000, 777]) | 0\n",
-      "2511     | torch.Size([1, 1000, 757]) | 0\n",
-      "2512     | torch.Size([1, 1000, 754]) | 0\n",
-      "2513     | torch.Size([1, 1000, 754]) | 0\n",
-      "2514     | torch.Size([1, 1000, 810]) | 0\n",
-      "2515     | torch.Size([1, 1000, 806]) | 1\n",
-      "2516     | torch.Size([1, 1000, 627]) | 1\n",
-      "2517     | torch.Size([1, 1000, 778]) | 1\n",
-      "2518     | torch.Size([1, 1000, 754]) | 1\n",
-      "2519     | torch.Size([1, 1000, 754]) | 1\n",
-      "2520     | torch.Size([1, 1000, 754]) | 1\n",
-      "2521     | torch.Size([1, 1000, 768]) | 1\n",
-      "2522     | torch.Size([1, 1000, 781]) | 1\n",
-      "2523     | torch.Size([1, 1000, 754]) | 1\n",
-      "2524     | torch.Size([1, 1000, 809]) | 1\n",
-      "2525     | torch.Size([1, 1000, 777]) | 1\n",
-      "2526     | torch.Size([1, 1000, 754]) | 1\n",
-      "2527     | torch.Size([1, 1000, 754]) | 1\n",
-      "2528     | torch.Size([1, 1000, 754]) | 1\n",
-      "2529     | torch.Size([1, 1000, 777]) | 1\n",
-      "2530     | torch.Size([1, 1000, 754]) | 1\n",
-      "2531     | torch.Size([1, 1000, 777]) | 1\n",
-      "2532     | torch.Size([1, 1000, 779]) | 1\n",
-      "2533     | torch.Size([1, 1000, 762]) | 1\n",
-      "2534     | torch.Size([1, 1000, 777]) | 1\n",
-      "2535     | torch.Size([1, 1000, 754]) | 1\n",
-      "2536     | torch.Size([1, 1000, 777]) | 1\n",
-      "2537     | torch.Size([1, 1000, 788]) | 1\n",
-      "2538     | torch.Size([1, 1000, 754]) | 1\n",
-      "2539     | torch.Size([1, 1000, 776]) | 1\n",
-      "2540     | torch.Size([1, 1000, 754]) | 1\n",
-      "2541     | torch.Size([1, 1000, 776]) | 1\n",
-      "2542     | torch.Size([1, 1000, 754]) | 1\n",
-      "2543     | torch.Size([1, 1000, 781]) | 1\n",
-      "2544     | torch.Size([1, 1000, 762]) | 1\n",
-      "2545     | torch.Size([1, 1000, 754]) | 1\n",
-      "2546     | torch.Size([1, 1000, 782]) | 1\n",
-      "2547     | torch.Size([1, 1000, 754]) | 1\n",
-      "2548     | torch.Size([1, 1000, 754]) | 1\n",
-      "2549     | torch.Size([1, 1000, 754]) | 1\n",
-      "2550     | torch.Size([1, 1000, 762]) | 1\n",
-      "2551     | torch.Size([1, 1000, 802]) | 1\n",
-      "2552     | torch.Size([1, 1000, 784]) | 1\n",
-      "2553     | torch.Size([1, 1000, 754]) | 1\n",
-      "2554     | torch.Size([1, 1000, 754]) | 1\n",
-      "2555     | torch.Size([1, 1000, 775]) | 1\n",
-      "2556     | torch.Size([1, 1000, 776]) | 1\n",
-      "2557     | torch.Size([1, 1000, 754]) | 1\n",
-      "2558     | torch.Size([1, 1000, 791]) | 1\n",
-      "2559     | torch.Size([1, 1000, 849]) | 1\n",
-      "2560     | torch.Size([1, 1000, 754]) | 1\n",
-      "2561     | torch.Size([1, 1000, 842]) | 1\n",
-      "2562     | torch.Size([1, 1000, 787]) | 1\n",
-      "2563     | torch.Size([1, 1000, 785]) | 1\n",
-      "2564     | torch.Size([1, 1000, 770]) | 1\n",
-      "2565     | torch.Size([1, 1000, 762]) | 1\n",
-      "2566     | torch.Size([1, 1000, 754]) | 1\n",
-      "2567     | torch.Size([1, 1000, 771]) | 1\n",
-      "2568     | torch.Size([1, 1000, 754]) | 1\n",
-      "2569     | torch.Size([1, 1000, 772]) | 1\n",
-      "2570     | torch.Size([1, 1000, 786]) | 1\n",
-      "2571     | torch.Size([1, 1000, 754]) | 1\n",
-      "2572     | torch.Size([1, 1000, 777]) | 1\n",
-      "2573     | torch.Size([1, 1000, 781]) | 1\n",
-      "2574     | torch.Size([1, 1000, 786]) | 1\n",
-      "2575     | torch.Size([1, 1000, 754]) | 1\n",
-      "2576     | torch.Size([1, 1000, 754]) | 1\n",
-      "2577     | torch.Size([1, 1000, 824]) | 1\n",
-      "2578     | torch.Size([1, 1000, 754]) | 1\n",
-      "2579     | torch.Size([1, 1000, 777]) | 1\n",
-      "2580     | torch.Size([1, 1000, 777]) | 1\n",
-      "2581     | torch.Size([1, 1000, 784]) | 1\n",
-      "2582     | torch.Size([1, 1000, 754]) | 1\n",
-      "2583     | torch.Size([1, 1000, 783]) | 1\n",
-      "2584     | torch.Size([1, 1000, 846]) | 1\n",
-      "2585     | torch.Size([1, 1000, 762]) | 1\n",
-      "2586     | torch.Size([1, 1000, 762]) | 1\n",
-      "2587     | torch.Size([1, 1000, 779]) | 1\n",
-      "2588     | torch.Size([1, 1000, 754]) | 1\n",
-      "2589     | torch.Size([1, 1000, 754]) | 1\n",
-      "2590     | torch.Size([1, 1000, 754]) | 1\n",
-      "2591     | torch.Size([1, 1000, 777]) | 1\n",
-      "2592     | torch.Size([1, 1000, 754]) | 1\n",
-      "2593     | torch.Size([1, 1000, 754]) | 1\n",
-      "2594     | torch.Size([1, 1000, 781]) | 1\n",
-      "2595     | torch.Size([1, 1000, 754]) | 1\n",
-      "2596     | torch.Size([1, 1000, 786]) | 1\n",
-      "2597     | torch.Size([1, 1000, 777]) | 1\n",
-      "2598     | torch.Size([1, 1000, 754]) | 1\n",
-      "2599     | torch.Size([1, 1000, 754]) | 1\n",
-      "2600     | torch.Size([1, 1000, 781]) | 1\n",
-      "2601     | torch.Size([1, 1000, 754]) | 1\n",
-      "2602     | torch.Size([1, 1000, 784]) | 1\n",
-      "2603     | torch.Size([1, 1000, 754]) | 1\n",
-      "2604     | torch.Size([1, 1000, 754]) | 1\n",
-      "2605     | torch.Size([1, 1000, 848]) | 1\n",
-      "2606     | torch.Size([1, 1000, 754]) | 1\n",
-      "2607     | torch.Size([1, 1000, 803]) | 1\n",
-      "2608     | torch.Size([1, 1000, 754]) | 1\n",
-      "2609     | torch.Size([1, 1000, 755]) | 1\n",
-      "2610     | torch.Size([1, 1000, 780]) | 1\n",
-      "2611     | torch.Size([1, 1000, 754]) | 1\n",
-      "2612     | torch.Size([1, 1000, 762]) | 1\n",
-      "2613     | torch.Size([1, 1000, 802]) | 1\n",
-      "2614     | torch.Size([1, 1000, 786]) | 1\n",
-      "2615     | torch.Size([1, 1000, 845]) | 1\n",
-      "2616     | torch.Size([1, 1000, 762]) | 1\n",
-      "2617     | torch.Size([1, 1000, 754]) | 1\n",
-      "2618     | torch.Size([1, 1000, 792]) | 1\n",
-      "2619     | torch.Size([1, 1000, 754]) | 1\n",
-      "2620     | torch.Size([1, 1000, 762]) | 1\n",
-      "2621     | torch.Size([1, 1000, 779]) | 1\n",
-      "2622     | torch.Size([1, 1000, 849]) | 1\n",
-      "2623     | torch.Size([1, 1000, 795]) | 1\n",
-      "2624     | torch.Size([1, 1000, 768]) | 1\n",
-      "2625     | torch.Size([1, 1000, 754]) | 1\n",
-      "2626     | torch.Size([1, 1000, 786]) | 1\n",
-      "2627     | torch.Size([1, 1000, 793]) | 1\n",
-      "2628     | torch.Size([1, 1000, 754]) | 1\n",
-      "2629     | torch.Size([1, 1000, 754]) | 1\n",
-      "2630     | torch.Size([1, 1000, 754]) | 1\n",
-      "2631     | torch.Size([1, 1000, 793]) | 1\n",
-      "2632     | torch.Size([1, 1000, 786]) | 1\n",
-      "2633     | torch.Size([1, 1000, 766]) | 1\n",
-      "2634     | torch.Size([1, 1000, 754]) | 1\n",
-      "2635     | torch.Size([1, 1000, 785]) | 1\n",
-      "2636     | torch.Size([1, 1000, 762]) | 1\n",
-      "2637     | torch.Size([1, 1000, 754]) | 1\n",
-      "2638     | torch.Size([1, 1000, 773]) | 1\n",
-      "2639     | torch.Size([1, 1000, 787]) | 1\n",
-      "2640     | torch.Size([1, 1000, 754]) | 1\n",
-      "2641     | torch.Size([1, 1000, 754]) | 1\n",
-      "2642     | torch.Size([1, 1000, 799]) | 1\n",
-      "2643     | torch.Size([1, 1000, 806]) | 1\n",
-      "2644     | torch.Size([1, 1000, 777]) | 1\n",
-      "2645     | torch.Size([1, 1000, 777]) | 1\n",
-      "2646     | torch.Size([1, 1000, 754]) | 1\n",
-      "2647     | torch.Size([1, 1000, 777]) | 1\n",
-      "2648     | torch.Size([1, 1000, 754]) | 1\n",
-      "2649     | torch.Size([1, 1000, 762]) | 1\n",
-      "2650     | torch.Size([1, 1000, 786]) | 1\n",
-      "2651     | torch.Size([1, 1000, 754]) | 1\n",
-      "2652     | torch.Size([1, 1000, 762]) | 1\n",
-      "2653     | torch.Size([1, 1000, 777]) | 1\n",
-      "2654     | torch.Size([1, 1000, 754]) | 1\n",
-      "2655     | torch.Size([1, 1000, 783]) | 1\n",
-      "2656     | torch.Size([1, 1000, 762]) | 1\n",
-      "2657     | torch.Size([1, 1000, 754]) | 1\n",
-      "2658     | torch.Size([1, 1000, 754]) | 1\n",
-      "2659     | torch.Size([1, 1000, 782]) | 1\n",
-      "2660     | torch.Size([1, 1000, 754]) | 1\n",
-      "2661     | torch.Size([1, 1000, 754]) | 1\n",
-      "2662     | torch.Size([1, 1000, 754]) | 1\n",
-      "2663     | torch.Size([1, 1000, 762]) | 1\n",
-      "2664     | torch.Size([1, 1000, 754]) | 1\n",
-      "2665     | torch.Size([1, 1000, 754]) | 1\n",
-      "2666     | torch.Size([1, 1000, 777]) | 1\n",
-      "2667     | torch.Size([1, 1000, 777]) | 1\n",
-      "2668     | torch.Size([1, 1000, 786]) | 1\n",
-      "2669     | torch.Size([1, 1000, 778]) | 1\n",
-      "2670     | torch.Size([1, 1000, 754]) | 1\n",
-      "2671     | torch.Size([1, 1000, 782]) | 1\n",
-      "2672     | torch.Size([1, 1000, 776]) | 1\n",
-      "2673     | torch.Size([1, 1000, 808]) | 1\n",
-      "2674     | torch.Size([1, 1000, 808]) | 1\n",
-      "2675     | torch.Size([1, 1000, 787]) | 1\n",
-      "2676     | torch.Size([1, 1000, 754]) | 1\n",
-      "2677     | torch.Size([1, 1000, 754]) | 1\n",
-      "2678     | torch.Size([1, 1000, 754]) | 1\n",
-      "2679     | torch.Size([1, 1000, 754]) | 1\n",
-      "2680     | torch.Size([1, 1000, 754]) | 1\n",
-      "2681     | torch.Size([1, 1000, 754]) | 1\n",
-      "2682     | torch.Size([1, 1000, 777]) | 1\n",
-      "2683     | torch.Size([1, 1000, 777]) | 1\n",
-      "2684     | torch.Size([1, 1000, 777]) | 1\n",
-      "2685     | torch.Size([1, 1000, 788]) | 1\n",
-      "2686     | torch.Size([1, 1000, 782]) | 1\n",
-      "2687     | torch.Size([1, 1000, 789]) | 1\n",
-      "2688     | torch.Size([1, 1000, 786]) | 1\n",
-      "2689     | torch.Size([1, 1000, 754]) | 1\n",
-      "2690     | torch.Size([1, 1000, 754]) | 1\n",
-      "2691     | torch.Size([1, 1000, 777]) | 1\n",
-      "2692     | torch.Size([1, 1000, 754]) | 1\n",
-      "2693     | torch.Size([1, 1000, 783]) | 1\n",
-      "2694     | torch.Size([1, 1000, 754]) | 1\n",
-      "2695     | torch.Size([1, 1000, 779]) | 1\n",
-      "2696     | torch.Size([1, 1000, 790]) | 1\n",
-      "2697     | torch.Size([1, 1000, 839]) | 1\n",
-      "2698     | torch.Size([1, 1000, 774]) | 1\n",
-      "2699     | torch.Size([1, 1000, 754]) | 1\n",
-      "2700     | torch.Size([1, 1000, 754]) | 1\n",
-      "2701     | torch.Size([1, 1000, 754]) | 1\n",
-      "2702     | torch.Size([1, 1000, 763]) | 1\n",
-      "2703     | torch.Size([1, 1000, 777]) | 1\n",
-      "2704     | torch.Size([1, 1000, 844]) | 1\n",
-      "2705     | torch.Size([1, 1000, 762]) | 1\n",
-      "2706     | torch.Size([1, 1000, 784]) | 1\n",
-      "2707     | torch.Size([1, 1000, 771]) | 1\n",
-      "2708     | torch.Size([1, 1000, 754]) | 1\n",
-      "2709     | torch.Size([1, 1000, 777]) | 1\n",
-      "2710     | torch.Size([1, 1000, 783]) | 1\n",
-      "2711     | torch.Size([1, 1000, 811]) | 1\n",
-      "2712     | torch.Size([1, 1000, 777]) | 1\n",
-      "2713     | torch.Size([1, 1000, 754]) | 1\n",
-      "2714     | torch.Size([1, 1000, 799]) | 1\n",
-      "2715     | torch.Size([1, 1000, 802]) | 1\n",
-      "2716     | torch.Size([1, 1000, 754]) | 1\n",
-      "2717     | torch.Size([1, 1000, 783]) | 1\n",
-      "2718     | torch.Size([1, 1000, 754]) | 1\n",
-      "2719     | torch.Size([1, 1000, 789]) | 1\n",
-      "2720     | torch.Size([1, 1000, 754]) | 1\n",
-      "2721     | torch.Size([1, 1000, 754]) | 1\n",
-      "2722     | torch.Size([1, 1000, 807]) | 1\n",
-      "2723     | torch.Size([1, 1000, 754]) | 1\n",
-      "2724     | torch.Size([1, 1000, 791]) | 1\n",
-      "2725     | torch.Size([1, 1000, 779]) | 1\n",
-      "2726     | torch.Size([1, 1000, 754]) | 1\n",
-      "2727     | torch.Size([1, 1000, 778]) | 1\n",
-      "2728     | torch.Size([1, 1000, 870]) | 1\n",
-      "2729     | torch.Size([1, 1000, 767]) | 1\n",
-      "2730     | torch.Size([1, 1000, 754]) | 1\n",
-      "2731     | torch.Size([1, 1000, 754]) | 1\n",
-      "2732     | torch.Size([1, 1000, 771]) | 1\n",
-      "2733     | torch.Size([1, 1000, 780]) | 1\n",
-      "2734     | torch.Size([1, 1000, 754]) | 1\n",
-      "2735     | torch.Size([1, 1000, 782]) | 1\n",
-      "2736     | torch.Size([1, 1000, 771]) | 1\n",
-      "2737     | torch.Size([1, 1000, 780]) | 1\n",
-      "2738     | torch.Size([1, 1000, 754]) | 1\n",
-      "2739     | torch.Size([1, 1000, 803]) | 1\n",
-      "2740     | torch.Size([1, 1000, 791]) | 1\n",
-      "2741     | torch.Size([1, 1000, 777]) | 1\n",
-      "2742     | torch.Size([1, 1000, 777]) | 1\n",
-      "2743     | torch.Size([1, 1000, 777]) | 1\n",
-      "2744     | torch.Size([1, 1000, 827]) | 1\n",
-      "2745     | torch.Size([1, 1000, 754]) | 1\n",
-      "2746     | torch.Size([1, 1000, 772]) | 1\n",
-      "2747     | torch.Size([1, 1000, 754]) | 1\n",
-      "2748     | torch.Size([1, 1000, 754]) | 1\n",
-      "2749     | torch.Size([1, 1000, 754]) | 1\n",
-      "2750     | torch.Size([1, 1000, 754]) | 1\n",
-      "2751     | torch.Size([1, 1000, 754]) | 1\n",
-      "2752     | torch.Size([1, 1000, 780]) | 1\n",
-      "2753     | torch.Size([1, 1000, 783]) | 1\n",
-      "2754     | torch.Size([1, 1000, 843]) | 1\n",
-      "2755     | torch.Size([1, 1000, 789]) | 1\n",
-      "2756     | torch.Size([1, 1000, 762]) | 1\n",
-      "2757     | torch.Size([1, 1000, 788]) | 1\n",
-      "2758     | torch.Size([1, 1000, 808]) | 1\n",
-      "2759     | torch.Size([1, 1000, 867]) | 1\n",
-      "2760     | torch.Size([1, 1000, 777]) | 1\n",
-      "2761     | torch.Size([1, 1000, 754]) | 1\n",
-      "2762     | torch.Size([1, 1000, 754]) | 1\n",
-      "2763     | torch.Size([1, 1000, 762]) | 1\n",
-      "2764     | torch.Size([1, 1000, 754]) | 1\n",
-      "2765     | torch.Size([1, 1000, 769]) | 1\n",
-      "2766     | torch.Size([1, 1000, 772]) | 1\n",
-      "2767     | torch.Size([1, 1000, 754]) | 1\n",
-      "2768     | torch.Size([1, 1000, 754]) | 1\n",
-      "2769     | torch.Size([1, 1000, 806]) | 1\n",
-      "2770     | torch.Size([1, 1000, 772]) | 1\n",
-      "2771     | torch.Size([1, 1000, 777]) | 1\n",
-      "2772     | torch.Size([1, 1000, 754]) | 1\n",
-      "2773     | torch.Size([1, 1000, 786]) | 1\n",
-      "2774     | torch.Size([1, 1000, 754]) | 1\n",
-      "2775     | torch.Size([1, 1000, 781]) | 1\n",
-      "2776     | torch.Size([1, 1000, 754]) | 1\n",
-      "2777     | torch.Size([1, 1000, 802]) | 1\n",
-      "2778     | torch.Size([1, 1000, 753]) | 1\n",
-      "2779     | torch.Size([1, 1000, 754]) | 1\n",
-      "2780     | torch.Size([1, 1000, 754]) | 1\n",
-      "2781     | torch.Size([1, 1000, 777]) | 1\n",
-      "2782     | torch.Size([1, 1000, 777]) | 1\n",
-      "2783     | torch.Size([1, 1000, 754]) | 1\n",
-      "2784     | torch.Size([1, 1000, 808]) | 1\n",
-      "2785     | torch.Size([1, 1000, 754]) | 1\n",
-      "2786     | torch.Size([1, 1000, 783]) | 1\n",
-      "2787     | torch.Size([1, 1000, 764]) | 1\n",
-      "2788     | torch.Size([1, 1000, 784]) | 1\n",
-      "2789     | torch.Size([1, 1000, 754]) | 1\n",
-      "2790     | torch.Size([1, 1000, 754]) | 1\n",
-      "2791     | torch.Size([1, 1000, 754]) | 1\n",
-      "2792     | torch.Size([1, 1000, 754]) | 1\n",
-      "2793     | torch.Size([1, 1000, 777]) | 1\n",
-      "2794     | torch.Size([1, 1000, 762]) | 1\n",
-      "2795     | torch.Size([1, 1000, 771]) | 1\n",
-      "2796     | torch.Size([1, 1000, 777]) | 1\n",
-      "2797     | torch.Size([1, 1000, 754]) | 1\n",
-      "2798     | torch.Size([1, 1000, 754]) | 1\n",
-      "2799     | torch.Size([1, 1000, 777]) | 1\n",
-      "2800     | torch.Size([1, 1000, 754]) | 1\n",
-      "2801     | torch.Size([1, 1000, 754]) | 1\n",
-      "2802     | torch.Size([1, 1000, 784]) | 1\n",
-      "2803     | torch.Size([1, 1000, 754]) | 1\n",
-      "2804     | torch.Size([1, 1000, 808]) | 1\n",
-      "2805     | torch.Size([1, 1000, 770]) | 1\n",
-      "2806     | torch.Size([1, 1000, 777]) | 1\n",
-      "2807     | torch.Size([1, 1000, 777]) | 1\n",
-      "2808     | torch.Size([1, 1000, 754]) | 1\n",
-      "2809     | torch.Size([1, 1000, 786]) | 1\n",
-      "2810     | torch.Size([1, 1000, 808]) | 1\n",
-      "2811     | torch.Size([1, 1000, 777]) | 1\n",
-      "2812     | torch.Size([1, 1000, 798]) | 1\n",
-      "2813     | torch.Size([1, 1000, 771]) | 1\n",
-      "2814     | torch.Size([1, 1000, 754]) | 1\n",
-      "2815     | torch.Size([1, 1000, 776]) | 1\n",
-      "2816     | torch.Size([1, 1000, 843]) | 1\n",
-      "2817     | torch.Size([1, 1000, 776]) | 1\n",
-      "2818     | torch.Size([1, 1000, 754]) | 1\n",
-      "2819     | torch.Size([1, 1000, 754]) | 1\n",
-      "2820     | torch.Size([1, 1000, 784]) | 1\n",
-      "2821     | torch.Size([1, 1000, 754]) | 1\n",
-      "2822     | torch.Size([1, 1000, 797]) | 1\n",
-      "2823     | torch.Size([1, 1000, 782]) | 1\n",
-      "2824     | torch.Size([1, 1000, 754]) | 1\n",
-      "2825     | torch.Size([1, 1000, 783]) | 1\n",
-      "2826     | torch.Size([1, 1000, 754]) | 1\n",
-      "2827     | torch.Size([1, 1000, 783]) | 1\n",
-      "2828     | torch.Size([1, 1000, 754]) | 1\n",
-      "2829     | torch.Size([1, 1000, 777]) | 1\n",
-      "2830     | torch.Size([1, 1000, 777]) | 1\n",
-      "2831     | torch.Size([1, 1000, 754]) | 1\n",
-      "2832     | torch.Size([1, 1000, 618]) | 1\n",
-      "2833     | torch.Size([1, 1000, 809]) | 1\n",
-      "2834     | torch.Size([1, 1000, 788]) | 1\n",
-      "2835     | torch.Size([1, 1000, 802]) | 1\n",
-      "2836     | torch.Size([1, 1000, 769]) | 1\n",
-      "2837     | torch.Size([1, 1000, 786]) | 1\n",
-      "2838     | torch.Size([1, 1000, 777]) | 1\n",
-      "2839     | torch.Size([1, 1000, 752]) | 1\n",
-      "2840     | torch.Size([1, 1000, 762]) | 1\n",
-      "2841     | torch.Size([1, 1000, 781]) | 1\n",
-      "2842     | torch.Size([1, 1000, 777]) | 1\n",
-      "2843     | torch.Size([1, 1000, 811]) | 1\n",
-      "2844     | torch.Size([1, 1000, 797]) | 1\n",
-      "2845     | torch.Size([1, 1000, 754]) | 1\n",
-      "2846     | torch.Size([1, 1000, 754]) | 1\n",
-      "2847     | torch.Size([1, 1000, 754]) | 1\n",
-      "2848     | torch.Size([1, 1000, 780]) | 1\n",
-      "2849     | torch.Size([1, 1000, 778]) | 1\n",
-      "2850     | torch.Size([1, 1000, 754]) | 1\n",
-      "2851     | torch.Size([1, 1000, 777]) | 1\n",
-      "2852     | torch.Size([1, 1000, 754]) | 1\n",
-      "2853     | torch.Size([1, 1000, 788]) | 1\n",
-      "2854     | torch.Size([1, 1000, 777]) | 1\n",
-      "2855     | torch.Size([1, 1000, 754]) | 1\n",
-      "2856     | torch.Size([1, 1000, 775]) | 1\n",
-      "2857     | torch.Size([1, 1000, 754]) | 1\n",
-      "2858     | torch.Size([1, 1000, 762]) | 1\n",
-      "2859     | torch.Size([1, 1000, 754]) | 1\n",
-      "2860     | torch.Size([1, 1000, 773]) | 1\n",
-      "2861     | torch.Size([1, 1000, 754]) | 1\n",
-      "2862     | torch.Size([1, 1000, 754]) | 1\n",
-      "2863     | torch.Size([1, 1000, 754]) | 1\n",
-      "2864     | torch.Size([1, 1000, 777]) | 1\n",
-      "2865     | torch.Size([1, 1000, 754]) | 1\n",
-      "2866     | torch.Size([1, 1000, 771]) | 1\n",
-      "2867     | torch.Size([1, 1000, 754]) | 1\n",
-      "2868     | torch.Size([1, 1000, 754]) | 1\n",
-      "2869     | torch.Size([1, 1000, 799]) | 1\n",
-      "2870     | torch.Size([1, 1000, 754]) | 1\n",
-      "2871     | torch.Size([1, 1000, 786]) | 1\n",
-      "2872     | torch.Size([1, 1000, 778]) | 1\n",
-      "2873     | torch.Size([1, 1000, 781]) | 1\n",
-      "2874     | torch.Size([1, 1000, 777]) | 1\n",
-      "2875     | torch.Size([1, 1000, 754]) | 1\n",
-      "2876     | torch.Size([1, 1000, 754]) | 1\n",
-      "2877     | torch.Size([1, 1000, 754]) | 1\n",
-      "2878     | torch.Size([1, 1000, 768]) | 1\n",
-      "2879     | torch.Size([1, 1000, 807]) | 1\n",
-      "2880     | torch.Size([1, 1000, 754]) | 1\n",
-      "2881     | torch.Size([1, 1000, 791]) | 1\n",
-      "2882     | torch.Size([1, 1000, 777]) | 1\n",
-      "2883     | torch.Size([1, 1000, 775]) | 1\n",
-      "2884     | torch.Size([1, 1000, 754]) | 1\n",
-      "2885     | torch.Size([1, 1000, 805]) | 1\n",
-      "2886     | torch.Size([1, 1000, 783]) | 1\n",
-      "2887     | torch.Size([1, 1000, 762]) | 1\n",
-      "2888     | torch.Size([1, 1000, 773]) | 1\n",
-      "2889     | torch.Size([1, 1000, 754]) | 1\n",
-      "2890     | torch.Size([1, 1000, 754]) | 1\n",
-      "2891     | torch.Size([1, 1000, 754]) | 1\n",
-      "2892     | torch.Size([1, 1000, 777]) | 1\n",
-      "2893     | torch.Size([1, 1000, 767]) | 1\n",
-      "2894     | torch.Size([1, 1000, 776]) | 1\n",
-      "2895     | torch.Size([1, 1000, 754]) | 1\n",
-      "2896     | torch.Size([1, 1000, 766]) | 1\n",
-      "2897     | torch.Size([1, 1000, 754]) | 1\n",
-      "2898     | torch.Size([1, 1000, 814]) | 1\n",
-      "2899     | torch.Size([1, 1000, 762]) | 1\n",
-      "2900     | torch.Size([1, 1000, 774]) | 1\n",
-      "2901     | torch.Size([1, 1000, 777]) | 1\n",
-      "2902     | torch.Size([1, 1000, 767]) | 1\n",
-      "2903     | torch.Size([1, 1000, 754]) | 1\n",
-      "2904     | torch.Size([1, 1000, 754]) | 1\n",
-      "2905     | torch.Size([1, 1000, 754]) | 1\n",
-      "2906     | torch.Size([1, 1000, 754]) | 1\n",
-      "2907     | torch.Size([1, 1000, 754]) | 1\n",
-      "2908     | torch.Size([1, 1000, 724]) | 1\n",
-      "2909     | torch.Size([1, 1000, 777]) | 1\n",
-      "2910     | torch.Size([1, 1000, 776]) | 1\n",
-      "2911     | torch.Size([1, 1000, 777]) | 1\n",
-      "2912     | torch.Size([1, 1000, 793]) | 1\n",
-      "2913     | torch.Size([1, 1000, 777]) | 1\n",
-      "2914     | torch.Size([1, 1000, 778]) | 1\n",
-      "2915     | torch.Size([1, 1000, 777]) | 1\n",
-      "2916     | torch.Size([1, 1000, 754]) | 1\n",
-      "2917     | torch.Size([1, 1000, 794]) | 1\n",
-      "2918     | torch.Size([1, 1000, 771]) | 1\n",
-      "2919     | torch.Size([1, 1000, 754]) | 1\n",
-      "2920     | torch.Size([1, 1000, 754]) | 1\n",
-      "2921     | torch.Size([1, 1000, 777]) | 1\n",
-      "2922     | torch.Size([1, 1000, 787]) | 1\n",
-      "2923     | torch.Size([1, 1000, 754]) | 1\n",
-      "2924     | torch.Size([1, 1000, 754]) | 1\n",
-      "2925     | torch.Size([1, 1000, 781]) | 1\n",
-      "2926     | torch.Size([1, 1000, 754]) | 1\n",
-      "2927     | torch.Size([1, 1000, 766]) | 1\n",
-      "2928     | torch.Size([1, 1000, 785]) | 1\n",
-      "2929     | torch.Size([1, 1000, 747]) | 1\n",
-      "2930     | torch.Size([1, 1000, 754]) | 1\n",
-      "2931     | torch.Size([1, 1000, 765]) | 1\n",
-      "2932     | torch.Size([1, 1000, 777]) | 1\n",
-      "2933     | torch.Size([1, 1000, 754]) | 1\n",
-      "2934     | torch.Size([1, 1000, 754]) | 1\n",
-      "2935     | torch.Size([1, 1000, 754]) | 1\n",
-      "2936     | torch.Size([1, 1000, 754]) | 1\n",
-      "2937     | torch.Size([1, 1000, 766]) | 1\n",
-      "2938     | torch.Size([1, 1000, 754]) | 1\n",
-      "2939     | torch.Size([1, 1000, 777]) | 1\n",
-      "2940     | torch.Size([1, 1000, 784]) | 1\n",
-      "2941     | torch.Size([1, 1000, 777]) | 1\n",
-      "2942     | torch.Size([1, 1000, 777]) | 1\n",
-      "2943     | torch.Size([1, 1000, 817]) | 1\n",
-      "2944     | torch.Size([1, 1000, 754]) | 1\n",
-      "2945     | torch.Size([1, 1000, 765]) | 1\n",
-      "2946     | torch.Size([1, 1000, 754]) | 1\n",
-      "2947     | torch.Size([1, 1000, 754]) | 1\n",
-      "2948     | torch.Size([1, 1000, 754]) | 1\n",
-      "2949     | torch.Size([1, 1000, 754]) | 1\n",
-      "2950     | torch.Size([1, 1000, 783]) | 1\n",
-      "2951     | torch.Size([1, 1000, 772]) | 1\n",
-      "2952     | torch.Size([1, 1000, 762]) | 1\n",
-      "2953     | torch.Size([1, 1000, 796]) | 1\n",
-      "2954     | torch.Size([1, 1000, 754]) | 1\n",
-      "2955     | torch.Size([1, 1000, 762]) | 1\n",
-      "2956     | torch.Size([1, 1000, 799]) | 1\n",
-      "2957     | torch.Size([1, 1000, 827]) | 1\n",
-      "2958     | torch.Size([1, 1000, 760]) | 1\n",
-      "2959     | torch.Size([1, 1000, 777]) | 1\n",
-      "2960     | torch.Size([1, 1000, 754]) | 1\n",
-      "2961     | torch.Size([1, 1000, 779]) | 1\n",
-      "2962     | torch.Size([1, 1000, 754]) | 1\n",
-      "2963     | torch.Size([1, 1000, 777]) | 1\n",
-      "2964     | torch.Size([1, 1000, 787]) | 1\n",
-      "2965     | torch.Size([1, 1000, 781]) | 1\n",
-      "2966     | torch.Size([1, 1000, 797]) | 1\n",
-      "2967     | torch.Size([1, 1000, 754]) | 1\n",
-      "2968     | torch.Size([1, 1000, 754]) | 1\n",
-      "2969     | torch.Size([1, 1000, 777]) | 1\n",
-      "2970     | torch.Size([1, 1000, 754]) | 1\n",
-      "2971     | torch.Size([1, 1000, 754]) | 1\n",
-      "2972     | torch.Size([1, 1000, 754]) | 1\n",
-      "2973     | torch.Size([1, 1000, 754]) | 1\n",
-      "2974     | torch.Size([1, 1000, 786]) | 1\n",
-      "2975     | torch.Size([1, 1000, 754]) | 1\n",
-      "2976     | torch.Size([1, 1000, 754]) | 1\n",
-      "2977     | torch.Size([1, 1000, 786]) | 1\n",
-      "2978     | torch.Size([1, 1000, 767]) | 1\n",
-      "2979     | torch.Size([1, 1000, 807]) | 1\n",
-      "2980     | torch.Size([1, 1000, 777]) | 1\n",
-      "2981     | torch.Size([1, 1000, 777]) | 1\n",
-      "2982     | torch.Size([1, 1000, 754]) | 1\n",
-      "2983     | torch.Size([1, 1000, 777]) | 1\n",
-      "2984     | torch.Size([1, 1000, 762]) | 1\n",
-      "2985     | torch.Size([1, 1000, 754]) | 1\n",
-      "2986     | torch.Size([1, 1000, 773]) | 1\n",
-      "2987     | torch.Size([1, 1000, 848]) | 1\n",
-      "2988     | torch.Size([1, 1000, 779]) | 1\n",
-      "2989     | torch.Size([1, 1000, 772]) | 1\n",
-      "2990     | torch.Size([1, 1000, 754]) | 1\n",
-      "2991     | torch.Size([1, 1000, 808]) | 1\n",
-      "2992     | torch.Size([1, 1000, 754]) | 1\n",
-      "2993     | torch.Size([1, 1000, 762]) | 1\n",
-      "2994     | torch.Size([1, 1000, 773]) | 1\n",
-      "2995     | torch.Size([1, 1000, 745]) | 1\n",
-      "2996     | torch.Size([1, 1000, 787]) | 1\n",
-      "2997     | torch.Size([1, 1000, 778]) | 1\n",
-      "2998     | torch.Size([1, 1000, 767]) | 1\n",
-      "2999     | torch.Size([1, 1000, 754]) | 1\n",
-      "3000     | torch.Size([1, 1000, 777]) | 1\n",
-      "3001     | torch.Size([1, 1000, 779]) | 1\n",
-      "3002     | torch.Size([1, 1000, 772]) | 1\n",
-      "3003     | torch.Size([1, 1000, 802]) | 1\n",
-      "3004     | torch.Size([1, 1000, 772]) | 1\n",
-      "3005     | torch.Size([1, 1000, 770]) | 1\n",
-      "3006     | torch.Size([1, 1000, 762]) | 1\n",
-      "3007     | torch.Size([1, 1000, 754]) | 1\n",
-      "3008     | torch.Size([1, 1000, 754]) | 1\n",
-      "3009     | torch.Size([1, 1000, 777]) | 1\n",
-      "3010     | torch.Size([1, 1000, 785]) | 1\n",
-      "3011     | torch.Size([1, 1000, 778]) | 1\n",
-      "3012     | torch.Size([1, 1000, 774]) | 1\n",
-      "3013     | torch.Size([1, 1000, 754]) | 1\n",
-      "3014     | torch.Size([1, 1000, 746]) | 1\n",
-      "3015     | torch.Size([1, 1000, 777]) | 1\n",
-      "3016     | torch.Size([1, 1000, 780]) | 1\n",
-      "3017     | torch.Size([1, 1000, 754]) | 1\n",
-      "3018     | torch.Size([1, 1000, 754]) | 1\n",
-      "3019     | torch.Size([1, 1000, 788]) | 1\n",
-      "3020     | torch.Size([1, 1000, 782]) | 1\n",
-      "3021     | torch.Size([1, 1000, 754]) | 1\n",
-      "3022     | torch.Size([1, 1000, 779]) | 1\n",
-      "3023     | torch.Size([1, 1000, 754]) | 1\n",
-      "3024     | torch.Size([1, 1000, 754]) | 1\n",
-      "3025     | torch.Size([1, 1000, 777]) | 1\n",
-      "3026     | torch.Size([1, 1000, 754]) | 1\n",
-      "3027     | torch.Size([1, 1000, 797]) | 1\n",
-      "3028     | torch.Size([1, 1000, 774]) | 1\n",
-      "3029     | torch.Size([1, 1000, 754]) | 1\n",
-      "3030     | torch.Size([1, 1000, 754]) | 1\n",
-      "3031     | torch.Size([1, 1000, 754]) | 1\n",
-      "3032     | torch.Size([1, 1000, 754]) | 1\n",
-      "3033     | torch.Size([1, 1000, 783]) | 1\n",
-      "3034     | torch.Size([1, 1000, 779]) | 1\n",
-      "3035     | torch.Size([1, 1000, 754]) | 1\n",
-      "3036     | torch.Size([1, 1000, 754]) | 1\n",
-      "3037     | torch.Size([1, 1000, 754]) | 1\n",
-      "3038     | torch.Size([1, 1000, 754]) | 1\n",
-      "3039     | torch.Size([1, 1000, 754]) | 1\n",
-      "3040     | torch.Size([1, 1000, 802]) | 1\n",
-      "3041     | torch.Size([1, 1000, 624]) | 1\n",
-      "3042     | torch.Size([1, 1000, 766]) | 1\n",
-      "3043     | torch.Size([1, 1000, 790]) | 1\n",
-      "3044     | torch.Size([1, 1000, 753]) | 1\n",
-      "3045     | torch.Size([1, 1000, 754]) | 1\n",
-      "3046     | torch.Size([1, 1000, 762]) | 1\n",
-      "3047     | torch.Size([1, 1000, 754]) | 1\n",
-      "3048     | torch.Size([1, 1000, 762]) | 1\n",
-      "3049     | torch.Size([1, 1000, 772]) | 1\n",
-      "3050     | torch.Size([1, 1000, 754]) | 1\n",
-      "3051     | torch.Size([1, 1000, 777]) | 1\n",
-      "3052     | torch.Size([1, 1000, 848]) | 1\n",
-      "3053     | torch.Size([1, 1000, 802]) | 1\n",
-      "3054     | torch.Size([1, 1000, 754]) | 1\n",
-      "3055     | torch.Size([1, 1000, 754]) | 1\n",
-      "3056     | torch.Size([1, 1000, 777]) | 1\n",
-      "3057     | torch.Size([1, 1000, 754]) | 1\n",
-      "3058     | torch.Size([1, 1000, 797]) | 1\n",
-      "3059     | torch.Size([1, 1000, 783]) | 1\n",
-      "3060     | torch.Size([1, 1000, 754]) | 1\n",
-      "3061     | torch.Size([1, 1000, 773]) | 1\n",
-      "3062     | torch.Size([1, 1000, 777]) | 1\n",
-      "3063     | torch.Size([1, 1000, 769]) | 1\n",
-      "3064     | torch.Size([1, 1000, 762]) | 1\n",
-      "3065     | torch.Size([1, 1000, 754]) | 1\n",
-      "3066     | torch.Size([1, 1000, 777]) | 1\n",
-      "3067     | torch.Size([1, 1000, 773]) | 1\n",
-      "3068     | torch.Size([1, 1000, 778]) | 1\n",
-      "3069     | torch.Size([1, 1000, 773]) | 1\n",
-      "3070     | torch.Size([1, 1000, 754]) | 1\n",
-      "3071     | torch.Size([1, 1000, 623]) | 1\n",
-      "3072     | torch.Size([1, 1000, 781]) | 1\n",
-      "3073     | torch.Size([1, 1000, 867]) | 1\n",
-      "3074     | torch.Size([1, 1000, 853]) | 1\n",
-      "3075     | torch.Size([1, 1000, 784]) | 1\n",
-      "3076     | torch.Size([1, 1000, 781]) | 1\n",
-      "3077     | torch.Size([1, 1000, 802]) | 1\n",
-      "3078     | torch.Size([1, 1000, 754]) | 1\n",
-      "3079     | torch.Size([1, 1000, 754]) | 1\n",
-      "3080     | torch.Size([1, 1000, 765]) | 1\n",
-      "3081     | torch.Size([1, 1000, 754]) | 1\n",
-      "3082     | torch.Size([1, 1000, 768]) | 1\n",
-      "3083     | torch.Size([1, 1000, 754]) | 1\n",
-      "3084     | torch.Size([1, 1000, 777]) | 1\n",
-      "3085     | torch.Size([1, 1000, 775]) | 1\n",
-      "3086     | torch.Size([1, 1000, 839]) | 1\n",
-      "3087     | torch.Size([1, 1000, 769]) | 1\n",
-      "3088     | torch.Size([1, 1000, 794]) | 1\n",
-      "3089     | torch.Size([1, 1000, 754]) | 1\n",
-      "3090     | torch.Size([1, 1000, 754]) | 1\n",
-      "3091     | torch.Size([1, 1000, 794]) | 1\n",
-      "3092     | torch.Size([1, 1000, 754]) | 1\n",
-      "3093     | torch.Size([1, 1000, 781]) | 1\n",
-      "3094     | torch.Size([1, 1000, 772]) | 1\n",
-      "3095     | torch.Size([1, 1000, 762]) | 1\n",
-      "3096     | torch.Size([1, 1000, 802]) | 1\n",
-      "3097     | torch.Size([1, 1000, 754]) | 1\n",
-      "3098     | torch.Size([1, 1000, 787]) | 1\n",
-      "3099     | torch.Size([1, 1000, 754]) | 1\n",
-      "3100     | torch.Size([1, 1000, 754]) | 1\n",
-      "3101     | torch.Size([1, 1000, 784]) | 1\n",
-      "3102     | torch.Size([1, 1000, 786]) | 1\n",
-      "3103     | torch.Size([1, 1000, 797]) | 1\n",
-      "3104     | torch.Size([1, 1000, 754]) | 1\n",
-      "3105     | torch.Size([1, 1000, 754]) | 1\n",
-      "3106     | torch.Size([1, 1000, 790]) | 1\n",
-      "3107     | torch.Size([1, 1000, 790]) | 1\n",
-      "3108     | torch.Size([1, 1000, 785]) | 1\n",
-      "3109     | torch.Size([1, 1000, 773]) | 1\n",
-      "3110     | torch.Size([1, 1000, 754]) | 1\n",
-      "3111     | torch.Size([1, 1000, 754]) | 1\n",
-      "3112     | torch.Size([1, 1000, 762]) | 1\n",
-      "3113     | torch.Size([1, 1000, 777]) | 1\n",
-      "3114     | torch.Size([1, 1000, 817]) | 1\n",
-      "3115     | torch.Size([1, 1000, 783]) | 1\n",
-      "3116     | torch.Size([1, 1000, 762]) | 1\n",
-      "3117     | torch.Size([1, 1000, 754]) | 1\n",
-      "3118     | torch.Size([1, 1000, 754]) | 1\n",
-      "3119     | torch.Size([1, 1000, 754]) | 1\n",
-      "3120     | torch.Size([1, 1000, 754]) | 1\n",
-      "3121     | torch.Size([1, 1000, 754]) | 1\n",
-      "3122     | torch.Size([1, 1000, 767]) | 1\n",
-      "3123     | torch.Size([1, 1000, 789]) | 1\n",
-      "3124     | torch.Size([1, 1000, 754]) | 1\n",
-      "3125     | torch.Size([1, 1000, 777]) | 1\n",
-      "3126     | torch.Size([1, 1000, 754]) | 1\n",
-      "3127     | torch.Size([1, 1000, 754]) | 1\n",
-      "3128     | torch.Size([1, 1000, 754]) | 1\n",
-      "3129     | torch.Size([1, 1000, 782]) | 1\n",
-      "3130     | torch.Size([1, 1000, 754]) | 1\n",
-      "3131     | torch.Size([1, 1000, 776]) | 1\n",
-      "3132     | torch.Size([1, 1000, 754]) | 1\n",
-      "3133     | torch.Size([1, 1000, 754]) | 1\n",
-      "3134     | torch.Size([1, 1000, 794]) | 1\n",
-      "3135     | torch.Size([1, 1000, 749]) | 1\n",
-      "3136     | torch.Size([1, 1000, 754]) | 1\n",
-      "3137     | torch.Size([1, 1000, 782]) | 1\n",
-      "3138     | torch.Size([1, 1000, 754]) | 1\n",
-      "3139     | torch.Size([1, 1000, 762]) | 1\n",
-      "3140     | torch.Size([1, 1000, 789]) | 1\n",
-      "3141     | torch.Size([1, 1000, 754]) | 1\n",
-      "3142     | torch.Size([1, 1000, 754]) | 1\n",
-      "3143     | torch.Size([1, 1000, 754]) | 1\n",
-      "3144     | torch.Size([1, 1000, 768]) | 1\n",
-      "3145     | torch.Size([1, 1000, 754]) | 1\n",
-      "3146     | torch.Size([1, 1000, 777]) | 1\n",
-      "3147     | torch.Size([1, 1000, 781]) | 1\n",
-      "3148     | torch.Size([1, 1000, 754]) | 1\n",
-      "3149     | torch.Size([1, 1000, 774]) | 1\n",
-      "3150     | torch.Size([1, 1000, 754]) | 1\n",
-      "3151     | torch.Size([1, 1000, 773]) | 1\n",
-      "3152     | torch.Size([1, 1000, 774]) | 1\n",
-      "3153     | torch.Size([1, 1000, 754]) | 1\n",
-      "3154     | torch.Size([1, 1000, 789]) | 1\n",
-      "3155     | torch.Size([1, 1000, 754]) | 1\n",
-      "3156     | torch.Size([1, 1000, 770]) | 1\n",
-      "3157     | torch.Size([1, 1000, 754]) | 1\n",
-      "3158     | torch.Size([1, 1000, 763]) | 1\n",
-      "3159     | torch.Size([1, 1000, 769]) | 1\n",
-      "3160     | torch.Size([1, 1000, 754]) | 1\n",
-      "3161     | torch.Size([1, 1000, 762]) | 1\n",
-      "3162     | torch.Size([1, 1000, 762]) | 1\n",
-      "3163     | torch.Size([1, 1000, 831]) | 1\n",
-      "3164     | torch.Size([1, 1000, 844]) | 1\n",
-      "3165     | torch.Size([1, 1000, 777]) | 1\n",
-      "3166     | torch.Size([1, 1000, 781]) | 1\n",
-      "3167     | torch.Size([1, 1000, 757]) | 1\n",
-      "3168     | torch.Size([1, 1000, 754]) | 1\n",
-      "3169     | torch.Size([1, 1000, 810]) | 1\n",
-      "3170     | torch.Size([1, 1000, 772]) | 1\n",
-      "3171     | torch.Size([1, 1000, 754]) | 1\n",
-      "3172     | torch.Size([1, 1000, 767]) | 1\n",
-      "3173     | torch.Size([1, 1000, 785]) | 1\n",
-      "3174     | torch.Size([1, 1000, 802]) | 1\n",
-      "3175     | torch.Size([1, 1000, 778]) | 1\n",
-      "3176     | torch.Size([1, 1000, 788]) | 1\n",
-      "3177     | torch.Size([1, 1000, 754]) | 1\n",
-      "3178     | torch.Size([1, 1000, 773]) | 1\n",
-      "3179     | torch.Size([1, 1000, 754]) | 1\n",
-      "3180     | torch.Size([1, 1000, 781]) | 1\n",
-      "3181     | torch.Size([1, 1000, 783]) | 1\n",
-      "3182     | torch.Size([1, 1000, 789]) | 1\n",
-      "3183     | torch.Size([1, 1000, 776]) | 1\n",
-      "3184     | torch.Size([1, 1000, 777]) | 1\n",
-      "3185     | torch.Size([1, 1000, 754]) | 1\n",
-      "3186     | torch.Size([1, 1000, 779]) | 1\n",
-      "3187     | torch.Size([1, 1000, 768]) | 1\n",
-      "3188     | torch.Size([1, 1000, 754]) | 1\n",
-      "3189     | torch.Size([1, 1000, 778]) | 1\n",
-      "3190     | torch.Size([1, 1000, 843]) | 1\n",
-      "3191     | torch.Size([1, 1000, 766]) | 1\n",
-      "3192     | torch.Size([1, 1000, 754]) | 1\n",
-      "3193     | torch.Size([1, 1000, 754]) | 1\n",
-      "3194     | torch.Size([1, 1000, 777]) | 1\n",
-      "3195     | torch.Size([1, 1000, 847]) | 1\n",
-      "3196     | torch.Size([1, 1000, 754]) | 1\n",
-      "3197     | torch.Size([1, 1000, 754]) | 1\n",
-      "3198     | torch.Size([1, 1000, 754]) | 1\n",
-      "3199     | torch.Size([1, 1000, 754]) | 1\n",
-      "3200     | torch.Size([1, 1000, 754]) | 1\n",
-      "3201     | torch.Size([1, 1000, 762]) | 1\n",
-      "3202     | torch.Size([1, 1000, 782]) | 1\n",
-      "3203     | torch.Size([1, 1000, 806]) | 1\n",
-      "3204     | torch.Size([1, 1000, 762]) | 1\n",
-      "3205     | torch.Size([1, 1000, 754]) | 1\n",
-      "3206     | torch.Size([1, 1000, 787]) | 1\n",
-      "3207     | torch.Size([1, 1000, 802]) | 1\n",
-      "3208     | torch.Size([1, 1000, 782]) | 1\n",
-      "3209     | torch.Size([1, 1000, 786]) | 1\n",
-      "3210     | torch.Size([1, 1000, 780]) | 1\n",
-      "3211     | torch.Size([1, 1000, 782]) | 1\n",
-      "3212     | torch.Size([1, 1000, 754]) | 1\n",
-      "3213     | torch.Size([1, 1000, 777]) | 1\n",
-      "3214     | torch.Size([1, 1000, 754]) | 1\n",
-      "3215     | torch.Size([1, 1000, 754]) | 1\n",
-      "3216     | torch.Size([1, 1000, 780]) | 1\n",
-      "3217     | torch.Size([1, 1000, 785]) | 1\n",
-      "3218     | torch.Size([1, 1000, 754]) | 1\n",
-      "3219     | torch.Size([1, 1000, 777]) | 1\n",
-      "3220     | torch.Size([1, 1000, 754]) | 1\n",
-      "3221     | torch.Size([1, 1000, 781]) | 1\n",
-      "3222     | torch.Size([1, 1000, 773]) | 1\n",
-      "3223     | torch.Size([1, 1000, 796]) | 1\n",
-      "3224     | torch.Size([1, 1000, 754]) | 1\n",
-      "3225     | torch.Size([1, 1000, 754]) | 1\n",
-      "3226     | torch.Size([1, 1000, 754]) | 1\n",
-      "3227     | torch.Size([1, 1000, 754]) | 1\n",
-      "3228     | torch.Size([1, 1000, 800]) | 1\n",
-      "3229     | torch.Size([1, 1000, 831]) | 1\n",
-      "3230     | torch.Size([1, 1000, 762]) | 1\n",
-      "3231     | torch.Size([1, 1000, 782]) | 1\n",
-      "3232     | torch.Size([1, 1000, 754]) | 1\n",
-      "3233     | torch.Size([1, 1000, 781]) | 1\n",
-      "3234     | torch.Size([1, 1000, 776]) | 1\n",
-      "3235     | torch.Size([1, 1000, 802]) | 1\n",
-      "3236     | torch.Size([1, 1000, 754]) | 1\n",
-      "3237     | torch.Size([1, 1000, 754]) | 1\n",
-      "3238     | torch.Size([1, 1000, 754]) | 1\n",
-      "3239     | torch.Size([1, 1000, 754]) | 1\n",
-      "3240     | torch.Size([1, 1000, 762]) | 1\n",
-      "3241     | torch.Size([1, 1000, 754]) | 1\n",
-      "3242     | torch.Size([1, 1000, 754]) | 1\n",
-      "3243     | torch.Size([1, 1000, 754]) | 1\n",
-      "3244     | torch.Size([1, 1000, 754]) | 1\n",
-      "3245     | torch.Size([1, 1000, 842]) | 1\n",
-      "3246     | torch.Size([1, 1000, 787]) | 1\n",
-      "3247     | torch.Size([1, 1000, 754]) | 1\n",
-      "3248     | torch.Size([1, 1000, 754]) | 1\n",
-      "3249     | torch.Size([1, 1000, 808]) | 1\n",
-      "3250     | torch.Size([1, 1000, 778]) | 1\n",
-      "3251     | torch.Size([1, 1000, 812]) | 1\n",
-      "3252     | torch.Size([1, 1000, 754]) | 1\n",
-      "3253     | torch.Size([1, 1000, 754]) | 1\n",
-      "3254     | torch.Size([1, 1000, 780]) | 1\n",
-      "3255     | torch.Size([1, 1000, 809]) | 1\n",
-      "3256     | torch.Size([1, 1000, 798]) | 1\n",
-      "3257     | torch.Size([1, 1000, 781]) | 1\n",
-      "3258     | torch.Size([1, 1000, 848]) | 1\n",
-      "3259     | torch.Size([1, 1000, 802]) | 1\n",
-      "3260     | torch.Size([1, 1000, 754]) | 1\n",
-      "3261     | torch.Size([1, 1000, 762]) | 1\n",
-      "3262     | torch.Size([1, 1000, 780]) | 1\n",
-      "3263     | torch.Size([1, 1000, 787]) | 1\n",
-      "3264     | torch.Size([1, 1000, 798]) | 1\n",
-      "3265     | torch.Size([1, 1000, 788]) | 1\n",
-      "3266     | torch.Size([1, 1000, 776]) | 1\n",
-      "3267     | torch.Size([1, 1000, 778]) | 1\n",
-      "3268     | torch.Size([1, 1000, 754]) | 1\n",
-      "3269     | torch.Size([1, 1000, 754]) | 1\n",
-      "3270     | torch.Size([1, 1000, 794]) | 1\n",
-      "3271     | torch.Size([1, 1000, 754]) | 1\n",
-      "3272     | torch.Size([1, 1000, 835]) | 1\n",
-      "3273     | torch.Size([1, 1000, 754]) | 1\n",
-      "3274     | torch.Size([1, 1000, 777]) | 1\n",
-      "3275     | torch.Size([1, 1000, 847]) | 1\n",
-      "3276     | torch.Size([1, 1000, 754]) | 1\n",
-      "3277     | torch.Size([1, 1000, 754]) | 1\n",
-      "3278     | torch.Size([1, 1000, 846]) | 1\n",
-      "3279     | torch.Size([1, 1000, 777]) | 1\n",
-      "3280     | torch.Size([1, 1000, 754]) | 1\n",
-      "3281     | torch.Size([1, 1000, 754]) | 1\n",
-      "3282     | torch.Size([1, 1000, 773]) | 1\n",
-      "3283     | torch.Size([1, 1000, 808]) | 1\n",
-      "3284     | torch.Size([1, 1000, 754]) | 1\n",
-      "3285     | torch.Size([1, 1000, 785]) | 1\n",
-      "3286     | torch.Size([1, 1000, 754]) | 1\n",
-      "3287     | torch.Size([1, 1000, 790]) | 1\n",
-      "3288     | torch.Size([1, 1000, 754]) | 1\n",
-      "3289     | torch.Size([1, 1000, 774]) | 1\n",
-      "3290     | torch.Size([1, 1000, 777]) | 1\n",
-      "3291     | torch.Size([1, 1000, 754]) | 1\n",
-      "3292     | torch.Size([1, 1000, 754]) | 1\n",
-      "3293     | torch.Size([1, 1000, 754]) | 1\n",
-      "3294     | torch.Size([1, 1000, 754]) | 1\n",
-      "3295     | torch.Size([1, 1000, 783]) | 1\n",
-      "3296     | torch.Size([1, 1000, 762]) | 1\n",
-      "3297     | torch.Size([1, 1000, 789]) | 1\n",
-      "3298     | torch.Size([1, 1000, 805]) | 1\n",
-      "3299     | torch.Size([1, 1000, 754]) | 1\n",
-      "3300     | torch.Size([1, 1000, 774]) | 1\n",
-      "3301     | torch.Size([1, 1000, 788]) | 1\n",
-      "3302     | torch.Size([1, 1000, 790]) | 1\n",
-      "3303     | torch.Size([1, 1000, 754]) | 1\n",
-      "3304     | torch.Size([1, 1000, 754]) | 1\n",
-      "3305     | torch.Size([1, 1000, 777]) | 1\n",
-      "3306     | torch.Size([1, 1000, 773]) | 1\n",
-      "3307     | torch.Size([1, 1000, 754]) | 1\n",
-      "3308     | torch.Size([1, 1000, 754]) | 1\n",
-      "3309     | torch.Size([1, 1000, 791]) | 1\n",
-      "3310     | torch.Size([1, 1000, 809]) | 1\n",
-      "3311     | torch.Size([1, 1000, 812]) | 1\n",
-      "3312     | torch.Size([1, 1000, 754]) | 1\n",
-      "3313     | torch.Size([1, 1000, 754]) | 1\n",
-      "3314     | torch.Size([1, 1000, 754]) | 1\n",
-      "3315     | torch.Size([1, 1000, 754]) | 1\n",
-      "3316     | torch.Size([1, 1000, 754]) | 1\n",
-      "3317     | torch.Size([1, 1000, 754]) | 1\n",
-      "3318     | torch.Size([1, 1000, 773]) | 1\n",
-      "3319     | torch.Size([1, 1000, 754]) | 1\n",
-      "3320     | torch.Size([1, 1000, 797]) | 1\n",
-      "3321     | torch.Size([1, 1000, 754]) | 1\n",
-      "3322     | torch.Size([1, 1000, 836]) | 1\n",
-      "3323     | torch.Size([1, 1000, 776]) | 1\n",
-      "3324     | torch.Size([1, 1000, 754]) | 1\n",
-      "3325     | torch.Size([1, 1000, 785]) | 1\n",
-      "3326     | torch.Size([1, 1000, 754]) | 1\n",
-      "3327     | torch.Size([1, 1000, 853]) | 1\n",
-      "3328     | torch.Size([1, 1000, 754]) | 1\n",
-      "3329     | torch.Size([1, 1000, 754]) | 1\n",
-      "3330     | torch.Size([1, 1000, 754]) | 1\n",
-      "3331     | torch.Size([1, 1000, 777]) | 1\n",
-      "3332     | torch.Size([1, 1000, 754]) | 1\n",
-      "3333     | torch.Size([1, 1000, 761]) | 1\n",
-      "3334     | torch.Size([1, 1000, 779]) | 1\n",
-      "3335     | torch.Size([1, 1000, 780]) | 1\n",
-      "3336     | torch.Size([1, 1000, 777]) | 1\n",
-      "3337     | torch.Size([1, 1000, 781]) | 1\n",
-      "3338     | torch.Size([1, 1000, 768]) | 1\n",
-      "3339     | torch.Size([1, 1000, 807]) | 1\n",
-      "3340     | torch.Size([1, 1000, 784]) | 1\n",
-      "3341     | torch.Size([1, 1000, 784]) | 1\n",
-      "3342     | torch.Size([1, 1000, 754]) | 1\n",
-      "3343     | torch.Size([1, 1000, 741]) | 1\n",
-      "3344     | torch.Size([1, 1000, 754]) | 1\n",
-      "3345     | torch.Size([1, 1000, 754]) | 1\n",
-      "3346     | torch.Size([1, 1000, 754]) | 1\n",
-      "3347     | torch.Size([1, 1000, 790]) | 1\n",
-      "3348     | torch.Size([1, 1000, 768]) | 1\n",
-      "3349     | torch.Size([1, 1000, 775]) | 1\n",
-      "3350     | torch.Size([1, 1000, 754]) | 1\n",
-      "3351     | torch.Size([1, 1000, 754]) | 1\n",
-      "3352     | torch.Size([1, 1000, 800]) | 1\n",
-      "3353     | torch.Size([1, 1000, 754]) | 1\n",
-      "3354     | torch.Size([1, 1000, 754]) | 1\n",
-      "3355     | torch.Size([1, 1000, 772]) | 1\n",
-      "3356     | torch.Size([1, 1000, 762]) | 1\n",
-      "3357     | torch.Size([1, 1000, 810]) | 1\n",
-      "3358     | torch.Size([1, 1000, 762]) | 1\n",
-      "3359     | torch.Size([1, 1000, 777]) | 1\n",
-      "3360     | torch.Size([1, 1000, 777]) | 1\n",
-      "3361     | torch.Size([1, 1000, 754]) | 1\n",
-      "3362     | torch.Size([1, 1000, 780]) | 1\n",
-      "3363     | torch.Size([1, 1000, 808]) | 1\n",
-      "3364     | torch.Size([1, 1000, 754]) | 1\n",
-      "3365     | torch.Size([1, 1000, 780]) | 1\n",
-      "3366     | torch.Size([1, 1000, 784]) | 1\n",
-      "3367     | torch.Size([1, 1000, 777]) | 1\n",
-      "3368     | torch.Size([1, 1000, 767]) | 1\n",
-      "3369     | torch.Size([1, 1000, 777]) | 1\n",
-      "3370     | torch.Size([1, 1000, 755]) | 1\n",
-      "3371     | torch.Size([1, 1000, 788]) | 1\n",
-      "3372     | torch.Size([1, 1000, 802]) | 1\n",
-      "3373     | torch.Size([1, 1000, 754]) | 1\n",
-      "3374     | torch.Size([1, 1000, 795]) | 1\n",
-      "3375     | torch.Size([1, 1000, 797]) | 1\n",
-      "3376     | torch.Size([1, 1000, 754]) | 1\n",
-      "3377     | torch.Size([1, 1000, 810]) | 1\n",
-      "3378     | torch.Size([1, 1000, 780]) | 1\n",
-      "3379     | torch.Size([1, 1000, 754]) | 1\n",
-      "3380     | torch.Size([1, 1000, 754]) | 1\n",
-      "3381     | torch.Size([1, 1000, 777]) | 1\n",
-      "3382     | torch.Size([1, 1000, 784]) | 1\n",
-      "3383     | torch.Size([1, 1000, 796]) | 1\n",
-      "3384     | torch.Size([1, 1000, 778]) | 1\n",
-      "3385     | torch.Size([1, 1000, 823]) | 1\n",
-      "3386     | torch.Size([1, 1000, 779]) | 1\n",
-      "3387     | torch.Size([1, 1000, 784]) | 1\n",
-      "3388     | torch.Size([1, 1000, 754]) | 1\n",
-      "3389     | torch.Size([1, 1000, 762]) | 1\n",
-      "3390     | torch.Size([1, 1000, 778]) | 1\n",
-      "3391     | torch.Size([1, 1000, 802]) | 1\n",
-      "3392     | torch.Size([1, 1000, 754]) | 1\n",
-      "3393     | torch.Size([1, 1000, 754]) | 1\n",
-      "3394     | torch.Size([1, 1000, 806]) | 1\n",
-      "3395     | torch.Size([1, 1000, 787]) | 1\n",
-      "3396     | torch.Size([1, 1000, 791]) | 1\n",
-      "3397     | torch.Size([1, 1000, 776]) | 1\n",
-      "3398     | torch.Size([1, 1000, 809]) | 1\n",
-      "3399     | torch.Size([1, 1000, 754]) | 1\n",
-      "3400     | torch.Size([1, 1000, 767]) | 1\n",
-      "3401     | torch.Size([1, 1000, 809]) | 1\n",
-      "3402     | torch.Size([1, 1000, 774]) | 1\n",
-      "3403     | torch.Size([1, 1000, 754]) | 1\n",
-      "3404     | torch.Size([1, 1000, 777]) | 1\n",
-      "3405     | torch.Size([1, 1000, 778]) | 1\n",
-      "3406     | torch.Size([1, 1000, 754]) | 1\n",
-      "3407     | torch.Size([1, 1000, 777]) | 1\n",
-      "3408     | torch.Size([1, 1000, 843]) | 1\n",
-      "3409     | torch.Size([1, 1000, 754]) | 1\n",
-      "3410     | torch.Size([1, 1000, 789]) | 1\n",
-      "3411     | torch.Size([1, 1000, 754]) | 1\n",
-      "3412     | torch.Size([1, 1000, 754]) | 1\n",
-      "3413     | torch.Size([1, 1000, 754]) | 1\n",
-      "3414     | torch.Size([1, 1000, 754]) | 1\n",
-      "3415     | torch.Size([1, 1000, 777]) | 1\n",
-      "3416     | torch.Size([1, 1000, 754]) | 1\n",
-      "3417     | torch.Size([1, 1000, 766]) | 1\n",
-      "3418     | torch.Size([1, 1000, 754]) | 1\n",
-      "3419     | torch.Size([1, 1000, 768]) | 1\n",
-      "3420     | torch.Size([1, 1000, 754]) | 1\n",
-      "3421     | torch.Size([1, 1000, 781]) | 1\n",
-      "3422     | torch.Size([1, 1000, 754]) | 1\n",
-      "3423     | torch.Size([1, 1000, 762]) | 1\n",
-      "3424     | torch.Size([1, 1000, 796]) | 1\n",
-      "3425     | torch.Size([1, 1000, 754]) | 1\n",
-      "3426     | torch.Size([1, 1000, 777]) | 1\n",
-      "3427     | torch.Size([1, 1000, 823]) | 1\n",
-      "3428     | torch.Size([1, 1000, 803]) | 1\n",
-      "3429     | torch.Size([1, 1000, 776]) | 1\n",
-      "3430     | torch.Size([1, 1000, 795]) | 1\n",
-      "3431     | torch.Size([1, 1000, 845]) | 1\n",
-      "3432     | torch.Size([1, 1000, 762]) | 1\n",
-      "3433     | torch.Size([1, 1000, 754]) | 1\n",
-      "3434     | torch.Size([1, 1000, 762]) | 1\n",
-      "3435     | torch.Size([1, 1000, 775]) | 1\n",
-      "3436     | torch.Size([1, 1000, 802]) | 1\n",
-      "3437     | torch.Size([1, 1000, 754]) | 1\n",
-      "3438     | torch.Size([1, 1000, 844]) | 1\n",
-      "3439     | torch.Size([1, 1000, 771]) | 1\n",
-      "3440     | torch.Size([1, 1000, 754]) | 1\n",
-      "3441     | torch.Size([1, 1000, 754]) | 1\n",
-      "3442     | torch.Size([1, 1000, 762]) | 1\n",
-      "3443     | torch.Size([1, 1000, 779]) | 1\n",
-      "3444     | torch.Size([1, 1000, 782]) | 1\n",
-      "3445     | torch.Size([1, 1000, 775]) | 1\n",
-      "3446     | torch.Size([1, 1000, 810]) | 1\n",
-      "3447     | torch.Size([1, 1000, 777]) | 1\n",
-      "3448     | torch.Size([1, 1000, 771]) | 1\n",
-      "3449     | torch.Size([1, 1000, 718]) | 1\n",
-      "3450     | torch.Size([1, 1000, 777]) | 1\n",
-      "3451     | torch.Size([1, 1000, 754]) | 1\n",
-      "3452     | torch.Size([1, 1000, 754]) | 1\n",
-      "3453     | torch.Size([1, 1000, 867]) | 1\n",
-      "3454     | torch.Size([1, 1000, 802]) | 1\n",
-      "3455     | torch.Size([1, 1000, 754]) | 1\n",
-      "3456     | torch.Size([1, 1000, 777]) | 1\n",
-      "3457     | torch.Size([1, 1000, 777]) | 1\n",
-      "3458     | torch.Size([1, 1000, 769]) | 1\n",
-      "3459     | torch.Size([1, 1000, 762]) | 1\n",
-      "3460     | torch.Size([1, 1000, 775]) | 1\n",
-      "3461     | torch.Size([1, 1000, 796]) | 1\n",
-      "3462     | torch.Size([1, 1000, 754]) | 1\n",
-      "3463     | torch.Size([1, 1000, 785]) | 1\n",
-      "3464     | torch.Size([1, 1000, 808]) | 1\n",
-      "3465     | torch.Size([1, 1000, 778]) | 1\n",
-      "3466     | torch.Size([1, 1000, 754]) | 1\n",
-      "3467     | torch.Size([1, 1000, 777]) | 1\n",
-      "3468     | torch.Size([1, 1000, 754]) | 1\n",
-      "3469     | torch.Size([1, 1000, 763]) | 1\n",
-      "3470     | torch.Size([1, 1000, 777]) | 1\n",
-      "3471     | torch.Size([1, 1000, 754]) | 1\n",
-      "3472     | torch.Size([1, 1000, 754]) | 1\n",
-      "3473     | torch.Size([1, 1000, 776]) | 1\n",
-      "3474     | torch.Size([1, 1000, 754]) | 1\n",
-      "3475     | torch.Size([1, 1000, 754]) | 1\n",
-      "3476     | torch.Size([1, 1000, 611]) | 1\n",
-      "3477     | torch.Size([1, 1000, 754]) | 1\n",
-      "3478     | torch.Size([1, 1000, 801]) | 1\n",
-      "3479     | torch.Size([1, 1000, 783]) | 1\n",
-      "3480     | torch.Size([1, 1000, 778]) | 1\n",
-      "3481     | torch.Size([1, 1000, 785]) | 1\n",
-      "3482     | torch.Size([1, 1000, 771]) | 1\n",
-      "3483     | torch.Size([1, 1000, 754]) | 1\n",
-      "3484     | torch.Size([1, 1000, 773]) | 1\n",
-      "3485     | torch.Size([1, 1000, 802]) | 1\n",
-      "3486     | torch.Size([1, 1000, 754]) | 1\n",
-      "3487     | torch.Size([1, 1000, 754]) | 1\n",
-      "3488     | torch.Size([1, 1000, 784]) | 1\n",
-      "3489     | torch.Size([1, 1000, 783]) | 1\n",
-      "3490     | torch.Size([1, 1000, 777]) | 1\n",
-      "3491     | torch.Size([1, 1000, 754]) | 1\n",
-      "3492     | torch.Size([1, 1000, 754]) | 1\n",
-      "3493     | torch.Size([1, 1000, 754]) | 1\n",
-      "3494     | torch.Size([1, 1000, 754]) | 1\n",
-      "3495     | torch.Size([1, 1000, 782]) | 1\n",
-      "3496     | torch.Size([1, 1000, 754]) | 1\n",
-      "3497     | torch.Size([1, 1000, 754]) | 1\n",
-      "3498     | torch.Size([1, 1000, 754]) | 1\n",
-      "3499     | torch.Size([1, 1000, 776]) | 1\n",
-      "3500     | torch.Size([1, 1000, 774]) | 1\n",
-      "3501     | torch.Size([1, 1000, 780]) | 1\n",
-      "3502     | torch.Size([1, 1000, 781]) | 1\n",
-      "3503     | torch.Size([1, 1000, 840]) | 1\n",
-      "3504     | torch.Size([1, 1000, 754]) | 1\n",
-      "3505     | torch.Size([1, 1000, 810]) | 1\n",
-      "3506     | torch.Size([1, 1000, 771]) | 1\n",
-      "3507     | torch.Size([1, 1000, 831]) | 1\n",
-      "3508     | torch.Size([1, 1000, 754]) | 1\n",
-      "3509     | torch.Size([1, 1000, 754]) | 1\n",
-      "3510     | torch.Size([1, 1000, 780]) | 1\n",
-      "3511     | torch.Size([1, 1000, 772]) | 1\n",
-      "3512     | torch.Size([1, 1000, 762]) | 1\n",
-      "3513     | torch.Size([1, 1000, 777]) | 1\n",
-      "3514     | torch.Size([1, 1000, 762]) | 1\n",
-      "3515     | torch.Size([1, 1000, 777]) | 1\n",
-      "3516     | torch.Size([1, 1000, 777]) | 1\n",
-      "3517     | torch.Size([1, 1000, 771]) | 1\n",
-      "3518     | torch.Size([1, 1000, 777]) | 1\n",
-      "3519     | torch.Size([1, 1000, 766]) | 1\n",
-      "3520     | torch.Size([1, 1000, 785]) | 1\n",
-      "3521     | torch.Size([1, 1000, 754]) | 1\n",
-      "3522     | torch.Size([1, 1000, 809]) | 1\n",
-      "3523     | torch.Size([1, 1000, 787]) | 1\n",
-      "3524     | torch.Size([1, 1000, 787]) | 1\n",
-      "3525     | torch.Size([1, 1000, 769]) | 1\n",
-      "3526     | torch.Size([1, 1000, 802]) | 1\n",
-      "3527     | torch.Size([1, 1000, 782]) | 1\n",
-      "3528     | torch.Size([1, 1000, 772]) | 1\n",
-      "3529     | torch.Size([1, 1000, 781]) | 1\n",
-      "3530     | torch.Size([1, 1000, 823]) | 1\n",
-      "3531     | torch.Size([1, 1000, 754]) | 1\n",
-      "3532     | torch.Size([1, 1000, 754]) | 1\n",
-      "3533     | torch.Size([1, 1000, 802]) | 1\n",
-      "3534     | torch.Size([1, 1000, 754]) | 1\n",
-      "3535     | torch.Size([1, 1000, 843]) | 1\n",
-      "3536     | torch.Size([1, 1000, 754]) | 1\n",
-      "3537     | torch.Size([1, 1000, 767]) | 1\n",
-      "3538     | torch.Size([1, 1000, 777]) | 1\n",
-      "3539     | torch.Size([1, 1000, 754]) | 1\n",
-      "3540     | torch.Size([1, 1000, 754]) | 1\n",
-      "3541     | torch.Size([1, 1000, 754]) | 1\n",
-      "3542     | torch.Size([1, 1000, 754]) | 1\n",
-      "3543     | torch.Size([1, 1000, 778]) | 1\n",
-      "3544     | torch.Size([1, 1000, 754]) | 1\n",
-      "3545     | torch.Size([1, 1000, 777]) | 1\n",
-      "3546     | torch.Size([1, 1000, 762]) | 1\n",
-      "3547     | torch.Size([1, 1000, 754]) | 1\n",
-      "3548     | torch.Size([1, 1000, 774]) | 1\n",
-      "3549     | torch.Size([1, 1000, 770]) | 1\n",
-      "3550     | torch.Size([1, 1000, 754]) | 1\n",
-      "3551     | torch.Size([1, 1000, 843]) | 1\n",
-      "3552     | torch.Size([1, 1000, 754]) | 1\n",
-      "3553     | torch.Size([1, 1000, 772]) | 1\n",
-      "3554     | torch.Size([1, 1000, 754]) | 1\n",
-      "3555     | torch.Size([1, 1000, 784]) | 1\n",
-      "3556     | torch.Size([1, 1000, 846]) | 1\n",
-      "3557     | torch.Size([1, 1000, 779]) | 1\n",
-      "3558     | torch.Size([1, 1000, 754]) | 1\n",
-      "3559     | torch.Size([1, 1000, 754]) | 1\n",
-      "3560     | torch.Size([1, 1000, 754]) | 1\n",
-      "3561     | torch.Size([1, 1000, 754]) | 1\n",
-      "3562     | torch.Size([1, 1000, 767]) | 1\n",
-      "3563     | torch.Size([1, 1000, 754]) | 1\n",
-      "3564     | torch.Size([1, 1000, 792]) | 1\n",
-      "3565     | torch.Size([1, 1000, 792]) | 1\n",
-      "3566     | torch.Size([1, 1000, 754]) | 1\n",
-      "3567     | torch.Size([1, 1000, 754]) | 1\n",
-      "3568     | torch.Size([1, 1000, 778]) | 1\n",
-      "3569     | torch.Size([1, 1000, 843]) | 1\n",
-      "3570     | torch.Size([1, 1000, 780]) | 1\n",
-      "3571     | torch.Size([1, 1000, 762]) | 1\n",
-      "3572     | torch.Size([1, 1000, 786]) | 1\n",
-      "3573     | torch.Size([1, 1000, 731]) | 1\n",
-      "3574     | torch.Size([1, 1000, 787]) | 1\n",
-      "3575     | torch.Size([1, 1000, 778]) | 1\n",
-      "3576     | torch.Size([1, 1000, 754]) | 1\n",
-      "3577     | torch.Size([1, 1000, 802]) | 1\n",
-      "3578     | torch.Size([1, 1000, 787]) | 1\n",
-      "3579     | torch.Size([1, 1000, 754]) | 1\n",
-      "3580     | torch.Size([1, 1000, 772]) | 1\n",
-      "3581     | torch.Size([1, 1000, 785]) | 1\n",
-      "3582     | torch.Size([1, 1000, 754]) | 1\n",
-      "3583     | torch.Size([1, 1000, 754]) | 1\n",
-      "3584     | torch.Size([1, 1000, 765]) | 1\n",
-      "3585     | torch.Size([1, 1000, 754]) | 1\n",
-      "3586     | torch.Size([1, 1000, 754]) | 1\n",
-      "3587     | torch.Size([1, 1000, 754]) | 1\n",
-      "3588     | torch.Size([1, 1000, 754]) | 1\n",
-      "3589     | torch.Size([1, 1000, 776]) | 1\n",
-      "3590     | torch.Size([1, 1000, 754]) | 1\n",
-      "3591     | torch.Size([1, 1000, 754]) | 1\n",
-      "3592     | torch.Size([1, 1000, 777]) | 1\n",
-      "3593     | torch.Size([1, 1000, 754]) | 1\n",
-      "3594     | torch.Size([1, 1000, 772]) | 1\n",
-      "3595     | torch.Size([1, 1000, 807]) | 1\n",
-      "3596     | torch.Size([1, 1000, 754]) | 1\n",
-      "3597     | torch.Size([1, 1000, 764]) | 1\n",
-      "3598     | torch.Size([1, 1000, 772]) | 1\n",
-      "3599     | torch.Size([1, 1000, 777]) | 1\n",
-      "3600     | torch.Size([1, 1000, 847]) | 1\n",
-      "3601     | torch.Size([1, 1000, 754]) | 1\n",
-      "3602     | torch.Size([1, 1000, 762]) | 1\n",
-      "3603     | torch.Size([1, 1000, 782]) | 1\n",
-      "3604     | torch.Size([1, 1000, 787]) | 1\n",
-      "3605     | torch.Size([1, 1000, 754]) | 1\n",
-      "3606     | torch.Size([1, 1000, 811]) | 1\n",
-      "3607     | torch.Size([1, 1000, 754]) | 1\n",
-      "3608     | torch.Size([1, 1000, 754]) | 1\n",
-      "3609     | torch.Size([1, 1000, 754]) | 1\n",
-      "3610     | torch.Size([1, 1000, 754]) | 1\n",
-      "3611     | torch.Size([1, 1000, 762]) | 1\n",
-      "3612     | torch.Size([1, 1000, 786]) | 1\n",
-      "3613     | torch.Size([1, 1000, 768]) | 1\n",
-      "3614     | torch.Size([1, 1000, 754]) | 1\n",
-      "3615     | torch.Size([1, 1000, 754]) | 1\n",
-      "3616     | torch.Size([1, 1000, 785]) | 1\n",
-      "3617     | torch.Size([1, 1000, 754]) | 1\n",
-      "3618     | torch.Size([1, 1000, 767]) | 1\n",
-      "3619     | torch.Size([1, 1000, 803]) | 1\n",
-      "3620     | torch.Size([1, 1000, 778]) | 1\n",
-      "3621     | torch.Size([1, 1000, 775]) | 1\n",
-      "3622     | torch.Size([1, 1000, 764]) | 1\n",
-      "3623     | torch.Size([1, 1000, 754]) | 1\n",
-      "3624     | torch.Size([1, 1000, 754]) | 1\n",
-      "3625     | torch.Size([1, 1000, 802]) | 1\n",
-      "3626     | torch.Size([1, 1000, 754]) | 1\n",
-      "3627     | torch.Size([1, 1000, 754]) | 1\n",
-      "3628     | torch.Size([1, 1000, 754]) | 1\n",
-      "3629     | torch.Size([1, 1000, 754]) | 1\n",
-      "3630     | torch.Size([1, 1000, 768]) | 1\n",
-      "3631     | torch.Size([1, 1000, 772]) | 1\n",
-      "3632     | torch.Size([1, 1000, 782]) | 1\n",
-      "3633     | torch.Size([1, 1000, 781]) | 1\n",
-      "3634     | torch.Size([1, 1000, 754]) | 1\n",
-      "3635     | torch.Size([1, 1000, 777]) | 1\n",
-      "3636     | torch.Size([1, 1000, 754]) | 1\n",
-      "3637     | torch.Size([1, 1000, 791]) | 1\n",
-      "3638     | torch.Size([1, 1000, 791]) | 1\n",
-      "3639     | torch.Size([1, 1000, 783]) | 1\n",
-      "3640     | torch.Size([1, 1000, 754]) | 1\n",
-      "3641     | torch.Size([1, 1000, 776]) | 1\n",
-      "3642     | torch.Size([1, 1000, 771]) | 1\n",
-      "3643     | torch.Size([1, 1000, 777]) | 1\n",
-      "3644     | torch.Size([1, 1000, 777]) | 1\n",
-      "3645     | torch.Size([1, 1000, 754]) | 1\n",
-      "3646     | torch.Size([1, 1000, 754]) | 1\n",
-      "3647     | torch.Size([1, 1000, 754]) | 1\n",
-      "3648     | torch.Size([1, 1000, 777]) | 1\n",
-      "3649     | torch.Size([1, 1000, 754]) | 1\n",
-      "3650     | torch.Size([1, 1000, 754]) | 1\n",
-      "3651     | torch.Size([1, 1000, 806]) | 1\n",
-      "3652     | torch.Size([1, 1000, 754]) | 1\n",
-      "3653     | torch.Size([1, 1000, 754]) | 1\n",
-      "3654     | torch.Size([1, 1000, 777]) | 1\n",
-      "3655     | torch.Size([1, 1000, 779]) | 1\n",
-      "3656     | torch.Size([1, 1000, 777]) | 1\n",
-      "3657     | torch.Size([1, 1000, 808]) | 1\n",
-      "3658     | torch.Size([1, 1000, 762]) | 1\n",
-      "3659     | torch.Size([1, 1000, 770]) | 1\n",
-      "3660     | torch.Size([1, 1000, 754]) | 1\n",
-      "3661     | torch.Size([1, 1000, 754]) | 1\n",
-      "3662     | torch.Size([1, 1000, 754]) | 1\n",
-      "3663     | torch.Size([1, 1000, 802]) | 1\n",
-      "3664     | torch.Size([1, 1000, 786]) | 1\n",
-      "3665     | torch.Size([1, 1000, 779]) | 1\n",
-      "3666     | torch.Size([1, 1000, 777]) | 1\n",
-      "3667     | torch.Size([1, 1000, 754]) | 1\n",
-      "3668     | torch.Size([1, 1000, 784]) | 1\n",
-      "3669     | torch.Size([1, 1000, 850]) | 1\n",
-      "3670     | torch.Size([1, 1000, 777]) | 1\n",
-      "3671     | torch.Size([1, 1000, 777]) | 1\n",
-      "3672     | torch.Size([1, 1000, 842]) | 1\n",
-      "3673     | torch.Size([1, 1000, 792]) | 1\n",
-      "3674     | torch.Size([1, 1000, 763]) | 1\n",
-      "3675     | torch.Size([1, 1000, 777]) | 1\n",
-      "3676     | torch.Size([1, 1000, 762]) | 1\n",
-      "3677     | torch.Size([1, 1000, 808]) | 1\n",
-      "3678     | torch.Size([1, 1000, 779]) | 1\n",
-      "3679     | torch.Size([1, 1000, 754]) | 1\n",
-      "3680     | torch.Size([1, 1000, 809]) | 1\n",
-      "3681     | torch.Size([1, 1000, 762]) | 1\n",
-      "3682     | torch.Size([1, 1000, 788]) | 1\n",
-      "3683     | torch.Size([1, 1000, 777]) | 1\n",
-      "3684     | torch.Size([1, 1000, 754]) | 1\n",
-      "3685     | torch.Size([1, 1000, 754]) | 1\n",
-      "3686     | torch.Size([1, 1000, 754]) | 1\n",
-      "3687     | torch.Size([1, 1000, 754]) | 1\n",
-      "3688     | torch.Size([1, 1000, 762]) | 1\n",
-      "3689     | torch.Size([1, 1000, 796]) | 1\n",
-      "3690     | torch.Size([1, 1000, 782]) | 1\n",
-      "3691     | torch.Size([1, 1000, 802]) | 1\n",
-      "3692     | torch.Size([1, 1000, 754]) | 1\n",
-      "3693     | torch.Size([1, 1000, 754]) | 1\n",
-      "3694     | torch.Size([1, 1000, 754]) | 1\n",
-      "3695     | torch.Size([1, 1000, 777]) | 1\n",
-      "3696     | torch.Size([1, 1000, 778]) | 1\n",
-      "3697     | torch.Size([1, 1000, 754]) | 1\n",
-      "3698     | torch.Size([1, 1000, 786]) | 1\n",
-      "3699     | torch.Size([1, 1000, 784]) | 1\n",
-      "3700     | torch.Size([1, 1000, 777]) | 1\n",
-      "3701     | torch.Size([1, 1000, 754]) | 1\n",
-      "3702     | torch.Size([1, 1000, 762]) | 1\n",
-      "3703     | torch.Size([1, 1000, 796]) | 1\n",
-      "3704     | torch.Size([1, 1000, 754]) | 1\n",
-      "3705     | torch.Size([1, 1000, 765]) | 1\n",
-      "3706     | torch.Size([1, 1000, 762]) | 1\n",
-      "3707     | torch.Size([1, 1000, 781]) | 1\n",
-      "3708     | torch.Size([1, 1000, 754]) | 1\n",
-      "3709     | torch.Size([1, 1000, 869]) | 1\n",
-      "3710     | torch.Size([1, 1000, 754]) | 1\n",
-      "3711     | torch.Size([1, 1000, 781]) | 1\n",
-      "3712     | torch.Size([1, 1000, 777]) | 1\n",
-      "3713     | torch.Size([1, 1000, 754]) | 1\n",
-      "3714     | torch.Size([1, 1000, 808]) | 1\n",
-      "3715     | torch.Size([1, 1000, 754]) | 1\n",
-      "3716     | torch.Size([1, 1000, 804]) | 1\n",
-      "3717     | torch.Size([1, 1000, 778]) | 1\n",
-      "3718     | torch.Size([1, 1000, 754]) | 1\n",
-      "3719     | torch.Size([1, 1000, 780]) | 1\n",
-      "3720     | torch.Size([1, 1000, 781]) | 1\n",
-      "3721     | torch.Size([1, 1000, 783]) | 1\n",
-      "3722     | torch.Size([1, 1000, 762]) | 1\n",
-      "3723     | torch.Size([1, 1000, 783]) | 1\n",
-      "3724     | torch.Size([1, 1000, 777]) | 1\n",
-      "3725     | torch.Size([1, 1000, 777]) | 1\n",
-      "3726     | torch.Size([1, 1000, 754]) | 1\n",
-      "3727     | torch.Size([1, 1000, 793]) | 1\n",
-      "3728     | torch.Size([1, 1000, 754]) | 1\n",
-      "3729     | torch.Size([1, 1000, 799]) | 1\n",
-      "3730     | torch.Size([1, 1000, 799]) | 1\n",
-      "3731     | torch.Size([1, 1000, 777]) | 1\n",
-      "3732     | torch.Size([1, 1000, 754]) | 1\n",
-      "3733     | torch.Size([1, 1000, 754]) | 1\n",
-      "3734     | torch.Size([1, 1000, 764]) | 1\n",
-      "3735     | torch.Size([1, 1000, 787]) | 1\n",
-      "3736     | torch.Size([1, 1000, 765]) | 1\n",
-      "3737     | torch.Size([1, 1000, 778]) | 1\n",
-      "3738     | torch.Size([1, 1000, 781]) | 1\n",
-      "3739     | torch.Size([1, 1000, 785]) | 1\n",
-      "3740     | torch.Size([1, 1000, 811]) | 1\n",
-      "3741     | torch.Size([1, 1000, 754]) | 1\n",
-      "3742     | torch.Size([1, 1000, 777]) | 1\n",
-      "3743     | torch.Size([1, 1000, 754]) | 1\n",
-      "3744     | torch.Size([1, 1000, 754]) | 1\n",
-      "3745     | torch.Size([1, 1000, 754]) | 1\n",
-      "3746     | torch.Size([1, 1000, 763]) | 1\n",
-      "3747     | torch.Size([1, 1000, 802]) | 1\n",
-      "3748     | torch.Size([1, 1000, 762]) | 1\n",
-      "3749     | torch.Size([1, 1000, 775]) | 1\n",
-      "3750     | torch.Size([1, 1000, 754]) | 1\n",
-      "3751     | torch.Size([1, 1000, 771]) | 1\n",
-      "3752     | torch.Size([1, 1000, 754]) | 1\n",
-      "3753     | torch.Size([1, 1000, 777]) | 1\n",
-      "3754     | torch.Size([1, 1000, 844]) | 1\n",
-      "3755     | torch.Size([1, 1000, 760]) | 1\n",
-      "3756     | torch.Size([1, 1000, 754]) | 1\n",
-      "3757     | torch.Size([1, 1000, 783]) | 1\n",
-      "3758     | torch.Size([1, 1000, 780]) | 1\n",
-      "3759     | torch.Size([1, 1000, 754]) | 1\n",
-      "3760     | torch.Size([1, 1000, 754]) | 1\n",
-      "3761     | torch.Size([1, 1000, 777]) | 1\n",
-      "3762     | torch.Size([1, 1000, 812]) | 1\n",
-      "3763     | torch.Size([1, 1000, 802]) | 1\n",
-      "3764     | torch.Size([1, 1000, 754]) | 1\n",
-      "3765     | torch.Size([1, 1000, 809]) | 1\n",
-      "3766     | torch.Size([1, 1000, 800]) | 1\n",
-      "3767     | torch.Size([1, 1000, 777]) | 1\n",
-      "3768     | torch.Size([1, 1000, 779]) | 1\n",
-      "3769     | torch.Size([1, 1000, 807]) | 1\n",
-      "3770     | torch.Size([1, 1000, 802]) | 1\n",
-      "3771     | torch.Size([1, 1000, 784]) | 1\n",
-      "3772     | torch.Size([1, 1000, 786]) | 1\n",
-      "3773     | torch.Size([1, 1000, 754]) | 1\n",
-      "3774     | torch.Size([1, 1000, 754]) | 1\n",
-      "3775     | torch.Size([1, 1000, 754]) | 1\n",
-      "3776     | torch.Size([1, 1000, 851]) | 1\n",
-      "3777     | torch.Size([1, 1000, 783]) | 1\n",
-      "3778     | torch.Size([1, 1000, 779]) | 1\n",
-      "3779     | torch.Size([1, 1000, 772]) | 1\n",
-      "3780     | torch.Size([1, 1000, 814]) | 1\n",
-      "3781     | torch.Size([1, 1000, 841]) | 1\n",
-      "3782     | torch.Size([1, 1000, 754]) | 1\n",
-      "3783     | torch.Size([1, 1000, 787]) | 1\n",
-      "3784     | torch.Size([1, 1000, 777]) | 1\n",
-      "3785     | torch.Size([1, 1000, 754]) | 1\n",
-      "3786     | torch.Size([1, 1000, 790]) | 1\n",
-      "3787     | torch.Size([1, 1000, 784]) | 1\n",
-      "3788     | torch.Size([1, 1000, 762]) | 1\n",
-      "3789     | torch.Size([1, 1000, 762]) | 1\n",
-      "3790     | torch.Size([1, 1000, 784]) | 1\n",
-      "3791     | torch.Size([1, 1000, 802]) | 1\n",
-      "3792     | torch.Size([1, 1000, 754]) | 1\n",
-      "3793     | torch.Size([1, 1000, 784]) | 1\n",
-      "3794     | torch.Size([1, 1000, 782]) | 1\n",
-      "3795     | torch.Size([1, 1000, 754]) | 1\n",
-      "3796     | torch.Size([1, 1000, 783]) | 1\n",
-      "3797     | torch.Size([1, 1000, 762]) | 1\n",
-      "3798     | torch.Size([1, 1000, 783]) | 1\n",
-      "3799     | torch.Size([1, 1000, 754]) | 1\n",
-      "3800     | torch.Size([1, 1000, 754]) | 1\n",
-      "3801     | torch.Size([1, 1000, 754]) | 1\n",
-      "3802     | torch.Size([1, 1000, 777]) | 1\n",
-      "3803     | torch.Size([1, 1000, 763]) | 1\n",
-      "3804     | torch.Size([1, 1000, 769]) | 1\n",
-      "3805     | torch.Size([1, 1000, 754]) | 1\n",
-      "3806     | torch.Size([1, 1000, 776]) | 1\n",
-      "3807     | torch.Size([1, 1000, 754]) | 1\n",
-      "3808     | torch.Size([1, 1000, 784]) | 1\n",
-      "3809     | torch.Size([1, 1000, 754]) | 1\n",
-      "3810     | torch.Size([1, 1000, 777]) | 1\n",
-      "3811     | torch.Size([1, 1000, 754]) | 1\n",
-      "3812     | torch.Size([1, 1000, 848]) | 1\n",
-      "3813     | torch.Size([1, 1000, 770]) | 1\n",
-      "3814     | torch.Size([1, 1000, 787]) | 1\n",
-      "3815     | torch.Size([1, 1000, 754]) | 1\n",
-      "3816     | torch.Size([1, 1000, 798]) | 1\n",
-      "3817     | torch.Size([1, 1000, 777]) | 1\n",
-      "3818     | torch.Size([1, 1000, 765]) | 1\n",
-      "3819     | torch.Size([1, 1000, 781]) | 1\n",
-      "3820     | torch.Size([1, 1000, 754]) | 1\n",
-      "3821     | torch.Size([1, 1000, 754]) | 1\n",
-      "3822     | torch.Size([1, 1000, 754]) | 1\n",
-      "3823     | torch.Size([1, 1000, 754]) | 1\n",
-      "3824     | torch.Size([1, 1000, 783]) | 1\n",
-      "3825     | torch.Size([1, 1000, 781]) | 1\n",
-      "3826     | torch.Size([1, 1000, 754]) | 1\n",
-      "3827     | torch.Size([1, 1000, 763]) | 1\n",
-      "3828     | torch.Size([1, 1000, 754]) | 1\n",
-      "3829     | torch.Size([1, 1000, 777]) | 1\n",
-      "3830     | torch.Size([1, 1000, 754]) | 1\n",
-      "3831     | torch.Size([1, 1000, 777]) | 1\n",
-      "3832     | torch.Size([1, 1000, 765]) | 1\n",
-      "3833     | torch.Size([1, 1000, 769]) | 1\n",
-      "3834     | torch.Size([1, 1000, 762]) | 1\n",
-      "3835     | torch.Size([1, 1000, 786]) | 1\n",
-      "3836     | torch.Size([1, 1000, 762]) | 1\n",
-      "3837     | torch.Size([1, 1000, 754]) | 1\n",
-      "3838     | torch.Size([1, 1000, 802]) | 1\n",
-      "3839     | torch.Size([1, 1000, 762]) | 1\n",
-      "3840     | torch.Size([1, 1000, 807]) | 1\n",
-      "3841     | torch.Size([1, 1000, 787]) | 1\n",
-      "3842     | torch.Size([1, 1000, 777]) | 1\n",
-      "3843     | torch.Size([1, 1000, 777]) | 1\n",
-      "3844     | torch.Size([1, 1000, 754]) | 1\n",
-      "3845     | torch.Size([1, 1000, 754]) | 1\n",
-      "3846     | torch.Size([1, 1000, 754]) | 1\n",
-      "3847     | torch.Size([1, 1000, 785]) | 1\n",
-      "3848     | torch.Size([1, 1000, 754]) | 1\n",
-      "3849     | torch.Size([1, 1000, 778]) | 1\n",
-      "3850     | torch.Size([1, 1000, 775]) | 1\n",
-      "3851     | torch.Size([1, 1000, 754]) | 1\n",
-      "3852     | torch.Size([1, 1000, 797]) | 1\n",
-      "3853     | torch.Size([1, 1000, 802]) | 1\n",
-      "3854     | torch.Size([1, 1000, 754]) | 1\n",
-      "3855     | torch.Size([1, 1000, 781]) | 1\n",
-      "3856     | torch.Size([1, 1000, 780]) | 1\n",
-      "3857     | torch.Size([1, 1000, 801]) | 1\n",
-      "3858     | torch.Size([1, 1000, 789]) | 1\n",
-      "3859     | torch.Size([1, 1000, 798]) | 1\n",
-      "3860     | torch.Size([1, 1000, 802]) | 1\n",
-      "3861     | torch.Size([1, 1000, 754]) | 1\n",
-      "3862     | torch.Size([1, 1000, 797]) | 1\n",
-      "3863     | torch.Size([1, 1000, 786]) | 1\n",
-      "3864     | torch.Size([1, 1000, 754]) | 1\n",
-      "3865     | torch.Size([1, 1000, 785]) | 1\n",
-      "3866     | torch.Size([1, 1000, 756]) | 1\n",
-      "3867     | torch.Size([1, 1000, 806]) | 1\n",
-      "3868     | torch.Size([1, 1000, 777]) | 1\n",
-      "3869     | torch.Size([1, 1000, 754]) | 1\n",
-      "3870     | torch.Size([1, 1000, 785]) | 1\n",
-      "3871     | torch.Size([1, 1000, 777]) | 1\n",
-      "3872     | torch.Size([1, 1000, 754]) | 1\n",
-      "3873     | torch.Size([1, 1000, 754]) | 1\n",
-      "3874     | torch.Size([1, 1000, 808]) | 1\n",
-      "3875     | torch.Size([1, 1000, 754]) | 1\n",
-      "3876     | torch.Size([1, 1000, 754]) | 1\n",
-      "3877     | torch.Size([1, 1000, 782]) | 1\n",
-      "3878     | torch.Size([1, 1000, 777]) | 1\n",
-      "3879     | torch.Size([1, 1000, 786]) | 1\n",
-      "3880     | torch.Size([1, 1000, 778]) | 1\n",
-      "3881     | torch.Size([1, 1000, 777]) | 1\n",
-      "3882     | torch.Size([1, 1000, 781]) | 1\n",
-      "3883     | torch.Size([1, 1000, 840]) | 1\n",
-      "3884     | torch.Size([1, 1000, 754]) | 1\n",
-      "3885     | torch.Size([1, 1000, 762]) | 1\n",
-      "3886     | torch.Size([1, 1000, 777]) | 1\n",
-      "3887     | torch.Size([1, 1000, 778]) | 1\n",
-      "3888     | torch.Size([1, 1000, 773]) | 1\n",
-      "3889     | torch.Size([1, 1000, 807]) | 1\n",
-      "3890     | torch.Size([1, 1000, 766]) | 1\n",
-      "3891     | torch.Size([1, 1000, 779]) | 1\n",
-      "3892     | torch.Size([1, 1000, 783]) | 1\n",
-      "3893     | torch.Size([1, 1000, 846]) | 1\n",
-      "3894     | torch.Size([1, 1000, 790]) | 1\n",
-      "3895     | torch.Size([1, 1000, 802]) | 1\n",
-      "3896     | torch.Size([1, 1000, 777]) | 1\n",
-      "3897     | torch.Size([1, 1000, 797]) | 1\n",
-      "3898     | torch.Size([1, 1000, 807]) | 1\n",
-      "3899     | torch.Size([1, 1000, 612]) | 1\n",
-      "3900     | torch.Size([1, 1000, 754]) | 1\n",
-      "3901     | torch.Size([1, 1000, 754]) | 1\n",
-      "3902     | torch.Size([1, 1000, 769]) | 1\n",
-      "3903     | torch.Size([1, 1000, 777]) | 1\n",
-      "3904     | torch.Size([1, 1000, 804]) | 1\n",
-      "3905     | torch.Size([1, 1000, 762]) | 1\n",
-      "3906     | torch.Size([1, 1000, 754]) | 1\n",
-      "3907     | torch.Size([1, 1000, 845]) | 1\n",
-      "3908     | torch.Size([1, 1000, 754]) | 1\n",
-      "3909     | torch.Size([1, 1000, 789]) | 1\n",
-      "3910     | torch.Size([1, 1000, 777]) | 1\n",
-      "3911     | torch.Size([1, 1000, 754]) | 1\n",
-      "3912     | torch.Size([1, 1000, 754]) | 1\n",
-      "3913     | torch.Size([1, 1000, 783]) | 1\n",
-      "3914     | torch.Size([1, 1000, 781]) | 1\n",
-      "3915     | torch.Size([1, 1000, 772]) | 1\n",
-      "3916     | torch.Size([1, 1000, 754]) | 1\n",
-      "3917     | torch.Size([1, 1000, 785]) | 1\n",
-      "3918     | torch.Size([1, 1000, 754]) | 1\n",
-      "3919     | torch.Size([1, 1000, 754]) | 1\n",
-      "3920     | torch.Size([1, 1000, 787]) | 1\n",
-      "3921     | torch.Size([1, 1000, 754]) | 1\n",
-      "3922     | torch.Size([1, 1000, 777]) | 1\n",
-      "3923     | torch.Size([1, 1000, 769]) | 1\n",
-      "3924     | torch.Size([1, 1000, 777]) | 1\n",
-      "3925     | torch.Size([1, 1000, 773]) | 1\n",
-      "3926     | torch.Size([1, 1000, 782]) | 1\n",
-      "3927     | torch.Size([1, 1000, 754]) | 1\n",
-      "3928     | torch.Size([1, 1000, 771]) | 1\n",
-      "3929     | torch.Size([1, 1000, 790]) | 1\n",
-      "3930     | torch.Size([1, 1000, 780]) | 1\n",
-      "3931     | torch.Size([1, 1000, 754]) | 1\n",
-      "3932     | torch.Size([1, 1000, 783]) | 1\n",
-      "3933     | torch.Size([1, 1000, 777]) | 1\n",
-      "3934     | torch.Size([1, 1000, 786]) | 1\n",
-      "3935     | torch.Size([1, 1000, 754]) | 1\n",
-      "3936     | torch.Size([1, 1000, 785]) | 1\n",
-      "3937     | torch.Size([1, 1000, 754]) | 1\n",
-      "3938     | torch.Size([1, 1000, 754]) | 1\n",
-      "3939     | torch.Size([1, 1000, 797]) | 1\n",
-      "3940     | torch.Size([1, 1000, 754]) | 1\n",
-      "3941     | torch.Size([1, 1000, 777]) | 1\n",
-      "3942     | torch.Size([1, 1000, 777]) | 1\n",
-      "3943     | torch.Size([1, 1000, 754]) | 1\n",
-      "3944     | torch.Size([1, 1000, 843]) | 1\n",
-      "3945     | torch.Size([1, 1000, 777]) | 1\n",
-      "3946     | torch.Size([1, 1000, 754]) | 1\n",
-      "3947     | torch.Size([1, 1000, 802]) | 1\n",
-      "3948     | torch.Size([1, 1000, 762]) | 1\n",
-      "3949     | torch.Size([1, 1000, 779]) | 1\n",
-      "3950     | torch.Size([1, 1000, 754]) | 1\n",
-      "3951     | torch.Size([1, 1000, 787]) | 1\n",
-      "3952     | torch.Size([1, 1000, 754]) | 1\n",
-      "3953     | torch.Size([1, 1000, 772]) | 1\n",
-      "3954     | torch.Size([1, 1000, 754]) | 1\n",
-      "3955     | torch.Size([1, 1000, 754]) | 1\n",
-      "3956     | torch.Size([1, 1000, 754]) | 1\n",
-      "3957     | torch.Size([1, 1000, 754]) | 1\n",
-      "3958     | torch.Size([1, 1000, 823]) | 1\n",
-      "3959     | torch.Size([1, 1000, 845]) | 1\n",
-      "3960     | torch.Size([1, 1000, 754]) | 1\n",
-      "3961     | torch.Size([1, 1000, 754]) | 1\n",
-      "3962     | torch.Size([1, 1000, 770]) | 1\n",
-      "3963     | torch.Size([1, 1000, 762]) | 1\n",
-      "3964     | torch.Size([1, 1000, 779]) | 1\n",
-      "3965     | torch.Size([1, 1000, 765]) | 1\n",
-      "3966     | torch.Size([1, 1000, 776]) | 1\n",
-      "3967     | torch.Size([1, 1000, 811]) | 1\n",
-      "3968     | torch.Size([1, 1000, 754]) | 1\n",
-      "3969     | torch.Size([1, 1000, 762]) | 1\n",
-      "3970     | torch.Size([1, 1000, 754]) | 1\n",
-      "3971     | torch.Size([1, 1000, 754]) | 1\n",
-      "3972     | torch.Size([1, 1000, 784]) | 1\n",
-      "3973     | torch.Size([1, 1000, 754]) | 1\n",
-      "3974     | torch.Size([1, 1000, 781]) | 1\n",
-      "3975     | torch.Size([1, 1000, 754]) | 1\n",
-      "3976     | torch.Size([1, 1000, 754]) | 1\n",
-      "3977     | torch.Size([1, 1000, 780]) | 1\n",
-      "3978     | torch.Size([1, 1000, 771]) | 1\n",
-      "3979     | torch.Size([1, 1000, 776]) | 1\n",
-      "3980     | torch.Size([1, 1000, 783]) | 1\n",
-      "3981     | torch.Size([1, 1000, 779]) | 1\n",
-      "3982     | torch.Size([1, 1000, 778]) | 1\n",
-      "3983     | torch.Size([1, 1000, 797]) | 1\n",
-      "3984     | torch.Size([1, 1000, 754]) | 1\n",
-      "3985     | torch.Size([1, 1000, 754]) | 1\n",
-      "3986     | torch.Size([1, 1000, 774]) | 1\n",
-      "3987     | torch.Size([1, 1000, 754]) | 1\n",
-      "3988     | torch.Size([1, 1000, 785]) | 1\n",
-      "3989     | torch.Size([1, 1000, 754]) | 1\n",
-      "3990     | torch.Size([1, 1000, 807]) | 1\n",
-      "3991     | torch.Size([1, 1000, 787]) | 1\n",
-      "3992     | torch.Size([1, 1000, 754]) | 1\n",
-      "3993     | torch.Size([1, 1000, 773]) | 1\n",
-      "3994     | torch.Size([1, 1000, 754]) | 1\n",
-      "3995     | torch.Size([1, 1000, 762]) | 1\n",
-      "3996     | torch.Size([1, 1000, 777]) | 1\n",
-      "3997     | torch.Size([1, 1000, 762]) | 1\n",
-      "3998     | torch.Size([1, 1000, 754]) | 1\n",
-      "3999     | torch.Size([1, 1000, 777]) | 1\n",
-      "4000     | torch.Size([1, 1000, 787]) | 1\n",
-      "4001     | torch.Size([1, 1000, 827]) | 1\n",
-      "4002     | torch.Size([1, 1000, 754]) | 1\n",
-      "4003     | torch.Size([1, 1000, 777]) | 1\n",
-      "4004     | torch.Size([1, 1000, 831]) | 1\n",
-      "4005     | torch.Size([1, 1000, 796]) | 1\n",
-      "4006     | torch.Size([1, 1000, 771]) | 1\n",
-      "4007     | torch.Size([1, 1000, 802]) | 1\n",
-      "4008     | torch.Size([1, 1000, 789]) | 1\n",
-      "4009     | torch.Size([1, 1000, 777]) | 1\n",
-      "4010     | torch.Size([1, 1000, 754]) | 1\n",
-      "4011     | torch.Size([1, 1000, 754]) | 1\n",
-      "4012     | torch.Size([1, 1000, 774]) | 1\n",
-      "4013     | torch.Size([1, 1000, 784]) | 1\n",
-      "4014     | torch.Size([1, 1000, 800]) | 1\n",
-      "4015     | torch.Size([1, 1000, 754]) | 1\n",
-      "4016     | torch.Size([1, 1000, 754]) | 1\n",
-      "4017     | torch.Size([1, 1000, 754]) | 1\n",
-      "4018     | torch.Size([1, 1000, 754]) | 1\n",
-      "4019     | torch.Size([1, 1000, 787]) | 1\n",
-      "4020     | torch.Size([1, 1000, 781]) | 1\n",
-      "4021     | torch.Size([1, 1000, 754]) | 1\n",
-      "4022     | torch.Size([1, 1000, 754]) | 1\n",
-      "4023     | torch.Size([1, 1000, 754]) | 1\n",
-      "4024     | torch.Size([1, 1000, 776]) | 1\n",
-      "4025     | torch.Size([1, 1000, 776]) | 1\n",
-      "4026     | torch.Size([1, 1000, 783]) | 1\n",
-      "4027     | torch.Size([1, 1000, 754]) | 1\n",
-      "4028     | torch.Size([1, 1000, 805]) | 1\n",
-      "4029     | torch.Size([1, 1000, 762]) | 1\n",
-      "4030     | torch.Size([1, 1000, 776]) | 1\n",
-      "4031     | torch.Size([1, 1000, 757]) | 1\n",
-      "4032     | torch.Size([1, 1000, 754]) | 1\n",
-      "4033     | torch.Size([1, 1000, 785]) | 1\n",
-      "4034     | torch.Size([1, 1000, 780]) | 1\n",
-      "4035     | torch.Size([1, 1000, 782]) | 1\n",
-      "4036     | torch.Size([1, 1000, 807]) | 1\n",
-      "4037     | torch.Size([1, 1000, 782]) | 1\n",
-      "4038     | torch.Size([1, 1000, 802]) | 1\n",
-      "4039     | torch.Size([1, 1000, 754]) | 1\n",
-      "4040     | torch.Size([1, 1000, 754]) | 1\n",
-      "4041     | torch.Size([1, 1000, 808]) | 1\n",
-      "4042     | torch.Size([1, 1000, 789]) | 1\n",
-      "4043     | torch.Size([1, 1000, 754]) | 1\n",
-      "4044     | torch.Size([1, 1000, 800]) | 1\n",
-      "4045     | torch.Size([1, 1000, 782]) | 1\n",
-      "4046     | torch.Size([1, 1000, 777]) | 1\n",
-      "4047     | torch.Size([1, 1000, 754]) | 1\n",
-      "4048     | torch.Size([1, 1000, 794]) | 1\n",
-      "4049     | torch.Size([1, 1000, 800]) | 1\n",
-      "4050     | torch.Size([1, 1000, 754]) | 1\n",
-      "4051     | torch.Size([1, 1000, 754]) | 1\n",
-      "4052     | torch.Size([1, 1000, 754]) | 1\n",
-      "4053     | torch.Size([1, 1000, 777]) | 1\n",
-      "4054     | torch.Size([1, 1000, 754]) | 1\n",
-      "4055     | torch.Size([1, 1000, 777]) | 1\n",
-      "4056     | torch.Size([1, 1000, 844]) | 1\n",
-      "4057     | torch.Size([1, 1000, 754]) | 1\n",
-      "4058     | torch.Size([1, 1000, 781]) | 1\n",
-      "4059     | torch.Size([1, 1000, 754]) | 1\n",
-      "4060     | torch.Size([1, 1000, 810]) | 1\n",
-      "4061     | torch.Size([1, 1000, 809]) | 1\n",
-      "4062     | torch.Size([1, 1000, 771]) | 1\n",
-      "4063     | torch.Size([1, 1000, 767]) | 1\n",
-      "4064     | torch.Size([1, 1000, 762]) | 1\n",
-      "4065     | torch.Size([1, 1000, 762]) | 1\n",
-      "4066     | torch.Size([1, 1000, 785]) | 1\n",
-      "4067     | torch.Size([1, 1000, 770]) | 1\n",
-      "4068     | torch.Size([1, 1000, 754]) | 1\n",
-      "4069     | torch.Size([1, 1000, 754]) | 1\n",
-      "4070     | torch.Size([1, 1000, 754]) | 1\n",
-      "4071     | torch.Size([1, 1000, 761]) | 1\n",
-      "4072     | torch.Size([1, 1000, 754]) | 1\n",
-      "4073     | torch.Size([1, 1000, 778]) | 1\n",
-      "4074     | torch.Size([1, 1000, 786]) | 1\n",
-      "4075     | torch.Size([1, 1000, 754]) | 1\n",
-      "4076     | torch.Size([1, 1000, 754]) | 1\n",
-      "4077     | torch.Size([1, 1000, 775]) | 1\n",
-      "4078     | torch.Size([1, 1000, 754]) | 1\n",
-      "4079     | torch.Size([1, 1000, 766]) | 1\n",
-      "4080     | torch.Size([1, 1000, 776]) | 1\n",
-      "4081     | torch.Size([1, 1000, 781]) | 1\n",
-      "4082     | torch.Size([1, 1000, 768]) | 1\n",
-      "4083     | torch.Size([1, 1000, 787]) | 1\n",
-      "4084     | torch.Size([1, 1000, 769]) | 1\n",
-      "4085     | torch.Size([1, 1000, 764]) | 1\n",
-      "4086     | torch.Size([1, 1000, 754]) | 1\n",
-      "4087     | torch.Size([1, 1000, 754]) | 1\n",
-      "4088     | torch.Size([1, 1000, 793]) | 1\n",
-      "4089     | torch.Size([1, 1000, 754]) | 1\n",
-      "4090     | torch.Size([1, 1000, 754]) | 1\n",
-      "4091     | torch.Size([1, 1000, 781]) | 1\n",
-      "4092     | torch.Size([1, 1000, 799]) | 1\n",
-      "4093     | torch.Size([1, 1000, 754]) | 1\n",
-      "4094     | torch.Size([1, 1000, 802]) | 1\n",
-      "4095     | torch.Size([1, 1000, 795]) | 1\n",
-      "4096     | torch.Size([1, 1000, 802]) | 1\n",
-      "4097     | torch.Size([1, 1000, 777]) | 1\n",
-      "4098     | torch.Size([1, 1000, 754]) | 1\n",
-      "4099     | torch.Size([1, 1000, 754]) | 1\n",
-      "4100     | torch.Size([1, 1000, 762]) | 1\n",
-      "4101     | torch.Size([1, 1000, 762]) | 1\n",
-      "4102     | torch.Size([1, 1000, 782]) | 1\n",
-      "4103     | torch.Size([1, 1000, 808]) | 1\n",
-      "4104     | torch.Size([1, 1000, 782]) | 1\n",
-      "4105     | torch.Size([1, 1000, 778]) | 1\n",
-      "4106     | torch.Size([1, 1000, 754]) | 1\n",
-      "4107     | torch.Size([1, 1000, 802]) | 1\n",
-      "4108     | torch.Size([1, 1000, 763]) | 1\n",
-      "4109     | torch.Size([1, 1000, 802]) | 1\n",
-      "4110     | torch.Size([1, 1000, 777]) | 1\n",
-      "4111     | torch.Size([1, 1000, 776]) | 1\n",
-      "4112     | torch.Size([1, 1000, 777]) | 1\n",
-      "4113     | torch.Size([1, 1000, 754]) | 1\n",
-      "4114     | torch.Size([1, 1000, 754]) | 1\n",
-      "4115     | torch.Size([1, 1000, 754]) | 1\n",
-      "4116     | torch.Size([1, 1000, 802]) | 1\n",
-      "4117     | torch.Size([1, 1000, 754]) | 1\n",
-      "4118     | torch.Size([1, 1000, 779]) | 1\n",
-      "4119     | torch.Size([1, 1000, 771]) | 1\n",
-      "4120     | torch.Size([1, 1000, 778]) | 1\n",
-      "4121     | torch.Size([1, 1000, 777]) | 1\n",
-      "4122     | torch.Size([1, 1000, 771]) | 1\n",
-      "4123     | torch.Size([1, 1000, 765]) | 1\n",
-      "4124     | torch.Size([1, 1000, 777]) | 1\n",
-      "4125     | torch.Size([1, 1000, 775]) | 1\n",
-      "4126     | torch.Size([1, 1000, 781]) | 1\n",
-      "4127     | torch.Size([1, 1000, 793]) | 1\n",
-      "4128     | torch.Size([1, 1000, 754]) | 1\n",
-      "4129     | torch.Size([1, 1000, 777]) | 1\n",
-      "4130     | torch.Size([1, 1000, 792]) | 1\n",
-      "4131     | torch.Size([1, 1000, 754]) | 1\n",
-      "4132     | torch.Size([1, 1000, 754]) | 1\n",
-      "4133     | torch.Size([1, 1000, 754]) | 1\n",
-      "4134     | torch.Size([1, 1000, 762]) | 1\n",
-      "4135     | torch.Size([1, 1000, 754]) | 1\n",
-      "4136     | torch.Size([1, 1000, 754]) | 1\n",
-      "4137     | torch.Size([1, 1000, 754]) | 1\n",
-      "4138     | torch.Size([1, 1000, 806]) | 1\n",
-      "4139     | torch.Size([1, 1000, 797]) | 1\n",
-      "4140     | torch.Size([1, 1000, 754]) | 1\n",
-      "4141     | torch.Size([1, 1000, 754]) | 1\n",
-      "4142     | torch.Size([1, 1000, 754]) | 1\n",
-      "4143     | torch.Size([1, 1000, 726]) | 1\n",
-      "4144     | torch.Size([1, 1000, 754]) | 1\n",
-      "4145     | torch.Size([1, 1000, 766]) | 1\n",
-      "4146     | torch.Size([1, 1000, 754]) | 1\n",
-      "4147     | torch.Size([1, 1000, 771]) | 1\n",
-      "4148     | torch.Size([1, 1000, 781]) | 1\n",
-      "4149     | torch.Size([1, 1000, 770]) | 1\n",
-      "4150     | torch.Size([1, 1000, 782]) | 1\n",
-      "4151     | torch.Size([1, 1000, 754]) | 1\n",
-      "4152     | torch.Size([1, 1000, 754]) | 1\n",
-      "4153     | torch.Size([1, 1000, 835]) | 1\n",
-      "4154     | torch.Size([1, 1000, 785]) | 1\n",
-      "4155     | torch.Size([1, 1000, 754]) | 1\n",
-      "4156     | torch.Size([1, 1000, 754]) | 1\n",
-      "4157     | torch.Size([1, 1000, 781]) | 1\n",
-      "4158     | torch.Size([1, 1000, 754]) | 1\n",
-      "4159     | torch.Size([1, 1000, 777]) | 1\n",
-      "4160     | torch.Size([1, 1000, 754]) | 1\n",
-      "4161     | torch.Size([1, 1000, 777]) | 1\n",
-      "4162     | torch.Size([1, 1000, 772]) | 1\n",
-      "4163     | torch.Size([1, 1000, 754]) | 1\n",
-      "4164     | torch.Size([1, 1000, 787]) | 1\n",
-      "4165     | torch.Size([1, 1000, 754]) | 1\n",
-      "4166     | torch.Size([1, 1000, 781]) | 1\n",
-      "4167     | torch.Size([1, 1000, 754]) | 1\n",
-      "4168     | torch.Size([1, 1000, 754]) | 1\n",
-      "4169     | torch.Size([1, 1000, 754]) | 1\n",
-      "4170     | torch.Size([1, 1000, 782]) | 1\n",
-      "4171     | torch.Size([1, 1000, 782]) | 1\n",
-      "4172     | torch.Size([1, 1000, 754]) | 1\n",
-      "4173     | torch.Size([1, 1000, 841]) | 1\n",
-      "4174     | torch.Size([1, 1000, 754]) | 1\n",
-      "4175     | torch.Size([1, 1000, 778]) | 1\n",
-      "4176     | torch.Size([1, 1000, 754]) | 1\n",
-      "4177     | torch.Size([1, 1000, 754]) | 1\n",
-      "4178     | torch.Size([1, 1000, 782]) | 1\n",
-      "4179     | torch.Size([1, 1000, 754]) | 1\n",
-      "4180     | torch.Size([1, 1000, 784]) | 1\n",
-      "4181     | torch.Size([1, 1000, 782]) | 1\n",
-      "4182     | torch.Size([1, 1000, 725]) | 1\n",
-      "4183     | torch.Size([1, 1000, 754]) | 1\n",
-      "4184     | torch.Size([1, 1000, 621]) | 1\n",
-      "4185     | torch.Size([1, 1000, 754]) | 1\n",
-      "4186     | torch.Size([1, 1000, 754]) | 1\n",
-      "4187     | torch.Size([1, 1000, 783]) | 1\n",
-      "4188     | torch.Size([1, 1000, 766]) | 1\n",
-      "4189     | torch.Size([1, 1000, 777]) | 1\n",
-      "4190     | torch.Size([1, 1000, 754]) | 1\n",
-      "4191     | torch.Size([1, 1000, 754]) | 1\n",
-      "4192     | torch.Size([1, 1000, 754]) | 1\n",
-      "4193     | torch.Size([1, 1000, 784]) | 1\n",
-      "4194     | torch.Size([1, 1000, 754]) | 1\n",
-      "4195     | torch.Size([1, 1000, 782]) | 1\n",
-      "4196     | torch.Size([1, 1000, 780]) | 1\n",
-      "4197     | torch.Size([1, 1000, 802]) | 1\n",
-      "4198     | torch.Size([1, 1000, 776]) | 1\n",
-      "4199     | torch.Size([1, 1000, 778]) | 1\n",
-      "4200     | torch.Size([1, 1000, 777]) | 1\n",
-      "4201     | torch.Size([1, 1000, 754]) | 1\n",
-      "4202     | torch.Size([1, 1000, 783]) | 1\n",
-      "4203     | torch.Size([1, 1000, 792]) | 1\n",
-      "4204     | torch.Size([1, 1000, 777]) | 1\n",
-      "4205     | torch.Size([1, 1000, 754]) | 1\n",
-      "4206     | torch.Size([1, 1000, 754]) | 1\n",
-      "4207     | torch.Size([1, 1000, 845]) | 1\n",
-      "4208     | torch.Size([1, 1000, 810]) | 1\n",
-      "4209     | torch.Size([1, 1000, 781]) | 1\n",
-      "4210     | torch.Size([1, 1000, 754]) | 1\n",
-      "4211     | torch.Size([1, 1000, 762]) | 1\n",
-      "4212     | torch.Size([1, 1000, 799]) | 1\n",
-      "4213     | torch.Size([1, 1000, 754]) | 1\n",
-      "4214     | torch.Size([1, 1000, 777]) | 1\n",
-      "4215     | torch.Size([1, 1000, 754]) | 1\n",
-      "4216     | torch.Size([1, 1000, 754]) | 1\n",
-      "4217     | torch.Size([1, 1000, 811]) | 1\n",
-      "4218     | torch.Size([1, 1000, 777]) | 1\n",
-      "4219     | torch.Size([1, 1000, 782]) | 1\n",
-      "4220     | torch.Size([1, 1000, 783]) | 1\n",
-      "4221     | torch.Size([1, 1000, 779]) | 1\n",
-      "4222     | torch.Size([1, 1000, 777]) | 1\n",
-      "4223     | torch.Size([1, 1000, 724]) | 1\n",
-      "4224     | torch.Size([1, 1000, 787]) | 1\n",
-      "4225     | torch.Size([1, 1000, 762]) | 1\n",
-      "4226     | torch.Size([1, 1000, 754]) | 1\n",
-      "4227     | torch.Size([1, 1000, 754]) | 1\n",
-      "4228     | torch.Size([1, 1000, 754]) | 1\n",
-      "4229     | torch.Size([1, 1000, 754]) | 1\n",
-      "4230     | torch.Size([1, 1000, 773]) | 1\n",
-      "4231     | torch.Size([1, 1000, 780]) | 1\n",
-      "4232     | torch.Size([1, 1000, 796]) | 1\n",
-      "4233     | torch.Size([1, 1000, 779]) | 1\n",
-      "4234     | torch.Size([1, 1000, 802]) | 1\n",
-      "4235     | torch.Size([1, 1000, 762]) | 1\n",
-      "4236     | torch.Size([1, 1000, 811]) | 1\n",
-      "4237     | torch.Size([1, 1000, 782]) | 1\n",
-      "4238     | torch.Size([1, 1000, 777]) | 1\n",
-      "4239     | torch.Size([1, 1000, 774]) | 1\n",
-      "4240     | torch.Size([1, 1000, 754]) | 1\n",
-      "4241     | torch.Size([1, 1000, 788]) | 1\n",
-      "4242     | torch.Size([1, 1000, 808]) | 1\n",
-      "4243     | torch.Size([1, 1000, 805]) | 1\n",
-      "4244     | torch.Size([1, 1000, 777]) | 1\n",
-      "4245     | torch.Size([1, 1000, 784]) | 1\n",
-      "4246     | torch.Size([1, 1000, 797]) | 1\n",
-      "4247     | torch.Size([1, 1000, 774]) | 1\n",
-      "4248     | torch.Size([1, 1000, 754]) | 1\n",
-      "4249     | torch.Size([1, 1000, 754]) | 1\n",
-      "4250     | torch.Size([1, 1000, 754]) | 1\n",
-      "4251     | torch.Size([1, 1000, 754]) | 1\n",
-      "4252     | torch.Size([1, 1000, 754]) | 1\n",
-      "4253     | torch.Size([1, 1000, 754]) | 1\n",
-      "4254     | torch.Size([1, 1000, 777]) | 1\n",
-      "4255     | torch.Size([1, 1000, 777]) | 1\n",
-      "4256     | torch.Size([1, 1000, 762]) | 1\n",
-      "4257     | torch.Size([1, 1000, 781]) | 1\n",
-      "4258     | torch.Size([1, 1000, 788]) | 1\n",
-      "4259     | torch.Size([1, 1000, 777]) | 1\n",
-      "4260     | torch.Size([1, 1000, 786]) | 1\n",
-      "4261     | torch.Size([1, 1000, 754]) | 1\n",
-      "4262     | torch.Size([1, 1000, 754]) | 1\n",
-      "4263     | torch.Size([1, 1000, 777]) | 1\n",
-      "4264     | torch.Size([1, 1000, 754]) | 1\n",
-      "4265     | torch.Size([1, 1000, 780]) | 1\n",
-      "4266     | torch.Size([1, 1000, 777]) | 1\n",
-      "4267     | torch.Size([1, 1000, 769]) | 1\n",
-      "4268     | torch.Size([1, 1000, 784]) | 1\n",
-      "4269     | torch.Size([1, 1000, 754]) | 1\n",
-      "4270     | torch.Size([1, 1000, 777]) | 1\n",
-      "4271     | torch.Size([1, 1000, 754]) | 1\n",
-      "4272     | torch.Size([1, 1000, 754]) | 1\n",
-      "4273     | torch.Size([1, 1000, 784]) | 1\n",
-      "4274     | torch.Size([1, 1000, 762]) | 1\n",
-      "4275     | torch.Size([1, 1000, 788]) | 1\n",
-      "4276     | torch.Size([1, 1000, 777]) | 1\n",
-      "4277     | torch.Size([1, 1000, 776]) | 1\n",
-      "4278     | torch.Size([1, 1000, 791]) | 1\n",
-      "4279     | torch.Size([1, 1000, 791]) | 1\n",
-      "4280     | torch.Size([1, 1000, 771]) | 1\n",
-      "4281     | torch.Size([1, 1000, 754]) | 1\n",
-      "4282     | torch.Size([1, 1000, 792]) | 1\n",
-      "4283     | torch.Size([1, 1000, 754]) | 1\n",
-      "4284     | torch.Size([1, 1000, 773]) | 1\n",
-      "4285     | torch.Size([1, 1000, 754]) | 1\n",
-      "4286     | torch.Size([1, 1000, 754]) | 1\n",
-      "4287     | torch.Size([1, 1000, 754]) | 1\n",
-      "4288     | torch.Size([1, 1000, 754]) | 1\n",
-      "4289     | torch.Size([1, 1000, 754]) | 1\n",
-      "4290     | torch.Size([1, 1000, 764]) | 1\n",
-      "4291     | torch.Size([1, 1000, 794]) | 1\n",
-      "4292     | torch.Size([1, 1000, 777]) | 1\n",
-      "4293     | torch.Size([1, 1000, 754]) | 1\n",
-      "4294     | torch.Size([1, 1000, 777]) | 1\n",
-      "4295     | torch.Size([1, 1000, 725]) | 1\n",
-      "4296     | torch.Size([1, 1000, 777]) | 1\n",
-      "4297     | torch.Size([1, 1000, 796]) | 1\n",
-      "4298     | torch.Size([1, 1000, 754]) | 1\n",
-      "4299     | torch.Size([1, 1000, 754]) | 1\n",
-      "4300     | torch.Size([1, 1000, 754]) | 1\n",
-      "4301     | torch.Size([1, 1000, 754]) | 1\n",
-      "4302     | torch.Size([1, 1000, 754]) | 1\n",
-      "4303     | torch.Size([1, 1000, 754]) | 1\n",
-      "4304     | torch.Size([1, 1000, 754]) | 1\n",
-      "4305     | torch.Size([1, 1000, 788]) | 1\n",
-      "4306     | torch.Size([1, 1000, 779]) | 1\n",
-      "4307     | torch.Size([1, 1000, 780]) | 1\n",
-      "4308     | torch.Size([1, 1000, 754]) | 1\n",
-      "4309     | torch.Size([1, 1000, 762]) | 1\n",
-      "4310     | torch.Size([1, 1000, 761]) | 1\n",
-      "4311     | torch.Size([1, 1000, 778]) | 1\n",
-      "4312     | torch.Size([1, 1000, 754]) | 1\n",
-      "4313     | torch.Size([1, 1000, 796]) | 1\n",
-      "4314     | torch.Size([1, 1000, 754]) | 1\n",
-      "4315     | torch.Size([1, 1000, 754]) | 1\n",
-      "4316     | torch.Size([1, 1000, 780]) | 1\n",
-      "4317     | torch.Size([1, 1000, 771]) | 1\n",
-      "4318     | torch.Size([1, 1000, 782]) | 1\n",
-      "4319     | torch.Size([1, 1000, 754]) | 1\n",
-      "4320     | torch.Size([1, 1000, 754]) | 1\n",
-      "4321     | torch.Size([1, 1000, 770]) | 1\n",
-      "4322     | torch.Size([1, 1000, 754]) | 1\n",
-      "4323     | torch.Size([1, 1000, 754]) | 1\n",
-      "4324     | torch.Size([1, 1000, 772]) | 1\n",
-      "4325     | torch.Size([1, 1000, 805]) | 1\n",
-      "4326     | torch.Size([1, 1000, 779]) | 1\n",
-      "4327     | torch.Size([1, 1000, 762]) | 1\n",
-      "4328     | torch.Size([1, 1000, 797]) | 1\n",
-      "4329     | torch.Size([1, 1000, 754]) | 1\n",
-      "4330     | torch.Size([1, 1000, 798]) | 1\n",
-      "4331     | torch.Size([1, 1000, 783]) | 1\n",
-      "4332     | torch.Size([1, 1000, 776]) | 1\n",
-      "4333     | torch.Size([1, 1000, 777]) | 1\n",
-      "4334     | torch.Size([1, 1000, 775]) | 1\n",
-      "4335     | torch.Size([1, 1000, 782]) | 1\n",
-      "4336     | torch.Size([1, 1000, 850]) | 1\n",
-      "4337     | torch.Size([1, 1000, 845]) | 1\n",
-      "4338     | torch.Size([1, 1000, 780]) | 1\n",
-      "4339     | torch.Size([1, 1000, 754]) | 1\n",
-      "4340     | torch.Size([1, 1000, 750]) | 1\n",
-      "4341     | torch.Size([1, 1000, 771]) | 1\n",
-      "4342     | torch.Size([1, 1000, 802]) | 1\n",
-      "4343     | torch.Size([1, 1000, 754]) | 1\n",
-      "4344     | torch.Size([1, 1000, 799]) | 1\n",
-      "4345     | torch.Size([1, 1000, 754]) | 1\n",
-      "4346     | torch.Size([1, 1000, 754]) | 1\n",
-      "4347     | torch.Size([1, 1000, 754]) | 1\n",
-      "4348     | torch.Size([1, 1000, 781]) | 1\n",
-      "4349     | torch.Size([1, 1000, 773]) | 1\n",
-      "4350     | torch.Size([1, 1000, 753]) | 1\n",
-      "4351     | torch.Size([1, 1000, 754]) | 1\n",
-      "4352     | torch.Size([1, 1000, 780]) | 1\n",
-      "4353     | torch.Size([1, 1000, 796]) | 1\n",
-      "4354     | torch.Size([1, 1000, 762]) | 1\n",
-      "4355     | torch.Size([1, 1000, 784]) | 1\n",
-      "4356     | torch.Size([1, 1000, 802]) | 1\n",
-      "4357     | torch.Size([1, 1000, 754]) | 1\n",
-      "4358     | torch.Size([1, 1000, 754]) | 1\n",
-      "4359     | torch.Size([1, 1000, 754]) | 1\n",
-      "4360     | torch.Size([1, 1000, 754]) | 1\n",
-      "4361     | torch.Size([1, 1000, 771]) | 1\n",
-      "4362     | torch.Size([1, 1000, 754]) | 1\n",
-      "4363     | torch.Size([1, 1000, 754]) | 1\n",
-      "4364     | torch.Size([1, 1000, 780]) | 1\n",
-      "4365     | torch.Size([1, 1000, 776]) | 1\n",
-      "4366     | torch.Size([1, 1000, 793]) | 1\n",
-      "4367     | torch.Size([1, 1000, 754]) | 1\n",
-      "4368     | torch.Size([1, 1000, 754]) | 1\n",
-      "4369     | torch.Size([1, 1000, 754]) | 1\n",
-      "4370     | torch.Size([1, 1000, 761]) | 1\n",
-      "4371     | torch.Size([1, 1000, 774]) | 1\n",
-      "4372     | torch.Size([1, 1000, 777]) | 1\n",
-      "4373     | torch.Size([1, 1000, 766]) | 1\n",
-      "4374     | torch.Size([1, 1000, 784]) | 1\n",
-      "4375     | torch.Size([1, 1000, 754]) | 1\n",
-      "4376     | torch.Size([1, 1000, 754]) | 1\n",
-      "4377     | torch.Size([1, 1000, 796]) | 1\n",
-      "4378     | torch.Size([1, 1000, 762]) | 1\n",
-      "4379     | torch.Size([1, 1000, 777]) | 1\n",
-      "4380     | torch.Size([1, 1000, 762]) | 1\n",
-      "4381     | torch.Size([1, 1000, 777]) | 1\n",
-      "4382     | torch.Size([1, 1000, 787]) | 1\n",
-      "4383     | torch.Size([1, 1000, 754]) | 1\n",
-      "4384     | torch.Size([1, 1000, 768]) | 1\n",
-      "4385     | torch.Size([1, 1000, 843]) | 1\n",
-      "4386     | torch.Size([1, 1000, 806]) | 1\n",
-      "4387     | torch.Size([1, 1000, 787]) | 1\n",
-      "4388     | torch.Size([1, 1000, 818]) | 1\n",
-      "4389     | torch.Size([1, 1000, 762]) | 1\n",
-      "4390     | torch.Size([1, 1000, 776]) | 1\n",
-      "4391     | torch.Size([1, 1000, 847]) | 1\n",
-      "4392     | torch.Size([1, 1000, 781]) | 1\n",
-      "4393     | torch.Size([1, 1000, 754]) | 1\n",
-      "4394     | torch.Size([1, 1000, 754]) | 1\n",
-      "4395     | torch.Size([1, 1000, 828]) | 1\n",
-      "4396     | torch.Size([1, 1000, 799]) | 1\n",
-      "4397     | torch.Size([1, 1000, 754]) | 1\n",
-      "4398     | torch.Size([1, 1000, 777]) | 1\n",
-      "4399     | torch.Size([1, 1000, 762]) | 1\n",
-      "4400     | torch.Size([1, 1000, 754]) | 1\n",
-      "4401     | torch.Size([1, 1000, 777]) | 1\n",
-      "4402     | torch.Size([1, 1000, 754]) | 1\n",
-      "4403     | torch.Size([1, 1000, 762]) | 1\n",
-      "4404     | torch.Size([1, 1000, 767]) | 1\n",
-      "4405     | torch.Size([1, 1000, 782]) | 1\n",
-      "4406     | torch.Size([1, 1000, 754]) | 1\n",
-      "4407     | torch.Size([1, 1000, 622]) | 1\n",
-      "4408     | torch.Size([1, 1000, 773]) | 1\n",
-      "4409     | torch.Size([1, 1000, 802]) | 1\n",
-      "4410     | torch.Size([1, 1000, 793]) | 1\n",
-      "4411     | torch.Size([1, 1000, 777]) | 1\n",
-      "4412     | torch.Size([1, 1000, 754]) | 1\n",
-      "4413     | torch.Size([1, 1000, 754]) | 1\n",
-      "4414     | torch.Size([1, 1000, 777]) | 1\n",
-      "4415     | torch.Size([1, 1000, 765]) | 1\n",
-      "4416     | torch.Size([1, 1000, 784]) | 1\n",
-      "4417     | torch.Size([1, 1000, 782]) | 1\n",
-      "4418     | torch.Size([1, 1000, 762]) | 1\n",
-      "4419     | torch.Size([1, 1000, 781]) | 1\n",
-      "4420     | torch.Size([1, 1000, 776]) | 1\n",
-      "4421     | torch.Size([1, 1000, 785]) | 1\n",
-      "4422     | torch.Size([1, 1000, 754]) | 1\n",
-      "4423     | torch.Size([1, 1000, 810]) | 1\n",
-      "4424     | torch.Size([1, 1000, 754]) | 1\n",
-      "4425     | torch.Size([1, 1000, 848]) | 1\n",
-      "4426     | torch.Size([1, 1000, 786]) | 1\n",
-      "4427     | torch.Size([1, 1000, 787]) | 1\n",
-      "4428     | torch.Size([1, 1000, 774]) | 1\n",
-      "4429     | torch.Size([1, 1000, 822]) | 1\n",
-      "4430     | torch.Size([1, 1000, 782]) | 1\n",
-      "4431     | torch.Size([1, 1000, 754]) | 1\n",
-      "4432     | torch.Size([1, 1000, 786]) | 1\n",
-      "4433     | torch.Size([1, 1000, 771]) | 1\n",
-      "4434     | torch.Size([1, 1000, 769]) | 1\n",
-      "4435     | torch.Size([1, 1000, 754]) | 1\n",
-      "4436     | torch.Size([1, 1000, 754]) | 1\n",
-      "4437     | torch.Size([1, 1000, 771]) | 1\n",
-      "4438     | torch.Size([1, 1000, 761]) | 1\n",
-      "4439     | torch.Size([1, 1000, 777]) | 1\n",
-      "4440     | torch.Size([1, 1000, 754]) | 1\n",
-      "4441     | torch.Size([1, 1000, 782]) | 1\n",
-      "4442     | torch.Size([1, 1000, 797]) | 1\n",
-      "4443     | torch.Size([1, 1000, 771]) | 1\n",
-      "4444     | torch.Size([1, 1000, 753]) | 1\n",
-      "4445     | torch.Size([1, 1000, 754]) | 1\n",
-      "4446     | torch.Size([1, 1000, 754]) | 1\n",
-      "4447     | torch.Size([1, 1000, 754]) | 1\n",
-      "4448     | torch.Size([1, 1000, 762]) | 1\n",
-      "4449     | torch.Size([1, 1000, 754]) | 1\n",
-      "4450     | torch.Size([1, 1000, 772]) | 1\n",
-      "4451     | torch.Size([1, 1000, 754]) | 1\n",
-      "4452     | torch.Size([1, 1000, 754]) | 1\n",
-      "4453     | torch.Size([1, 1000, 802]) | 1\n",
-      "4454     | torch.Size([1, 1000, 754]) | 1\n",
-      "4455     | torch.Size([1, 1000, 777]) | 1\n",
-      "4456     | torch.Size([1, 1000, 782]) | 1\n",
-      "4457     | torch.Size([1, 1000, 767]) | 1\n",
-      "4458     | torch.Size([1, 1000, 786]) | 1\n",
-      "4459     | torch.Size([1, 1000, 780]) | 1\n",
-      "4460     | torch.Size([1, 1000, 754]) | 1\n",
-      "4461     | torch.Size([1, 1000, 754]) | 1\n",
-      "4462     | torch.Size([1, 1000, 754]) | 1\n",
-      "4463     | torch.Size([1, 1000, 754]) | 1\n",
-      "4464     | torch.Size([1, 1000, 777]) | 1\n",
-      "4465     | torch.Size([1, 1000, 754]) | 1\n",
-      "4466     | torch.Size([1, 1000, 802]) | 1\n",
-      "4467     | torch.Size([1, 1000, 754]) | 1\n",
-      "4468     | torch.Size([1, 1000, 782]) | 1\n",
-      "4469     | torch.Size([1, 1000, 754]) | 1\n",
-      "4470     | torch.Size([1, 1000, 754]) | 1\n",
-      "4471     | torch.Size([1, 1000, 783]) | 1\n",
-      "4472     | torch.Size([1, 1000, 754]) | 1\n",
-      "4473     | torch.Size([1, 1000, 778]) | 1\n",
-      "4474     | torch.Size([1, 1000, 754]) | 1\n",
-      "4475     | torch.Size([1, 1000, 777]) | 1\n",
-      "4476     | torch.Size([1, 1000, 783]) | 1\n",
-      "4477     | torch.Size([1, 1000, 754]) | 1\n",
-      "4478     | torch.Size([1, 1000, 797]) | 1\n",
-      "4479     | torch.Size([1, 1000, 798]) | 1\n",
-      "4480     | torch.Size([1, 1000, 754]) | 1\n",
-      "4481     | torch.Size([1, 1000, 771]) | 1\n",
-      "4482     | torch.Size([1, 1000, 776]) | 1\n",
-      "4483     | torch.Size([1, 1000, 777]) | 1\n",
-      "4484     | torch.Size([1, 1000, 782]) | 1\n",
-      "4485     | torch.Size([1, 1000, 785]) | 1\n",
-      "4486     | torch.Size([1, 1000, 780]) | 1\n",
-      "4487     | torch.Size([1, 1000, 754]) | 1\n",
-      "4488     | torch.Size([1, 1000, 783]) | 1\n",
-      "4489     | torch.Size([1, 1000, 754]) | 1\n",
-      "4490     | torch.Size([1, 1000, 788]) | 1\n",
-      "4491     | torch.Size([1, 1000, 754]) | 1\n",
-      "4492     | torch.Size([1, 1000, 788]) | 1\n",
-      "4493     | torch.Size([1, 1000, 754]) | 1\n",
-      "4494     | torch.Size([1, 1000, 754]) | 1\n",
-      "4495     | torch.Size([1, 1000, 754]) | 1\n",
-      "4496     | torch.Size([1, 1000, 768]) | 1\n",
-      "4497     | torch.Size([1, 1000, 762]) | 1\n",
-      "4498     | torch.Size([1, 1000, 754]) | 1\n",
-      "4499     | torch.Size([1, 1000, 806]) | 1\n",
-      "4500     | torch.Size([1, 1000, 795]) | 1\n",
-      "4501     | torch.Size([1, 1000, 777]) | 1\n",
-      "4502     | torch.Size([1, 1000, 807]) | 1\n",
-      "4503     | torch.Size([1, 1000, 841]) | 1\n",
-      "4504     | torch.Size([1, 1000, 754]) | 1\n",
-      "4505     | torch.Size([1, 1000, 776]) | 1\n",
-      "4506     | torch.Size([1, 1000, 754]) | 1\n",
-      "4507     | torch.Size([1, 1000, 777]) | 1\n",
-      "4508     | torch.Size([1, 1000, 754]) | 1\n",
-      "4509     | torch.Size([1, 1000, 774]) | 1\n",
-      "4510     | torch.Size([1, 1000, 754]) | 1\n",
-      "4511     | torch.Size([1, 1000, 754]) | 1\n",
-      "4512     | torch.Size([1, 1000, 798]) | 1\n",
-      "4513     | torch.Size([1, 1000, 777]) | 1\n",
-      "4514     | torch.Size([1, 1000, 762]) | 1\n",
-      "4515     | torch.Size([1, 1000, 754]) | 1\n",
-      "4516     | torch.Size([1, 1000, 762]) | 1\n",
-      "4517     | torch.Size([1, 1000, 754]) | 1\n",
-      "4518     | torch.Size([1, 1000, 754]) | 1\n",
-      "4519     | torch.Size([1, 1000, 782]) | 1\n",
-      "4520     | torch.Size([1, 1000, 754]) | 1\n",
-      "4521     | torch.Size([1, 1000, 786]) | 1\n",
-      "4522     | torch.Size([1, 1000, 771]) | 1\n",
-      "4523     | torch.Size([1, 1000, 785]) | 1\n",
-      "4524     | torch.Size([1, 1000, 777]) | 1\n",
-      "4525     | torch.Size([1, 1000, 754]) | 1\n",
-      "4526     | torch.Size([1, 1000, 808]) | 1\n",
-      "4527     | torch.Size([1, 1000, 777]) | 1\n",
-      "4528     | torch.Size([1, 1000, 768]) | 1\n",
-      "4529     | torch.Size([1, 1000, 754]) | 1\n",
-      "4530     | torch.Size([1, 1000, 778]) | 1\n",
-      "4531     | torch.Size([1, 1000, 767]) | 1\n",
-      "4532     | torch.Size([1, 1000, 781]) | 1\n",
-      "4533     | torch.Size([1, 1000, 778]) | 1\n",
-      "4534     | torch.Size([1, 1000, 777]) | 1\n",
-      "4535     | torch.Size([1, 1000, 754]) | 1\n",
-      "4536     | torch.Size([1, 1000, 808]) | 1\n",
-      "4537     | torch.Size([1, 1000, 754]) | 1\n",
-      "4538     | torch.Size([1, 1000, 787]) | 1\n",
-      "4539     | torch.Size([1, 1000, 754]) | 1\n",
-      "4540     | torch.Size([1, 1000, 754]) | 1\n",
-      "4541     | torch.Size([1, 1000, 754]) | 1\n",
-      "4542     | torch.Size([1, 1000, 754]) | 1\n",
-      "4543     | torch.Size([1, 1000, 754]) | 1\n",
-      "4544     | torch.Size([1, 1000, 776]) | 1\n",
-      "4545     | torch.Size([1, 1000, 762]) | 1\n",
-      "4546     | torch.Size([1, 1000, 766]) | 1\n",
-      "4547     | torch.Size([1, 1000, 754]) | 1\n",
-      "4548     | torch.Size([1, 1000, 754]) | 1\n",
-      "4549     | torch.Size([1, 1000, 784]) | 1\n",
-      "4550     | torch.Size([1, 1000, 754]) | 1\n",
-      "4551     | torch.Size([1, 1000, 754]) | 1\n",
-      "4552     | torch.Size([1, 1000, 754]) | 1\n",
-      "4553     | torch.Size([1, 1000, 781]) | 1\n",
-      "4554     | torch.Size([1, 1000, 777]) | 1\n",
-      "4555     | torch.Size([1, 1000, 777]) | 1\n",
-      "4556     | torch.Size([1, 1000, 754]) | 1\n",
-      "4557     | torch.Size([1, 1000, 773]) | 1\n",
-      "4558     | torch.Size([1, 1000, 785]) | 1\n",
-      "4559     | torch.Size([1, 1000, 777]) | 1\n",
-      "4560     | torch.Size([1, 1000, 762]) | 1\n",
-      "4561     | torch.Size([1, 1000, 754]) | 1\n",
-      "4562     | torch.Size([1, 1000, 802]) | 1\n",
-      "4563     | torch.Size([1, 1000, 772]) | 1\n",
-      "4564     | torch.Size([1, 1000, 783]) | 1\n",
-      "4565     | torch.Size([1, 1000, 754]) | 1\n",
-      "4566     | torch.Size([1, 1000, 754]) | 1\n",
-      "4567     | torch.Size([1, 1000, 792]) | 1\n",
-      "4568     | torch.Size([1, 1000, 754]) | 1\n",
-      "4569     | torch.Size([1, 1000, 754]) | 1\n",
-      "4570     | torch.Size([1, 1000, 789]) | 1\n",
-      "4571     | torch.Size([1, 1000, 776]) | 1\n",
-      "4572     | torch.Size([1, 1000, 754]) | 1\n",
-      "4573     | torch.Size([1, 1000, 777]) | 1\n",
-      "4574     | torch.Size([1, 1000, 783]) | 1\n",
-      "4575     | torch.Size([1, 1000, 777]) | 1\n",
-      "4576     | torch.Size([1, 1000, 777]) | 1\n",
-      "4577     | torch.Size([1, 1000, 775]) | 1\n",
-      "4578     | torch.Size([1, 1000, 777]) | 1\n",
-      "4579     | torch.Size([1, 1000, 754]) | 1\n",
-      "4580     | torch.Size([1, 1000, 754]) | 1\n",
-      "4581     | torch.Size([1, 1000, 847]) | 1\n",
-      "4582     | torch.Size([1, 1000, 777]) | 1\n",
-      "4583     | torch.Size([1, 1000, 754]) | 1\n",
-      "4584     | torch.Size([1, 1000, 787]) | 1\n",
-      "4585     | torch.Size([1, 1000, 777]) | 1\n",
-      "4586     | torch.Size([1, 1000, 754]) | 1\n",
-      "4587     | torch.Size([1, 1000, 754]) | 1\n",
-      "4588     | torch.Size([1, 1000, 797]) | 1\n",
-      "4589     | torch.Size([1, 1000, 780]) | 1\n",
-      "4590     | torch.Size([1, 1000, 772]) | 1\n",
-      "4591     | torch.Size([1, 1000, 802]) | 1\n",
-      "4592     | torch.Size([1, 1000, 754]) | 1\n",
-      "4593     | torch.Size([1, 1000, 754]) | 1\n",
-      "4594     | torch.Size([1, 1000, 754]) | 1\n",
-      "4595     | torch.Size([1, 1000, 754]) | 1\n",
-      "4596     | torch.Size([1, 1000, 772]) | 1\n",
-      "4597     | torch.Size([1, 1000, 780]) | 1\n",
-      "4598     | torch.Size([1, 1000, 845]) | 1\n",
-      "4599     | torch.Size([1, 1000, 754]) | 1\n",
-      "4600     | torch.Size([1, 1000, 762]) | 1\n",
-      "4601     | torch.Size([1, 1000, 754]) | 1\n",
-      "4602     | torch.Size([1, 1000, 754]) | 1\n",
-      "4603     | torch.Size([1, 1000, 754]) | 1\n",
-      "4604     | torch.Size([1, 1000, 844]) | 1\n",
-      "4605     | torch.Size([1, 1000, 783]) | 1\n",
-      "4606     | torch.Size([1, 1000, 780]) | 1\n",
-      "4607     | torch.Size([1, 1000, 725]) | 1\n",
-      "4608     | torch.Size([1, 1000, 774]) | 1\n",
-      "4609     | torch.Size([1, 1000, 798]) | 1\n",
-      "4610     | torch.Size([1, 1000, 781]) | 1\n",
-      "4611     | torch.Size([1, 1000, 754]) | 1\n",
-      "4612     | torch.Size([1, 1000, 754]) | 1\n",
-      "4613     | torch.Size([1, 1000, 772]) | 1\n",
-      "4614     | torch.Size([1, 1000, 771]) | 1\n",
-      "4615     | torch.Size([1, 1000, 754]) | 1\n",
-      "4616     | torch.Size([1, 1000, 780]) | 1\n",
-      "4617     | torch.Size([1, 1000, 754]) | 1\n",
-      "4618     | torch.Size([1, 1000, 796]) | 1\n",
-      "4619     | torch.Size([1, 1000, 754]) | 1\n",
-      "4620     | torch.Size([1, 1000, 754]) | 1\n",
-      "4621     | torch.Size([1, 1000, 754]) | 1\n",
-      "4622     | torch.Size([1, 1000, 762]) | 1\n",
-      "4623     | torch.Size([1, 1000, 754]) | 1\n",
-      "4624     | torch.Size([1, 1000, 771]) | 1\n",
-      "4625     | torch.Size([1, 1000, 786]) | 1\n",
-      "4626     | torch.Size([1, 1000, 789]) | 1\n",
-      "4627     | torch.Size([1, 1000, 777]) | 1\n",
-      "4628     | torch.Size([1, 1000, 754]) | 1\n",
-      "4629     | torch.Size([1, 1000, 787]) | 1\n",
-      "4630     | torch.Size([1, 1000, 754]) | 1\n",
-      "4631     | torch.Size([1, 1000, 754]) | 1\n",
-      "4632     | torch.Size([1, 1000, 754]) | 1\n",
-      "4633     | torch.Size([1, 1000, 754]) | 1\n",
-      "4634     | torch.Size([1, 1000, 762]) | 1\n",
-      "4635     | torch.Size([1, 1000, 762]) | 1\n",
-      "4636     | torch.Size([1, 1000, 754]) | 1\n",
-      "4637     | torch.Size([1, 1000, 809]) | 1\n",
-      "4638     | torch.Size([1, 1000, 770]) | 1\n",
-      "4639     | torch.Size([1, 1000, 766]) | 1\n",
-      "4640     | torch.Size([1, 1000, 767]) | 1\n",
-      "4641     | torch.Size([1, 1000, 754]) | 1\n",
-      "4642     | torch.Size([1, 1000, 785]) | 1\n",
-      "4643     | torch.Size([1, 1000, 754]) | 1\n",
-      "4644     | torch.Size([1, 1000, 802]) | 1\n",
-      "4645     | torch.Size([1, 1000, 797]) | 1\n",
-      "4646     | torch.Size([1, 1000, 754]) | 1\n",
-      "4647     | torch.Size([1, 1000, 777]) | 1\n",
-      "4648     | torch.Size([1, 1000, 798]) | 1\n",
-      "4649     | torch.Size([1, 1000, 777]) | 1\n",
-      "4650     | torch.Size([1, 1000, 754]) | 1\n",
-      "4651     | torch.Size([1, 1000, 802]) | 1\n",
-      "4652     | torch.Size([1, 1000, 754]) | 1\n",
-      "4653     | torch.Size([1, 1000, 775]) | 1\n",
-      "4654     | torch.Size([1, 1000, 775]) | 1\n",
-      "4655     | torch.Size([1, 1000, 772]) | 1\n",
-      "4656     | torch.Size([1, 1000, 802]) | 1\n",
-      "4657     | torch.Size([1, 1000, 788]) | 1\n",
-      "4658     | torch.Size([1, 1000, 754]) | 1\n",
-      "4659     | torch.Size([1, 1000, 785]) | 1\n",
-      "4660     | torch.Size([1, 1000, 785]) | 1\n",
-      "4661     | torch.Size([1, 1000, 797]) | 1\n",
-      "4662     | torch.Size([1, 1000, 762]) | 1\n",
-      "4663     | torch.Size([1, 1000, 802]) | 1\n",
-      "4664     | torch.Size([1, 1000, 802]) | 1\n",
-      "4665     | torch.Size([1, 1000, 842]) | 1\n",
-      "4666     | torch.Size([1, 1000, 784]) | 1\n",
-      "4667     | torch.Size([1, 1000, 771]) | 1\n",
-      "4668     | torch.Size([1, 1000, 802]) | 1\n",
-      "4669     | torch.Size([1, 1000, 768]) | 1\n",
-      "4670     | torch.Size([1, 1000, 788]) | 1\n",
-      "4671     | torch.Size([1, 1000, 783]) | 1\n",
-      "4672     | torch.Size([1, 1000, 754]) | 1\n",
-      "4673     | torch.Size([1, 1000, 771]) | 1\n",
-      "4674     | torch.Size([1, 1000, 777]) | 1\n",
-      "4675     | torch.Size([1, 1000, 754]) | 1\n",
-      "4676     | torch.Size([1, 1000, 768]) | 1\n",
-      "4677     | torch.Size([1, 1000, 777]) | 1\n",
-      "4678     | torch.Size([1, 1000, 784]) | 1\n",
-      "4679     | torch.Size([1, 1000, 799]) | 1\n",
-      "4680     | torch.Size([1, 1000, 781]) | 1\n",
-      "4681     | torch.Size([1, 1000, 777]) | 1\n",
-      "4682     | torch.Size([1, 1000, 754]) | 1\n",
-      "4683     | torch.Size([1, 1000, 762]) | 1\n",
-      "4684     | torch.Size([1, 1000, 762]) | 1\n",
-      "4685     | torch.Size([1, 1000, 810]) | 1\n",
-      "4686     | torch.Size([1, 1000, 781]) | 1\n",
-      "4687     | torch.Size([1, 1000, 810]) | 1\n",
-      "4688     | torch.Size([1, 1000, 754]) | 1\n",
-      "4689     | torch.Size([1, 1000, 773]) | 1\n",
-      "4690     | torch.Size([1, 1000, 777]) | 1\n",
-      "4691     | torch.Size([1, 1000, 754]) | 1\n",
-      "4692     | torch.Size([1, 1000, 790]) | 1\n",
-      "4693     | torch.Size([1, 1000, 754]) | 1\n",
-      "4694     | torch.Size([1, 1000, 808]) | 1\n",
-      "4695     | torch.Size([1, 1000, 789]) | 1\n",
-      "4696     | torch.Size([1, 1000, 754]) | 1\n",
-      "4697     | torch.Size([1, 1000, 754]) | 1\n",
-      "4698     | torch.Size([1, 1000, 796]) | 1\n",
-      "4699     | torch.Size([1, 1000, 781]) | 1\n",
-      "4700     | torch.Size([1, 1000, 754]) | 1\n",
-      "4701     | torch.Size([1, 1000, 776]) | 1\n",
-      "4702     | torch.Size([1, 1000, 777]) | 1\n",
-      "4703     | torch.Size([1, 1000, 754]) | 1\n",
-      "4704     | torch.Size([1, 1000, 777]) | 1\n",
-      "4705     | torch.Size([1, 1000, 754]) | 1\n",
-      "4706     | torch.Size([1, 1000, 762]) | 1\n",
-      "4707     | torch.Size([1, 1000, 754]) | 1\n",
-      "4708     | torch.Size([1, 1000, 789]) | 1\n",
-      "4709     | torch.Size([1, 1000, 777]) | 1\n",
-      "4710     | torch.Size([1, 1000, 754]) | 1\n",
-      "4711     | torch.Size([1, 1000, 781]) | 1\n",
-      "4712     | torch.Size([1, 1000, 754]) | 1\n",
-      "4713     | torch.Size([1, 1000, 826]) | 1\n",
-      "4714     | torch.Size([1, 1000, 779]) | 1\n",
-      "4715     | torch.Size([1, 1000, 787]) | 1\n",
-      "4716     | torch.Size([1, 1000, 785]) | 1\n",
-      "4717     | torch.Size([1, 1000, 783]) | 1\n",
-      "4718     | torch.Size([1, 1000, 786]) | 1\n",
-      "4719     | torch.Size([1, 1000, 781]) | 1\n",
-      "4720     | torch.Size([1, 1000, 754]) | 1\n",
-      "4721     | torch.Size([1, 1000, 754]) | 1\n",
-      "4722     | torch.Size([1, 1000, 777]) | 1\n",
-      "4723     | torch.Size([1, 1000, 786]) | 1\n",
-      "4724     | torch.Size([1, 1000, 777]) | 1\n",
-      "4725     | torch.Size([1, 1000, 768]) | 1\n",
-      "4726     | torch.Size([1, 1000, 773]) | 1\n",
-      "4727     | torch.Size([1, 1000, 754]) | 1\n",
-      "4728     | torch.Size([1, 1000, 770]) | 1\n",
-      "4729     | torch.Size([1, 1000, 754]) | 1\n",
-      "4730     | torch.Size([1, 1000, 783]) | 1\n",
-      "4731     | torch.Size([1, 1000, 754]) | 1\n",
-      "4732     | torch.Size([1, 1000, 783]) | 1\n",
-      "4733     | torch.Size([1, 1000, 793]) | 1\n",
-      "4734     | torch.Size([1, 1000, 754]) | 1\n",
-      "4735     | torch.Size([1, 1000, 786]) | 1\n",
-      "4736     | torch.Size([1, 1000, 776]) | 1\n",
-      "4737     | torch.Size([1, 1000, 754]) | 1\n",
-      "4738     | torch.Size([1, 1000, 779]) | 1\n",
-      "4739     | torch.Size([1, 1000, 754]) | 1\n",
-      "4740     | torch.Size([1, 1000, 782]) | 1\n",
-      "4741     | torch.Size([1, 1000, 754]) | 1\n",
-      "4742     | torch.Size([1, 1000, 754]) | 1\n",
-      "4743     | torch.Size([1, 1000, 777]) | 1\n",
-      "4744     | torch.Size([1, 1000, 775]) | 1\n",
-      "4745     | torch.Size([1, 1000, 802]) | 1\n",
-      "4746     | torch.Size([1, 1000, 762]) | 1\n",
-      "4747     | torch.Size([1, 1000, 754]) | 1\n",
-      "4748     | torch.Size([1, 1000, 787]) | 1\n",
-      "4749     | torch.Size([1, 1000, 754]) | 1\n",
-      "4750     | torch.Size([1, 1000, 777]) | 1\n",
-      "4751     | torch.Size([1, 1000, 754]) | 1\n",
-      "4752     | torch.Size([1, 1000, 762]) | 1\n",
-      "4753     | torch.Size([1, 1000, 754]) | 1\n",
-      "4754     | torch.Size([1, 1000, 754]) | 1\n",
-      "4755     | torch.Size([1, 1000, 754]) | 1\n",
-      "4756     | torch.Size([1, 1000, 806]) | 1\n",
-      "4757     | torch.Size([1, 1000, 770]) | 1\n",
-      "4758     | torch.Size([1, 1000, 754]) | 1\n",
-      "4759     | torch.Size([1, 1000, 754]) | 1\n",
-      "4760     | torch.Size([1, 1000, 754]) | 1\n",
-      "4761     | torch.Size([1, 1000, 754]) | 1\n",
-      "4762     | torch.Size([1, 1000, 754]) | 1\n",
-      "4763     | torch.Size([1, 1000, 802]) | 1\n",
-      "4764     | torch.Size([1, 1000, 798]) | 1\n",
-      "4765     | torch.Size([1, 1000, 778]) | 1\n",
-      "4766     | torch.Size([1, 1000, 614]) | 1\n",
-      "4767     | torch.Size([1, 1000, 787]) | 1\n",
-      "4768     | torch.Size([1, 1000, 754]) | 1\n",
-      "4769     | torch.Size([1, 1000, 781]) | 1\n",
-      "4770     | torch.Size([1, 1000, 776]) | 1\n",
-      "4771     | torch.Size([1, 1000, 754]) | 1\n",
-      "4772     | torch.Size([1, 1000, 754]) | 1\n",
-      "4773     | torch.Size([1, 1000, 781]) | 1\n",
-      "4774     | torch.Size([1, 1000, 762]) | 1\n",
-      "4775     | torch.Size([1, 1000, 754]) | 1\n",
-      "4776     | torch.Size([1, 1000, 807]) | 1\n",
-      "4777     | torch.Size([1, 1000, 796]) | 1\n",
-      "4778     | torch.Size([1, 1000, 754]) | 1\n",
-      "4779     | torch.Size([1, 1000, 788]) | 1\n",
-      "4780     | torch.Size([1, 1000, 766]) | 1\n",
-      "4781     | torch.Size([1, 1000, 762]) | 1\n",
-      "4782     | torch.Size([1, 1000, 766]) | 1\n",
-      "4783     | torch.Size([1, 1000, 754]) | 1\n",
-      "4784     | torch.Size([1, 1000, 774]) | 1\n",
-      "4785     | torch.Size([1, 1000, 780]) | 1\n",
-      "4786     | torch.Size([1, 1000, 762]) | 1\n",
-      "4787     | torch.Size([1, 1000, 762]) | 1\n",
-      "4788     | torch.Size([1, 1000, 777]) | 1\n",
-      "4789     | torch.Size([1, 1000, 784]) | 1\n",
-      "4790     | torch.Size([1, 1000, 754]) | 1\n",
-      "4791     | torch.Size([1, 1000, 802]) | 1\n",
-      "4792     | torch.Size([1, 1000, 777]) | 1\n",
-      "4793     | torch.Size([1, 1000, 754]) | 1\n",
-      "4794     | torch.Size([1, 1000, 777]) | 1\n",
-      "4795     | torch.Size([1, 1000, 780]) | 1\n",
-      "4796     | torch.Size([1, 1000, 781]) | 1\n",
-      "4797     | torch.Size([1, 1000, 788]) | 1\n",
-      "4798     | torch.Size([1, 1000, 754]) | 1\n",
-      "4799     | torch.Size([1, 1000, 754]) | 1\n",
-      "4800     | torch.Size([1, 1000, 786]) | 1\n",
-      "4801     | torch.Size([1, 1000, 777]) | 1\n",
-      "4802     | torch.Size([1, 1000, 810]) | 1\n",
-      "4803     | torch.Size([1, 1000, 786]) | 1\n",
-      "4804     | torch.Size([1, 1000, 767]) | 1\n",
-      "4805     | torch.Size([1, 1000, 774]) | 1\n",
-      "4806     | torch.Size([1, 1000, 777]) | 1\n",
-      "4807     | torch.Size([1, 1000, 762]) | 1\n",
-      "4808     | torch.Size([1, 1000, 754]) | 1\n",
-      "4809     | torch.Size([1, 1000, 786]) | 1\n",
-      "4810     | torch.Size([1, 1000, 789]) | 1\n",
-      "4811     | torch.Size([1, 1000, 776]) | 1\n",
-      "4812     | torch.Size([1, 1000, 762]) | 1\n",
-      "4813     | torch.Size([1, 1000, 754]) | 1\n",
-      "4814     | torch.Size([1, 1000, 783]) | 1\n",
-      "4815     | torch.Size([1, 1000, 772]) | 1\n",
-      "4816     | torch.Size([1, 1000, 777]) | 1\n",
-      "4817     | torch.Size([1, 1000, 762]) | 1\n",
-      "4818     | torch.Size([1, 1000, 778]) | 1\n",
-      "4819     | torch.Size([1, 1000, 788]) | 1\n",
-      "4820     | torch.Size([1, 1000, 769]) | 1\n",
-      "4821     | torch.Size([1, 1000, 754]) | 1\n",
-      "4822     | torch.Size([1, 1000, 809]) | 1\n",
-      "4823     | torch.Size([1, 1000, 754]) | 1\n",
-      "4824     | torch.Size([1, 1000, 766]) | 1\n",
-      "4825     | torch.Size([1, 1000, 754]) | 1\n",
-      "4826     | torch.Size([1, 1000, 754]) | 1\n",
-      "4827     | torch.Size([1, 1000, 796]) | 1\n",
-      "4828     | torch.Size([1, 1000, 754]) | 1\n",
-      "4829     | torch.Size([1, 1000, 788]) | 1\n",
-      "4830     | torch.Size([1, 1000, 779]) | 1\n",
-      "4831     | torch.Size([1, 1000, 754]) | 1\n",
-      "4832     | torch.Size([1, 1000, 754]) | 1\n",
-      "4833     | torch.Size([1, 1000, 754]) | 1\n",
-      "4834     | torch.Size([1, 1000, 754]) | 1\n",
-      "4835     | torch.Size([1, 1000, 754]) | 1\n",
-      "4836     | torch.Size([1, 1000, 772]) | 1\n",
-      "4837     | torch.Size([1, 1000, 754]) | 1\n",
-      "4838     | torch.Size([1, 1000, 777]) | 1\n",
-      "4839     | torch.Size([1, 1000, 781]) | 1\n",
-      "4840     | torch.Size([1, 1000, 754]) | 1\n",
-      "4841     | torch.Size([1, 1000, 754]) | 1\n",
-      "4842     | torch.Size([1, 1000, 754]) | 1\n",
-      "4843     | torch.Size([1, 1000, 754]) | 1\n",
-      "4844     | torch.Size([1, 1000, 862]) | 1\n",
-      "4845     | torch.Size([1, 1000, 754]) | 1\n",
-      "4846     | torch.Size([1, 1000, 782]) | 1\n",
-      "4847     | torch.Size([1, 1000, 788]) | 1\n",
-      "4848     | torch.Size([1, 1000, 754]) | 1\n",
-      "4849     | torch.Size([1, 1000, 754]) | 1\n",
-      "4850     | torch.Size([1, 1000, 754]) | 1\n",
-      "4851     | torch.Size([1, 1000, 772]) | 1\n",
-      "4852     | torch.Size([1, 1000, 802]) | 1\n",
-      "4853     | torch.Size([1, 1000, 777]) | 1\n",
-      "4854     | torch.Size([1, 1000, 754]) | 1\n",
-      "4855     | torch.Size([1, 1000, 845]) | 1\n",
-      "4856     | torch.Size([1, 1000, 752]) | 1\n",
-      "4857     | torch.Size([1, 1000, 754]) | 1\n",
-      "4858     | torch.Size([1, 1000, 754]) | 1\n",
-      "4859     | torch.Size([1, 1000, 784]) | 1\n",
-      "4860     | torch.Size([1, 1000, 778]) | 1\n",
-      "4861     | torch.Size([1, 1000, 754]) | 1\n",
-      "4862     | torch.Size([1, 1000, 780]) | 1\n",
-      "4863     | torch.Size([1, 1000, 776]) | 1\n",
-      "4864     | torch.Size([1, 1000, 754]) | 1\n",
-      "4865     | torch.Size([1, 1000, 754]) | 1\n",
-      "4866     | torch.Size([1, 1000, 754]) | 1\n",
-      "4867     | torch.Size([1, 1000, 754]) | 1\n",
-      "4868     | torch.Size([1, 1000, 754]) | 1\n",
-      "4869     | torch.Size([1, 1000, 791]) | 1\n",
-      "4870     | torch.Size([1, 1000, 754]) | 1\n",
-      "4871     | torch.Size([1, 1000, 762]) | 1\n",
-      "4872     | torch.Size([1, 1000, 754]) | 1\n",
-      "4873     | torch.Size([1, 1000, 781]) | 1\n",
-      "4874     | torch.Size([1, 1000, 754]) | 1\n",
-      "4875     | torch.Size([1, 1000, 811]) | 1\n",
-      "4876     | torch.Size([1, 1000, 775]) | 1\n",
-      "4877     | torch.Size([1, 1000, 754]) | 1\n",
-      "4878     | torch.Size([1, 1000, 754]) | 1\n",
-      "4879     | torch.Size([1, 1000, 777]) | 1\n",
-      "4880     | torch.Size([1, 1000, 767]) | 1\n",
-      "4881     | torch.Size([1, 1000, 781]) | 1\n",
-      "4882     | torch.Size([1, 1000, 790]) | 1\n",
-      "4883     | torch.Size([1, 1000, 754]) | 1\n",
-      "4884     | torch.Size([1, 1000, 802]) | 1\n",
-      "4885     | torch.Size([1, 1000, 762]) | 1\n",
-      "4886     | torch.Size([1, 1000, 785]) | 1\n",
-      "4887     | torch.Size([1, 1000, 754]) | 1\n",
-      "4888     | torch.Size([1, 1000, 781]) | 1\n",
-      "4889     | torch.Size([1, 1000, 762]) | 1\n",
-      "4890     | torch.Size([1, 1000, 754]) | 1\n",
-      "4891     | torch.Size([1, 1000, 861]) | 1\n",
-      "4892     | torch.Size([1, 1000, 754]) | 1\n",
-      "4893     | torch.Size([1, 1000, 754]) | 1\n",
-      "4894     | torch.Size([1, 1000, 787]) | 1\n",
-      "4895     | torch.Size([1, 1000, 780]) | 1\n",
-      "4896     | torch.Size([1, 1000, 766]) | 1\n",
-      "4897     | torch.Size([1, 1000, 762]) | 1\n",
-      "4898     | torch.Size([1, 1000, 754]) | 1\n",
-      "4899     | torch.Size([1, 1000, 773]) | 1\n",
-      "4900     | torch.Size([1, 1000, 754]) | 1\n",
-      "4901     | torch.Size([1, 1000, 771]) | 1\n",
-      "4902     | torch.Size([1, 1000, 754]) | 1\n",
-      "4903     | torch.Size([1, 1000, 754]) | 1\n",
-      "4904     | torch.Size([1, 1000, 754]) | 1\n",
-      "4905     | torch.Size([1, 1000, 794]) | 1\n",
-      "4906     | torch.Size([1, 1000, 812]) | 1\n",
-      "4907     | torch.Size([1, 1000, 768]) | 1\n",
-      "4908     | torch.Size([1, 1000, 843]) | 1\n",
-      "4909     | torch.Size([1, 1000, 754]) | 1\n",
-      "4910     | torch.Size([1, 1000, 793]) | 1\n",
-      "4911     | torch.Size([1, 1000, 845]) | 1\n",
-      "4912     | torch.Size([1, 1000, 754]) | 1\n",
-      "4913     | torch.Size([1, 1000, 754]) | 1\n",
-      "4914     | torch.Size([1, 1000, 762]) | 1\n",
-      "4915     | torch.Size([1, 1000, 754]) | 1\n",
-      "4916     | torch.Size([1, 1000, 779]) | 1\n",
-      "4917     | torch.Size([1, 1000, 754]) | 1\n",
-      "4918     | torch.Size([1, 1000, 754]) | 1\n",
-      "4919     | torch.Size([1, 1000, 846]) | 1\n",
-      "4920     | torch.Size([1, 1000, 778]) | 1\n",
-      "4921     | torch.Size([1, 1000, 754]) | 1\n",
-      "4922     | torch.Size([1, 1000, 754]) | 1\n",
-      "4923     | torch.Size([1, 1000, 790]) | 1\n",
-      "4924     | torch.Size([1, 1000, 802]) | 1\n",
-      "4925     | torch.Size([1, 1000, 777]) | 1\n",
-      "4926     | torch.Size([1, 1000, 777]) | 1\n",
-      "4927     | torch.Size([1, 1000, 754]) | 1\n",
-      "4928     | torch.Size([1, 1000, 785]) | 1\n",
-      "4929     | torch.Size([1, 1000, 808]) | 1\n",
-      "4930     | torch.Size([1, 1000, 762]) | 1\n",
-      "4931     | torch.Size([1, 1000, 754]) | 1\n",
-      "4932     | torch.Size([1, 1000, 754]) | 1\n",
-      "4933     | torch.Size([1, 1000, 785]) | 1\n",
-      "4934     | torch.Size([1, 1000, 754]) | 1\n",
-      "4935     | torch.Size([1, 1000, 754]) | 1\n",
-      "4936     | torch.Size([1, 1000, 760]) | 1\n",
-      "4937     | torch.Size([1, 1000, 786]) | 1\n",
-      "4938     | torch.Size([1, 1000, 754]) | 1\n",
-      "4939     | torch.Size([1, 1000, 754]) | 1\n",
-      "4940     | torch.Size([1, 1000, 802]) | 1\n",
-      "4941     | torch.Size([1, 1000, 773]) | 1\n",
-      "4942     | torch.Size([1, 1000, 754]) | 1\n",
-      "4943     | torch.Size([1, 1000, 754]) | 1\n",
-      "4944     | torch.Size([1, 1000, 779]) | 1\n",
-      "4945     | torch.Size([1, 1000, 754]) | 1\n",
-      "4946     | torch.Size([1, 1000, 790]) | 1\n",
-      "4947     | torch.Size([1, 1000, 785]) | 1\n",
-      "4948     | torch.Size([1, 1000, 754]) | 1\n",
-      "4949     | torch.Size([1, 1000, 754]) | 1\n",
-      "4950     | torch.Size([1, 1000, 802]) | 1\n",
-      "4951     | torch.Size([1, 1000, 790]) | 1\n",
-      "4952     | torch.Size([1, 1000, 762]) | 1\n",
-      "4953     | torch.Size([1, 1000, 754]) | 1\n",
-      "4954     | torch.Size([1, 1000, 774]) | 1\n",
-      "4955     | torch.Size([1, 1000, 776]) | 1\n",
-      "4956     | torch.Size([1, 1000, 754]) | 1\n",
-      "4957     | torch.Size([1, 1000, 782]) | 1\n",
-      "4958     | torch.Size([1, 1000, 775]) | 1\n",
-      "4959     | torch.Size([1, 1000, 795]) | 1\n",
-      "4960     | torch.Size([1, 1000, 754]) | 1\n",
-      "4961     | torch.Size([1, 1000, 845]) | 1\n",
-      "4962     | torch.Size([1, 1000, 754]) | 1\n",
-      "4963     | torch.Size([1, 1000, 808]) | 1\n",
-      "4964     | torch.Size([1, 1000, 784]) | 1\n",
-      "4965     | torch.Size([1, 1000, 754]) | 1\n",
-      "4966     | torch.Size([1, 1000, 754]) | 1\n",
-      "4967     | torch.Size([1, 1000, 785]) | 1\n",
-      "4968     | torch.Size([1, 1000, 802]) | 1\n",
-      "4969     | torch.Size([1, 1000, 754]) | 1\n",
-      "4970     | torch.Size([1, 1000, 786]) | 1\n",
-      "4971     | torch.Size([1, 1000, 807]) | 1\n",
-      "4972     | torch.Size([1, 1000, 754]) | 1\n",
-      "4973     | torch.Size([1, 1000, 777]) | 1\n",
-      "4974     | torch.Size([1, 1000, 754]) | 1\n",
-      "4975     | torch.Size([1, 1000, 777]) | 1\n",
-      "4976     | torch.Size([1, 1000, 782]) | 1\n",
-      "4977     | torch.Size([1, 1000, 754]) | 1\n",
-      "4978     | torch.Size([1, 1000, 764]) | 1\n",
-      "4979     | torch.Size([1, 1000, 763]) | 1\n",
-      "4980     | torch.Size([1, 1000, 754]) | 1\n",
-      "4981     | torch.Size([1, 1000, 776]) | 1\n",
-      "4982     | torch.Size([1, 1000, 754]) | 1\n",
-      "4983     | torch.Size([1, 1000, 754]) | 1\n",
-      "4984     | torch.Size([1, 1000, 754]) | 1\n",
-      "4985     | torch.Size([1, 1000, 754]) | 1\n",
-      "4986     | torch.Size([1, 1000, 754]) | 1\n",
-      "4987     | torch.Size([1, 1000, 754]) | 1\n",
-      "4988     | torch.Size([1, 1000, 772]) | 1\n",
-      "4989     | torch.Size([1, 1000, 772]) | 1\n",
-      "4990     | torch.Size([1, 1000, 754]) | 1\n",
-      "4991     | torch.Size([1, 1000, 762]) | 1\n",
-      "4992     | torch.Size([1, 1000, 762]) | 1\n",
-      "4993     | torch.Size([1, 1000, 842]) | 1\n",
-      "4994     | torch.Size([1, 1000, 762]) | 1\n",
-      "4995     | torch.Size([1, 1000, 777]) | 1\n",
-      "4996     | torch.Size([1, 1000, 845]) | 1\n",
-      "4997     | torch.Size([1, 1000, 780]) | 1\n",
-      "4998     | torch.Size([1, 1000, 777]) | 1\n",
-      "4999     | torch.Size([1, 1000, 845]) | 1\n",
-      "5000     | torch.Size([1, 1000, 737]) | 1\n",
-      "5001     | torch.Size([1, 1000, 754]) | 1\n",
-      "5002     | torch.Size([1, 1000, 802]) | 1\n",
-      "5003     | torch.Size([1, 1000, 776]) | 1\n",
-      "5004     | torch.Size([1, 1000, 754]) | 1\n",
-      "5005     | torch.Size([1, 1000, 754]) | 1\n",
-      "5006     | torch.Size([1, 1000, 778]) | 1\n",
-      "5007     | torch.Size([1, 1000, 776]) | 1\n",
-      "5008     | torch.Size([1, 1000, 807]) | 1\n",
-      "5009     | torch.Size([1, 1000, 849]) | 1\n",
-      "5010     | torch.Size([1, 1000, 762]) | 1\n",
-      "5011     | torch.Size([1, 1000, 754]) | 1\n",
-      "5012     | torch.Size([1, 1000, 754]) | 1\n",
-      "5013     | torch.Size([1, 1000, 777]) | 1\n",
-      "5014     | torch.Size([1, 1000, 774]) | 1\n",
-      "5015     | torch.Size([1, 1000, 777]) | 1\n",
-      "5016     | torch.Size([1, 1000, 777]) | 1\n",
-      "5017     | torch.Size([1, 1000, 764]) | 1\n",
-      "5018     | torch.Size([1, 1000, 841]) | 1\n",
-      "5019     | torch.Size([1, 1000, 753]) | 1\n",
-      "5020     | torch.Size([1, 1000, 754]) | 2\n",
-      "5021     | torch.Size([1, 1000, 802]) | 2\n",
-      "5022     | torch.Size([1, 1000, 754]) | 2\n",
-      "5023     | torch.Size([1, 1000, 754]) | 2\n",
-      "5024     | torch.Size([1, 1000, 754]) | 2\n",
-      "5025     | torch.Size([1, 1000, 754]) | 2\n",
-      "5026     | torch.Size([1, 1000, 754]) | 2\n",
-      "5027     | torch.Size([1, 1000, 754]) | 2\n",
-      "5028     | torch.Size([1, 1000, 754]) | 2\n",
-      "5029     | torch.Size([1, 1000, 754]) | 2\n",
-      "5030     | torch.Size([1, 1000, 754]) | 2\n",
-      "5031     | torch.Size([1, 1000, 754]) | 2\n",
-      "5032     | torch.Size([1, 1000, 754]) | 2\n",
-      "5033     | torch.Size([1, 1000, 754]) | 2\n",
-      "5034     | torch.Size([1, 1000, 754]) | 2\n",
-      "5035     | torch.Size([1, 1000, 802]) | 2\n",
-      "5036     | torch.Size([1, 1000, 802]) | 2\n",
-      "5037     | torch.Size([1, 1000, 754]) | 2\n",
-      "5038     | torch.Size([1, 1000, 754]) | 2\n",
-      "5039     | torch.Size([1, 1000, 754]) | 2\n",
-      "5040     | torch.Size([1, 1000, 754]) | 2\n",
-      "5041     | torch.Size([1, 1000, 754]) | 2\n",
-      "5042     | torch.Size([1, 1000, 754]) | 2\n",
-      "5043     | torch.Size([1, 1000, 754]) | 2\n",
-      "5044     | torch.Size([1, 1000, 754]) | 2\n",
-      "5045     | torch.Size([1, 1000, 754]) | 2\n",
-      "5046     | torch.Size([1, 1000, 754]) | 2\n",
-      "5047     | torch.Size([1, 1000, 754]) | 2\n",
-      "5048     | torch.Size([1, 1000, 754]) | 2\n",
-      "5049     | torch.Size([1, 1000, 754]) | 2\n",
-      "5050     | torch.Size([1, 1000, 754]) | 2\n",
-      "5051     | torch.Size([1, 1000, 754]) | 2\n",
-      "5052     | torch.Size([1, 1000, 754]) | 2\n",
-      "5053     | torch.Size([1, 1000, 754]) | 2\n",
-      "5054     | torch.Size([1, 1000, 754]) | 2\n",
-      "5055     | torch.Size([1, 1000, 754]) | 2\n",
-      "5056     | torch.Size([1, 1000, 754]) | 2\n",
-      "5057     | torch.Size([1, 1000, 754]) | 2\n",
-      "5058     | torch.Size([1, 1000, 754]) | 2\n",
-      "5059     | torch.Size([1, 1000, 754]) | 2\n",
-      "5060     | torch.Size([1, 1000, 754]) | 2\n",
-      "5061     | torch.Size([1, 1000, 754]) | 2\n",
-      "5062     | torch.Size([1, 1000, 754]) | 2\n",
-      "5063     | torch.Size([1, 1000, 754]) | 2\n",
-      "5064     | torch.Size([1, 1000, 754]) | 2\n",
-      "5065     | torch.Size([1, 1000, 754]) | 2\n",
-      "5066     | torch.Size([1, 1000, 754]) | 2\n",
-      "5067     | torch.Size([1, 1000, 754]) | 2\n",
-      "5068     | torch.Size([1, 1000, 754]) | 2\n",
-      "5069     | torch.Size([1, 1000, 754]) | 2\n",
-      "5070     | torch.Size([1, 1000, 754]) | 2\n",
-      "5071     | torch.Size([1, 1000, 754]) | 2\n",
-      "5072     | torch.Size([1, 1000, 754]) | 2\n",
-      "5073     | torch.Size([1, 1000, 754]) | 2\n",
-      "5074     | torch.Size([1, 1000, 754]) | 2\n",
-      "5075     | torch.Size([1, 1000, 754]) | 2\n",
-      "5076     | torch.Size([1, 1000, 754]) | 2\n",
-      "5077     | torch.Size([1, 1000, 754]) | 2\n",
-      "5078     | torch.Size([1, 1000, 754]) | 2\n",
-      "5079     | torch.Size([1, 1000, 754]) | 2\n",
-      "5080     | torch.Size([1, 1000, 802]) | 2\n",
-      "5081     | torch.Size([1, 1000, 754]) | 2\n",
-      "5082     | torch.Size([1, 1000, 754]) | 2\n",
-      "5083     | torch.Size([1, 1000, 754]) | 2\n",
-      "5084     | torch.Size([1, 1000, 754]) | 2\n",
-      "5085     | torch.Size([1, 1000, 754]) | 2\n",
-      "5086     | torch.Size([1, 1000, 754]) | 2\n",
-      "5087     | torch.Size([1, 1000, 754]) | 2\n",
-      "5088     | torch.Size([1, 1000, 754]) | 2\n",
-      "5089     | torch.Size([1, 1000, 754]) | 2\n",
-      "5090     | torch.Size([1, 1000, 754]) | 2\n",
-      "5091     | torch.Size([1, 1000, 754]) | 2\n",
-      "5092     | torch.Size([1, 1000, 754]) | 2\n",
-      "5093     | torch.Size([1, 1000, 754]) | 2\n",
-      "5094     | torch.Size([1, 1000, 754]) | 2\n",
-      "5095     | torch.Size([1, 1000, 754]) | 2\n",
-      "5096     | torch.Size([1, 1000, 754]) | 2\n",
-      "5097     | torch.Size([1, 1000, 754]) | 2\n",
-      "5098     | torch.Size([1, 1000, 754]) | 2\n",
-      "5099     | torch.Size([1, 1000, 754]) | 2\n",
-      "5100     | torch.Size([1, 1000, 754]) | 2\n",
-      "5101     | torch.Size([1, 1000, 754]) | 2\n",
-      "5102     | torch.Size([1, 1000, 754]) | 2\n",
-      "5103     | torch.Size([1, 1000, 754]) | 2\n",
-      "5104     | torch.Size([1, 1000, 754]) | 2\n",
-      "5105     | torch.Size([1, 1000, 754]) | 2\n",
-      "5106     | torch.Size([1, 1000, 754]) | 2\n",
-      "5107     | torch.Size([1, 1000, 754]) | 2\n",
-      "5108     | torch.Size([1, 1000, 754]) | 2\n",
-      "5109     | torch.Size([1, 1000, 754]) | 2\n",
-      "5110     | torch.Size([1, 1000, 754]) | 2\n",
-      "5111     | torch.Size([1, 1000, 754]) | 2\n",
-      "5112     | torch.Size([1, 1000, 754]) | 2\n",
-      "5113     | torch.Size([1, 1000, 754]) | 2\n",
-      "5114     | torch.Size([1, 1000, 754]) | 2\n",
-      "5115     | torch.Size([1, 1000, 754]) | 2\n",
-      "5116     | torch.Size([1, 1000, 754]) | 2\n",
-      "5117     | torch.Size([1, 1000, 754]) | 2\n",
-      "5118     | torch.Size([1, 1000, 754]) | 2\n",
-      "5119     | torch.Size([1, 1000, 754]) | 2\n",
-      "5120     | torch.Size([1, 1000, 754]) | 2\n",
-      "5121     | torch.Size([1, 1000, 754]) | 2\n",
-      "5122     | torch.Size([1, 1000, 754]) | 2\n",
-      "5123     | torch.Size([1, 1000, 754]) | 2\n",
-      "5124     | torch.Size([1, 1000, 754]) | 2\n",
-      "5125     | torch.Size([1, 1000, 754]) | 2\n",
-      "5126     | torch.Size([1, 1000, 754]) | 2\n",
-      "5127     | torch.Size([1, 1000, 754]) | 2\n",
-      "5128     | torch.Size([1, 1000, 754]) | 2\n",
-      "5129     | torch.Size([1, 1000, 802]) | 2\n",
-      "5130     | torch.Size([1, 1000, 754]) | 2\n",
-      "5131     | torch.Size([1, 1000, 754]) | 2\n",
-      "5132     | torch.Size([1, 1000, 754]) | 2\n",
-      "5133     | torch.Size([1, 1000, 754]) | 2\n",
-      "5134     | torch.Size([1, 1000, 754]) | 2\n",
-      "5135     | torch.Size([1, 1000, 754]) | 2\n",
-      "5136     | torch.Size([1, 1000, 754]) | 2\n",
-      "5137     | torch.Size([1, 1000, 754]) | 2\n",
-      "5138     | torch.Size([1, 1000, 754]) | 2\n",
-      "5139     | torch.Size([1, 1000, 754]) | 2\n",
-      "5140     | torch.Size([1, 1000, 754]) | 2\n",
-      "5141     | torch.Size([1, 1000, 754]) | 2\n",
-      "5142     | torch.Size([1, 1000, 754]) | 2\n",
-      "5143     | torch.Size([1, 1000, 754]) | 2\n",
-      "5144     | torch.Size([1, 1000, 754]) | 2\n",
-      "5145     | torch.Size([1, 1000, 754]) | 2\n",
-      "5146     | torch.Size([1, 1000, 754]) | 2\n",
-      "5147     | torch.Size([1, 1000, 754]) | 2\n",
-      "5148     | torch.Size([1, 1000, 754]) | 2\n",
-      "5149     | torch.Size([1, 1000, 754]) | 2\n",
-      "5150     | torch.Size([1, 1000, 754]) | 2\n",
-      "5151     | torch.Size([1, 1000, 754]) | 2\n",
-      "5152     | torch.Size([1, 1000, 754]) | 2\n",
-      "5153     | torch.Size([1, 1000, 754]) | 2\n",
-      "5154     | torch.Size([1, 1000, 754]) | 2\n",
-      "5155     | torch.Size([1, 1000, 754]) | 2\n",
-      "5156     | torch.Size([1, 1000, 754]) | 2\n",
-      "5157     | torch.Size([1, 1000, 754]) | 2\n",
-      "5158     | torch.Size([1, 1000, 754]) | 2\n",
-      "5159     | torch.Size([1, 1000, 754]) | 2\n",
-      "5160     | torch.Size([1, 1000, 754]) | 2\n",
-      "5161     | torch.Size([1, 1000, 754]) | 2\n",
-      "5162     | torch.Size([1, 1000, 754]) | 2\n",
-      "5163     | torch.Size([1, 1000, 754]) | 2\n",
-      "5164     | torch.Size([1, 1000, 754]) | 2\n",
-      "5165     | torch.Size([1, 1000, 754]) | 2\n",
-      "5166     | torch.Size([1, 1000, 754]) | 2\n",
-      "5167     | torch.Size([1, 1000, 754]) | 2\n",
-      "5168     | torch.Size([1, 1000, 754]) | 2\n",
-      "5169     | torch.Size([1, 1000, 754]) | 2\n",
-      "5170     | torch.Size([1, 1000, 754]) | 2\n",
-      "5171     | torch.Size([1, 1000, 754]) | 2\n",
-      "5172     | torch.Size([1, 1000, 754]) | 2\n",
-      "5173     | torch.Size([1, 1000, 754]) | 2\n",
-      "5174     | torch.Size([1, 1000, 754]) | 2\n",
-      "5175     | torch.Size([1, 1000, 754]) | 2\n",
-      "5176     | torch.Size([1, 1000, 754]) | 2\n",
-      "5177     | torch.Size([1, 1000, 754]) | 2\n",
-      "5178     | torch.Size([1, 1000, 754]) | 2\n",
-      "5179     | torch.Size([1, 1000, 754]) | 2\n",
-      "5180     | torch.Size([1, 1000, 754]) | 2\n",
-      "5181     | torch.Size([1, 1000, 754]) | 2\n",
-      "5182     | torch.Size([1, 1000, 754]) | 2\n",
-      "5183     | torch.Size([1, 1000, 754]) | 2\n",
-      "5184     | torch.Size([1, 1000, 754]) | 2\n",
-      "5185     | torch.Size([1, 1000, 754]) | 2\n",
-      "5186     | torch.Size([1, 1000, 754]) | 2\n",
-      "5187     | torch.Size([1, 1000, 754]) | 2\n",
-      "5188     | torch.Size([1, 1000, 754]) | 2\n",
-      "5189     | torch.Size([1, 1000, 754]) | 2\n",
-      "5190     | torch.Size([1, 1000, 754]) | 2\n",
-      "5191     | torch.Size([1, 1000, 754]) | 2\n",
-      "5192     | torch.Size([1, 1000, 754]) | 2\n",
-      "5193     | torch.Size([1, 1000, 802]) | 2\n",
-      "5194     | torch.Size([1, 1000, 754]) | 2\n",
-      "5195     | torch.Size([1, 1000, 754]) | 2\n",
-      "5196     | torch.Size([1, 1000, 754]) | 2\n",
-      "5197     | torch.Size([1, 1000, 754]) | 2\n",
-      "5198     | torch.Size([1, 1000, 754]) | 2\n",
-      "5199     | torch.Size([1, 1000, 754]) | 2\n",
-      "5200     | torch.Size([1, 1000, 762]) | 2\n",
-      "5201     | torch.Size([1, 1000, 754]) | 2\n",
-      "5202     | torch.Size([1, 1000, 754]) | 2\n",
-      "5203     | torch.Size([1, 1000, 754]) | 2\n",
-      "5204     | torch.Size([1, 1000, 754]) | 2\n",
-      "5205     | torch.Size([1, 1000, 754]) | 2\n",
-      "5206     | torch.Size([1, 1000, 754]) | 2\n",
-      "5207     | torch.Size([1, 1000, 754]) | 2\n",
-      "5208     | torch.Size([1, 1000, 754]) | 2\n",
-      "5209     | torch.Size([1, 1000, 754]) | 2\n",
-      "5210     | torch.Size([1, 1000, 754]) | 2\n",
-      "5211     | torch.Size([1, 1000, 802]) | 2\n",
-      "5212     | torch.Size([1, 1000, 754]) | 2\n",
-      "5213     | torch.Size([1, 1000, 754]) | 2\n",
-      "5214     | torch.Size([1, 1000, 754]) | 2\n",
-      "5215     | torch.Size([1, 1000, 754]) | 2\n",
-      "5216     | torch.Size([1, 1000, 754]) | 2\n",
-      "5217     | torch.Size([1, 1000, 754]) | 2\n",
-      "5218     | torch.Size([1, 1000, 754]) | 2\n",
-      "5219     | torch.Size([1, 1000, 802]) | 2\n",
-      "5220     | torch.Size([1, 1000, 754]) | 2\n",
-      "5221     | torch.Size([1, 1000, 754]) | 2\n",
-      "5222     | torch.Size([1, 1000, 754]) | 2\n",
-      "5223     | torch.Size([1, 1000, 754]) | 2\n",
-      "5224     | torch.Size([1, 1000, 754]) | 2\n",
-      "5225     | torch.Size([1, 1000, 754]) | 2\n",
-      "5226     | torch.Size([1, 1000, 754]) | 2\n",
-      "5227     | torch.Size([1, 1000, 754]) | 2\n",
-      "5228     | torch.Size([1, 1000, 754]) | 2\n",
-      "5229     | torch.Size([1, 1000, 754]) | 2\n",
-      "5230     | torch.Size([1, 1000, 754]) | 2\n",
-      "5231     | torch.Size([1, 1000, 754]) | 2\n",
-      "5232     | torch.Size([1, 1000, 754]) | 2\n",
-      "5233     | torch.Size([1, 1000, 754]) | 2\n",
-      "5234     | torch.Size([1, 1000, 754]) | 2\n",
-      "5235     | torch.Size([1, 1000, 754]) | 2\n",
-      "5236     | torch.Size([1, 1000, 754]) | 2\n",
-      "5237     | torch.Size([1, 1000, 754]) | 2\n",
-      "5238     | torch.Size([1, 1000, 754]) | 2\n",
-      "5239     | torch.Size([1, 1000, 754]) | 2\n",
-      "5240     | torch.Size([1, 1000, 754]) | 2\n",
-      "5241     | torch.Size([1, 1000, 754]) | 2\n",
-      "5242     | torch.Size([1, 1000, 754]) | 2\n",
-      "5243     | torch.Size([1, 1000, 754]) | 2\n",
-      "5244     | torch.Size([1, 1000, 754]) | 2\n",
-      "5245     | torch.Size([1, 1000, 754]) | 2\n",
-      "5246     | torch.Size([1, 1000, 754]) | 2\n",
-      "5247     | torch.Size([1, 1000, 754]) | 2\n",
-      "5248     | torch.Size([1, 1000, 754]) | 2\n",
-      "5249     | torch.Size([1, 1000, 754]) | 2\n",
-      "5250     | torch.Size([1, 1000, 754]) | 2\n",
-      "5251     | torch.Size([1, 1000, 754]) | 2\n",
-      "5252     | torch.Size([1, 1000, 754]) | 2\n",
-      "5253     | torch.Size([1, 1000, 754]) | 2\n",
-      "5254     | torch.Size([1, 1000, 754]) | 2\n",
-      "5255     | torch.Size([1, 1000, 754]) | 2\n",
-      "5256     | torch.Size([1, 1000, 754]) | 2\n",
-      "5257     | torch.Size([1, 1000, 754]) | 2\n",
-      "5258     | torch.Size([1, 1000, 754]) | 2\n",
-      "5259     | torch.Size([1, 1000, 754]) | 2\n",
-      "5260     | torch.Size([1, 1000, 754]) | 2\n",
-      "5261     | torch.Size([1, 1000, 754]) | 2\n",
-      "5262     | torch.Size([1, 1000, 754]) | 2\n",
-      "5263     | torch.Size([1, 1000, 754]) | 2\n",
-      "5264     | torch.Size([1, 1000, 754]) | 2\n",
-      "5265     | torch.Size([1, 1000, 754]) | 2\n",
-      "5266     | torch.Size([1, 1000, 754]) | 2\n",
-      "5267     | torch.Size([1, 1000, 754]) | 2\n",
-      "5268     | torch.Size([1, 1000, 754]) | 2\n",
-      "5269     | torch.Size([1, 1000, 754]) | 2\n",
-      "5270     | torch.Size([1, 1000, 754]) | 2\n",
-      "5271     | torch.Size([1, 1000, 754]) | 2\n",
-      "5272     | torch.Size([1, 1000, 754]) | 2\n",
-      "5273     | torch.Size([1, 1000, 754]) | 2\n",
-      "5274     | torch.Size([1, 1000, 754]) | 2\n",
-      "5275     | torch.Size([1, 1000, 754]) | 2\n",
-      "5276     | torch.Size([1, 1000, 754]) | 2\n",
-      "5277     | torch.Size([1, 1000, 754]) | 2\n",
-      "5278     | torch.Size([1, 1000, 754]) | 2\n",
-      "5279     | torch.Size([1, 1000, 754]) | 2\n",
-      "5280     | torch.Size([1, 1000, 754]) | 2\n",
-      "5281     | torch.Size([1, 1000, 754]) | 2\n",
-      "5282     | torch.Size([1, 1000, 754]) | 2\n",
-      "5283     | torch.Size([1, 1000, 754]) | 2\n",
-      "5284     | torch.Size([1, 1000, 754]) | 2\n",
-      "5285     | torch.Size([1, 1000, 754]) | 2\n",
-      "5286     | torch.Size([1, 1000, 754]) | 2\n",
-      "5287     | torch.Size([1, 1000, 754]) | 2\n",
-      "5288     | torch.Size([1, 1000, 754]) | 2\n",
-      "5289     | torch.Size([1, 1000, 754]) | 2\n",
-      "5290     | torch.Size([1, 1000, 754]) | 2\n",
-      "5291     | torch.Size([1, 1000, 754]) | 2\n",
-      "5292     | torch.Size([1, 1000, 754]) | 2\n",
-      "5293     | torch.Size([1, 1000, 754]) | 2\n",
-      "5294     | torch.Size([1, 1000, 754]) | 2\n",
-      "5295     | torch.Size([1, 1000, 754]) | 2\n",
-      "5296     | torch.Size([1, 1000, 754]) | 2\n",
-      "5297     | torch.Size([1, 1000, 754]) | 2\n",
-      "5298     | torch.Size([1, 1000, 754]) | 2\n",
-      "5299     | torch.Size([1, 1000, 754]) | 2\n",
-      "5300     | torch.Size([1, 1000, 754]) | 2\n",
-      "5301     | torch.Size([1, 1000, 754]) | 2\n",
-      "5302     | torch.Size([1, 1000, 754]) | 2\n",
-      "5303     | torch.Size([1, 1000, 754]) | 2\n",
-      "5304     | torch.Size([1, 1000, 754]) | 2\n",
-      "5305     | torch.Size([1, 1000, 754]) | 2\n",
-      "5306     | torch.Size([1, 1000, 754]) | 2\n",
-      "5307     | torch.Size([1, 1000, 754]) | 2\n",
-      "5308     | torch.Size([1, 1000, 754]) | 2\n",
-      "5309     | torch.Size([1, 1000, 754]) | 2\n",
-      "5310     | torch.Size([1, 1000, 754]) | 2\n",
-      "5311     | torch.Size([1, 1000, 754]) | 2\n",
-      "5312     | torch.Size([1, 1000, 754]) | 2\n",
-      "5313     | torch.Size([1, 1000, 802]) | 2\n",
-      "5314     | torch.Size([1, 1000, 754]) | 2\n",
-      "5315     | torch.Size([1, 1000, 754]) | 2\n",
-      "5316     | torch.Size([1, 1000, 754]) | 2\n",
-      "5317     | torch.Size([1, 1000, 762]) | 2\n",
-      "5318     | torch.Size([1, 1000, 754]) | 2\n",
-      "5319     | torch.Size([1, 1000, 754]) | 2\n",
-      "5320     | torch.Size([1, 1000, 754]) | 2\n",
-      "5321     | torch.Size([1, 1000, 754]) | 2\n",
-      "5322     | torch.Size([1, 1000, 754]) | 2\n",
-      "5323     | torch.Size([1, 1000, 754]) | 2\n",
-      "5324     | torch.Size([1, 1000, 754]) | 2\n",
-      "5325     | torch.Size([1, 1000, 802]) | 2\n",
-      "5326     | torch.Size([1, 1000, 754]) | 2\n",
-      "5327     | torch.Size([1, 1000, 754]) | 2\n",
-      "5328     | torch.Size([1, 1000, 754]) | 2\n",
-      "5329     | torch.Size([1, 1000, 754]) | 2\n",
-      "5330     | torch.Size([1, 1000, 754]) | 2\n",
-      "5331     | torch.Size([1, 1000, 754]) | 2\n",
-      "5332     | torch.Size([1, 1000, 754]) | 2\n",
-      "5333     | torch.Size([1, 1000, 754]) | 2\n",
-      "5334     | torch.Size([1, 1000, 754]) | 2\n",
-      "5335     | torch.Size([1, 1000, 754]) | 2\n",
-      "5336     | torch.Size([1, 1000, 754]) | 2\n",
-      "5337     | torch.Size([1, 1000, 754]) | 2\n",
-      "5338     | torch.Size([1, 1000, 754]) | 2\n",
-      "5339     | torch.Size([1, 1000, 754]) | 2\n",
-      "5340     | torch.Size([1, 1000, 754]) | 2\n",
-      "5341     | torch.Size([1, 1000, 754]) | 2\n",
-      "5342     | torch.Size([1, 1000, 754]) | 2\n",
-      "5343     | torch.Size([1, 1000, 754]) | 2\n",
-      "5344     | torch.Size([1, 1000, 754]) | 2\n",
-      "5345     | torch.Size([1, 1000, 802]) | 2\n",
-      "5346     | torch.Size([1, 1000, 754]) | 2\n",
-      "5347     | torch.Size([1, 1000, 754]) | 2\n",
-      "5348     | torch.Size([1, 1000, 754]) | 2\n",
-      "5349     | torch.Size([1, 1000, 754]) | 2\n",
-      "5350     | torch.Size([1, 1000, 754]) | 2\n",
-      "5351     | torch.Size([1, 1000, 754]) | 2\n",
-      "5352     | torch.Size([1, 1000, 754]) | 2\n",
-      "5353     | torch.Size([1, 1000, 754]) | 2\n",
-      "5354     | torch.Size([1, 1000, 754]) | 2\n",
-      "5355     | torch.Size([1, 1000, 754]) | 2\n",
-      "5356     | torch.Size([1, 1000, 754]) | 2\n",
-      "5357     | torch.Size([1, 1000, 754]) | 2\n",
-      "5358     | torch.Size([1, 1000, 754]) | 2\n",
-      "5359     | torch.Size([1, 1000, 754]) | 2\n",
-      "5360     | torch.Size([1, 1000, 754]) | 2\n",
-      "5361     | torch.Size([1, 1000, 754]) | 2\n",
-      "5362     | torch.Size([1, 1000, 754]) | 2\n",
-      "5363     | torch.Size([1, 1000, 754]) | 2\n",
-      "5364     | torch.Size([1, 1000, 754]) | 2\n",
-      "5365     | torch.Size([1, 1000, 754]) | 2\n",
-      "5366     | torch.Size([1, 1000, 754]) | 2\n",
-      "5367     | torch.Size([1, 1000, 754]) | 2\n",
-      "5368     | torch.Size([1, 1000, 754]) | 2\n",
-      "5369     | torch.Size([1, 1000, 754]) | 2\n",
-      "5370     | torch.Size([1, 1000, 754]) | 2\n",
-      "5371     | torch.Size([1, 1000, 754]) | 2\n",
-      "5372     | torch.Size([1, 1000, 754]) | 2\n",
-      "5373     | torch.Size([1, 1000, 754]) | 2\n",
-      "5374     | torch.Size([1, 1000, 754]) | 2\n",
-      "5375     | torch.Size([1, 1000, 754]) | 2\n",
-      "5376     | torch.Size([1, 1000, 754]) | 2\n",
-      "5377     | torch.Size([1, 1000, 802]) | 2\n",
-      "5378     | torch.Size([1, 1000, 754]) | 2\n",
-      "5379     | torch.Size([1, 1000, 754]) | 2\n",
-      "5380     | torch.Size([1, 1000, 754]) | 2\n",
-      "5381     | torch.Size([1, 1000, 754]) | 2\n",
-      "5382     | torch.Size([1, 1000, 802]) | 2\n",
-      "5383     | torch.Size([1, 1000, 754]) | 2\n",
-      "5384     | torch.Size([1, 1000, 754]) | 2\n",
-      "5385     | torch.Size([1, 1000, 754]) | 2\n",
-      "5386     | torch.Size([1, 1000, 754]) | 2\n",
-      "5387     | torch.Size([1, 1000, 754]) | 2\n",
-      "5388     | torch.Size([1, 1000, 754]) | 2\n",
-      "5389     | torch.Size([1, 1000, 754]) | 2\n",
-      "5390     | torch.Size([1, 1000, 754]) | 2\n",
-      "5391     | torch.Size([1, 1000, 754]) | 2\n",
-      "5392     | torch.Size([1, 1000, 754]) | 2\n",
-      "5393     | torch.Size([1, 1000, 754]) | 2\n",
-      "5394     | torch.Size([1, 1000, 754]) | 2\n",
-      "5395     | torch.Size([1, 1000, 754]) | 2\n",
-      "5396     | torch.Size([1, 1000, 754]) | 2\n",
-      "5397     | torch.Size([1, 1000, 754]) | 2\n",
-      "5398     | torch.Size([1, 1000, 754]) | 2\n",
-      "5399     | torch.Size([1, 1000, 754]) | 2\n",
-      "5400     | torch.Size([1, 1000, 802]) | 2\n",
-      "5401     | torch.Size([1, 1000, 754]) | 2\n",
-      "5402     | torch.Size([1, 1000, 754]) | 2\n",
-      "5403     | torch.Size([1, 1000, 754]) | 2\n",
-      "5404     | torch.Size([1, 1000, 754]) | 2\n",
-      "5405     | torch.Size([1, 1000, 754]) | 2\n",
-      "5406     | torch.Size([1, 1000, 754]) | 2\n",
-      "5407     | torch.Size([1, 1000, 754]) | 2\n",
-      "5408     | torch.Size([1, 1000, 754]) | 2\n",
-      "5409     | torch.Size([1, 1000, 754]) | 2\n",
-      "5410     | torch.Size([1, 1000, 754]) | 2\n",
-      "5411     | torch.Size([1, 1000, 754]) | 2\n",
-      "5412     | torch.Size([1, 1000, 754]) | 2\n",
-      "5413     | torch.Size([1, 1000, 754]) | 2\n",
-      "5414     | torch.Size([1, 1000, 754]) | 2\n",
-      "5415     | torch.Size([1, 1000, 754]) | 2\n",
-      "5416     | torch.Size([1, 1000, 754]) | 2\n",
-      "5417     | torch.Size([1, 1000, 754]) | 2\n",
-      "5418     | torch.Size([1, 1000, 754]) | 2\n",
-      "5419     | torch.Size([1, 1000, 754]) | 2\n",
-      "5420     | torch.Size([1, 1000, 754]) | 2\n",
-      "5421     | torch.Size([1, 1000, 754]) | 2\n",
-      "5422     | torch.Size([1, 1000, 754]) | 2\n",
-      "5423     | torch.Size([1, 1000, 754]) | 2\n",
-      "5424     | torch.Size([1, 1000, 754]) | 2\n",
-      "5425     | torch.Size([1, 1000, 754]) | 2\n",
-      "5426     | torch.Size([1, 1000, 754]) | 2\n",
-      "5427     | torch.Size([1, 1000, 754]) | 2\n",
-      "5428     | torch.Size([1, 1000, 754]) | 2\n",
-      "5429     | torch.Size([1, 1000, 754]) | 2\n",
-      "5430     | torch.Size([1, 1000, 754]) | 2\n",
-      "5431     | torch.Size([1, 1000, 754]) | 2\n",
-      "5432     | torch.Size([1, 1000, 754]) | 2\n",
-      "5433     | torch.Size([1, 1000, 754]) | 2\n",
-      "5434     | torch.Size([1, 1000, 754]) | 2\n",
-      "5435     | torch.Size([1, 1000, 754]) | 2\n",
-      "5436     | torch.Size([1, 1000, 754]) | 2\n",
-      "5437     | torch.Size([1, 1000, 754]) | 2\n",
-      "5438     | torch.Size([1, 1000, 754]) | 2\n",
-      "5439     | torch.Size([1, 1000, 754]) | 2\n",
-      "5440     | torch.Size([1, 1000, 754]) | 2\n",
-      "5441     | torch.Size([1, 1000, 754]) | 2\n",
-      "5442     | torch.Size([1, 1000, 802]) | 2\n",
-      "5443     | torch.Size([1, 1000, 754]) | 2\n",
-      "5444     | torch.Size([1, 1000, 754]) | 2\n",
-      "5445     | torch.Size([1, 1000, 754]) | 2\n",
-      "5446     | torch.Size([1, 1000, 754]) | 2\n",
-      "5447     | torch.Size([1, 1000, 754]) | 2\n",
-      "5448     | torch.Size([1, 1000, 754]) | 2\n",
-      "5449     | torch.Size([1, 1000, 754]) | 2\n",
-      "5450     | torch.Size([1, 1000, 754]) | 2\n",
-      "5451     | torch.Size([1, 1000, 754]) | 2\n",
-      "5452     | torch.Size([1, 1000, 754]) | 2\n",
-      "5453     | torch.Size([1, 1000, 754]) | 2\n",
-      "5454     | torch.Size([1, 1000, 754]) | 2\n",
-      "5455     | torch.Size([1, 1000, 754]) | 2\n",
-      "5456     | torch.Size([1, 1000, 754]) | 2\n",
-      "5457     | torch.Size([1, 1000, 754]) | 2\n",
-      "5458     | torch.Size([1, 1000, 754]) | 2\n",
-      "5459     | torch.Size([1, 1000, 754]) | 2\n",
-      "5460     | torch.Size([1, 1000, 754]) | 2\n",
-      "5461     | torch.Size([1, 1000, 754]) | 2\n",
-      "5462     | torch.Size([1, 1000, 754]) | 2\n",
-      "5463     | torch.Size([1, 1000, 802]) | 2\n",
-      "5464     | torch.Size([1, 1000, 754]) | 2\n",
-      "5465     | torch.Size([1, 1000, 754]) | 2\n",
-      "5466     | torch.Size([1, 1000, 754]) | 2\n",
-      "5467     | torch.Size([1, 1000, 754]) | 2\n",
-      "5468     | torch.Size([1, 1000, 754]) | 2\n",
-      "5469     | torch.Size([1, 1000, 754]) | 2\n",
-      "5470     | torch.Size([1, 1000, 754]) | 2\n",
-      "5471     | torch.Size([1, 1000, 754]) | 2\n",
-      "5472     | torch.Size([1, 1000, 754]) | 2\n",
-      "5473     | torch.Size([1, 1000, 754]) | 2\n",
-      "5474     | torch.Size([1, 1000, 754]) | 2\n",
-      "5475     | torch.Size([1, 1000, 754]) | 2\n",
-      "5476     | torch.Size([1, 1000, 754]) | 2\n",
-      "5477     | torch.Size([1, 1000, 754]) | 2\n",
-      "5478     | torch.Size([1, 1000, 762]) | 2\n",
-      "5479     | torch.Size([1, 1000, 754]) | 2\n",
-      "5480     | torch.Size([1, 1000, 754]) | 2\n",
-      "5481     | torch.Size([1, 1000, 754]) | 2\n",
-      "5482     | torch.Size([1, 1000, 754]) | 2\n",
-      "5483     | torch.Size([1, 1000, 754]) | 2\n",
-      "5484     | torch.Size([1, 1000, 754]) | 2\n",
-      "5485     | torch.Size([1, 1000, 754]) | 2\n",
-      "5486     | torch.Size([1, 1000, 754]) | 2\n",
-      "5487     | torch.Size([1, 1000, 754]) | 2\n",
-      "5488     | torch.Size([1, 1000, 754]) | 2\n",
-      "5489     | torch.Size([1, 1000, 754]) | 2\n",
-      "5490     | torch.Size([1, 1000, 754]) | 2\n",
-      "5491     | torch.Size([1, 1000, 754]) | 2\n",
-      "5492     | torch.Size([1, 1000, 802]) | 2\n",
-      "5493     | torch.Size([1, 1000, 754]) | 2\n",
-      "5494     | torch.Size([1, 1000, 754]) | 2\n",
-      "5495     | torch.Size([1, 1000, 754]) | 2\n",
-      "5496     | torch.Size([1, 1000, 754]) | 2\n",
-      "5497     | torch.Size([1, 1000, 754]) | 2\n",
-      "5498     | torch.Size([1, 1000, 754]) | 2\n",
-      "5499     | torch.Size([1, 1000, 754]) | 2\n",
-      "5500     | torch.Size([1, 1000, 754]) | 2\n",
-      "5501     | torch.Size([1, 1000, 754]) | 2\n",
-      "5502     | torch.Size([1, 1000, 754]) | 2\n",
-      "5503     | torch.Size([1, 1000, 754]) | 2\n",
-      "5504     | torch.Size([1, 1000, 754]) | 2\n",
-      "5505     | torch.Size([1, 1000, 754]) | 2\n",
-      "5506     | torch.Size([1, 1000, 754]) | 2\n",
-      "5507     | torch.Size([1, 1000, 754]) | 2\n",
-      "5508     | torch.Size([1, 1000, 754]) | 2\n",
-      "5509     | torch.Size([1, 1000, 754]) | 2\n",
-      "5510     | torch.Size([1, 1000, 754]) | 2\n",
-      "5511     | torch.Size([1, 1000, 754]) | 2\n",
-      "5512     | torch.Size([1, 1000, 754]) | 2\n",
-      "5513     | torch.Size([1, 1000, 754]) | 2\n",
-      "5514     | torch.Size([1, 1000, 754]) | 2\n",
-      "5515     | torch.Size([1, 1000, 754]) | 2\n",
-      "5516     | torch.Size([1, 1000, 754]) | 2\n",
-      "5517     | torch.Size([1, 1000, 754]) | 2\n",
-      "5518     | torch.Size([1, 1000, 754]) | 2\n",
-      "5519     | torch.Size([1, 1000, 754]) | 2\n",
-      "5520     | torch.Size([1, 1000, 754]) | 2\n",
-      "5521     | torch.Size([1, 1000, 754]) | 2\n",
-      "5522     | torch.Size([1, 1000, 754]) | 2\n",
-      "5523     | torch.Size([1, 1000, 754]) | 2\n",
-      "5524     | torch.Size([1, 1000, 754]) | 2\n",
-      "5525     | torch.Size([1, 1000, 754]) | 2\n",
-      "5526     | torch.Size([1, 1000, 754]) | 2\n",
-      "5527     | torch.Size([1, 1000, 754]) | 2\n",
-      "5528     | torch.Size([1, 1000, 754]) | 2\n",
-      "5529     | torch.Size([1, 1000, 754]) | 2\n",
-      "5530     | torch.Size([1, 1000, 754]) | 2\n",
-      "5531     | torch.Size([1, 1000, 754]) | 2\n",
-      "5532     | torch.Size([1, 1000, 754]) | 2\n",
-      "5533     | torch.Size([1, 1000, 802]) | 2\n",
-      "5534     | torch.Size([1, 1000, 754]) | 2\n",
-      "5535     | torch.Size([1, 1000, 754]) | 2\n",
-      "5536     | torch.Size([1, 1000, 754]) | 2\n",
-      "5537     | torch.Size([1, 1000, 754]) | 2\n",
-      "5538     | torch.Size([1, 1000, 754]) | 2\n",
-      "5539     | torch.Size([1, 1000, 754]) | 2\n",
-      "5540     | torch.Size([1, 1000, 754]) | 2\n",
-      "5541     | torch.Size([1, 1000, 754]) | 2\n",
-      "5542     | torch.Size([1, 1000, 754]) | 2\n",
-      "5543     | torch.Size([1, 1000, 754]) | 2\n",
-      "5544     | torch.Size([1, 1000, 754]) | 2\n",
-      "5545     | torch.Size([1, 1000, 754]) | 2\n",
-      "5546     | torch.Size([1, 1000, 754]) | 2\n",
-      "5547     | torch.Size([1, 1000, 754]) | 2\n",
-      "5548     | torch.Size([1, 1000, 754]) | 2\n",
-      "5549     | torch.Size([1, 1000, 754]) | 2\n",
-      "5550     | torch.Size([1, 1000, 754]) | 2\n",
-      "5551     | torch.Size([1, 1000, 754]) | 2\n",
-      "5552     | torch.Size([1, 1000, 754]) | 2\n",
-      "5553     | torch.Size([1, 1000, 754]) | 2\n",
-      "5554     | torch.Size([1, 1000, 754]) | 2\n",
-      "5555     | torch.Size([1, 1000, 754]) | 2\n",
-      "5556     | torch.Size([1, 1000, 754]) | 2\n",
-      "5557     | torch.Size([1, 1000, 754]) | 2\n",
-      "5558     | torch.Size([1, 1000, 754]) | 2\n",
-      "5559     | torch.Size([1, 1000, 754]) | 2\n",
-      "5560     | torch.Size([1, 1000, 754]) | 2\n",
-      "5561     | torch.Size([1, 1000, 754]) | 2\n",
-      "5562     | torch.Size([1, 1000, 754]) | 2\n",
-      "5563     | torch.Size([1, 1000, 754]) | 2\n",
-      "5564     | torch.Size([1, 1000, 754]) | 2\n",
-      "5565     | torch.Size([1, 1000, 754]) | 2\n",
-      "5566     | torch.Size([1, 1000, 754]) | 2\n",
-      "5567     | torch.Size([1, 1000, 754]) | 2\n",
-      "5568     | torch.Size([1, 1000, 754]) | 2\n",
-      "5569     | torch.Size([1, 1000, 754]) | 2\n",
-      "5570     | torch.Size([1, 1000, 754]) | 2\n",
-      "5571     | torch.Size([1, 1000, 754]) | 2\n",
-      "5572     | torch.Size([1, 1000, 754]) | 2\n",
-      "5573     | torch.Size([1, 1000, 754]) | 2\n",
-      "5574     | torch.Size([1, 1000, 754]) | 2\n",
-      "5575     | torch.Size([1, 1000, 754]) | 2\n",
-      "5576     | torch.Size([1, 1000, 754]) | 2\n",
-      "5577     | torch.Size([1, 1000, 754]) | 2\n",
-      "5578     | torch.Size([1, 1000, 754]) | 2\n",
-      "5579     | torch.Size([1, 1000, 754]) | 2\n",
-      "5580     | torch.Size([1, 1000, 754]) | 2\n",
-      "5581     | torch.Size([1, 1000, 754]) | 2\n",
-      "5582     | torch.Size([1, 1000, 802]) | 2\n",
-      "5583     | torch.Size([1, 1000, 754]) | 2\n",
-      "5584     | torch.Size([1, 1000, 754]) | 2\n",
-      "5585     | torch.Size([1, 1000, 754]) | 2\n",
-      "5586     | torch.Size([1, 1000, 754]) | 2\n",
-      "5587     | torch.Size([1, 1000, 754]) | 2\n",
-      "5588     | torch.Size([1, 1000, 754]) | 2\n",
-      "5589     | torch.Size([1, 1000, 754]) | 2\n",
-      "5590     | torch.Size([1, 1000, 754]) | 2\n",
-      "5591     | torch.Size([1, 1000, 754]) | 2\n",
-      "5592     | torch.Size([1, 1000, 754]) | 2\n",
-      "5593     | torch.Size([1, 1000, 754]) | 2\n",
-      "5594     | torch.Size([1, 1000, 754]) | 2\n",
-      "5595     | torch.Size([1, 1000, 754]) | 2\n",
-      "5596     | torch.Size([1, 1000, 754]) | 2\n",
-      "5597     | torch.Size([1, 1000, 754]) | 2\n",
-      "5598     | torch.Size([1, 1000, 754]) | 2\n",
-      "5599     | torch.Size([1, 1000, 754]) | 2\n",
-      "5600     | torch.Size([1, 1000, 754]) | 2\n",
-      "5601     | torch.Size([1, 1000, 754]) | 2\n",
-      "5602     | torch.Size([1, 1000, 754]) | 2\n",
-      "5603     | torch.Size([1, 1000, 754]) | 2\n",
-      "5604     | torch.Size([1, 1000, 754]) | 2\n",
-      "5605     | torch.Size([1, 1000, 754]) | 2\n",
-      "5606     | torch.Size([1, 1000, 754]) | 2\n",
-      "5607     | torch.Size([1, 1000, 754]) | 2\n",
-      "5608     | torch.Size([1, 1000, 754]) | 2\n",
-      "5609     | torch.Size([1, 1000, 754]) | 2\n",
-      "5610     | torch.Size([1, 1000, 754]) | 2\n",
-      "5611     | torch.Size([1, 1000, 754]) | 2\n",
-      "5612     | torch.Size([1, 1000, 754]) | 2\n",
-      "5613     | torch.Size([1, 1000, 754]) | 2\n",
-      "5614     | torch.Size([1, 1000, 754]) | 2\n",
-      "5615     | torch.Size([1, 1000, 754]) | 2\n",
-      "5616     | torch.Size([1, 1000, 754]) | 2\n",
-      "5617     | torch.Size([1, 1000, 754]) | 2\n",
-      "5618     | torch.Size([1, 1000, 754]) | 2\n",
-      "5619     | torch.Size([1, 1000, 754]) | 2\n",
-      "5620     | torch.Size([1, 1000, 754]) | 2\n",
-      "5621     | torch.Size([1, 1000, 754]) | 2\n",
-      "5622     | torch.Size([1, 1000, 754]) | 2\n",
-      "5623     | torch.Size([1, 1000, 754]) | 2\n",
-      "5624     | torch.Size([1, 1000, 754]) | 2\n",
-      "5625     | torch.Size([1, 1000, 754]) | 2\n",
-      "5626     | torch.Size([1, 1000, 754]) | 2\n",
-      "5627     | torch.Size([1, 1000, 754]) | 2\n",
-      "5628     | torch.Size([1, 1000, 754]) | 2\n",
-      "5629     | torch.Size([1, 1000, 754]) | 2\n",
-      "5630     | torch.Size([1, 1000, 754]) | 2\n",
-      "5631     | torch.Size([1, 1000, 754]) | 2\n",
-      "5632     | torch.Size([1, 1000, 754]) | 2\n",
-      "5633     | torch.Size([1, 1000, 754]) | 2\n",
-      "5634     | torch.Size([1, 1000, 754]) | 2\n",
-      "5635     | torch.Size([1, 1000, 754]) | 2\n",
-      "5636     | torch.Size([1, 1000, 754]) | 2\n",
-      "5637     | torch.Size([1, 1000, 754]) | 2\n",
-      "5638     | torch.Size([1, 1000, 754]) | 2\n",
-      "5639     | torch.Size([1, 1000, 754]) | 2\n",
-      "5640     | torch.Size([1, 1000, 754]) | 2\n",
-      "5641     | torch.Size([1, 1000, 754]) | 2\n",
-      "5642     | torch.Size([1, 1000, 754]) | 2\n",
-      "5643     | torch.Size([1, 1000, 754]) | 2\n",
-      "5644     | torch.Size([1, 1000, 754]) | 2\n",
-      "5645     | torch.Size([1, 1000, 754]) | 2\n",
-      "5646     | torch.Size([1, 1000, 754]) | 2\n",
-      "5647     | torch.Size([1, 1000, 754]) | 2\n",
-      "5648     | torch.Size([1, 1000, 754]) | 2\n",
-      "5649     | torch.Size([1, 1000, 754]) | 2\n",
-      "5650     | torch.Size([1, 1000, 754]) | 2\n",
-      "5651     | torch.Size([1, 1000, 754]) | 2\n",
-      "5652     | torch.Size([1, 1000, 754]) | 2\n",
-      "5653     | torch.Size([1, 1000, 754]) | 2\n",
-      "5654     | torch.Size([1, 1000, 754]) | 2\n",
-      "5655     | torch.Size([1, 1000, 754]) | 2\n",
-      "5656     | torch.Size([1, 1000, 754]) | 2\n",
-      "5657     | torch.Size([1, 1000, 754]) | 2\n",
-      "5658     | torch.Size([1, 1000, 754]) | 2\n",
-      "5659     | torch.Size([1, 1000, 754]) | 2\n",
-      "5660     | torch.Size([1, 1000, 754]) | 2\n",
-      "5661     | torch.Size([1, 1000, 754]) | 2\n",
-      "5662     | torch.Size([1, 1000, 802]) | 2\n",
-      "5663     | torch.Size([1, 1000, 754]) | 2\n",
-      "5664     | torch.Size([1, 1000, 754]) | 2\n",
-      "5665     | torch.Size([1, 1000, 754]) | 2\n",
-      "5666     | torch.Size([1, 1000, 754]) | 2\n",
-      "5667     | torch.Size([1, 1000, 754]) | 2\n",
-      "5668     | torch.Size([1, 1000, 754]) | 2\n",
-      "5669     | torch.Size([1, 1000, 754]) | 2\n",
-      "5670     | torch.Size([1, 1000, 754]) | 2\n",
-      "5671     | torch.Size([1, 1000, 754]) | 2\n",
-      "5672     | torch.Size([1, 1000, 754]) | 2\n",
-      "5673     | torch.Size([1, 1000, 754]) | 2\n",
-      "5674     | torch.Size([1, 1000, 754]) | 2\n",
-      "5675     | torch.Size([1, 1000, 754]) | 2\n",
-      "5676     | torch.Size([1, 1000, 754]) | 2\n",
-      "5677     | torch.Size([1, 1000, 754]) | 2\n",
-      "5678     | torch.Size([1, 1000, 754]) | 2\n",
-      "5679     | torch.Size([1, 1000, 754]) | 2\n",
-      "5680     | torch.Size([1, 1000, 754]) | 2\n",
-      "5681     | torch.Size([1, 1000, 754]) | 2\n",
-      "5682     | torch.Size([1, 1000, 754]) | 2\n",
-      "5683     | torch.Size([1, 1000, 754]) | 2\n",
-      "5684     | torch.Size([1, 1000, 754]) | 2\n",
-      "5685     | torch.Size([1, 1000, 754]) | 2\n",
-      "5686     | torch.Size([1, 1000, 754]) | 2\n",
-      "5687     | torch.Size([1, 1000, 754]) | 2\n",
-      "5688     | torch.Size([1, 1000, 754]) | 2\n",
-      "5689     | torch.Size([1, 1000, 754]) | 2\n",
-      "5690     | torch.Size([1, 1000, 754]) | 2\n",
-      "5691     | torch.Size([1, 1000, 754]) | 2\n",
-      "5692     | torch.Size([1, 1000, 754]) | 2\n",
-      "5693     | torch.Size([1, 1000, 754]) | 2\n",
-      "5694     | torch.Size([1, 1000, 754]) | 2\n",
-      "5695     | torch.Size([1, 1000, 754]) | 2\n",
-      "5696     | torch.Size([1, 1000, 754]) | 2\n",
-      "5697     | torch.Size([1, 1000, 754]) | 2\n",
-      "5698     | torch.Size([1, 1000, 754]) | 2\n",
-      "5699     | torch.Size([1, 1000, 754]) | 2\n",
-      "5700     | torch.Size([1, 1000, 754]) | 2\n",
-      "5701     | torch.Size([1, 1000, 754]) | 2\n",
-      "5702     | torch.Size([1, 1000, 754]) | 2\n",
-      "5703     | torch.Size([1, 1000, 754]) | 2\n",
-      "5704     | torch.Size([1, 1000, 754]) | 2\n",
-      "5705     | torch.Size([1, 1000, 754]) | 2\n",
-      "5706     | torch.Size([1, 1000, 754]) | 2\n",
-      "5707     | torch.Size([1, 1000, 754]) | 2\n",
-      "5708     | torch.Size([1, 1000, 754]) | 2\n",
-      "5709     | torch.Size([1, 1000, 754]) | 2\n",
-      "5710     | torch.Size([1, 1000, 754]) | 2\n",
-      "5711     | torch.Size([1, 1000, 754]) | 2\n",
-      "5712     | torch.Size([1, 1000, 754]) | 2\n",
-      "5713     | torch.Size([1, 1000, 754]) | 2\n",
-      "5714     | torch.Size([1, 1000, 754]) | 2\n",
-      "5715     | torch.Size([1, 1000, 754]) | 2\n",
-      "5716     | torch.Size([1, 1000, 754]) | 2\n",
-      "5717     | torch.Size([1, 1000, 754]) | 2\n",
-      "5718     | torch.Size([1, 1000, 754]) | 2\n",
-      "5719     | torch.Size([1, 1000, 754]) | 2\n",
-      "5720     | torch.Size([1, 1000, 754]) | 2\n",
-      "5721     | torch.Size([1, 1000, 754]) | 2\n",
-      "5722     | torch.Size([1, 1000, 754]) | 2\n",
-      "5723     | torch.Size([1, 1000, 754]) | 2\n",
-      "5724     | torch.Size([1, 1000, 754]) | 2\n",
-      "5725     | torch.Size([1, 1000, 754]) | 2\n",
-      "5726     | torch.Size([1, 1000, 754]) | 2\n",
-      "5727     | torch.Size([1, 1000, 754]) | 2\n",
-      "5728     | torch.Size([1, 1000, 754]) | 2\n",
-      "5729     | torch.Size([1, 1000, 754]) | 2\n",
-      "5730     | torch.Size([1, 1000, 754]) | 2\n",
-      "5731     | torch.Size([1, 1000, 754]) | 2\n",
-      "5732     | torch.Size([1, 1000, 754]) | 2\n",
-      "5733     | torch.Size([1, 1000, 754]) | 2\n",
-      "5734     | torch.Size([1, 1000, 754]) | 2\n",
-      "5735     | torch.Size([1, 1000, 754]) | 2\n",
-      "5736     | torch.Size([1, 1000, 754]) | 2\n",
-      "5737     | torch.Size([1, 1000, 754]) | 2\n",
-      "5738     | torch.Size([1, 1000, 754]) | 2\n",
-      "5739     | torch.Size([1, 1000, 754]) | 2\n",
-      "5740     | torch.Size([1, 1000, 754]) | 2\n",
-      "5741     | torch.Size([1, 1000, 754]) | 2\n",
-      "5742     | torch.Size([1, 1000, 802]) | 2\n",
-      "5743     | torch.Size([1, 1000, 754]) | 2\n",
-      "5744     | torch.Size([1, 1000, 754]) | 2\n",
-      "5745     | torch.Size([1, 1000, 754]) | 2\n",
-      "5746     | torch.Size([1, 1000, 754]) | 2\n",
-      "5747     | torch.Size([1, 1000, 754]) | 2\n",
-      "5748     | torch.Size([1, 1000, 754]) | 2\n",
-      "5749     | torch.Size([1, 1000, 754]) | 2\n",
-      "5750     | torch.Size([1, 1000, 754]) | 2\n",
-      "5751     | torch.Size([1, 1000, 754]) | 2\n",
-      "5752     | torch.Size([1, 1000, 754]) | 2\n",
-      "5753     | torch.Size([1, 1000, 754]) | 2\n",
-      "5754     | torch.Size([1, 1000, 754]) | 2\n",
-      "5755     | torch.Size([1, 1000, 754]) | 2\n",
-      "5756     | torch.Size([1, 1000, 754]) | 2\n",
-      "5757     | torch.Size([1, 1000, 754]) | 2\n",
-      "5758     | torch.Size([1, 1000, 754]) | 2\n",
-      "5759     | torch.Size([1, 1000, 754]) | 2\n",
-      "5760     | torch.Size([1, 1000, 754]) | 2\n",
-      "5761     | torch.Size([1, 1000, 754]) | 2\n",
-      "5762     | torch.Size([1, 1000, 754]) | 2\n",
-      "5763     | torch.Size([1, 1000, 754]) | 2\n",
-      "5764     | torch.Size([1, 1000, 754]) | 2\n",
-      "5765     | torch.Size([1, 1000, 754]) | 2\n",
-      "5766     | torch.Size([1, 1000, 754]) | 2\n",
-      "5767     | torch.Size([1, 1000, 754]) | 2\n",
-      "5768     | torch.Size([1, 1000, 754]) | 2\n",
-      "5769     | torch.Size([1, 1000, 802]) | 2\n",
-      "5770     | torch.Size([1, 1000, 754]) | 2\n",
-      "5771     | torch.Size([1, 1000, 754]) | 2\n",
-      "5772     | torch.Size([1, 1000, 754]) | 2\n",
-      "5773     | torch.Size([1, 1000, 754]) | 2\n",
-      "5774     | torch.Size([1, 1000, 754]) | 2\n",
-      "5775     | torch.Size([1, 1000, 754]) | 2\n",
-      "5776     | torch.Size([1, 1000, 754]) | 2\n",
-      "5777     | torch.Size([1, 1000, 754]) | 2\n",
-      "5778     | torch.Size([1, 1000, 754]) | 2\n",
-      "5779     | torch.Size([1, 1000, 754]) | 2\n",
-      "5780     | torch.Size([1, 1000, 754]) | 2\n",
-      "5781     | torch.Size([1, 1000, 754]) | 2\n",
-      "5782     | torch.Size([1, 1000, 754]) | 2\n",
-      "5783     | torch.Size([1, 1000, 754]) | 2\n",
-      "5784     | torch.Size([1, 1000, 754]) | 2\n",
-      "5785     | torch.Size([1, 1000, 754]) | 2\n",
-      "5786     | torch.Size([1, 1000, 754]) | 2\n",
-      "5787     | torch.Size([1, 1000, 754]) | 2\n",
-      "5788     | torch.Size([1, 1000, 754]) | 2\n",
-      "5789     | torch.Size([1, 1000, 754]) | 2\n",
-      "5790     | torch.Size([1, 1000, 754]) | 2\n",
-      "5791     | torch.Size([1, 1000, 754]) | 2\n",
-      "5792     | torch.Size([1, 1000, 754]) | 2\n",
-      "5793     | torch.Size([1, 1000, 754]) | 2\n",
-      "5794     | torch.Size([1, 1000, 754]) | 2\n",
-      "5795     | torch.Size([1, 1000, 754]) | 2\n",
-      "5796     | torch.Size([1, 1000, 754]) | 2\n",
-      "5797     | torch.Size([1, 1000, 754]) | 2\n",
-      "5798     | torch.Size([1, 1000, 754]) | 2\n",
-      "5799     | torch.Size([1, 1000, 754]) | 2\n",
-      "5800     | torch.Size([1, 1000, 754]) | 2\n",
-      "5801     | torch.Size([1, 1000, 754]) | 2\n",
-      "5802     | torch.Size([1, 1000, 754]) | 2\n",
-      "5803     | torch.Size([1, 1000, 754]) | 2\n",
-      "5804     | torch.Size([1, 1000, 754]) | 2\n",
-      "5805     | torch.Size([1, 1000, 754]) | 2\n",
-      "5806     | torch.Size([1, 1000, 754]) | 2\n",
-      "5807     | torch.Size([1, 1000, 754]) | 2\n",
-      "5808     | torch.Size([1, 1000, 754]) | 2\n",
-      "5809     | torch.Size([1, 1000, 754]) | 2\n",
-      "5810     | torch.Size([1, 1000, 754]) | 2\n",
-      "5811     | torch.Size([1, 1000, 754]) | 2\n",
-      "5812     | torch.Size([1, 1000, 754]) | 2\n",
-      "5813     | torch.Size([1, 1000, 754]) | 2\n",
-      "5814     | torch.Size([1, 1000, 754]) | 2\n",
-      "5815     | torch.Size([1, 1000, 754]) | 2\n",
-      "5816     | torch.Size([1, 1000, 754]) | 2\n",
-      "5817     | torch.Size([1, 1000, 754]) | 2\n",
-      "5818     | torch.Size([1, 1000, 754]) | 2\n",
-      "5819     | torch.Size([1, 1000, 754]) | 2\n",
-      "5820     | torch.Size([1, 1000, 754]) | 2\n",
-      "5821     | torch.Size([1, 1000, 754]) | 2\n",
-      "5822     | torch.Size([1, 1000, 754]) | 2\n",
-      "5823     | torch.Size([1, 1000, 754]) | 2\n",
-      "5824     | torch.Size([1, 1000, 754]) | 2\n",
-      "5825     | torch.Size([1, 1000, 802]) | 2\n",
-      "5826     | torch.Size([1, 1000, 754]) | 2\n",
-      "5827     | torch.Size([1, 1000, 754]) | 2\n",
-      "5828     | torch.Size([1, 1000, 754]) | 2\n",
-      "5829     | torch.Size([1, 1000, 754]) | 2\n",
-      "5830     | torch.Size([1, 1000, 754]) | 2\n",
-      "5831     | torch.Size([1, 1000, 802]) | 2\n",
-      "5832     | torch.Size([1, 1000, 754]) | 2\n",
-      "5833     | torch.Size([1, 1000, 754]) | 2\n",
-      "5834     | torch.Size([1, 1000, 754]) | 2\n",
-      "5835     | torch.Size([1, 1000, 754]) | 2\n",
-      "5836     | torch.Size([1, 1000, 754]) | 2\n",
-      "5837     | torch.Size([1, 1000, 754]) | 2\n",
-      "5838     | torch.Size([1, 1000, 754]) | 2\n",
-      "5839     | torch.Size([1, 1000, 754]) | 2\n",
-      "5840     | torch.Size([1, 1000, 754]) | 2\n",
-      "5841     | torch.Size([1, 1000, 754]) | 2\n",
-      "5842     | torch.Size([1, 1000, 754]) | 2\n",
-      "5843     | torch.Size([1, 1000, 754]) | 2\n",
-      "5844     | torch.Size([1, 1000, 754]) | 2\n",
-      "5845     | torch.Size([1, 1000, 754]) | 2\n",
-      "5846     | torch.Size([1, 1000, 754]) | 2\n",
-      "5847     | torch.Size([1, 1000, 754]) | 2\n",
-      "5848     | torch.Size([1, 1000, 754]) | 2\n",
-      "5849     | torch.Size([1, 1000, 754]) | 2\n",
-      "5850     | torch.Size([1, 1000, 754]) | 2\n",
-      "5851     | torch.Size([1, 1000, 754]) | 2\n",
-      "5852     | torch.Size([1, 1000, 754]) | 2\n",
-      "5853     | torch.Size([1, 1000, 754]) | 2\n",
-      "5854     | torch.Size([1, 1000, 754]) | 2\n",
-      "5855     | torch.Size([1, 1000, 754]) | 2\n",
-      "5856     | torch.Size([1, 1000, 754]) | 2\n",
-      "5857     | torch.Size([1, 1000, 754]) | 2\n",
-      "5858     | torch.Size([1, 1000, 754]) | 2\n",
-      "5859     | torch.Size([1, 1000, 754]) | 2\n",
-      "5860     | torch.Size([1, 1000, 754]) | 2\n",
-      "5861     | torch.Size([1, 1000, 754]) | 2\n",
-      "5862     | torch.Size([1, 1000, 754]) | 2\n",
-      "5863     | torch.Size([1, 1000, 754]) | 2\n",
-      "5864     | torch.Size([1, 1000, 754]) | 2\n",
-      "5865     | torch.Size([1, 1000, 754]) | 2\n",
-      "5866     | torch.Size([1, 1000, 754]) | 2\n",
-      "5867     | torch.Size([1, 1000, 754]) | 2\n",
-      "5868     | torch.Size([1, 1000, 754]) | 2\n",
-      "5869     | torch.Size([1, 1000, 754]) | 2\n",
-      "5870     | torch.Size([1, 1000, 754]) | 2\n",
-      "5871     | torch.Size([1, 1000, 754]) | 2\n",
-      "5872     | torch.Size([1, 1000, 754]) | 2\n",
-      "5873     | torch.Size([1, 1000, 754]) | 2\n",
-      "5874     | torch.Size([1, 1000, 754]) | 2\n",
-      "5875     | torch.Size([1, 1000, 754]) | 2\n",
-      "5876     | torch.Size([1, 1000, 754]) | 2\n",
-      "5877     | torch.Size([1, 1000, 754]) | 2\n",
-      "5878     | torch.Size([1, 1000, 754]) | 2\n",
-      "5879     | torch.Size([1, 1000, 754]) | 2\n",
-      "5880     | torch.Size([1, 1000, 754]) | 2\n",
-      "5881     | torch.Size([1, 1000, 754]) | 2\n",
-      "5882     | torch.Size([1, 1000, 754]) | 2\n",
-      "5883     | torch.Size([1, 1000, 754]) | 2\n",
-      "5884     | torch.Size([1, 1000, 754]) | 2\n",
-      "5885     | torch.Size([1, 1000, 754]) | 2\n",
-      "5886     | torch.Size([1, 1000, 754]) | 2\n",
-      "5887     | torch.Size([1, 1000, 754]) | 2\n",
-      "5888     | torch.Size([1, 1000, 754]) | 2\n",
-      "5889     | torch.Size([1, 1000, 754]) | 2\n",
-      "5890     | torch.Size([1, 1000, 754]) | 2\n",
-      "5891     | torch.Size([1, 1000, 754]) | 2\n",
-      "5892     | torch.Size([1, 1000, 754]) | 2\n",
-      "5893     | torch.Size([1, 1000, 754]) | 2\n",
-      "5894     | torch.Size([1, 1000, 802]) | 2\n",
-      "5895     | torch.Size([1, 1000, 754]) | 2\n",
-      "5896     | torch.Size([1, 1000, 754]) | 2\n",
-      "5897     | torch.Size([1, 1000, 754]) | 2\n",
-      "5898     | torch.Size([1, 1000, 754]) | 2\n",
-      "5899     | torch.Size([1, 1000, 754]) | 2\n",
-      "5900     | torch.Size([1, 1000, 754]) | 2\n",
-      "5901     | torch.Size([1, 1000, 754]) | 2\n",
-      "5902     | torch.Size([1, 1000, 754]) | 2\n",
-      "5903     | torch.Size([1, 1000, 754]) | 2\n",
-      "5904     | torch.Size([1, 1000, 754]) | 2\n",
-      "5905     | torch.Size([1, 1000, 754]) | 2\n",
-      "5906     | torch.Size([1, 1000, 754]) | 2\n",
-      "5907     | torch.Size([1, 1000, 754]) | 2\n",
-      "5908     | torch.Size([1, 1000, 754]) | 2\n",
-      "5909     | torch.Size([1, 1000, 754]) | 2\n",
-      "5910     | torch.Size([1, 1000, 754]) | 2\n",
-      "5911     | torch.Size([1, 1000, 754]) | 2\n",
-      "5912     | torch.Size([1, 1000, 754]) | 2\n",
-      "5913     | torch.Size([1, 1000, 754]) | 2\n",
-      "5914     | torch.Size([1, 1000, 754]) | 2\n",
-      "5915     | torch.Size([1, 1000, 754]) | 2\n",
-      "5916     | torch.Size([1, 1000, 754]) | 2\n",
-      "5917     | torch.Size([1, 1000, 754]) | 2\n",
-      "5918     | torch.Size([1, 1000, 754]) | 2\n",
-      "5919     | torch.Size([1, 1000, 754]) | 2\n",
-      "5920     | torch.Size([1, 1000, 754]) | 2\n",
-      "5921     | torch.Size([1, 1000, 754]) | 2\n",
-      "5922     | torch.Size([1, 1000, 754]) | 2\n",
-      "5923     | torch.Size([1, 1000, 754]) | 2\n",
-      "5924     | torch.Size([1, 1000, 754]) | 2\n",
-      "5925     | torch.Size([1, 1000, 754]) | 2\n",
-      "5926     | torch.Size([1, 1000, 754]) | 2\n",
-      "5927     | torch.Size([1, 1000, 754]) | 2\n",
-      "5928     | torch.Size([1, 1000, 754]) | 2\n",
-      "5929     | torch.Size([1, 1000, 754]) | 2\n",
-      "5930     | torch.Size([1, 1000, 754]) | 2\n",
-      "5931     | torch.Size([1, 1000, 754]) | 2\n",
-      "5932     | torch.Size([1, 1000, 754]) | 2\n",
-      "5933     | torch.Size([1, 1000, 754]) | 2\n",
-      "5934     | torch.Size([1, 1000, 754]) | 2\n",
-      "5935     | torch.Size([1, 1000, 754]) | 2\n",
-      "5936     | torch.Size([1, 1000, 754]) | 2\n",
-      "5937     | torch.Size([1, 1000, 754]) | 2\n",
-      "5938     | torch.Size([1, 1000, 754]) | 2\n",
-      "5939     | torch.Size([1, 1000, 754]) | 2\n",
-      "5940     | torch.Size([1, 1000, 754]) | 2\n",
-      "5941     | torch.Size([1, 1000, 754]) | 2\n",
-      "5942     | torch.Size([1, 1000, 754]) | 2\n",
-      "5943     | torch.Size([1, 1000, 754]) | 2\n",
-      "5944     | torch.Size([1, 1000, 754]) | 2\n",
-      "5945     | torch.Size([1, 1000, 754]) | 2\n",
-      "5946     | torch.Size([1, 1000, 754]) | 2\n",
-      "5947     | torch.Size([1, 1000, 754]) | 2\n",
-      "5948     | torch.Size([1, 1000, 754]) | 2\n",
-      "5949     | torch.Size([1, 1000, 754]) | 2\n",
-      "5950     | torch.Size([1, 1000, 754]) | 2\n",
-      "5951     | torch.Size([1, 1000, 754]) | 2\n",
-      "5952     | torch.Size([1, 1000, 754]) | 2\n",
-      "5953     | torch.Size([1, 1000, 754]) | 2\n",
-      "5954     | torch.Size([1, 1000, 754]) | 2\n",
-      "5955     | torch.Size([1, 1000, 754]) | 2\n",
-      "5956     | torch.Size([1, 1000, 754]) | 2\n",
-      "5957     | torch.Size([1, 1000, 754]) | 2\n",
-      "5958     | torch.Size([1, 1000, 754]) | 2\n",
-      "5959     | torch.Size([1, 1000, 754]) | 2\n",
-      "5960     | torch.Size([1, 1000, 802]) | 2\n",
-      "5961     | torch.Size([1, 1000, 754]) | 2\n",
-      "5962     | torch.Size([1, 1000, 754]) | 2\n",
-      "5963     | torch.Size([1, 1000, 754]) | 2\n",
-      "5964     | torch.Size([1, 1000, 754]) | 2\n",
-      "5965     | torch.Size([1, 1000, 754]) | 2\n",
-      "5966     | torch.Size([1, 1000, 754]) | 2\n",
-      "5967     | torch.Size([1, 1000, 754]) | 2\n",
-      "5968     | torch.Size([1, 1000, 754]) | 2\n",
-      "5969     | torch.Size([1, 1000, 754]) | 2\n",
-      "5970     | torch.Size([1, 1000, 754]) | 2\n",
-      "5971     | torch.Size([1, 1000, 754]) | 2\n",
-      "5972     | torch.Size([1, 1000, 754]) | 2\n",
-      "5973     | torch.Size([1, 1000, 754]) | 2\n",
-      "5974     | torch.Size([1, 1000, 754]) | 2\n",
-      "5975     | torch.Size([1, 1000, 754]) | 2\n",
-      "5976     | torch.Size([1, 1000, 754]) | 2\n",
-      "5977     | torch.Size([1, 1000, 754]) | 2\n",
-      "5978     | torch.Size([1, 1000, 754]) | 2\n",
-      "5979     | torch.Size([1, 1000, 754]) | 2\n",
-      "5980     | torch.Size([1, 1000, 754]) | 2\n",
-      "5981     | torch.Size([1, 1000, 754]) | 2\n",
-      "5982     | torch.Size([1, 1000, 762]) | 2\n",
-      "5983     | torch.Size([1, 1000, 754]) | 2\n",
-      "5984     | torch.Size([1, 1000, 754]) | 2\n",
-      "5985     | torch.Size([1, 1000, 754]) | 2\n",
-      "5986     | torch.Size([1, 1000, 754]) | 2\n",
-      "5987     | torch.Size([1, 1000, 754]) | 2\n",
-      "5988     | torch.Size([1, 1000, 754]) | 2\n",
-      "5989     | torch.Size([1, 1000, 754]) | 2\n",
-      "5990     | torch.Size([1, 1000, 754]) | 2\n",
-      "5991     | torch.Size([1, 1000, 754]) | 2\n",
-      "5992     | torch.Size([1, 1000, 802]) | 2\n",
-      "5993     | torch.Size([1, 1000, 754]) | 2\n",
-      "5994     | torch.Size([1, 1000, 754]) | 2\n",
-      "5995     | torch.Size([1, 1000, 754]) | 2\n",
-      "5996     | torch.Size([1, 1000, 754]) | 2\n",
-      "5997     | torch.Size([1, 1000, 754]) | 2\n",
-      "5998     | torch.Size([1, 1000, 754]) | 2\n",
-      "5999     | torch.Size([1, 1000, 754]) | 2\n",
-      "6000     | torch.Size([1, 1000, 754]) | 2\n",
-      "6001     | torch.Size([1, 1000, 754]) | 2\n",
-      "6002     | torch.Size([1, 1000, 754]) | 2\n",
-      "6003     | torch.Size([1, 1000, 754]) | 2\n",
-      "6004     | torch.Size([1, 1000, 754]) | 2\n",
-      "6005     | torch.Size([1, 1000, 754]) | 2\n",
-      "6006     | torch.Size([1, 1000, 754]) | 2\n",
-      "6007     | torch.Size([1, 1000, 754]) | 2\n",
-      "6008     | torch.Size([1, 1000, 754]) | 2\n",
-      "6009     | torch.Size([1, 1000, 754]) | 2\n",
-      "6010     | torch.Size([1, 1000, 754]) | 2\n",
-      "6011     | torch.Size([1, 1000, 754]) | 2\n",
-      "6012     | torch.Size([1, 1000, 754]) | 2\n",
-      "6013     | torch.Size([1, 1000, 754]) | 2\n",
-      "6014     | torch.Size([1, 1000, 754]) | 2\n",
-      "6015     | torch.Size([1, 1000, 754]) | 2\n",
-      "6016     | torch.Size([1, 1000, 754]) | 2\n",
-      "6017     | torch.Size([1, 1000, 754]) | 2\n",
-      "6018     | torch.Size([1, 1000, 771]) | 2\n",
-      "6019     | torch.Size([1, 1000, 754]) | 2\n",
-      "6020     | torch.Size([1, 1000, 754]) | 2\n",
-      "6021     | torch.Size([1, 1000, 754]) | 2\n",
-      "6022     | torch.Size([1, 1000, 754]) | 2\n",
-      "6023     | torch.Size([1, 1000, 754]) | 2\n",
-      "6024     | torch.Size([1, 1000, 754]) | 2\n",
-      "6025     | torch.Size([1, 1000, 754]) | 2\n",
-      "6026     | torch.Size([1, 1000, 754]) | 2\n",
-      "6027     | torch.Size([1, 1000, 754]) | 2\n",
-      "6028     | torch.Size([1, 1000, 754]) | 2\n",
-      "6029     | torch.Size([1, 1000, 754]) | 2\n",
-      "6030     | torch.Size([1, 1000, 754]) | 2\n",
-      "6031     | torch.Size([1, 1000, 754]) | 2\n",
-      "6032     | torch.Size([1, 1000, 754]) | 2\n",
-      "6033     | torch.Size([1, 1000, 754]) | 2\n",
-      "6034     | torch.Size([1, 1000, 754]) | 2\n",
-      "6035     | torch.Size([1, 1000, 754]) | 2\n",
-      "6036     | torch.Size([1, 1000, 754]) | 2\n",
-      "6037     | torch.Size([1, 1000, 754]) | 2\n",
-      "6038     | torch.Size([1, 1000, 754]) | 2\n",
-      "6039     | torch.Size([1, 1000, 762]) | 2\n",
-      "6040     | torch.Size([1, 1000, 754]) | 2\n",
-      "6041     | torch.Size([1, 1000, 754]) | 2\n",
-      "6042     | torch.Size([1, 1000, 754]) | 2\n",
-      "6043     | torch.Size([1, 1000, 754]) | 2\n",
-      "6044     | torch.Size([1, 1000, 754]) | 2\n",
-      "6045     | torch.Size([1, 1000, 754]) | 2\n",
-      "6046     | torch.Size([1, 1000, 754]) | 2\n",
-      "6047     | torch.Size([1, 1000, 754]) | 2\n",
-      "6048     | torch.Size([1, 1000, 754]) | 2\n",
-      "6049     | torch.Size([1, 1000, 754]) | 2\n",
-      "6050     | torch.Size([1, 1000, 754]) | 2\n",
-      "6051     | torch.Size([1, 1000, 754]) | 2\n",
-      "6052     | torch.Size([1, 1000, 754]) | 2\n",
-      "6053     | torch.Size([1, 1000, 754]) | 2\n",
-      "6054     | torch.Size([1, 1000, 754]) | 2\n",
-      "6055     | torch.Size([1, 1000, 754]) | 2\n",
-      "6056     | torch.Size([1, 1000, 754]) | 2\n",
-      "6057     | torch.Size([1, 1000, 802]) | 2\n",
-      "6058     | torch.Size([1, 1000, 754]) | 2\n",
-      "6059     | torch.Size([1, 1000, 754]) | 2\n",
-      "6060     | torch.Size([1, 1000, 754]) | 2\n",
-      "6061     | torch.Size([1, 1000, 754]) | 2\n",
-      "6062     | torch.Size([1, 1000, 802]) | 2\n",
-      "6063     | torch.Size([1, 1000, 754]) | 2\n",
-      "6064     | torch.Size([1, 1000, 754]) | 2\n",
-      "6065     | torch.Size([1, 1000, 754]) | 2\n",
-      "6066     | torch.Size([1, 1000, 754]) | 2\n",
-      "6067     | torch.Size([1, 1000, 754]) | 2\n",
-      "6068     | torch.Size([1, 1000, 754]) | 2\n",
-      "6069     | torch.Size([1, 1000, 754]) | 2\n",
-      "6070     | torch.Size([1, 1000, 754]) | 2\n",
-      "6071     | torch.Size([1, 1000, 754]) | 2\n",
-      "6072     | torch.Size([1, 1000, 754]) | 2\n",
-      "6073     | torch.Size([1, 1000, 754]) | 2\n",
-      "6074     | torch.Size([1, 1000, 754]) | 2\n",
-      "6075     | torch.Size([1, 1000, 754]) | 2\n",
-      "6076     | torch.Size([1, 1000, 754]) | 2\n",
-      "6077     | torch.Size([1, 1000, 754]) | 2\n",
-      "6078     | torch.Size([1, 1000, 754]) | 2\n",
-      "6079     | torch.Size([1, 1000, 754]) | 2\n",
-      "6080     | torch.Size([1, 1000, 754]) | 2\n",
-      "6081     | torch.Size([1, 1000, 754]) | 2\n",
-      "6082     | torch.Size([1, 1000, 754]) | 2\n",
-      "6083     | torch.Size([1, 1000, 754]) | 2\n",
-      "6084     | torch.Size([1, 1000, 754]) | 2\n",
-      "6085     | torch.Size([1, 1000, 754]) | 2\n",
-      "6086     | torch.Size([1, 1000, 754]) | 2\n",
-      "6087     | torch.Size([1, 1000, 754]) | 2\n",
-      "6088     | torch.Size([1, 1000, 754]) | 2\n",
-      "6089     | torch.Size([1, 1000, 754]) | 2\n",
-      "6090     | torch.Size([1, 1000, 754]) | 2\n",
-      "6091     | torch.Size([1, 1000, 754]) | 2\n",
-      "6092     | torch.Size([1, 1000, 754]) | 2\n",
-      "6093     | torch.Size([1, 1000, 754]) | 2\n",
-      "6094     | torch.Size([1, 1000, 754]) | 2\n",
-      "6095     | torch.Size([1, 1000, 754]) | 2\n",
-      "6096     | torch.Size([1, 1000, 754]) | 2\n",
-      "6097     | torch.Size([1, 1000, 754]) | 2\n",
-      "6098     | torch.Size([1, 1000, 754]) | 2\n",
-      "6099     | torch.Size([1, 1000, 802]) | 2\n",
-      "6100     | torch.Size([1, 1000, 754]) | 2\n",
-      "6101     | torch.Size([1, 1000, 754]) | 2\n",
-      "6102     | torch.Size([1, 1000, 754]) | 2\n",
-      "6103     | torch.Size([1, 1000, 754]) | 2\n",
-      "6104     | torch.Size([1, 1000, 754]) | 2\n",
-      "6105     | torch.Size([1, 1000, 754]) | 2\n",
-      "6106     | torch.Size([1, 1000, 754]) | 2\n",
-      "6107     | torch.Size([1, 1000, 754]) | 2\n",
-      "6108     | torch.Size([1, 1000, 754]) | 2\n",
-      "6109     | torch.Size([1, 1000, 754]) | 2\n",
-      "6110     | torch.Size([1, 1000, 754]) | 2\n",
-      "6111     | torch.Size([1, 1000, 754]) | 2\n",
-      "6112     | torch.Size([1, 1000, 754]) | 2\n",
-      "6113     | torch.Size([1, 1000, 754]) | 2\n",
-      "6114     | torch.Size([1, 1000, 754]) | 2\n",
-      "6115     | torch.Size([1, 1000, 754]) | 2\n",
-      "6116     | torch.Size([1, 1000, 754]) | 2\n",
-      "6117     | torch.Size([1, 1000, 754]) | 2\n",
-      "6118     | torch.Size([1, 1000, 754]) | 2\n",
-      "6119     | torch.Size([1, 1000, 754]) | 2\n",
-      "6120     | torch.Size([1, 1000, 754]) | 2\n",
-      "6121     | torch.Size([1, 1000, 754]) | 2\n",
-      "6122     | torch.Size([1, 1000, 754]) | 2\n",
-      "6123     | torch.Size([1, 1000, 754]) | 2\n",
-      "6124     | torch.Size([1, 1000, 754]) | 2\n",
-      "6125     | torch.Size([1, 1000, 754]) | 2\n",
-      "6126     | torch.Size([1, 1000, 754]) | 2\n",
-      "6127     | torch.Size([1, 1000, 754]) | 2\n",
-      "6128     | torch.Size([1, 1000, 754]) | 2\n",
-      "6129     | torch.Size([1, 1000, 754]) | 2\n",
-      "6130     | torch.Size([1, 1000, 754]) | 2\n",
-      "6131     | torch.Size([1, 1000, 754]) | 2\n",
-      "6132     | torch.Size([1, 1000, 754]) | 2\n",
-      "6133     | torch.Size([1, 1000, 754]) | 2\n",
-      "6134     | torch.Size([1, 1000, 802]) | 2\n",
-      "6135     | torch.Size([1, 1000, 754]) | 2\n",
-      "6136     | torch.Size([1, 1000, 754]) | 2\n",
-      "6137     | torch.Size([1, 1000, 754]) | 2\n",
-      "6138     | torch.Size([1, 1000, 754]) | 2\n",
-      "6139     | torch.Size([1, 1000, 754]) | 2\n",
-      "6140     | torch.Size([1, 1000, 754]) | 2\n",
-      "6141     | torch.Size([1, 1000, 754]) | 2\n",
-      "6142     | torch.Size([1, 1000, 754]) | 2\n",
-      "6143     | torch.Size([1, 1000, 754]) | 2\n",
-      "6144     | torch.Size([1, 1000, 754]) | 2\n",
-      "6145     | torch.Size([1, 1000, 754]) | 2\n",
-      "6146     | torch.Size([1, 1000, 754]) | 2\n",
-      "6147     | torch.Size([1, 1000, 754]) | 2\n",
-      "6148     | torch.Size([1, 1000, 754]) | 2\n",
-      "6149     | torch.Size([1, 1000, 754]) | 2\n",
-      "6150     | torch.Size([1, 1000, 754]) | 2\n",
-      "6151     | torch.Size([1, 1000, 754]) | 2\n",
-      "6152     | torch.Size([1, 1000, 754]) | 2\n",
-      "6153     | torch.Size([1, 1000, 754]) | 2\n",
-      "6154     | torch.Size([1, 1000, 754]) | 2\n",
-      "6155     | torch.Size([1, 1000, 754]) | 2\n",
-      "6156     | torch.Size([1, 1000, 754]) | 2\n",
-      "6157     | torch.Size([1, 1000, 754]) | 2\n",
-      "6158     | torch.Size([1, 1000, 754]) | 2\n",
-      "6159     | torch.Size([1, 1000, 754]) | 2\n",
-      "6160     | torch.Size([1, 1000, 754]) | 2\n",
-      "6161     | torch.Size([1, 1000, 754]) | 2\n",
-      "6162     | torch.Size([1, 1000, 754]) | 2\n",
-      "6163     | torch.Size([1, 1000, 754]) | 2\n",
-      "6164     | torch.Size([1, 1000, 754]) | 2\n",
-      "6165     | torch.Size([1, 1000, 754]) | 2\n",
-      "6166     | torch.Size([1, 1000, 754]) | 2\n",
-      "6167     | torch.Size([1, 1000, 754]) | 2\n",
-      "6168     | torch.Size([1, 1000, 754]) | 2\n",
-      "6169     | torch.Size([1, 1000, 754]) | 2\n",
-      "6170     | torch.Size([1, 1000, 754]) | 2\n",
-      "6171     | torch.Size([1, 1000, 754]) | 2\n",
-      "6172     | torch.Size([1, 1000, 754]) | 2\n",
-      "6173     | torch.Size([1, 1000, 762]) | 2\n",
-      "6174     | torch.Size([1, 1000, 754]) | 2\n",
-      "6175     | torch.Size([1, 1000, 754]) | 2\n",
-      "6176     | torch.Size([1, 1000, 754]) | 2\n",
-      "6177     | torch.Size([1, 1000, 754]) | 2\n",
-      "6178     | torch.Size([1, 1000, 754]) | 2\n",
-      "6179     | torch.Size([1, 1000, 754]) | 2\n",
-      "6180     | torch.Size([1, 1000, 754]) | 2\n",
-      "6181     | torch.Size([1, 1000, 754]) | 2\n",
-      "6182     | torch.Size([1, 1000, 754]) | 2\n",
-      "6183     | torch.Size([1, 1000, 754]) | 2\n",
-      "6184     | torch.Size([1, 1000, 754]) | 2\n",
-      "6185     | torch.Size([1, 1000, 754]) | 2\n",
-      "6186     | torch.Size([1, 1000, 754]) | 2\n",
-      "6187     | torch.Size([1, 1000, 754]) | 2\n",
-      "6188     | torch.Size([1, 1000, 754]) | 2\n",
-      "6189     | torch.Size([1, 1000, 754]) | 2\n",
-      "6190     | torch.Size([1, 1000, 754]) | 2\n",
-      "6191     | torch.Size([1, 1000, 754]) | 2\n",
-      "6192     | torch.Size([1, 1000, 754]) | 2\n",
-      "6193     | torch.Size([1, 1000, 754]) | 2\n",
-      "6194     | torch.Size([1, 1000, 754]) | 2\n",
-      "6195     | torch.Size([1, 1000, 754]) | 2\n",
-      "6196     | torch.Size([1, 1000, 754]) | 2\n",
-      "6197     | torch.Size([1, 1000, 754]) | 2\n",
-      "6198     | torch.Size([1, 1000, 754]) | 2\n",
-      "6199     | torch.Size([1, 1000, 754]) | 2\n",
-      "6200     | torch.Size([1, 1000, 754]) | 2\n",
-      "6201     | torch.Size([1, 1000, 762]) | 2\n",
-      "6202     | torch.Size([1, 1000, 754]) | 2\n",
-      "6203     | torch.Size([1, 1000, 754]) | 2\n",
-      "6204     | torch.Size([1, 1000, 754]) | 2\n",
-      "6205     | torch.Size([1, 1000, 754]) | 2\n",
-      "6206     | torch.Size([1, 1000, 754]) | 2\n",
-      "6207     | torch.Size([1, 1000, 762]) | 2\n",
-      "6208     | torch.Size([1, 1000, 754]) | 2\n",
-      "6209     | torch.Size([1, 1000, 754]) | 2\n",
-      "6210     | torch.Size([1, 1000, 754]) | 2\n",
-      "6211     | torch.Size([1, 1000, 754]) | 2\n",
-      "6212     | torch.Size([1, 1000, 754]) | 2\n",
-      "6213     | torch.Size([1, 1000, 754]) | 2\n",
-      "6214     | torch.Size([1, 1000, 754]) | 2\n",
-      "6215     | torch.Size([1, 1000, 754]) | 2\n",
-      "6216     | torch.Size([1, 1000, 754]) | 2\n",
-      "6217     | torch.Size([1, 1000, 754]) | 2\n",
-      "6218     | torch.Size([1, 1000, 754]) | 2\n",
-      "6219     | torch.Size([1, 1000, 754]) | 2\n",
-      "6220     | torch.Size([1, 1000, 754]) | 2\n",
-      "6221     | torch.Size([1, 1000, 754]) | 2\n",
-      "6222     | torch.Size([1, 1000, 754]) | 2\n",
-      "6223     | torch.Size([1, 1000, 754]) | 2\n",
-      "6224     | torch.Size([1, 1000, 754]) | 2\n",
-      "6225     | torch.Size([1, 1000, 754]) | 2\n",
-      "6226     | torch.Size([1, 1000, 754]) | 2\n",
-      "6227     | torch.Size([1, 1000, 754]) | 2\n",
-      "6228     | torch.Size([1, 1000, 754]) | 2\n",
-      "6229     | torch.Size([1, 1000, 754]) | 2\n",
-      "6230     | torch.Size([1, 1000, 754]) | 2\n",
-      "6231     | torch.Size([1, 1000, 754]) | 2\n",
-      "6232     | torch.Size([1, 1000, 754]) | 2\n",
-      "6233     | torch.Size([1, 1000, 754]) | 2\n",
-      "6234     | torch.Size([1, 1000, 754]) | 2\n",
-      "6235     | torch.Size([1, 1000, 754]) | 2\n",
-      "6236     | torch.Size([1, 1000, 754]) | 2\n",
-      "6237     | torch.Size([1, 1000, 754]) | 2\n",
-      "6238     | torch.Size([1, 1000, 754]) | 2\n",
-      "6239     | torch.Size([1, 1000, 754]) | 2\n",
-      "6240     | torch.Size([1, 1000, 754]) | 2\n",
-      "6241     | torch.Size([1, 1000, 754]) | 2\n",
-      "6242     | torch.Size([1, 1000, 754]) | 2\n",
-      "6243     | torch.Size([1, 1000, 754]) | 2\n",
-      "6244     | torch.Size([1, 1000, 754]) | 2\n",
-      "6245     | torch.Size([1, 1000, 754]) | 2\n",
-      "6246     | torch.Size([1, 1000, 754]) | 2\n",
-      "6247     | torch.Size([1, 1000, 754]) | 2\n",
-      "6248     | torch.Size([1, 1000, 754]) | 2\n",
-      "6249     | torch.Size([1, 1000, 754]) | 2\n",
-      "6250     | torch.Size([1, 1000, 754]) | 2\n",
-      "6251     | torch.Size([1, 1000, 754]) | 2\n",
-      "6252     | torch.Size([1, 1000, 754]) | 2\n",
-      "6253     | torch.Size([1, 1000, 754]) | 2\n",
-      "6254     | torch.Size([1, 1000, 754]) | 2\n",
-      "6255     | torch.Size([1, 1000, 754]) | 2\n",
-      "6256     | torch.Size([1, 1000, 754]) | 2\n",
-      "6257     | torch.Size([1, 1000, 754]) | 2\n",
-      "6258     | torch.Size([1, 1000, 754]) | 2\n",
-      "6259     | torch.Size([1, 1000, 754]) | 2\n",
-      "6260     | torch.Size([1, 1000, 754]) | 2\n",
-      "6261     | torch.Size([1, 1000, 754]) | 2\n",
-      "6262     | torch.Size([1, 1000, 754]) | 2\n",
-      "6263     | torch.Size([1, 1000, 754]) | 2\n",
-      "6264     | torch.Size([1, 1000, 754]) | 2\n",
-      "6265     | torch.Size([1, 1000, 754]) | 2\n",
-      "6266     | torch.Size([1, 1000, 754]) | 2\n",
-      "6267     | torch.Size([1, 1000, 754]) | 2\n",
-      "6268     | torch.Size([1, 1000, 754]) | 2\n",
-      "6269     | torch.Size([1, 1000, 754]) | 2\n",
-      "6270     | torch.Size([1, 1000, 754]) | 2\n",
-      "6271     | torch.Size([1, 1000, 754]) | 2\n",
-      "6272     | torch.Size([1, 1000, 754]) | 2\n",
-      "6273     | torch.Size([1, 1000, 754]) | 2\n",
-      "6274     | torch.Size([1, 1000, 754]) | 2\n",
-      "6275     | torch.Size([1, 1000, 754]) | 2\n",
-      "6276     | torch.Size([1, 1000, 754]) | 2\n",
-      "6277     | torch.Size([1, 1000, 754]) | 2\n",
-      "6278     | torch.Size([1, 1000, 754]) | 2\n",
-      "6279     | torch.Size([1, 1000, 754]) | 2\n",
-      "6280     | torch.Size([1, 1000, 754]) | 2\n",
-      "6281     | torch.Size([1, 1000, 754]) | 2\n",
-      "6282     | torch.Size([1, 1000, 802]) | 2\n",
-      "6283     | torch.Size([1, 1000, 754]) | 2\n",
-      "6284     | torch.Size([1, 1000, 754]) | 2\n",
-      "6285     | torch.Size([1, 1000, 754]) | 2\n",
-      "6286     | torch.Size([1, 1000, 754]) | 2\n",
-      "6287     | torch.Size([1, 1000, 754]) | 2\n",
-      "6288     | torch.Size([1, 1000, 754]) | 2\n",
-      "6289     | torch.Size([1, 1000, 754]) | 2\n",
-      "6290     | torch.Size([1, 1000, 754]) | 2\n",
-      "6291     | torch.Size([1, 1000, 754]) | 2\n",
-      "6292     | torch.Size([1, 1000, 754]) | 2\n",
-      "6293     | torch.Size([1, 1000, 754]) | 2\n",
-      "6294     | torch.Size([1, 1000, 754]) | 2\n",
-      "6295     | torch.Size([1, 1000, 754]) | 2\n",
-      "6296     | torch.Size([1, 1000, 754]) | 2\n",
-      "6297     | torch.Size([1, 1000, 754]) | 2\n",
-      "6298     | torch.Size([1, 1000, 754]) | 2\n",
-      "6299     | torch.Size([1, 1000, 754]) | 2\n",
-      "6300     | torch.Size([1, 1000, 754]) | 2\n",
-      "6301     | torch.Size([1, 1000, 754]) | 2\n",
-      "6302     | torch.Size([1, 1000, 754]) | 2\n",
-      "6303     | torch.Size([1, 1000, 754]) | 2\n",
-      "6304     | torch.Size([1, 1000, 754]) | 2\n",
-      "6305     | torch.Size([1, 1000, 754]) | 2\n",
-      "6306     | torch.Size([1, 1000, 754]) | 2\n",
-      "6307     | torch.Size([1, 1000, 754]) | 2\n",
-      "6308     | torch.Size([1, 1000, 754]) | 2\n",
-      "6309     | torch.Size([1, 1000, 754]) | 2\n",
-      "6310     | torch.Size([1, 1000, 754]) | 2\n",
-      "6311     | torch.Size([1, 1000, 754]) | 2\n",
-      "6312     | torch.Size([1, 1000, 754]) | 2\n",
-      "6313     | torch.Size([1, 1000, 754]) | 2\n",
-      "6314     | torch.Size([1, 1000, 754]) | 2\n",
-      "6315     | torch.Size([1, 1000, 754]) | 2\n",
-      "6316     | torch.Size([1, 1000, 754]) | 2\n",
-      "6317     | torch.Size([1, 1000, 754]) | 2\n",
-      "6318     | torch.Size([1, 1000, 754]) | 2\n",
-      "6319     | torch.Size([1, 1000, 754]) | 2\n",
-      "6320     | torch.Size([1, 1000, 754]) | 2\n",
-      "6321     | torch.Size([1, 1000, 754]) | 2\n",
-      "6322     | torch.Size([1, 1000, 754]) | 2\n",
-      "6323     | torch.Size([1, 1000, 754]) | 2\n",
-      "6324     | torch.Size([1, 1000, 754]) | 2\n",
-      "6325     | torch.Size([1, 1000, 754]) | 2\n",
-      "6326     | torch.Size([1, 1000, 754]) | 2\n",
-      "6327     | torch.Size([1, 1000, 754]) | 2\n",
-      "6328     | torch.Size([1, 1000, 762]) | 2\n",
-      "6329     | torch.Size([1, 1000, 754]) | 2\n",
-      "6330     | torch.Size([1, 1000, 754]) | 2\n",
-      "6331     | torch.Size([1, 1000, 754]) | 2\n",
-      "6332     | torch.Size([1, 1000, 754]) | 2\n",
-      "6333     | torch.Size([1, 1000, 754]) | 2\n",
-      "6334     | torch.Size([1, 1000, 754]) | 2\n",
-      "6335     | torch.Size([1, 1000, 754]) | 2\n",
-      "6336     | torch.Size([1, 1000, 754]) | 2\n",
-      "6337     | torch.Size([1, 1000, 754]) | 2\n",
-      "6338     | torch.Size([1, 1000, 754]) | 2\n",
-      "6339     | torch.Size([1, 1000, 754]) | 2\n",
-      "6340     | torch.Size([1, 1000, 754]) | 2\n",
-      "6341     | torch.Size([1, 1000, 754]) | 2\n",
-      "6342     | torch.Size([1, 1000, 754]) | 2\n",
-      "6343     | torch.Size([1, 1000, 754]) | 2\n",
-      "6344     | torch.Size([1, 1000, 754]) | 2\n",
-      "6345     | torch.Size([1, 1000, 754]) | 2\n",
-      "6346     | torch.Size([1, 1000, 754]) | 2\n",
-      "6347     | torch.Size([1, 1000, 802]) | 2\n",
-      "6348     | torch.Size([1, 1000, 754]) | 2\n",
-      "6349     | torch.Size([1, 1000, 754]) | 2\n",
-      "6350     | torch.Size([1, 1000, 754]) | 2\n",
-      "6351     | torch.Size([1, 1000, 754]) | 2\n",
-      "6352     | torch.Size([1, 1000, 754]) | 2\n",
-      "6353     | torch.Size([1, 1000, 754]) | 2\n",
-      "6354     | torch.Size([1, 1000, 754]) | 2\n",
-      "6355     | torch.Size([1, 1000, 754]) | 2\n",
-      "6356     | torch.Size([1, 1000, 754]) | 2\n",
-      "6357     | torch.Size([1, 1000, 754]) | 2\n",
-      "6358     | torch.Size([1, 1000, 754]) | 2\n",
-      "6359     | torch.Size([1, 1000, 754]) | 2\n",
-      "6360     | torch.Size([1, 1000, 754]) | 2\n",
-      "6361     | torch.Size([1, 1000, 754]) | 2\n",
-      "6362     | torch.Size([1, 1000, 802]) | 2\n",
-      "6363     | torch.Size([1, 1000, 754]) | 2\n",
-      "6364     | torch.Size([1, 1000, 754]) | 2\n",
-      "6365     | torch.Size([1, 1000, 754]) | 2\n",
-      "6366     | torch.Size([1, 1000, 754]) | 2\n",
-      "6367     | torch.Size([1, 1000, 754]) | 2\n",
-      "6368     | torch.Size([1, 1000, 802]) | 2\n",
-      "6369     | torch.Size([1, 1000, 754]) | 2\n",
-      "6370     | torch.Size([1, 1000, 754]) | 2\n",
-      "6371     | torch.Size([1, 1000, 754]) | 2\n",
-      "6372     | torch.Size([1, 1000, 762]) | 2\n",
-      "6373     | torch.Size([1, 1000, 754]) | 2\n",
-      "6374     | torch.Size([1, 1000, 754]) | 2\n",
-      "6375     | torch.Size([1, 1000, 754]) | 2\n",
-      "6376     | torch.Size([1, 1000, 802]) | 2\n",
-      "6377     | torch.Size([1, 1000, 754]) | 2\n",
-      "6378     | torch.Size([1, 1000, 754]) | 2\n",
-      "6379     | torch.Size([1, 1000, 754]) | 2\n",
-      "6380     | torch.Size([1, 1000, 754]) | 2\n",
-      "6381     | torch.Size([1, 1000, 754]) | 2\n",
-      "6382     | torch.Size([1, 1000, 754]) | 2\n",
-      "6383     | torch.Size([1, 1000, 754]) | 2\n",
-      "6384     | torch.Size([1, 1000, 754]) | 2\n",
-      "6385     | torch.Size([1, 1000, 754]) | 2\n",
-      "6386     | torch.Size([1, 1000, 754]) | 2\n",
-      "6387     | torch.Size([1, 1000, 754]) | 2\n",
-      "6388     | torch.Size([1, 1000, 754]) | 2\n",
-      "6389     | torch.Size([1, 1000, 754]) | 2\n",
-      "6390     | torch.Size([1, 1000, 754]) | 2\n",
-      "6391     | torch.Size([1, 1000, 754]) | 2\n",
-      "6392     | torch.Size([1, 1000, 754]) | 2\n",
-      "6393     | torch.Size([1, 1000, 754]) | 2\n",
-      "6394     | torch.Size([1, 1000, 754]) | 2\n",
-      "6395     | torch.Size([1, 1000, 754]) | 2\n",
-      "6396     | torch.Size([1, 1000, 754]) | 2\n",
-      "6397     | torch.Size([1, 1000, 802]) | 2\n",
-      "6398     | torch.Size([1, 1000, 754]) | 2\n",
-      "6399     | torch.Size([1, 1000, 754]) | 2\n",
-      "6400     | torch.Size([1, 1000, 754]) | 2\n",
-      "6401     | torch.Size([1, 1000, 754]) | 2\n",
-      "6402     | torch.Size([1, 1000, 754]) | 2\n",
-      "6403     | torch.Size([1, 1000, 754]) | 2\n",
-      "6404     | torch.Size([1, 1000, 754]) | 2\n",
-      "6405     | torch.Size([1, 1000, 754]) | 2\n",
-      "6406     | torch.Size([1, 1000, 754]) | 2\n",
-      "6407     | torch.Size([1, 1000, 754]) | 2\n",
-      "6408     | torch.Size([1, 1000, 754]) | 2\n",
-      "6409     | torch.Size([1, 1000, 754]) | 2\n",
-      "6410     | torch.Size([1, 1000, 754]) | 2\n",
-      "6411     | torch.Size([1, 1000, 754]) | 2\n",
-      "6412     | torch.Size([1, 1000, 754]) | 2\n",
-      "6413     | torch.Size([1, 1000, 754]) | 2\n",
-      "6414     | torch.Size([1, 1000, 754]) | 2\n",
-      "6415     | torch.Size([1, 1000, 754]) | 2\n",
-      "6416     | torch.Size([1, 1000, 754]) | 2\n",
-      "6417     | torch.Size([1, 1000, 754]) | 2\n",
-      "6418     | torch.Size([1, 1000, 754]) | 2\n",
-      "6419     | torch.Size([1, 1000, 754]) | 2\n",
-      "6420     | torch.Size([1, 1000, 754]) | 2\n",
-      "6421     | torch.Size([1, 1000, 754]) | 2\n",
-      "6422     | torch.Size([1, 1000, 754]) | 2\n",
-      "6423     | torch.Size([1, 1000, 754]) | 2\n",
-      "6424     | torch.Size([1, 1000, 754]) | 2\n",
-      "6425     | torch.Size([1, 1000, 754]) | 2\n",
-      "6426     | torch.Size([1, 1000, 754]) | 2\n",
-      "6427     | torch.Size([1, 1000, 754]) | 2\n",
-      "6428     | torch.Size([1, 1000, 754]) | 2\n",
-      "6429     | torch.Size([1, 1000, 754]) | 2\n",
-      "6430     | torch.Size([1, 1000, 754]) | 2\n",
-      "6431     | torch.Size([1, 1000, 754]) | 2\n",
-      "6432     | torch.Size([1, 1000, 754]) | 2\n",
-      "6433     | torch.Size([1, 1000, 754]) | 2\n",
-      "6434     | torch.Size([1, 1000, 754]) | 2\n",
-      "6435     | torch.Size([1, 1000, 754]) | 2\n",
-      "6436     | torch.Size([1, 1000, 754]) | 2\n",
-      "6437     | torch.Size([1, 1000, 754]) | 2\n",
-      "6438     | torch.Size([1, 1000, 754]) | 2\n",
-      "6439     | torch.Size([1, 1000, 754]) | 2\n",
-      "6440     | torch.Size([1, 1000, 754]) | 2\n",
-      "6441     | torch.Size([1, 1000, 754]) | 2\n",
-      "6442     | torch.Size([1, 1000, 754]) | 2\n",
-      "6443     | torch.Size([1, 1000, 754]) | 2\n",
-      "6444     | torch.Size([1, 1000, 754]) | 2\n",
-      "6445     | torch.Size([1, 1000, 754]) | 2\n",
-      "6446     | torch.Size([1, 1000, 754]) | 2\n",
-      "6447     | torch.Size([1, 1000, 754]) | 2\n",
-      "6448     | torch.Size([1, 1000, 754]) | 2\n",
-      "6449     | torch.Size([1, 1000, 754]) | 2\n",
-      "6450     | torch.Size([1, 1000, 754]) | 2\n",
-      "6451     | torch.Size([1, 1000, 754]) | 2\n",
-      "6452     | torch.Size([1, 1000, 754]) | 2\n",
-      "6453     | torch.Size([1, 1000, 754]) | 2\n",
-      "6454     | torch.Size([1, 1000, 754]) | 2\n",
-      "6455     | torch.Size([1, 1000, 754]) | 2\n",
-      "6456     | torch.Size([1, 1000, 754]) | 2\n",
-      "6457     | torch.Size([1, 1000, 754]) | 2\n",
-      "6458     | torch.Size([1, 1000, 754]) | 2\n",
-      "6459     | torch.Size([1, 1000, 754]) | 2\n",
-      "6460     | torch.Size([1, 1000, 754]) | 2\n",
-      "6461     | torch.Size([1, 1000, 754]) | 2\n",
-      "6462     | torch.Size([1, 1000, 754]) | 2\n",
-      "6463     | torch.Size([1, 1000, 754]) | 2\n",
-      "6464     | torch.Size([1, 1000, 754]) | 2\n",
-      "6465     | torch.Size([1, 1000, 754]) | 2\n",
-      "6466     | torch.Size([1, 1000, 754]) | 2\n",
-      "6467     | torch.Size([1, 1000, 754]) | 2\n",
-      "6468     | torch.Size([1, 1000, 754]) | 2\n",
-      "6469     | torch.Size([1, 1000, 754]) | 2\n",
-      "6470     | torch.Size([1, 1000, 802]) | 2\n",
-      "6471     | torch.Size([1, 1000, 754]) | 2\n",
-      "6472     | torch.Size([1, 1000, 754]) | 2\n",
-      "6473     | torch.Size([1, 1000, 754]) | 2\n",
-      "6474     | torch.Size([1, 1000, 754]) | 2\n",
-      "6475     | torch.Size([1, 1000, 754]) | 2\n",
-      "6476     | torch.Size([1, 1000, 754]) | 2\n",
-      "6477     | torch.Size([1, 1000, 754]) | 2\n",
-      "6478     | torch.Size([1, 1000, 754]) | 2\n",
-      "6479     | torch.Size([1, 1000, 754]) | 2\n",
-      "6480     | torch.Size([1, 1000, 802]) | 2\n",
-      "6481     | torch.Size([1, 1000, 754]) | 2\n",
-      "6482     | torch.Size([1, 1000, 754]) | 2\n",
-      "6483     | torch.Size([1, 1000, 754]) | 2\n",
-      "6484     | torch.Size([1, 1000, 754]) | 2\n",
-      "6485     | torch.Size([1, 1000, 754]) | 2\n",
-      "6486     | torch.Size([1, 1000, 754]) | 2\n",
-      "6487     | torch.Size([1, 1000, 754]) | 2\n",
-      "6488     | torch.Size([1, 1000, 754]) | 2\n",
-      "6489     | torch.Size([1, 1000, 754]) | 2\n",
-      "6490     | torch.Size([1, 1000, 754]) | 2\n",
-      "6491     | torch.Size([1, 1000, 754]) | 2\n",
-      "6492     | torch.Size([1, 1000, 754]) | 2\n",
-      "6493     | torch.Size([1, 1000, 754]) | 2\n",
-      "6494     | torch.Size([1, 1000, 754]) | 2\n",
-      "6495     | torch.Size([1, 1000, 754]) | 2\n",
-      "6496     | torch.Size([1, 1000, 754]) | 2\n",
-      "6497     | torch.Size([1, 1000, 754]) | 2\n",
-      "6498     | torch.Size([1, 1000, 754]) | 2\n",
-      "6499     | torch.Size([1, 1000, 754]) | 2\n",
-      "6500     | torch.Size([1, 1000, 754]) | 2\n",
-      "6501     | torch.Size([1, 1000, 754]) | 2\n",
-      "6502     | torch.Size([1, 1000, 754]) | 2\n",
-      "6503     | torch.Size([1, 1000, 802]) | 2\n",
-      "6504     | torch.Size([1, 1000, 754]) | 2\n",
-      "6505     | torch.Size([1, 1000, 754]) | 2\n",
-      "6506     | torch.Size([1, 1000, 754]) | 2\n",
-      "6507     | torch.Size([1, 1000, 754]) | 2\n",
-      "6508     | torch.Size([1, 1000, 754]) | 2\n",
-      "6509     | torch.Size([1, 1000, 754]) | 2\n",
-      "6510     | torch.Size([1, 1000, 754]) | 2\n",
-      "6511     | torch.Size([1, 1000, 754]) | 2\n",
-      "6512     | torch.Size([1, 1000, 802]) | 2\n",
-      "6513     | torch.Size([1, 1000, 754]) | 2\n",
-      "6514     | torch.Size([1, 1000, 754]) | 2\n",
-      "6515     | torch.Size([1, 1000, 754]) | 2\n",
-      "6516     | torch.Size([1, 1000, 754]) | 2\n",
-      "6517     | torch.Size([1, 1000, 754]) | 2\n",
-      "6518     | torch.Size([1, 1000, 754]) | 2\n",
-      "6519     | torch.Size([1, 1000, 802]) | 2\n",
-      "6520     | torch.Size([1, 1000, 754]) | 2\n",
-      "6521     | torch.Size([1, 1000, 754]) | 2\n",
-      "6522     | torch.Size([1, 1000, 754]) | 2\n",
-      "6523     | torch.Size([1, 1000, 754]) | 2\n",
-      "6524     | torch.Size([1, 1000, 754]) | 2\n",
-      "6525     | torch.Size([1, 1000, 754]) | 2\n",
-      "6526     | torch.Size([1, 1000, 754]) | 2\n",
-      "6527     | torch.Size([1, 1000, 754]) | 2\n",
-      "6528     | torch.Size([1, 1000, 754]) | 2\n",
-      "6529     | torch.Size([1, 1000, 754]) | 2\n",
-      "6530     | torch.Size([1, 1000, 754]) | 2\n",
-      "6531     | torch.Size([1, 1000, 754]) | 2\n",
-      "6532     | torch.Size([1, 1000, 754]) | 2\n",
-      "6533     | torch.Size([1, 1000, 754]) | 2\n",
-      "6534     | torch.Size([1, 1000, 754]) | 2\n",
-      "6535     | torch.Size([1, 1000, 754]) | 2\n",
-      "6536     | torch.Size([1, 1000, 754]) | 2\n",
-      "6537     | torch.Size([1, 1000, 754]) | 2\n",
-      "6538     | torch.Size([1, 1000, 754]) | 2\n",
-      "6539     | torch.Size([1, 1000, 754]) | 2\n",
-      "6540     | torch.Size([1, 1000, 754]) | 2\n",
-      "6541     | torch.Size([1, 1000, 754]) | 2\n",
-      "6542     | torch.Size([1, 1000, 802]) | 2\n",
-      "6543     | torch.Size([1, 1000, 754]) | 2\n",
-      "6544     | torch.Size([1, 1000, 754]) | 2\n",
-      "6545     | torch.Size([1, 1000, 754]) | 2\n",
-      "6546     | torch.Size([1, 1000, 754]) | 2\n",
-      "6547     | torch.Size([1, 1000, 754]) | 2\n",
-      "6548     | torch.Size([1, 1000, 754]) | 2\n",
-      "6549     | torch.Size([1, 1000, 754]) | 2\n",
-      "6550     | torch.Size([1, 1000, 754]) | 2\n",
-      "6551     | torch.Size([1, 1000, 754]) | 2\n",
-      "6552     | torch.Size([1, 1000, 754]) | 2\n",
-      "6553     | torch.Size([1, 1000, 754]) | 2\n",
-      "6554     | torch.Size([1, 1000, 754]) | 2\n",
-      "6555     | torch.Size([1, 1000, 754]) | 2\n",
-      "6556     | torch.Size([1, 1000, 754]) | 2\n",
-      "6557     | torch.Size([1, 1000, 802]) | 2\n",
-      "6558     | torch.Size([1, 1000, 754]) | 2\n",
-      "6559     | torch.Size([1, 1000, 754]) | 2\n",
-      "6560     | torch.Size([1, 1000, 754]) | 2\n",
-      "6561     | torch.Size([1, 1000, 754]) | 2\n",
-      "6562     | torch.Size([1, 1000, 754]) | 2\n",
-      "6563     | torch.Size([1, 1000, 754]) | 2\n",
-      "6564     | torch.Size([1, 1000, 754]) | 2\n",
-      "6565     | torch.Size([1, 1000, 754]) | 2\n",
-      "6566     | torch.Size([1, 1000, 754]) | 2\n",
-      "6567     | torch.Size([1, 1000, 754]) | 2\n",
-      "6568     | torch.Size([1, 1000, 754]) | 2\n",
-      "6569     | torch.Size([1, 1000, 754]) | 2\n",
-      "6570     | torch.Size([1, 1000, 754]) | 2\n",
-      "6571     | torch.Size([1, 1000, 754]) | 2\n",
-      "6572     | torch.Size([1, 1000, 754]) | 2\n",
-      "6573     | torch.Size([1, 1000, 754]) | 2\n",
-      "6574     | torch.Size([1, 1000, 754]) | 2\n",
-      "6575     | torch.Size([1, 1000, 754]) | 2\n",
-      "6576     | torch.Size([1, 1000, 754]) | 2\n",
-      "6577     | torch.Size([1, 1000, 754]) | 2\n",
-      "6578     | torch.Size([1, 1000, 754]) | 2\n",
-      "6579     | torch.Size([1, 1000, 754]) | 2\n",
-      "6580     | torch.Size([1, 1000, 754]) | 2\n",
-      "6581     | torch.Size([1, 1000, 754]) | 2\n",
-      "6582     | torch.Size([1, 1000, 754]) | 2\n",
-      "6583     | torch.Size([1, 1000, 754]) | 2\n",
-      "6584     | torch.Size([1, 1000, 754]) | 2\n",
-      "6585     | torch.Size([1, 1000, 754]) | 2\n",
-      "6586     | torch.Size([1, 1000, 754]) | 2\n",
-      "6587     | torch.Size([1, 1000, 762]) | 2\n",
-      "6588     | torch.Size([1, 1000, 754]) | 2\n",
-      "6589     | torch.Size([1, 1000, 754]) | 2\n",
-      "6590     | torch.Size([1, 1000, 754]) | 2\n",
-      "6591     | torch.Size([1, 1000, 754]) | 2\n",
-      "6592     | torch.Size([1, 1000, 754]) | 2\n",
-      "6593     | torch.Size([1, 1000, 754]) | 2\n",
-      "6594     | torch.Size([1, 1000, 754]) | 2\n",
-      "6595     | torch.Size([1, 1000, 754]) | 2\n",
-      "6596     | torch.Size([1, 1000, 754]) | 2\n",
-      "6597     | torch.Size([1, 1000, 754]) | 2\n",
-      "6598     | torch.Size([1, 1000, 754]) | 2\n",
-      "6599     | torch.Size([1, 1000, 754]) | 2\n",
-      "6600     | torch.Size([1, 1000, 754]) | 2\n",
-      "6601     | torch.Size([1, 1000, 754]) | 2\n",
-      "6602     | torch.Size([1, 1000, 754]) | 2\n",
-      "6603     | torch.Size([1, 1000, 754]) | 2\n",
-      "6604     | torch.Size([1, 1000, 754]) | 2\n",
-      "6605     | torch.Size([1, 1000, 802]) | 2\n",
-      "6606     | torch.Size([1, 1000, 754]) | 2\n",
-      "6607     | torch.Size([1, 1000, 754]) | 2\n",
-      "6608     | torch.Size([1, 1000, 754]) | 2\n",
-      "6609     | torch.Size([1, 1000, 754]) | 2\n",
-      "6610     | torch.Size([1, 1000, 754]) | 2\n",
-      "6611     | torch.Size([1, 1000, 754]) | 2\n",
-      "6612     | torch.Size([1, 1000, 754]) | 2\n",
-      "6613     | torch.Size([1, 1000, 754]) | 2\n",
-      "6614     | torch.Size([1, 1000, 754]) | 2\n",
-      "6615     | torch.Size([1, 1000, 754]) | 2\n",
-      "6616     | torch.Size([1, 1000, 754]) | 2\n",
-      "6617     | torch.Size([1, 1000, 754]) | 2\n",
-      "6618     | torch.Size([1, 1000, 754]) | 2\n",
-      "6619     | torch.Size([1, 1000, 754]) | 2\n",
-      "6620     | torch.Size([1, 1000, 754]) | 2\n",
-      "6621     | torch.Size([1, 1000, 754]) | 2\n",
-      "6622     | torch.Size([1, 1000, 754]) | 2\n",
-      "6623     | torch.Size([1, 1000, 754]) | 2\n",
-      "6624     | torch.Size([1, 1000, 754]) | 2\n",
-      "6625     | torch.Size([1, 1000, 754]) | 2\n",
-      "6626     | torch.Size([1, 1000, 754]) | 2\n",
-      "6627     | torch.Size([1, 1000, 754]) | 2\n",
-      "6628     | torch.Size([1, 1000, 754]) | 2\n",
-      "6629     | torch.Size([1, 1000, 754]) | 2\n",
-      "6630     | torch.Size([1, 1000, 754]) | 2\n",
-      "6631     | torch.Size([1, 1000, 754]) | 2\n",
-      "6632     | torch.Size([1, 1000, 754]) | 2\n",
-      "6633     | torch.Size([1, 1000, 754]) | 2\n",
-      "6634     | torch.Size([1, 1000, 754]) | 2\n",
-      "6635     | torch.Size([1, 1000, 754]) | 2\n",
-      "6636     | torch.Size([1, 1000, 754]) | 2\n",
-      "6637     | torch.Size([1, 1000, 754]) | 2\n",
-      "6638     | torch.Size([1, 1000, 754]) | 2\n",
-      "6639     | torch.Size([1, 1000, 754]) | 2\n",
-      "6640     | torch.Size([1, 1000, 754]) | 2\n",
-      "6641     | torch.Size([1, 1000, 754]) | 2\n",
-      "6642     | torch.Size([1, 1000, 754]) | 2\n",
-      "6643     | torch.Size([1, 1000, 754]) | 2\n",
-      "6644     | torch.Size([1, 1000, 754]) | 2\n",
-      "6645     | torch.Size([1, 1000, 754]) | 2\n",
-      "6646     | torch.Size([1, 1000, 754]) | 2\n",
-      "6647     | torch.Size([1, 1000, 754]) | 2\n",
-      "6648     | torch.Size([1, 1000, 754]) | 2\n",
-      "6649     | torch.Size([1, 1000, 754]) | 2\n",
-      "6650     | torch.Size([1, 1000, 754]) | 2\n",
-      "6651     | torch.Size([1, 1000, 754]) | 2\n",
-      "6652     | torch.Size([1, 1000, 754]) | 2\n",
-      "6653     | torch.Size([1, 1000, 754]) | 2\n",
-      "6654     | torch.Size([1, 1000, 754]) | 2\n",
-      "6655     | torch.Size([1, 1000, 754]) | 2\n",
-      "6656     | torch.Size([1, 1000, 754]) | 2\n",
-      "6657     | torch.Size([1, 1000, 754]) | 2\n",
-      "6658     | torch.Size([1, 1000, 754]) | 2\n",
-      "6659     | torch.Size([1, 1000, 754]) | 2\n",
-      "6660     | torch.Size([1, 1000, 802]) | 2\n",
-      "6661     | torch.Size([1, 1000, 754]) | 2\n",
-      "6662     | torch.Size([1, 1000, 754]) | 2\n",
-      "6663     | torch.Size([1, 1000, 754]) | 2\n",
-      "6664     | torch.Size([1, 1000, 754]) | 2\n",
-      "6665     | torch.Size([1, 1000, 754]) | 2\n",
-      "6666     | torch.Size([1, 1000, 754]) | 2\n",
-      "6667     | torch.Size([1, 1000, 754]) | 2\n",
-      "6668     | torch.Size([1, 1000, 754]) | 2\n",
-      "6669     | torch.Size([1, 1000, 754]) | 2\n",
-      "6670     | torch.Size([1, 1000, 754]) | 2\n",
-      "6671     | torch.Size([1, 1000, 754]) | 2\n",
-      "6672     | torch.Size([1, 1000, 754]) | 2\n",
-      "6673     | torch.Size([1, 1000, 754]) | 2\n",
-      "6674     | torch.Size([1, 1000, 754]) | 2\n",
-      "6675     | torch.Size([1, 1000, 754]) | 2\n",
-      "6676     | torch.Size([1, 1000, 754]) | 2\n",
-      "6677     | torch.Size([1, 1000, 754]) | 2\n",
-      "6678     | torch.Size([1, 1000, 754]) | 2\n",
-      "6679     | torch.Size([1, 1000, 754]) | 2\n",
-      "6680     | torch.Size([1, 1000, 754]) | 2\n",
-      "6681     | torch.Size([1, 1000, 754]) | 2\n",
-      "6682     | torch.Size([1, 1000, 754]) | 2\n",
-      "6683     | torch.Size([1, 1000, 754]) | 2\n",
-      "6684     | torch.Size([1, 1000, 754]) | 2\n",
-      "6685     | torch.Size([1, 1000, 754]) | 2\n",
-      "6686     | torch.Size([1, 1000, 754]) | 2\n",
-      "6687     | torch.Size([1, 1000, 754]) | 2\n",
-      "6688     | torch.Size([1, 1000, 754]) | 2\n",
-      "6689     | torch.Size([1, 1000, 754]) | 2\n",
-      "6690     | torch.Size([1, 1000, 754]) | 2\n",
-      "6691     | torch.Size([1, 1000, 754]) | 2\n",
-      "6692     | torch.Size([1, 1000, 754]) | 2\n",
-      "6693     | torch.Size([1, 1000, 754]) | 2\n",
-      "6694     | torch.Size([1, 1000, 754]) | 2\n",
-      "6695     | torch.Size([1, 1000, 754]) | 2\n",
-      "6696     | torch.Size([1, 1000, 754]) | 2\n",
-      "6697     | torch.Size([1, 1000, 754]) | 2\n",
-      "6698     | torch.Size([1, 1000, 754]) | 2\n",
-      "6699     | torch.Size([1, 1000, 754]) | 2\n",
-      "6700     | torch.Size([1, 1000, 754]) | 2\n",
-      "6701     | torch.Size([1, 1000, 754]) | 2\n",
-      "6702     | torch.Size([1, 1000, 754]) | 2\n",
-      "6703     | torch.Size([1, 1000, 754]) | 2\n",
-      "6704     | torch.Size([1, 1000, 754]) | 2\n",
-      "6705     | torch.Size([1, 1000, 754]) | 2\n",
-      "6706     | torch.Size([1, 1000, 802]) | 2\n",
-      "6707     | torch.Size([1, 1000, 754]) | 2\n",
-      "6708     | torch.Size([1, 1000, 754]) | 2\n",
-      "6709     | torch.Size([1, 1000, 754]) | 2\n",
-      "6710     | torch.Size([1, 1000, 754]) | 2\n",
-      "6711     | torch.Size([1, 1000, 754]) | 2\n",
-      "6712     | torch.Size([1, 1000, 754]) | 2\n",
-      "6713     | torch.Size([1, 1000, 754]) | 2\n",
-      "6714     | torch.Size([1, 1000, 754]) | 2\n",
-      "6715     | torch.Size([1, 1000, 754]) | 2\n",
-      "6716     | torch.Size([1, 1000, 754]) | 2\n",
-      "6717     | torch.Size([1, 1000, 754]) | 2\n",
-      "6718     | torch.Size([1, 1000, 754]) | 2\n",
-      "6719     | torch.Size([1, 1000, 754]) | 2\n",
-      "6720     | torch.Size([1, 1000, 754]) | 2\n",
-      "6721     | torch.Size([1, 1000, 754]) | 2\n",
-      "6722     | torch.Size([1, 1000, 754]) | 2\n",
-      "6723     | torch.Size([1, 1000, 802]) | 2\n",
-      "6724     | torch.Size([1, 1000, 754]) | 2\n",
-      "6725     | torch.Size([1, 1000, 754]) | 2\n",
-      "6726     | torch.Size([1, 1000, 754]) | 2\n",
-      "6727     | torch.Size([1, 1000, 754]) | 2\n",
-      "6728     | torch.Size([1, 1000, 754]) | 2\n",
-      "6729     | torch.Size([1, 1000, 754]) | 2\n",
-      "6730     | torch.Size([1, 1000, 754]) | 2\n",
-      "6731     | torch.Size([1, 1000, 754]) | 2\n",
-      "6732     | torch.Size([1, 1000, 754]) | 2\n",
-      "6733     | torch.Size([1, 1000, 754]) | 2\n",
-      "6734     | torch.Size([1, 1000, 802]) | 2\n",
-      "6735     | torch.Size([1, 1000, 754]) | 2\n",
-      "6736     | torch.Size([1, 1000, 754]) | 2\n",
-      "6737     | torch.Size([1, 1000, 754]) | 2\n",
-      "6738     | torch.Size([1, 1000, 754]) | 2\n",
-      "6739     | torch.Size([1, 1000, 754]) | 2\n",
-      "6740     | torch.Size([1, 1000, 754]) | 2\n",
-      "6741     | torch.Size([1, 1000, 754]) | 2\n",
-      "6742     | torch.Size([1, 1000, 754]) | 2\n",
-      "6743     | torch.Size([1, 1000, 802]) | 2\n",
-      "6744     | torch.Size([1, 1000, 754]) | 2\n",
-      "6745     | torch.Size([1, 1000, 754]) | 2\n",
-      "6746     | torch.Size([1, 1000, 754]) | 2\n",
-      "6747     | torch.Size([1, 1000, 754]) | 2\n",
-      "6748     | torch.Size([1, 1000, 754]) | 2\n",
-      "6749     | torch.Size([1, 1000, 754]) | 2\n",
-      "6750     | torch.Size([1, 1000, 754]) | 2\n",
-      "6751     | torch.Size([1, 1000, 754]) | 2\n",
-      "6752     | torch.Size([1, 1000, 754]) | 2\n",
-      "6753     | torch.Size([1, 1000, 754]) | 2\n",
-      "6754     | torch.Size([1, 1000, 754]) | 2\n",
-      "6755     | torch.Size([1, 1000, 754]) | 2\n",
-      "6756     | torch.Size([1, 1000, 754]) | 2\n",
-      "6757     | torch.Size([1, 1000, 754]) | 2\n",
-      "6758     | torch.Size([1, 1000, 754]) | 2\n",
-      "6759     | torch.Size([1, 1000, 754]) | 2\n",
-      "6760     | torch.Size([1, 1000, 754]) | 2\n",
-      "6761     | torch.Size([1, 1000, 754]) | 2\n",
-      "6762     | torch.Size([1, 1000, 754]) | 2\n",
-      "6763     | torch.Size([1, 1000, 754]) | 2\n",
-      "6764     | torch.Size([1, 1000, 754]) | 2\n",
-      "6765     | torch.Size([1, 1000, 754]) | 2\n",
-      "6766     | torch.Size([1, 1000, 754]) | 2\n",
-      "6767     | torch.Size([1, 1000, 754]) | 2\n",
-      "6768     | torch.Size([1, 1000, 754]) | 2\n",
-      "6769     | torch.Size([1, 1000, 754]) | 2\n",
-      "6770     | torch.Size([1, 1000, 754]) | 2\n",
-      "6771     | torch.Size([1, 1000, 754]) | 2\n",
-      "6772     | torch.Size([1, 1000, 754]) | 2\n",
-      "6773     | torch.Size([1, 1000, 754]) | 2\n",
-      "6774     | torch.Size([1, 1000, 802]) | 2\n",
-      "6775     | torch.Size([1, 1000, 754]) | 2\n",
-      "6776     | torch.Size([1, 1000, 754]) | 2\n",
-      "6777     | torch.Size([1, 1000, 802]) | 2\n",
-      "6778     | torch.Size([1, 1000, 754]) | 2\n",
-      "6779     | torch.Size([1, 1000, 754]) | 2\n",
-      "6780     | torch.Size([1, 1000, 754]) | 2\n",
-      "6781     | torch.Size([1, 1000, 754]) | 2\n",
-      "6782     | torch.Size([1, 1000, 754]) | 2\n",
-      "6783     | torch.Size([1, 1000, 802]) | 2\n",
-      "6784     | torch.Size([1, 1000, 754]) | 2\n",
-      "6785     | torch.Size([1, 1000, 754]) | 2\n",
-      "6786     | torch.Size([1, 1000, 754]) | 2\n",
-      "6787     | torch.Size([1, 1000, 754]) | 2\n",
-      "6788     | torch.Size([1, 1000, 754]) | 2\n",
-      "6789     | torch.Size([1, 1000, 754]) | 2\n",
-      "6790     | torch.Size([1, 1000, 754]) | 2\n",
-      "6791     | torch.Size([1, 1000, 754]) | 2\n",
-      "6792     | torch.Size([1, 1000, 754]) | 2\n",
-      "6793     | torch.Size([1, 1000, 754]) | 2\n",
-      "6794     | torch.Size([1, 1000, 754]) | 2\n",
-      "6795     | torch.Size([1, 1000, 754]) | 2\n",
-      "6796     | torch.Size([1, 1000, 754]) | 2\n",
-      "6797     | torch.Size([1, 1000, 754]) | 2\n",
-      "6798     | torch.Size([1, 1000, 762]) | 2\n",
-      "6799     | torch.Size([1, 1000, 754]) | 2\n",
-      "6800     | torch.Size([1, 1000, 754]) | 2\n",
-      "6801     | torch.Size([1, 1000, 754]) | 2\n",
-      "6802     | torch.Size([1, 1000, 754]) | 2\n",
-      "6803     | torch.Size([1, 1000, 754]) | 2\n",
-      "6804     | torch.Size([1, 1000, 754]) | 2\n",
-      "6805     | torch.Size([1, 1000, 754]) | 2\n",
-      "6806     | torch.Size([1, 1000, 754]) | 2\n",
-      "6807     | torch.Size([1, 1000, 754]) | 2\n",
-      "6808     | torch.Size([1, 1000, 754]) | 2\n",
-      "6809     | torch.Size([1, 1000, 754]) | 2\n",
-      "6810     | torch.Size([1, 1000, 754]) | 2\n",
-      "6811     | torch.Size([1, 1000, 754]) | 2\n",
-      "6812     | torch.Size([1, 1000, 754]) | 2\n",
-      "6813     | torch.Size([1, 1000, 754]) | 2\n",
-      "6814     | torch.Size([1, 1000, 754]) | 2\n",
-      "6815     | torch.Size([1, 1000, 754]) | 2\n",
-      "6816     | torch.Size([1, 1000, 754]) | 2\n",
-      "6817     | torch.Size([1, 1000, 754]) | 2\n",
-      "6818     | torch.Size([1, 1000, 754]) | 2\n",
-      "6819     | torch.Size([1, 1000, 754]) | 2\n",
-      "6820     | torch.Size([1, 1000, 754]) | 2\n",
-      "6821     | torch.Size([1, 1000, 754]) | 2\n",
-      "6822     | torch.Size([1, 1000, 754]) | 2\n",
-      "6823     | torch.Size([1, 1000, 802]) | 2\n",
-      "6824     | torch.Size([1, 1000, 754]) | 2\n",
-      "6825     | torch.Size([1, 1000, 754]) | 2\n",
-      "6826     | torch.Size([1, 1000, 754]) | 2\n",
-      "6827     | torch.Size([1, 1000, 754]) | 2\n",
-      "6828     | torch.Size([1, 1000, 754]) | 2\n",
-      "6829     | torch.Size([1, 1000, 754]) | 2\n",
-      "6830     | torch.Size([1, 1000, 754]) | 2\n",
-      "6831     | torch.Size([1, 1000, 754]) | 2\n",
-      "6832     | torch.Size([1, 1000, 754]) | 2\n",
-      "6833     | torch.Size([1, 1000, 762]) | 2\n",
-      "6834     | torch.Size([1, 1000, 754]) | 2\n",
-      "6835     | torch.Size([1, 1000, 754]) | 2\n",
-      "6836     | torch.Size([1, 1000, 754]) | 2\n",
-      "6837     | torch.Size([1, 1000, 754]) | 2\n",
-      "6838     | torch.Size([1, 1000, 754]) | 2\n",
-      "6839     | torch.Size([1, 1000, 754]) | 2\n",
-      "6840     | torch.Size([1, 1000, 754]) | 2\n",
-      "6841     | torch.Size([1, 1000, 754]) | 2\n",
-      "6842     | torch.Size([1, 1000, 802]) | 2\n",
-      "6843     | torch.Size([1, 1000, 754]) | 2\n",
-      "6844     | torch.Size([1, 1000, 754]) | 2\n",
-      "6845     | torch.Size([1, 1000, 754]) | 2\n",
-      "6846     | torch.Size([1, 1000, 754]) | 2\n",
-      "6847     | torch.Size([1, 1000, 754]) | 2\n",
-      "6848     | torch.Size([1, 1000, 754]) | 2\n",
-      "6849     | torch.Size([1, 1000, 754]) | 2\n",
-      "6850     | torch.Size([1, 1000, 754]) | 2\n",
-      "6851     | torch.Size([1, 1000, 754]) | 2\n",
-      "6852     | torch.Size([1, 1000, 754]) | 2\n",
-      "6853     | torch.Size([1, 1000, 754]) | 2\n",
-      "6854     | torch.Size([1, 1000, 754]) | 2\n",
-      "6855     | torch.Size([1, 1000, 754]) | 2\n",
-      "6856     | torch.Size([1, 1000, 754]) | 2\n",
-      "6857     | torch.Size([1, 1000, 754]) | 2\n",
-      "6858     | torch.Size([1, 1000, 754]) | 2\n",
-      "6859     | torch.Size([1, 1000, 754]) | 2\n",
-      "6860     | torch.Size([1, 1000, 754]) | 2\n",
-      "6861     | torch.Size([1, 1000, 754]) | 2\n",
-      "6862     | torch.Size([1, 1000, 754]) | 2\n",
-      "6863     | torch.Size([1, 1000, 802]) | 2\n",
-      "6864     | torch.Size([1, 1000, 754]) | 2\n",
-      "6865     | torch.Size([1, 1000, 754]) | 2\n",
-      "6866     | torch.Size([1, 1000, 754]) | 2\n",
-      "6867     | torch.Size([1, 1000, 754]) | 2\n",
-      "6868     | torch.Size([1, 1000, 754]) | 2\n",
-      "6869     | torch.Size([1, 1000, 754]) | 2\n",
-      "6870     | torch.Size([1, 1000, 754]) | 2\n",
-      "6871     | torch.Size([1, 1000, 754]) | 2\n",
-      "6872     | torch.Size([1, 1000, 754]) | 2\n",
-      "6873     | torch.Size([1, 1000, 754]) | 2\n",
-      "6874     | torch.Size([1, 1000, 754]) | 2\n",
-      "6875     | torch.Size([1, 1000, 754]) | 2\n",
-      "6876     | torch.Size([1, 1000, 754]) | 2\n",
-      "6877     | torch.Size([1, 1000, 754]) | 2\n",
-      "6878     | torch.Size([1, 1000, 754]) | 2\n",
-      "6879     | torch.Size([1, 1000, 754]) | 2\n",
-      "6880     | torch.Size([1, 1000, 754]) | 2\n",
-      "6881     | torch.Size([1, 1000, 754]) | 2\n",
-      "6882     | torch.Size([1, 1000, 754]) | 2\n",
-      "6883     | torch.Size([1, 1000, 754]) | 2\n",
-      "6884     | torch.Size([1, 1000, 754]) | 2\n",
-      "6885     | torch.Size([1, 1000, 754]) | 2\n",
-      "6886     | torch.Size([1, 1000, 754]) | 2\n",
-      "6887     | torch.Size([1, 1000, 754]) | 2\n",
-      "6888     | torch.Size([1, 1000, 754]) | 2\n",
-      "6889     | torch.Size([1, 1000, 754]) | 2\n",
-      "6890     | torch.Size([1, 1000, 754]) | 2\n",
-      "6891     | torch.Size([1, 1000, 754]) | 2\n",
-      "6892     | torch.Size([1, 1000, 754]) | 2\n",
-      "6893     | torch.Size([1, 1000, 754]) | 2\n",
-      "6894     | torch.Size([1, 1000, 754]) | 2\n",
-      "6895     | torch.Size([1, 1000, 754]) | 2\n",
-      "6896     | torch.Size([1, 1000, 754]) | 2\n",
-      "6897     | torch.Size([1, 1000, 754]) | 2\n",
-      "6898     | torch.Size([1, 1000, 754]) | 2\n",
-      "6899     | torch.Size([1, 1000, 754]) | 2\n",
-      "6900     | torch.Size([1, 1000, 754]) | 2\n",
-      "6901     | torch.Size([1, 1000, 754]) | 2\n",
-      "6902     | torch.Size([1, 1000, 754]) | 2\n",
-      "6903     | torch.Size([1, 1000, 754]) | 2\n",
-      "6904     | torch.Size([1, 1000, 754]) | 2\n",
-      "6905     | torch.Size([1, 1000, 754]) | 2\n",
-      "6906     | torch.Size([1, 1000, 754]) | 2\n",
-      "6907     | torch.Size([1, 1000, 754]) | 2\n",
-      "6908     | torch.Size([1, 1000, 754]) | 2\n",
-      "6909     | torch.Size([1, 1000, 754]) | 2\n",
-      "6910     | torch.Size([1, 1000, 754]) | 2\n",
-      "6911     | torch.Size([1, 1000, 754]) | 2\n",
-      "6912     | torch.Size([1, 1000, 754]) | 2\n",
-      "6913     | torch.Size([1, 1000, 754]) | 2\n",
-      "6914     | torch.Size([1, 1000, 754]) | 2\n",
-      "6915     | torch.Size([1, 1000, 754]) | 2\n",
-      "6916     | torch.Size([1, 1000, 754]) | 2\n",
-      "6917     | torch.Size([1, 1000, 754]) | 2\n",
-      "6918     | torch.Size([1, 1000, 754]) | 2\n",
-      "6919     | torch.Size([1, 1000, 754]) | 2\n",
-      "6920     | torch.Size([1, 1000, 754]) | 2\n",
-      "6921     | torch.Size([1, 1000, 754]) | 2\n",
-      "6922     | torch.Size([1, 1000, 754]) | 2\n",
-      "6923     | torch.Size([1, 1000, 754]) | 2\n",
-      "6924     | torch.Size([1, 1000, 754]) | 2\n",
-      "6925     | torch.Size([1, 1000, 754]) | 2\n",
-      "6926     | torch.Size([1, 1000, 754]) | 2\n",
-      "6927     | torch.Size([1, 1000, 754]) | 2\n",
-      "6928     | torch.Size([1, 1000, 754]) | 2\n",
-      "6929     | torch.Size([1, 1000, 754]) | 2\n",
-      "6930     | torch.Size([1, 1000, 802]) | 2\n",
-      "6931     | torch.Size([1, 1000, 754]) | 2\n",
-      "6932     | torch.Size([1, 1000, 754]) | 2\n",
-      "6933     | torch.Size([1, 1000, 754]) | 2\n",
-      "6934     | torch.Size([1, 1000, 754]) | 2\n",
-      "6935     | torch.Size([1, 1000, 754]) | 2\n",
-      "6936     | torch.Size([1, 1000, 754]) | 2\n",
-      "6937     | torch.Size([1, 1000, 754]) | 2\n",
-      "6938     | torch.Size([1, 1000, 754]) | 2\n",
-      "6939     | torch.Size([1, 1000, 754]) | 2\n",
-      "6940     | torch.Size([1, 1000, 754]) | 2\n",
-      "6941     | torch.Size([1, 1000, 802]) | 2\n",
-      "6942     | torch.Size([1, 1000, 754]) | 2\n",
-      "6943     | torch.Size([1, 1000, 754]) | 2\n",
-      "6944     | torch.Size([1, 1000, 754]) | 2\n",
-      "6945     | torch.Size([1, 1000, 754]) | 2\n",
-      "6946     | torch.Size([1, 1000, 754]) | 2\n",
-      "6947     | torch.Size([1, 1000, 754]) | 2\n",
-      "6948     | torch.Size([1, 1000, 754]) | 2\n",
-      "6949     | torch.Size([1, 1000, 754]) | 2\n",
-      "6950     | torch.Size([1, 1000, 754]) | 2\n",
-      "6951     | torch.Size([1, 1000, 754]) | 2\n",
-      "6952     | torch.Size([1, 1000, 754]) | 2\n",
-      "6953     | torch.Size([1, 1000, 754]) | 2\n",
-      "6954     | torch.Size([1, 1000, 754]) | 2\n",
-      "6955     | torch.Size([1, 1000, 754]) | 2\n",
-      "6956     | torch.Size([1, 1000, 754]) | 2\n",
-      "6957     | torch.Size([1, 1000, 754]) | 2\n",
-      "6958     | torch.Size([1, 1000, 754]) | 2\n",
-      "6959     | torch.Size([1, 1000, 754]) | 2\n",
-      "6960     | torch.Size([1, 1000, 754]) | 2\n",
-      "6961     | torch.Size([1, 1000, 754]) | 2\n",
-      "6962     | torch.Size([1, 1000, 754]) | 2\n",
-      "6963     | torch.Size([1, 1000, 754]) | 2\n",
-      "6964     | torch.Size([1, 1000, 802]) | 2\n",
-      "6965     | torch.Size([1, 1000, 754]) | 2\n",
-      "6966     | torch.Size([1, 1000, 754]) | 2\n",
-      "6967     | torch.Size([1, 1000, 754]) | 2\n",
-      "6968     | torch.Size([1, 1000, 754]) | 2\n",
-      "6969     | torch.Size([1, 1000, 754]) | 2\n",
-      "6970     | torch.Size([1, 1000, 754]) | 2\n",
-      "6971     | torch.Size([1, 1000, 754]) | 2\n",
-      "6972     | torch.Size([1, 1000, 754]) | 2\n",
-      "6973     | torch.Size([1, 1000, 754]) | 2\n",
-      "6974     | torch.Size([1, 1000, 754]) | 2\n",
-      "6975     | torch.Size([1, 1000, 754]) | 2\n",
-      "6976     | torch.Size([1, 1000, 754]) | 2\n",
-      "6977     | torch.Size([1, 1000, 754]) | 2\n",
-      "6978     | torch.Size([1, 1000, 754]) | 2\n",
-      "6979     | torch.Size([1, 1000, 754]) | 2\n",
-      "6980     | torch.Size([1, 1000, 754]) | 2\n",
-      "6981     | torch.Size([1, 1000, 754]) | 2\n",
-      "6982     | torch.Size([1, 1000, 802]) | 2\n",
-      "6983     | torch.Size([1, 1000, 754]) | 2\n",
-      "6984     | torch.Size([1, 1000, 754]) | 2\n",
-      "6985     | torch.Size([1, 1000, 754]) | 2\n",
-      "6986     | torch.Size([1, 1000, 754]) | 2\n",
-      "6987     | torch.Size([1, 1000, 754]) | 2\n",
-      "6988     | torch.Size([1, 1000, 754]) | 2\n",
-      "6989     | torch.Size([1, 1000, 754]) | 2\n",
-      "6990     | torch.Size([1, 1000, 754]) | 2\n",
-      "6991     | torch.Size([1, 1000, 754]) | 2\n",
-      "6992     | torch.Size([1, 1000, 754]) | 2\n",
-      "6993     | torch.Size([1, 1000, 754]) | 2\n",
-      "6994     | torch.Size([1, 1000, 754]) | 2\n",
-      "6995     | torch.Size([1, 1000, 754]) | 2\n",
-      "6996     | torch.Size([1, 1000, 754]) | 2\n",
-      "6997     | torch.Size([1, 1000, 754]) | 2\n",
-      "6998     | torch.Size([1, 1000, 754]) | 2\n",
-      "6999     | torch.Size([1, 1000, 754]) | 2\n",
-      "7000     | torch.Size([1, 1000, 754]) | 2\n",
-      "7001     | torch.Size([1, 1000, 754]) | 2\n",
-      "7002     | torch.Size([1, 1000, 754]) | 2\n",
-      "7003     | torch.Size([1, 1000, 754]) | 2\n",
-      "7004     | torch.Size([1, 1000, 754]) | 2\n",
-      "7005     | torch.Size([1, 1000, 754]) | 2\n",
-      "7006     | torch.Size([1, 1000, 754]) | 2\n",
-      "7007     | torch.Size([1, 1000, 754]) | 2\n",
-      "7008     | torch.Size([1, 1000, 754]) | 2\n",
-      "7009     | torch.Size([1, 1000, 802]) | 2\n",
-      "7010     | torch.Size([1, 1000, 754]) | 2\n",
-      "7011     | torch.Size([1, 1000, 754]) | 2\n",
-      "7012     | torch.Size([1, 1000, 754]) | 2\n",
-      "7013     | torch.Size([1, 1000, 754]) | 2\n",
-      "7014     | torch.Size([1, 1000, 754]) | 2\n",
-      "7015     | torch.Size([1, 1000, 754]) | 2\n",
-      "7016     | torch.Size([1, 1000, 762]) | 2\n",
-      "7017     | torch.Size([1, 1000, 754]) | 2\n",
-      "7018     | torch.Size([1, 1000, 754]) | 2\n",
-      "7019     | torch.Size([1, 1000, 754]) | 2\n",
-      "7020     | torch.Size([1, 1000, 754]) | 2\n",
-      "7021     | torch.Size([1, 1000, 754]) | 2\n",
-      "7022     | torch.Size([1, 1000, 802]) | 2\n",
-      "7023     | torch.Size([1, 1000, 754]) | 2\n",
-      "7024     | torch.Size([1, 1000, 754]) | 2\n",
-      "7025     | torch.Size([1, 1000, 754]) | 2\n",
-      "7026     | torch.Size([1, 1000, 754]) | 2\n",
-      "7027     | torch.Size([1, 1000, 754]) | 2\n",
-      "7028     | torch.Size([1, 1000, 754]) | 2\n",
-      "7029     | torch.Size([1, 1000, 754]) | 2\n",
-      "7030     | torch.Size([1, 1000, 754]) | 2\n",
-      "7031     | torch.Size([1, 1000, 754]) | 2\n",
-      "7032     | torch.Size([1, 1000, 754]) | 2\n",
-      "7033     | torch.Size([1, 1000, 754]) | 2\n",
-      "7034     | torch.Size([1, 1000, 754]) | 2\n",
-      "7035     | torch.Size([1, 1000, 754]) | 2\n",
-      "7036     | torch.Size([1, 1000, 754]) | 2\n",
-      "7037     | torch.Size([1, 1000, 754]) | 2\n",
-      "7038     | torch.Size([1, 1000, 754]) | 2\n",
-      "7039     | torch.Size([1, 1000, 754]) | 2\n",
-      "7040     | torch.Size([1, 1000, 754]) | 2\n",
-      "7041     | torch.Size([1, 1000, 802]) | 2\n",
-      "7042     | torch.Size([1, 1000, 754]) | 2\n",
-      "7043     | torch.Size([1, 1000, 754]) | 2\n",
-      "7044     | torch.Size([1, 1000, 754]) | 2\n",
-      "7045     | torch.Size([1, 1000, 754]) | 2\n",
-      "7046     | torch.Size([1, 1000, 754]) | 2\n",
-      "7047     | torch.Size([1, 1000, 754]) | 2\n",
-      "7048     | torch.Size([1, 1000, 802]) | 2\n",
-      "7049     | torch.Size([1, 1000, 754]) | 2\n",
-      "7050     | torch.Size([1, 1000, 754]) | 2\n",
-      "7051     | torch.Size([1, 1000, 754]) | 2\n",
-      "7052     | torch.Size([1, 1000, 754]) | 2\n",
-      "7053     | torch.Size([1, 1000, 754]) | 2\n",
-      "7054     | torch.Size([1, 1000, 754]) | 2\n",
-      "7055     | torch.Size([1, 1000, 754]) | 2\n",
-      "7056     | torch.Size([1, 1000, 754]) | 2\n",
-      "7057     | torch.Size([1, 1000, 754]) | 2\n",
-      "7058     | torch.Size([1, 1000, 754]) | 2\n",
-      "7059     | torch.Size([1, 1000, 754]) | 2\n",
-      "7060     | torch.Size([1, 1000, 754]) | 2\n",
-      "7061     | torch.Size([1, 1000, 754]) | 2\n",
-      "7062     | torch.Size([1, 1000, 754]) | 2\n",
-      "7063     | torch.Size([1, 1000, 754]) | 2\n",
-      "7064     | torch.Size([1, 1000, 754]) | 2\n",
-      "7065     | torch.Size([1, 1000, 754]) | 2\n",
-      "7066     | torch.Size([1, 1000, 754]) | 2\n",
-      "7067     | torch.Size([1, 1000, 754]) | 2\n",
-      "7068     | torch.Size([1, 1000, 754]) | 2\n",
-      "7069     | torch.Size([1, 1000, 754]) | 2\n",
-      "7070     | torch.Size([1, 1000, 754]) | 2\n",
-      "7071     | torch.Size([1, 1000, 754]) | 2\n",
-      "7072     | torch.Size([1, 1000, 754]) | 2\n",
-      "7073     | torch.Size([1, 1000, 754]) | 2\n",
-      "7074     | torch.Size([1, 1000, 754]) | 2\n",
-      "7075     | torch.Size([1, 1000, 754]) | 2\n",
-      "7076     | torch.Size([1, 1000, 754]) | 2\n",
-      "7077     | torch.Size([1, 1000, 754]) | 2\n",
-      "7078     | torch.Size([1, 1000, 754]) | 2\n",
-      "7079     | torch.Size([1, 1000, 754]) | 2\n",
-      "7080     | torch.Size([1, 1000, 754]) | 2\n",
-      "7081     | torch.Size([1, 1000, 754]) | 2\n",
-      "7082     | torch.Size([1, 1000, 754]) | 2\n",
-      "7083     | torch.Size([1, 1000, 754]) | 2\n",
-      "7084     | torch.Size([1, 1000, 754]) | 2\n",
-      "7085     | torch.Size([1, 1000, 754]) | 2\n",
-      "7086     | torch.Size([1, 1000, 754]) | 2\n",
-      "7087     | torch.Size([1, 1000, 754]) | 2\n",
-      "7088     | torch.Size([1, 1000, 754]) | 2\n",
-      "7089     | torch.Size([1, 1000, 754]) | 2\n",
-      "7090     | torch.Size([1, 1000, 754]) | 2\n",
-      "7091     | torch.Size([1, 1000, 754]) | 2\n",
-      "7092     | torch.Size([1, 1000, 754]) | 2\n",
-      "7093     | torch.Size([1, 1000, 754]) | 2\n",
-      "7094     | torch.Size([1, 1000, 754]) | 2\n",
-      "7095     | torch.Size([1, 1000, 754]) | 2\n",
-      "7096     | torch.Size([1, 1000, 754]) | 2\n",
-      "7097     | torch.Size([1, 1000, 754]) | 2\n",
-      "7098     | torch.Size([1, 1000, 754]) | 2\n",
-      "7099     | torch.Size([1, 1000, 754]) | 2\n",
-      "7100     | torch.Size([1, 1000, 754]) | 2\n",
-      "7101     | torch.Size([1, 1000, 754]) | 2\n",
-      "7102     | torch.Size([1, 1000, 754]) | 2\n",
-      "7103     | torch.Size([1, 1000, 754]) | 2\n",
-      "7104     | torch.Size([1, 1000, 754]) | 2\n",
-      "7105     | torch.Size([1, 1000, 754]) | 2\n",
-      "7106     | torch.Size([1, 1000, 754]) | 2\n",
-      "7107     | torch.Size([1, 1000, 754]) | 2\n",
-      "7108     | torch.Size([1, 1000, 754]) | 2\n",
-      "7109     | torch.Size([1, 1000, 754]) | 2\n",
-      "7110     | torch.Size([1, 1000, 754]) | 2\n",
-      "7111     | torch.Size([1, 1000, 754]) | 2\n",
-      "7112     | torch.Size([1, 1000, 754]) | 2\n",
-      "7113     | torch.Size([1, 1000, 754]) | 2\n",
-      "7114     | torch.Size([1, 1000, 754]) | 2\n",
-      "7115     | torch.Size([1, 1000, 802]) | 2\n",
-      "7116     | torch.Size([1, 1000, 754]) | 2\n",
-      "7117     | torch.Size([1, 1000, 754]) | 2\n",
-      "7118     | torch.Size([1, 1000, 754]) | 2\n",
-      "7119     | torch.Size([1, 1000, 754]) | 2\n",
-      "7120     | torch.Size([1, 1000, 754]) | 2\n",
-      "7121     | torch.Size([1, 1000, 754]) | 2\n",
-      "7122     | torch.Size([1, 1000, 754]) | 2\n",
-      "7123     | torch.Size([1, 1000, 754]) | 2\n",
-      "7124     | torch.Size([1, 1000, 754]) | 2\n",
-      "7125     | torch.Size([1, 1000, 754]) | 2\n",
-      "7126     | torch.Size([1, 1000, 754]) | 2\n",
-      "7127     | torch.Size([1, 1000, 754]) | 2\n",
-      "7128     | torch.Size([1, 1000, 754]) | 2\n",
-      "7129     | torch.Size([1, 1000, 754]) | 2\n",
-      "7130     | torch.Size([1, 1000, 754]) | 2\n",
-      "7131     | torch.Size([1, 1000, 754]) | 2\n",
-      "7132     | torch.Size([1, 1000, 754]) | 2\n",
-      "7133     | torch.Size([1, 1000, 754]) | 2\n",
-      "7134     | torch.Size([1, 1000, 754]) | 2\n",
-      "7135     | torch.Size([1, 1000, 754]) | 2\n",
-      "7136     | torch.Size([1, 1000, 754]) | 2\n",
-      "7137     | torch.Size([1, 1000, 754]) | 2\n",
-      "7138     | torch.Size([1, 1000, 754]) | 2\n",
-      "7139     | torch.Size([1, 1000, 754]) | 2\n",
-      "7140     | torch.Size([1, 1000, 754]) | 2\n",
-      "7141     | torch.Size([1, 1000, 754]) | 2\n",
-      "7142     | torch.Size([1, 1000, 802]) | 2\n",
-      "7143     | torch.Size([1, 1000, 754]) | 2\n",
-      "7144     | torch.Size([1, 1000, 754]) | 2\n",
-      "7145     | torch.Size([1, 1000, 754]) | 2\n",
-      "7146     | torch.Size([1, 1000, 754]) | 2\n",
-      "7147     | torch.Size([1, 1000, 754]) | 2\n",
-      "7148     | torch.Size([1, 1000, 754]) | 2\n",
-      "7149     | torch.Size([1, 1000, 754]) | 2\n",
-      "7150     | torch.Size([1, 1000, 754]) | 2\n",
-      "7151     | torch.Size([1, 1000, 754]) | 2\n",
-      "7152     | torch.Size([1, 1000, 754]) | 2\n",
-      "7153     | torch.Size([1, 1000, 754]) | 2\n",
-      "7154     | torch.Size([1, 1000, 802]) | 2\n",
-      "7155     | torch.Size([1, 1000, 754]) | 2\n",
-      "7156     | torch.Size([1, 1000, 754]) | 2\n",
-      "7157     | torch.Size([1, 1000, 754]) | 2\n",
-      "7158     | torch.Size([1, 1000, 754]) | 2\n",
-      "7159     | torch.Size([1, 1000, 754]) | 2\n",
-      "7160     | torch.Size([1, 1000, 754]) | 2\n",
-      "7161     | torch.Size([1, 1000, 754]) | 2\n",
-      "7162     | torch.Size([1, 1000, 754]) | 2\n",
-      "7163     | torch.Size([1, 1000, 754]) | 2\n",
-      "7164     | torch.Size([1, 1000, 754]) | 2\n",
-      "7165     | torch.Size([1, 1000, 754]) | 2\n",
-      "7166     | torch.Size([1, 1000, 754]) | 2\n",
-      "7167     | torch.Size([1, 1000, 754]) | 2\n",
-      "7168     | torch.Size([1, 1000, 754]) | 2\n",
-      "7169     | torch.Size([1, 1000, 754]) | 2\n",
-      "7170     | torch.Size([1, 1000, 754]) | 2\n",
-      "7171     | torch.Size([1, 1000, 754]) | 2\n",
-      "7172     | torch.Size([1, 1000, 754]) | 2\n",
-      "7173     | torch.Size([1, 1000, 754]) | 2\n",
-      "7174     | torch.Size([1, 1000, 754]) | 2\n",
-      "7175     | torch.Size([1, 1000, 754]) | 2\n",
-      "7176     | torch.Size([1, 1000, 754]) | 2\n",
-      "7177     | torch.Size([1, 1000, 754]) | 2\n",
-      "7178     | torch.Size([1, 1000, 754]) | 2\n",
-      "7179     | torch.Size([1, 1000, 754]) | 2\n",
-      "7180     | torch.Size([1, 1000, 754]) | 2\n",
-      "7181     | torch.Size([1, 1000, 754]) | 2\n",
-      "7182     | torch.Size([1, 1000, 754]) | 2\n",
-      "7183     | torch.Size([1, 1000, 754]) | 2\n",
-      "7184     | torch.Size([1, 1000, 754]) | 2\n",
-      "7185     | torch.Size([1, 1000, 754]) | 2\n",
-      "7186     | torch.Size([1, 1000, 754]) | 2\n",
-      "7187     | torch.Size([1, 1000, 754]) | 2\n",
-      "7188     | torch.Size([1, 1000, 754]) | 2\n",
-      "7189     | torch.Size([1, 1000, 754]) | 2\n",
-      "7190     | torch.Size([1, 1000, 754]) | 2\n",
-      "7191     | torch.Size([1, 1000, 754]) | 2\n",
-      "7192     | torch.Size([1, 1000, 754]) | 2\n",
-      "7193     | torch.Size([1, 1000, 754]) | 2\n",
-      "7194     | torch.Size([1, 1000, 754]) | 2\n",
-      "7195     | torch.Size([1, 1000, 754]) | 2\n",
-      "7196     | torch.Size([1, 1000, 754]) | 2\n",
-      "7197     | torch.Size([1, 1000, 754]) | 2\n",
-      "7198     | torch.Size([1, 1000, 754]) | 2\n",
-      "7199     | torch.Size([1, 1000, 754]) | 2\n",
-      "7200     | torch.Size([1, 1000, 754]) | 2\n",
-      "7201     | torch.Size([1, 1000, 754]) | 2\n",
-      "7202     | torch.Size([1, 1000, 754]) | 2\n",
-      "7203     | torch.Size([1, 1000, 754]) | 2\n",
-      "7204     | torch.Size([1, 1000, 754]) | 2\n",
-      "7205     | torch.Size([1, 1000, 754]) | 2\n",
-      "7206     | torch.Size([1, 1000, 754]) | 2\n",
-      "7207     | torch.Size([1, 1000, 802]) | 2\n",
-      "7208     | torch.Size([1, 1000, 754]) | 2\n",
-      "7209     | torch.Size([1, 1000, 754]) | 2\n",
-      "7210     | torch.Size([1, 1000, 754]) | 2\n",
-      "7211     | torch.Size([1, 1000, 754]) | 2\n",
-      "7212     | torch.Size([1, 1000, 754]) | 2\n",
-      "7213     | torch.Size([1, 1000, 754]) | 2\n",
-      "7214     | torch.Size([1, 1000, 754]) | 2\n",
-      "7215     | torch.Size([1, 1000, 754]) | 2\n",
-      "7216     | torch.Size([1, 1000, 754]) | 2\n",
-      "7217     | torch.Size([1, 1000, 754]) | 2\n",
-      "7218     | torch.Size([1, 1000, 754]) | 2\n",
-      "7219     | torch.Size([1, 1000, 754]) | 2\n",
-      "7220     | torch.Size([1, 1000, 754]) | 2\n",
-      "7221     | torch.Size([1, 1000, 754]) | 2\n",
-      "7222     | torch.Size([1, 1000, 754]) | 2\n",
-      "7223     | torch.Size([1, 1000, 754]) | 2\n",
-      "7224     | torch.Size([1, 1000, 754]) | 2\n",
-      "7225     | torch.Size([1, 1000, 754]) | 2\n",
-      "7226     | torch.Size([1, 1000, 754]) | 2\n",
-      "7227     | torch.Size([1, 1000, 754]) | 2\n",
-      "7228     | torch.Size([1, 1000, 754]) | 2\n",
-      "7229     | torch.Size([1, 1000, 754]) | 2\n",
-      "7230     | torch.Size([1, 1000, 754]) | 2\n",
-      "7231     | torch.Size([1, 1000, 754]) | 2\n",
-      "7232     | torch.Size([1, 1000, 754]) | 2\n",
-      "7233     | torch.Size([1, 1000, 754]) | 2\n",
-      "7234     | torch.Size([1, 1000, 754]) | 2\n",
-      "7235     | torch.Size([1, 1000, 754]) | 2\n",
-      "7236     | torch.Size([1, 1000, 754]) | 2\n",
-      "7237     | torch.Size([1, 1000, 754]) | 2\n",
-      "7238     | torch.Size([1, 1000, 754]) | 2\n",
-      "7239     | torch.Size([1, 1000, 754]) | 2\n",
-      "7240     | torch.Size([1, 1000, 754]) | 2\n",
-      "7241     | torch.Size([1, 1000, 754]) | 2\n",
-      "7242     | torch.Size([1, 1000, 754]) | 2\n",
-      "7243     | torch.Size([1, 1000, 754]) | 2\n",
-      "7244     | torch.Size([1, 1000, 754]) | 2\n",
-      "7245     | torch.Size([1, 1000, 754]) | 2\n",
-      "7246     | torch.Size([1, 1000, 754]) | 2\n",
-      "7247     | torch.Size([1, 1000, 754]) | 2\n",
-      "7248     | torch.Size([1, 1000, 754]) | 2\n",
-      "7249     | torch.Size([1, 1000, 754]) | 2\n",
-      "7250     | torch.Size([1, 1000, 754]) | 2\n",
-      "7251     | torch.Size([1, 1000, 754]) | 2\n",
-      "7252     | torch.Size([1, 1000, 754]) | 2\n",
-      "7253     | torch.Size([1, 1000, 754]) | 2\n",
-      "7254     | torch.Size([1, 1000, 754]) | 2\n",
-      "7255     | torch.Size([1, 1000, 754]) | 2\n",
-      "7256     | torch.Size([1, 1000, 754]) | 2\n",
-      "7257     | torch.Size([1, 1000, 754]) | 2\n",
-      "7258     | torch.Size([1, 1000, 754]) | 2\n",
-      "7259     | torch.Size([1, 1000, 754]) | 2\n",
-      "7260     | torch.Size([1, 1000, 754]) | 2\n",
-      "7261     | torch.Size([1, 1000, 754]) | 2\n",
-      "7262     | torch.Size([1, 1000, 754]) | 2\n",
-      "7263     | torch.Size([1, 1000, 754]) | 2\n",
-      "7264     | torch.Size([1, 1000, 754]) | 2\n",
-      "7265     | torch.Size([1, 1000, 754]) | 2\n",
-      "7266     | torch.Size([1, 1000, 754]) | 2\n",
-      "7267     | torch.Size([1, 1000, 754]) | 2\n",
-      "7268     | torch.Size([1, 1000, 754]) | 2\n",
-      "7269     | torch.Size([1, 1000, 754]) | 2\n",
-      "7270     | torch.Size([1, 1000, 754]) | 2\n",
-      "7271     | torch.Size([1, 1000, 754]) | 2\n",
-      "7272     | torch.Size([1, 1000, 754]) | 2\n",
-      "7273     | torch.Size([1, 1000, 754]) | 2\n",
-      "7274     | torch.Size([1, 1000, 754]) | 2\n",
-      "7275     | torch.Size([1, 1000, 754]) | 2\n",
-      "7276     | torch.Size([1, 1000, 754]) | 2\n",
-      "7277     | torch.Size([1, 1000, 754]) | 2\n",
-      "7278     | torch.Size([1, 1000, 754]) | 2\n",
-      "7279     | torch.Size([1, 1000, 754]) | 2\n",
-      "7280     | torch.Size([1, 1000, 754]) | 2\n",
-      "7281     | torch.Size([1, 1000, 802]) | 2\n",
-      "7282     | torch.Size([1, 1000, 802]) | 2\n",
-      "7283     | torch.Size([1, 1000, 754]) | 2\n",
-      "7284     | torch.Size([1, 1000, 754]) | 2\n",
-      "7285     | torch.Size([1, 1000, 754]) | 2\n",
-      "7286     | torch.Size([1, 1000, 754]) | 2\n",
-      "7287     | torch.Size([1, 1000, 754]) | 2\n",
-      "7288     | torch.Size([1, 1000, 754]) | 2\n",
-      "7289     | torch.Size([1, 1000, 754]) | 2\n",
-      "7290     | torch.Size([1, 1000, 754]) | 2\n",
-      "7291     | torch.Size([1, 1000, 754]) | 2\n",
-      "7292     | torch.Size([1, 1000, 754]) | 2\n",
-      "7293     | torch.Size([1, 1000, 754]) | 2\n",
-      "7294     | torch.Size([1, 1000, 754]) | 2\n",
-      "7295     | torch.Size([1, 1000, 754]) | 2\n",
-      "7296     | torch.Size([1, 1000, 754]) | 2\n",
-      "7297     | torch.Size([1, 1000, 754]) | 2\n",
-      "7298     | torch.Size([1, 1000, 754]) | 2\n",
-      "7299     | torch.Size([1, 1000, 754]) | 2\n",
-      "7300     | torch.Size([1, 1000, 754]) | 2\n",
-      "7301     | torch.Size([1, 1000, 754]) | 2\n",
-      "7302     | torch.Size([1, 1000, 754]) | 2\n",
-      "7303     | torch.Size([1, 1000, 754]) | 2\n",
-      "7304     | torch.Size([1, 1000, 754]) | 2\n",
-      "7305     | torch.Size([1, 1000, 754]) | 2\n",
-      "7306     | torch.Size([1, 1000, 754]) | 2\n",
-      "7307     | torch.Size([1, 1000, 754]) | 2\n",
-      "7308     | torch.Size([1, 1000, 754]) | 2\n",
-      "7309     | torch.Size([1, 1000, 754]) | 2\n",
-      "7310     | torch.Size([1, 1000, 754]) | 2\n",
-      "7311     | torch.Size([1, 1000, 754]) | 2\n",
-      "7312     | torch.Size([1, 1000, 754]) | 2\n",
-      "7313     | torch.Size([1, 1000, 754]) | 2\n",
-      "7314     | torch.Size([1, 1000, 754]) | 2\n",
-      "7315     | torch.Size([1, 1000, 754]) | 2\n",
-      "7316     | torch.Size([1, 1000, 754]) | 2\n",
-      "7317     | torch.Size([1, 1000, 754]) | 2\n",
-      "7318     | torch.Size([1, 1000, 754]) | 2\n",
-      "7319     | torch.Size([1, 1000, 754]) | 2\n",
-      "7320     | torch.Size([1, 1000, 754]) | 2\n",
-      "7321     | torch.Size([1, 1000, 754]) | 2\n",
-      "7322     | torch.Size([1, 1000, 754]) | 2\n",
-      "7323     | torch.Size([1, 1000, 754]) | 2\n",
-      "7324     | torch.Size([1, 1000, 754]) | 2\n",
-      "7325     | torch.Size([1, 1000, 754]) | 2\n",
-      "7326     | torch.Size([1, 1000, 754]) | 2\n",
-      "7327     | torch.Size([1, 1000, 754]) | 2\n",
-      "7328     | torch.Size([1, 1000, 754]) | 2\n",
-      "7329     | torch.Size([1, 1000, 754]) | 2\n",
-      "7330     | torch.Size([1, 1000, 754]) | 2\n",
-      "7331     | torch.Size([1, 1000, 754]) | 2\n",
-      "7332     | torch.Size([1, 1000, 754]) | 2\n",
-      "7333     | torch.Size([1, 1000, 754]) | 2\n",
-      "7334     | torch.Size([1, 1000, 754]) | 2\n",
-      "7335     | torch.Size([1, 1000, 754]) | 2\n",
-      "7336     | torch.Size([1, 1000, 754]) | 2\n",
-      "7337     | torch.Size([1, 1000, 754]) | 2\n",
-      "7338     | torch.Size([1, 1000, 754]) | 2\n",
-      "7339     | torch.Size([1, 1000, 754]) | 2\n",
-      "7340     | torch.Size([1, 1000, 754]) | 2\n",
-      "7341     | torch.Size([1, 1000, 754]) | 2\n",
-      "7342     | torch.Size([1, 1000, 754]) | 2\n",
-      "7343     | torch.Size([1, 1000, 754]) | 2\n",
-      "7344     | torch.Size([1, 1000, 754]) | 2\n",
-      "7345     | torch.Size([1, 1000, 754]) | 2\n",
-      "7346     | torch.Size([1, 1000, 754]) | 2\n",
-      "7347     | torch.Size([1, 1000, 754]) | 2\n",
-      "7348     | torch.Size([1, 1000, 754]) | 2\n",
-      "7349     | torch.Size([1, 1000, 754]) | 2\n",
-      "7350     | torch.Size([1, 1000, 754]) | 2\n",
-      "7351     | torch.Size([1, 1000, 754]) | 2\n",
-      "7352     | torch.Size([1, 1000, 754]) | 2\n",
-      "7353     | torch.Size([1, 1000, 754]) | 2\n",
-      "7354     | torch.Size([1, 1000, 754]) | 2\n",
-      "7355     | torch.Size([1, 1000, 754]) | 2\n",
-      "7356     | torch.Size([1, 1000, 754]) | 2\n",
-      "7357     | torch.Size([1, 1000, 754]) | 2\n",
-      "7358     | torch.Size([1, 1000, 754]) | 2\n",
-      "7359     | torch.Size([1, 1000, 754]) | 2\n",
-      "7360     | torch.Size([1, 1000, 754]) | 2\n",
-      "7361     | torch.Size([1, 1000, 754]) | 2\n",
-      "7362     | torch.Size([1, 1000, 754]) | 2\n",
-      "7363     | torch.Size([1, 1000, 754]) | 2\n",
-      "7364     | torch.Size([1, 1000, 754]) | 2\n",
-      "7365     | torch.Size([1, 1000, 754]) | 2\n",
-      "7366     | torch.Size([1, 1000, 754]) | 2\n",
-      "7367     | torch.Size([1, 1000, 754]) | 2\n",
-      "7368     | torch.Size([1, 1000, 762]) | 2\n",
-      "7369     | torch.Size([1, 1000, 754]) | 2\n",
-      "7370     | torch.Size([1, 1000, 754]) | 2\n",
-      "7371     | torch.Size([1, 1000, 754]) | 2\n",
-      "7372     | torch.Size([1, 1000, 754]) | 2\n",
-      "7373     | torch.Size([1, 1000, 754]) | 2\n",
-      "7374     | torch.Size([1, 1000, 754]) | 2\n",
-      "7375     | torch.Size([1, 1000, 754]) | 2\n",
-      "7376     | torch.Size([1, 1000, 754]) | 2\n",
-      "7377     | torch.Size([1, 1000, 754]) | 2\n",
-      "7378     | torch.Size([1, 1000, 754]) | 2\n",
-      "7379     | torch.Size([1, 1000, 754]) | 2\n",
-      "7380     | torch.Size([1, 1000, 754]) | 2\n",
-      "7381     | torch.Size([1, 1000, 754]) | 2\n",
-      "7382     | torch.Size([1, 1000, 754]) | 2\n",
-      "7383     | torch.Size([1, 1000, 754]) | 2\n",
-      "7384     | torch.Size([1, 1000, 802]) | 2\n",
-      "7385     | torch.Size([1, 1000, 754]) | 2\n",
-      "7386     | torch.Size([1, 1000, 754]) | 2\n",
-      "7387     | torch.Size([1, 1000, 754]) | 2\n",
-      "7388     | torch.Size([1, 1000, 754]) | 2\n",
-      "7389     | torch.Size([1, 1000, 754]) | 2\n",
-      "7390     | torch.Size([1, 1000, 754]) | 2\n",
-      "7391     | torch.Size([1, 1000, 754]) | 2\n",
-      "7392     | torch.Size([1, 1000, 754]) | 2\n",
-      "7393     | torch.Size([1, 1000, 754]) | 2\n",
-      "7394     | torch.Size([1, 1000, 802]) | 2\n",
-      "7395     | torch.Size([1, 1000, 754]) | 2\n",
-      "7396     | torch.Size([1, 1000, 754]) | 2\n",
-      "7397     | torch.Size([1, 1000, 754]) | 2\n",
-      "7398     | torch.Size([1, 1000, 754]) | 2\n",
-      "7399     | torch.Size([1, 1000, 754]) | 2\n",
-      "7400     | torch.Size([1, 1000, 754]) | 2\n",
-      "7401     | torch.Size([1, 1000, 754]) | 2\n",
-      "7402     | torch.Size([1, 1000, 754]) | 2\n",
-      "7403     | torch.Size([1, 1000, 754]) | 2\n",
-      "7404     | torch.Size([1, 1000, 754]) | 2\n",
-      "7405     | torch.Size([1, 1000, 754]) | 2\n",
-      "7406     | torch.Size([1, 1000, 754]) | 2\n",
-      "7407     | torch.Size([1, 1000, 754]) | 2\n",
-      "7408     | torch.Size([1, 1000, 754]) | 2\n",
-      "7409     | torch.Size([1, 1000, 754]) | 2\n",
-      "7410     | torch.Size([1, 1000, 754]) | 2\n",
-      "7411     | torch.Size([1, 1000, 754]) | 2\n",
-      "7412     | torch.Size([1, 1000, 754]) | 2\n",
-      "7413     | torch.Size([1, 1000, 754]) | 2\n",
-      "7414     | torch.Size([1, 1000, 754]) | 2\n",
-      "7415     | torch.Size([1, 1000, 754]) | 2\n",
-      "7416     | torch.Size([1, 1000, 754]) | 2\n",
-      "7417     | torch.Size([1, 1000, 754]) | 2\n",
-      "7418     | torch.Size([1, 1000, 802]) | 2\n",
-      "7419     | torch.Size([1, 1000, 754]) | 2\n",
-      "7420     | torch.Size([1, 1000, 754]) | 2\n",
-      "7421     | torch.Size([1, 1000, 754]) | 2\n",
-      "7422     | torch.Size([1, 1000, 754]) | 2\n",
-      "7423     | torch.Size([1, 1000, 754]) | 2\n",
-      "7424     | torch.Size([1, 1000, 754]) | 2\n",
-      "7425     | torch.Size([1, 1000, 754]) | 2\n",
-      "7426     | torch.Size([1, 1000, 754]) | 2\n",
-      "7427     | torch.Size([1, 1000, 754]) | 2\n",
-      "7428     | torch.Size([1, 1000, 754]) | 2\n",
-      "7429     | torch.Size([1, 1000, 754]) | 2\n",
-      "7430     | torch.Size([1, 1000, 754]) | 2\n",
-      "7431     | torch.Size([1, 1000, 754]) | 2\n",
-      "7432     | torch.Size([1, 1000, 754]) | 2\n",
-      "7433     | torch.Size([1, 1000, 754]) | 2\n",
-      "7434     | torch.Size([1, 1000, 754]) | 2\n",
-      "7435     | torch.Size([1, 1000, 754]) | 2\n",
-      "7436     | torch.Size([1, 1000, 754]) | 2\n",
-      "7437     | torch.Size([1, 1000, 802]) | 2\n",
-      "7438     | torch.Size([1, 1000, 754]) | 2\n",
-      "7439     | torch.Size([1, 1000, 754]) | 2\n",
-      "7440     | torch.Size([1, 1000, 754]) | 2\n",
-      "7441     | torch.Size([1, 1000, 754]) | 2\n",
-      "7442     | torch.Size([1, 1000, 754]) | 2\n",
-      "7443     | torch.Size([1, 1000, 754]) | 2\n",
-      "7444     | torch.Size([1, 1000, 754]) | 2\n",
-      "7445     | torch.Size([1, 1000, 754]) | 2\n",
-      "7446     | torch.Size([1, 1000, 754]) | 2\n",
-      "7447     | torch.Size([1, 1000, 754]) | 2\n",
-      "7448     | torch.Size([1, 1000, 754]) | 2\n",
-      "7449     | torch.Size([1, 1000, 802]) | 2\n",
-      "7450     | torch.Size([1, 1000, 754]) | 2\n",
-      "7451     | torch.Size([1, 1000, 754]) | 2\n",
-      "7452     | torch.Size([1, 1000, 754]) | 2\n",
-      "7453     | torch.Size([1, 1000, 754]) | 2\n",
-      "7454     | torch.Size([1, 1000, 754]) | 2\n",
-      "7455     | torch.Size([1, 1000, 754]) | 2\n",
-      "7456     | torch.Size([1, 1000, 754]) | 2\n",
-      "7457     | torch.Size([1, 1000, 754]) | 2\n",
-      "7458     | torch.Size([1, 1000, 754]) | 2\n",
-      "7459     | torch.Size([1, 1000, 754]) | 2\n",
-      "7460     | torch.Size([1, 1000, 754]) | 2\n",
-      "7461     | torch.Size([1, 1000, 754]) | 2\n",
-      "7462     | torch.Size([1, 1000, 754]) | 2\n",
-      "7463     | torch.Size([1, 1000, 754]) | 2\n",
-      "7464     | torch.Size([1, 1000, 754]) | 2\n",
-      "7465     | torch.Size([1, 1000, 754]) | 2\n",
-      "7466     | torch.Size([1, 1000, 754]) | 2\n",
-      "7467     | torch.Size([1, 1000, 754]) | 2\n",
-      "7468     | torch.Size([1, 1000, 754]) | 2\n",
-      "7469     | torch.Size([1, 1000, 754]) | 2\n",
-      "7470     | torch.Size([1, 1000, 754]) | 2\n",
-      "7471     | torch.Size([1, 1000, 754]) | 2\n",
-      "7472     | torch.Size([1, 1000, 754]) | 2\n",
-      "7473     | torch.Size([1, 1000, 754]) | 2\n",
-      "7474     | torch.Size([1, 1000, 754]) | 2\n",
-      "7475     | torch.Size([1, 1000, 754]) | 2\n",
-      "7476     | torch.Size([1, 1000, 754]) | 2\n",
-      "7477     | torch.Size([1, 1000, 754]) | 2\n",
-      "7478     | torch.Size([1, 1000, 754]) | 2\n",
-      "7479     | torch.Size([1, 1000, 754]) | 2\n",
-      "7480     | torch.Size([1, 1000, 754]) | 2\n",
-      "7481     | torch.Size([1, 1000, 754]) | 2\n",
-      "7482     | torch.Size([1, 1000, 754]) | 2\n",
-      "7483     | torch.Size([1, 1000, 754]) | 2\n",
-      "7484     | torch.Size([1, 1000, 754]) | 2\n",
-      "7485     | torch.Size([1, 1000, 754]) | 2\n",
-      "7486     | torch.Size([1, 1000, 754]) | 2\n",
-      "7487     | torch.Size([1, 1000, 754]) | 2\n",
-      "7488     | torch.Size([1, 1000, 754]) | 2\n",
-      "7489     | torch.Size([1, 1000, 754]) | 2\n",
-      "7490     | torch.Size([1, 1000, 754]) | 2\n",
-      "7491     | torch.Size([1, 1000, 754]) | 2\n",
-      "7492     | torch.Size([1, 1000, 754]) | 2\n",
-      "7493     | torch.Size([1, 1000, 754]) | 2\n",
-      "7494     | torch.Size([1, 1000, 754]) | 2\n",
-      "7495     | torch.Size([1, 1000, 754]) | 2\n",
-      "7496     | torch.Size([1, 1000, 754]) | 2\n",
-      "7497     | torch.Size([1, 1000, 754]) | 2\n",
-      "7498     | torch.Size([1, 1000, 754]) | 2\n",
-      "7499     | torch.Size([1, 1000, 754]) | 2\n",
-      "7500     | torch.Size([1, 1000, 754]) | 2\n",
-      "7501     | torch.Size([1, 1000, 754]) | 2\n",
-      "7502     | torch.Size([1, 1000, 754]) | 2\n",
-      "7503     | torch.Size([1, 1000, 754]) | 2\n",
-      "7504     | torch.Size([1, 1000, 754]) | 2\n",
-      "7505     | torch.Size([1, 1000, 802]) | 2\n",
-      "7506     | torch.Size([1, 1000, 754]) | 2\n",
-      "7507     | torch.Size([1, 1000, 754]) | 2\n",
-      "7508     | torch.Size([1, 1000, 754]) | 2\n",
-      "7509     | torch.Size([1, 1000, 754]) | 2\n",
-      "7510     | torch.Size([1, 1000, 754]) | 2\n",
-      "7511     | torch.Size([1, 1000, 754]) | 2\n",
-      "7512     | torch.Size([1, 1000, 754]) | 2\n",
-      "7513     | torch.Size([1, 1000, 754]) | 2\n",
-      "7514     | torch.Size([1, 1000, 754]) | 2\n",
-      "7515     | torch.Size([1, 1000, 754]) | 2\n",
-      "7516     | torch.Size([1, 1000, 754]) | 2\n",
-      "7517     | torch.Size([1, 1000, 754]) | 2\n",
-      "7518     | torch.Size([1, 1000, 754]) | 2\n",
-      "7519     | torch.Size([1, 1000, 754]) | 2\n",
-      "7520     | torch.Size([1, 1000, 754]) | 2\n",
-      "7521     | torch.Size([1, 1000, 754]) | 2\n",
-      "7522     | torch.Size([1, 1000, 754]) | 2\n",
-      "7523     | torch.Size([1, 1000, 754]) | 2\n",
-      "7524     | torch.Size([1, 1000, 754]) | 2\n",
-      "7525     | torch.Size([1, 1000, 754]) | 2\n",
-      "7526     | torch.Size([1, 1000, 754]) | 2\n",
-      "7527     | torch.Size([1, 1000, 754]) | 2\n",
-      "7528     | torch.Size([1, 1000, 754]) | 2\n",
-      "7529     | torch.Size([1, 1000, 754]) | 2\n",
-      "7530     | torch.Size([1, 1000, 754]) | 2\n",
-      "7531     | torch.Size([1, 1000, 754]) | 2\n",
-      "7532     | torch.Size([1, 1000, 754]) | 2\n",
-      "7533     | torch.Size([1, 1000, 754]) | 2\n",
-      "7534     | torch.Size([1, 1000, 754]) | 2\n",
-      "7535     | torch.Size([1, 1000, 754]) | 2\n",
-      "7536     | torch.Size([1, 1000, 754]) | 3\n",
-      "7537     | torch.Size([1, 1000, 795]) | 3\n",
-      "7538     | torch.Size([1, 1000, 762]) | 3\n",
-      "7539     | torch.Size([1, 1000, 762]) | 3\n",
-      "7540     | torch.Size([1, 1000, 754]) | 3\n",
-      "7541     | torch.Size([1, 1000, 777]) | 3\n",
-      "7542     | torch.Size([1, 1000, 793]) | 3\n",
-      "7543     | torch.Size([1, 1000, 802]) | 3\n",
-      "7544     | torch.Size([1, 1000, 787]) | 3\n",
-      "7545     | torch.Size([1, 1000, 813]) | 3\n",
-      "7546     | torch.Size([1, 1000, 762]) | 3\n",
-      "7547     | torch.Size([1, 1000, 808]) | 3\n",
-      "7548     | torch.Size([1, 1000, 754]) | 3\n",
-      "7549     | torch.Size([1, 1000, 754]) | 3\n",
-      "7550     | torch.Size([1, 1000, 773]) | 3\n",
-      "7551     | torch.Size([1, 1000, 785]) | 3\n",
-      "7552     | torch.Size([1, 1000, 762]) | 3\n",
-      "7553     | torch.Size([1, 1000, 754]) | 3\n",
-      "7554     | torch.Size([1, 1000, 754]) | 3\n",
-      "7555     | torch.Size([1, 1000, 762]) | 3\n",
-      "7556     | torch.Size([1, 1000, 754]) | 3\n",
-      "7557     | torch.Size([1, 1000, 754]) | 3\n",
-      "7558     | torch.Size([1, 1000, 769]) | 3\n",
-      "7559     | torch.Size([1, 1000, 776]) | 3\n",
-      "7560     | torch.Size([1, 1000, 754]) | 3\n",
-      "7561     | torch.Size([1, 1000, 776]) | 3\n",
-      "7562     | torch.Size([1, 1000, 754]) | 3\n",
-      "7563     | torch.Size([1, 1000, 780]) | 3\n",
-      "7564     | torch.Size([1, 1000, 762]) | 3\n",
-      "7565     | torch.Size([1, 1000, 762]) | 3\n",
-      "7566     | torch.Size([1, 1000, 754]) | 3\n",
-      "7567     | torch.Size([1, 1000, 754]) | 3\n",
-      "7568     | torch.Size([1, 1000, 792]) | 3\n",
-      "7569     | torch.Size([1, 1000, 777]) | 3\n",
-      "7570     | torch.Size([1, 1000, 754]) | 3\n",
-      "7571     | torch.Size([1, 1000, 781]) | 3\n",
-      "7572     | torch.Size([1, 1000, 780]) | 3\n",
-      "7573     | torch.Size([1, 1000, 754]) | 3\n",
-      "7574     | torch.Size([1, 1000, 754]) | 3\n",
-      "7575     | torch.Size([1, 1000, 754]) | 3\n",
-      "7576     | torch.Size([1, 1000, 754]) | 3\n",
-      "7577     | torch.Size([1, 1000, 762]) | 3\n",
-      "7578     | torch.Size([1, 1000, 754]) | 3\n",
-      "7579     | torch.Size([1, 1000, 762]) | 3\n",
-      "7580     | torch.Size([1, 1000, 713]) | 3\n",
-      "7581     | torch.Size([1, 1000, 754]) | 3\n",
-      "7582     | torch.Size([1, 1000, 754]) | 3\n",
-      "7583     | torch.Size([1, 1000, 754]) | 3\n",
-      "7584     | torch.Size([1, 1000, 754]) | 3\n",
-      "7585     | torch.Size([1, 1000, 762]) | 3\n",
-      "7586     | torch.Size([1, 1000, 771]) | 3\n",
-      "7587     | torch.Size([1, 1000, 772]) | 3\n",
-      "7588     | torch.Size([1, 1000, 754]) | 3\n",
-      "7589     | torch.Size([1, 1000, 777]) | 3\n",
-      "7590     | torch.Size([1, 1000, 794]) | 3\n",
-      "7591     | torch.Size([1, 1000, 762]) | 3\n",
-      "7592     | torch.Size([1, 1000, 754]) | 3\n",
-      "7593     | torch.Size([1, 1000, 847]) | 3\n",
-      "7594     | torch.Size([1, 1000, 782]) | 3\n",
-      "7595     | torch.Size([1, 1000, 762]) | 3\n",
-      "7596     | torch.Size([1, 1000, 754]) | 3\n",
-      "7597     | torch.Size([1, 1000, 754]) | 3\n",
-      "7598     | torch.Size([1, 1000, 783]) | 3\n",
-      "7599     | torch.Size([1, 1000, 777]) | 3\n",
-      "7600     | torch.Size([1, 1000, 779]) | 3\n",
-      "7601     | torch.Size([1, 1000, 854]) | 3\n",
-      "7602     | torch.Size([1, 1000, 847]) | 3\n",
-      "7603     | torch.Size([1, 1000, 762]) | 3\n",
-      "7604     | torch.Size([1, 1000, 754]) | 3\n",
-      "7605     | torch.Size([1, 1000, 754]) | 3\n",
-      "7606     | torch.Size([1, 1000, 777]) | 3\n",
-      "7607     | torch.Size([1, 1000, 762]) | 3\n",
-      "7608     | torch.Size([1, 1000, 754]) | 3\n",
-      "7609     | torch.Size([1, 1000, 772]) | 3\n",
-      "7610     | torch.Size([1, 1000, 840]) | 3\n",
-      "7611     | torch.Size([1, 1000, 720]) | 3\n",
-      "7612     | torch.Size([1, 1000, 754]) | 3\n",
-      "7613     | torch.Size([1, 1000, 779]) | 3\n",
-      "7614     | torch.Size([1, 1000, 754]) | 3\n",
-      "7615     | torch.Size([1, 1000, 793]) | 3\n",
-      "7616     | torch.Size([1, 1000, 772]) | 3\n",
-      "7617     | torch.Size([1, 1000, 762]) | 3\n",
-      "7618     | torch.Size([1, 1000, 777]) | 3\n",
-      "7619     | torch.Size([1, 1000, 781]) | 3\n",
-      "7620     | torch.Size([1, 1000, 754]) | 3\n",
-      "7621     | torch.Size([1, 1000, 754]) | 3\n",
-      "7622     | torch.Size([1, 1000, 762]) | 3\n",
-      "7623     | torch.Size([1, 1000, 772]) | 3\n",
-      "7624     | torch.Size([1, 1000, 799]) | 3\n",
-      "7625     | torch.Size([1, 1000, 787]) | 3\n",
-      "7626     | torch.Size([1, 1000, 798]) | 3\n",
-      "7627     | torch.Size([1, 1000, 762]) | 3\n",
-      "7628     | torch.Size([1, 1000, 779]) | 3\n",
-      "7629     | torch.Size([1, 1000, 754]) | 3\n",
-      "7630     | torch.Size([1, 1000, 787]) | 3\n",
-      "7631     | torch.Size([1, 1000, 777]) | 3\n",
-      "7632     | torch.Size([1, 1000, 754]) | 3\n",
-      "7633     | torch.Size([1, 1000, 777]) | 3\n",
-      "7634     | torch.Size([1, 1000, 771]) | 3\n",
-      "7635     | torch.Size([1, 1000, 779]) | 3\n",
-      "7636     | torch.Size([1, 1000, 788]) | 3\n",
-      "7637     | torch.Size([1, 1000, 754]) | 3\n",
-      "7638     | torch.Size([1, 1000, 734]) | 3\n",
-      "7639     | torch.Size([1, 1000, 727]) | 3\n",
-      "7640     | torch.Size([1, 1000, 762]) | 3\n",
-      "7641     | torch.Size([1, 1000, 777]) | 3\n",
-      "7642     | torch.Size([1, 1000, 772]) | 3\n",
-      "7643     | torch.Size([1, 1000, 784]) | 3\n",
-      "7644     | torch.Size([1, 1000, 754]) | 3\n",
-      "7645     | torch.Size([1, 1000, 754]) | 3\n",
-      "7646     | torch.Size([1, 1000, 784]) | 3\n",
-      "7647     | torch.Size([1, 1000, 777]) | 3\n",
-      "7648     | torch.Size([1, 1000, 754]) | 3\n",
-      "7649     | torch.Size([1, 1000, 754]) | 3\n",
-      "7650     | torch.Size([1, 1000, 780]) | 3\n",
-      "7651     | torch.Size([1, 1000, 754]) | 3\n",
-      "7652     | torch.Size([1, 1000, 772]) | 3\n",
-      "7653     | torch.Size([1, 1000, 762]) | 3\n",
-      "7654     | torch.Size([1, 1000, 772]) | 3\n",
-      "7655     | torch.Size([1, 1000, 789]) | 3\n",
-      "7656     | torch.Size([1, 1000, 754]) | 3\n",
-      "7657     | torch.Size([1, 1000, 729]) | 3\n",
-      "7658     | torch.Size([1, 1000, 771]) | 3\n",
-      "7659     | torch.Size([1, 1000, 754]) | 3\n",
-      "7660     | torch.Size([1, 1000, 754]) | 3\n",
-      "7661     | torch.Size([1, 1000, 762]) | 3\n",
-      "7662     | torch.Size([1, 1000, 754]) | 3\n",
-      "7663     | torch.Size([1, 1000, 762]) | 3\n",
-      "7664     | torch.Size([1, 1000, 765]) | 3\n",
-      "7665     | torch.Size([1, 1000, 778]) | 3\n",
-      "7666     | torch.Size([1, 1000, 777]) | 3\n",
-      "7667     | torch.Size([1, 1000, 762]) | 3\n",
-      "7668     | torch.Size([1, 1000, 777]) | 3\n",
-      "7669     | torch.Size([1, 1000, 780]) | 3\n",
-      "7670     | torch.Size([1, 1000, 762]) | 3\n",
-      "7671     | torch.Size([1, 1000, 785]) | 3\n",
-      "7672     | torch.Size([1, 1000, 762]) | 3\n",
-      "7673     | torch.Size([1, 1000, 762]) | 3\n",
-      "7674     | torch.Size([1, 1000, 754]) | 3\n",
-      "7675     | torch.Size([1, 1000, 772]) | 3\n",
-      "7676     | torch.Size([1, 1000, 762]) | 3\n",
-      "7677     | torch.Size([1, 1000, 754]) | 3\n",
-      "7678     | torch.Size([1, 1000, 754]) | 3\n",
-      "7679     | torch.Size([1, 1000, 762]) | 3\n",
-      "7680     | torch.Size([1, 1000, 777]) | 3\n",
-      "7681     | torch.Size([1, 1000, 777]) | 3\n",
-      "7682     | torch.Size([1, 1000, 754]) | 3\n",
-      "7683     | torch.Size([1, 1000, 797]) | 3\n",
-      "7684     | torch.Size([1, 1000, 777]) | 3\n",
-      "7685     | torch.Size([1, 1000, 762]) | 3\n",
-      "7686     | torch.Size([1, 1000, 762]) | 3\n",
-      "7687     | torch.Size([1, 1000, 754]) | 3\n",
-      "7688     | torch.Size([1, 1000, 754]) | 3\n",
-      "7689     | torch.Size([1, 1000, 777]) | 3\n",
-      "7690     | torch.Size([1, 1000, 754]) | 3\n",
-      "7691     | torch.Size([1, 1000, 754]) | 3\n",
-      "7692     | torch.Size([1, 1000, 754]) | 3\n",
-      "7693     | torch.Size([1, 1000, 754]) | 3\n",
-      "7694     | torch.Size([1, 1000, 762]) | 3\n",
-      "7695     | torch.Size([1, 1000, 754]) | 3\n",
-      "7696     | torch.Size([1, 1000, 754]) | 3\n",
-      "7697     | torch.Size([1, 1000, 772]) | 3\n",
-      "7698     | torch.Size([1, 1000, 713]) | 3\n",
-      "7699     | torch.Size([1, 1000, 785]) | 3\n",
-      "7700     | torch.Size([1, 1000, 772]) | 3\n",
-      "7701     | torch.Size([1, 1000, 777]) | 3\n",
-      "7702     | torch.Size([1, 1000, 762]) | 3\n",
-      "7703     | torch.Size([1, 1000, 754]) | 3\n",
-      "7704     | torch.Size([1, 1000, 787]) | 3\n",
-      "7705     | torch.Size([1, 1000, 754]) | 3\n",
-      "7706     | torch.Size([1, 1000, 783]) | 3\n",
-      "7707     | torch.Size([1, 1000, 762]) | 3\n",
-      "7708     | torch.Size([1, 1000, 777]) | 3\n",
-      "7709     | torch.Size([1, 1000, 781]) | 3\n",
-      "7710     | torch.Size([1, 1000, 754]) | 3\n",
-      "7711     | torch.Size([1, 1000, 754]) | 3\n",
-      "7712     | torch.Size([1, 1000, 713]) | 3\n",
-      "7713     | torch.Size([1, 1000, 754]) | 3\n",
-      "7714     | torch.Size([1, 1000, 754]) | 3\n",
-      "7715     | torch.Size([1, 1000, 777]) | 3\n",
-      "7716     | torch.Size([1, 1000, 754]) | 3\n",
-      "7717     | torch.Size([1, 1000, 754]) | 3\n",
-      "7718     | torch.Size([1, 1000, 777]) | 3\n",
-      "7719     | torch.Size([1, 1000, 777]) | 3\n",
-      "7720     | torch.Size([1, 1000, 777]) | 3\n",
-      "7721     | torch.Size([1, 1000, 777]) | 3\n",
-      "7722     | torch.Size([1, 1000, 754]) | 3\n",
-      "7723     | torch.Size([1, 1000, 772]) | 3\n",
-      "7724     | torch.Size([1, 1000, 754]) | 3\n",
-      "7725     | torch.Size([1, 1000, 762]) | 3\n",
-      "7726     | torch.Size([1, 1000, 764]) | 3\n",
-      "7727     | torch.Size([1, 1000, 777]) | 3\n",
-      "7728     | torch.Size([1, 1000, 784]) | 3\n",
-      "7729     | torch.Size([1, 1000, 777]) | 3\n",
-      "7730     | torch.Size([1, 1000, 757]) | 3\n",
-      "7731     | torch.Size([1, 1000, 754]) | 3\n",
-      "7732     | torch.Size([1, 1000, 845]) | 3\n",
-      "7733     | torch.Size([1, 1000, 812]) | 3\n",
-      "7734     | torch.Size([1, 1000, 777]) | 3\n",
-      "7735     | torch.Size([1, 1000, 781]) | 3\n",
-      "7736     | torch.Size([1, 1000, 762]) | 3\n",
-      "7737     | torch.Size([1, 1000, 777]) | 3\n",
-      "7738     | torch.Size([1, 1000, 768]) | 3\n",
-      "7739     | torch.Size([1, 1000, 754]) | 3\n",
-      "7740     | torch.Size([1, 1000, 762]) | 3\n",
-      "7741     | torch.Size([1, 1000, 725]) | 3\n",
-      "7742     | torch.Size([1, 1000, 762]) | 3\n",
-      "7743     | torch.Size([1, 1000, 762]) | 3\n",
-      "7744     | torch.Size([1, 1000, 777]) | 3\n",
-      "7745     | torch.Size([1, 1000, 779]) | 3\n",
-      "7746     | torch.Size([1, 1000, 869]) | 3\n",
-      "7747     | torch.Size([1, 1000, 754]) | 3\n",
-      "7748     | torch.Size([1, 1000, 777]) | 3\n",
-      "7749     | torch.Size([1, 1000, 867]) | 3\n",
-      "7750     | torch.Size([1, 1000, 802]) | 3\n",
-      "7751     | torch.Size([1, 1000, 762]) | 3\n",
-      "7752     | torch.Size([1, 1000, 777]) | 3\n",
-      "7753     | torch.Size([1, 1000, 754]) | 3\n",
-      "7754     | torch.Size([1, 1000, 840]) | 3\n",
-      "7755     | torch.Size([1, 1000, 777]) | 3\n",
-      "7756     | torch.Size([1, 1000, 754]) | 3\n",
-      "7757     | torch.Size([1, 1000, 762]) | 3\n",
-      "7758     | torch.Size([1, 1000, 777]) | 3\n",
-      "7759     | torch.Size([1, 1000, 795]) | 3\n",
-      "7760     | torch.Size([1, 1000, 762]) | 3\n",
-      "7761     | torch.Size([1, 1000, 777]) | 3\n",
-      "7762     | torch.Size([1, 1000, 829]) | 3\n",
-      "7763     | torch.Size([1, 1000, 762]) | 3\n",
-      "7764     | torch.Size([1, 1000, 873]) | 3\n",
-      "7765     | torch.Size([1, 1000, 754]) | 3\n",
-      "7766     | torch.Size([1, 1000, 771]) | 3\n",
-      "7767     | torch.Size([1, 1000, 774]) | 3\n",
-      "7768     | torch.Size([1, 1000, 777]) | 3\n",
-      "7769     | torch.Size([1, 1000, 742]) | 3\n",
-      "7770     | torch.Size([1, 1000, 762]) | 3\n",
-      "7771     | torch.Size([1, 1000, 754]) | 3\n",
-      "7772     | torch.Size([1, 1000, 777]) | 3\n",
-      "7773     | torch.Size([1, 1000, 783]) | 3\n",
-      "7774     | torch.Size([1, 1000, 754]) | 3\n",
-      "7775     | torch.Size([1, 1000, 754]) | 3\n",
-      "7776     | torch.Size([1, 1000, 762]) | 3\n",
-      "7777     | torch.Size([1, 1000, 762]) | 3\n",
-      "7778     | torch.Size([1, 1000, 764]) | 3\n",
-      "7779     | torch.Size([1, 1000, 767]) | 3\n",
-      "7780     | torch.Size([1, 1000, 775]) | 3\n",
-      "7781     | torch.Size([1, 1000, 762]) | 3\n",
-      "7782     | torch.Size([1, 1000, 777]) | 3\n",
-      "7783     | torch.Size([1, 1000, 794]) | 3\n",
-      "7784     | torch.Size([1, 1000, 777]) | 3\n",
-      "7785     | torch.Size([1, 1000, 772]) | 3\n",
-      "7786     | torch.Size([1, 1000, 762]) | 3\n",
-      "7787     | torch.Size([1, 1000, 762]) | 3\n",
-      "7788     | torch.Size([1, 1000, 754]) | 3\n",
-      "7789     | torch.Size([1, 1000, 771]) | 3\n",
-      "7790     | torch.Size([1, 1000, 754]) | 3\n",
-      "7791     | torch.Size([1, 1000, 754]) | 3\n",
-      "7792     | torch.Size([1, 1000, 754]) | 3\n",
-      "7793     | torch.Size([1, 1000, 762]) | 3\n",
-      "7794     | torch.Size([1, 1000, 772]) | 3\n",
-      "7795     | torch.Size([1, 1000, 754]) | 3\n",
-      "7796     | torch.Size([1, 1000, 754]) | 3\n",
-      "7797     | torch.Size([1, 1000, 762]) | 3\n",
-      "7798     | torch.Size([1, 1000, 775]) | 3\n",
-      "7799     | torch.Size([1, 1000, 777]) | 3\n",
-      "7800     | torch.Size([1, 1000, 741]) | 3\n",
-      "7801     | torch.Size([1, 1000, 754]) | 3\n",
-      "7802     | torch.Size([1, 1000, 777]) | 3\n",
-      "7803     | torch.Size([1, 1000, 765]) | 3\n",
-      "7804     | torch.Size([1, 1000, 754]) | 3\n",
-      "7805     | torch.Size([1, 1000, 762]) | 3\n",
-      "7806     | torch.Size([1, 1000, 772]) | 3\n",
-      "7807     | torch.Size([1, 1000, 762]) | 3\n",
-      "7808     | torch.Size([1, 1000, 762]) | 3\n",
-      "7809     | torch.Size([1, 1000, 754]) | 3\n",
-      "7810     | torch.Size([1, 1000, 777]) | 3\n",
-      "7811     | torch.Size([1, 1000, 754]) | 3\n",
-      "7812     | torch.Size([1, 1000, 762]) | 3\n",
-      "7813     | torch.Size([1, 1000, 754]) | 3\n",
-      "7814     | torch.Size([1, 1000, 725]) | 3\n",
-      "7815     | torch.Size([1, 1000, 754]) | 3\n",
-      "7816     | torch.Size([1, 1000, 762]) | 3\n",
-      "7817     | torch.Size([1, 1000, 794]) | 3\n",
-      "7818     | torch.Size([1, 1000, 762]) | 3\n",
-      "7819     | torch.Size([1, 1000, 779]) | 3\n",
-      "7820     | torch.Size([1, 1000, 754]) | 3\n",
-      "7821     | torch.Size([1, 1000, 802]) | 3\n",
-      "7822     | torch.Size([1, 1000, 777]) | 3\n",
-      "7823     | torch.Size([1, 1000, 780]) | 3\n",
-      "7824     | torch.Size([1, 1000, 754]) | 3\n",
-      "7825     | torch.Size([1, 1000, 784]) | 3\n",
-      "7826     | torch.Size([1, 1000, 754]) | 3\n",
-      "7827     | torch.Size([1, 1000, 754]) | 3\n",
-      "7828     | torch.Size([1, 1000, 754]) | 3\n",
-      "7829     | torch.Size([1, 1000, 718]) | 3\n",
-      "7830     | torch.Size([1, 1000, 781]) | 3\n",
-      "7831     | torch.Size([1, 1000, 868]) | 3\n",
-      "7832     | torch.Size([1, 1000, 798]) | 3\n",
-      "7833     | torch.Size([1, 1000, 754]) | 3\n",
-      "7834     | torch.Size([1, 1000, 845]) | 3\n",
-      "7835     | torch.Size([1, 1000, 754]) | 3\n",
-      "7836     | torch.Size([1, 1000, 779]) | 3\n",
-      "7837     | torch.Size([1, 1000, 760]) | 3\n",
-      "7838     | torch.Size([1, 1000, 778]) | 3\n",
-      "7839     | torch.Size([1, 1000, 764]) | 3\n",
-      "7840     | torch.Size([1, 1000, 762]) | 3\n",
-      "7841     | torch.Size([1, 1000, 762]) | 3\n",
-      "7842     | torch.Size([1, 1000, 842]) | 3\n",
-      "7843     | torch.Size([1, 1000, 762]) | 3\n",
-      "7844     | torch.Size([1, 1000, 754]) | 3\n",
-      "7845     | torch.Size([1, 1000, 777]) | 3\n",
-      "7846     | torch.Size([1, 1000, 795]) | 3\n",
-      "7847     | torch.Size([1, 1000, 762]) | 3\n",
-      "7848     | torch.Size([1, 1000, 754]) | 3\n",
-      "7849     | torch.Size([1, 1000, 762]) | 3\n",
-      "7850     | torch.Size([1, 1000, 777]) | 3\n",
-      "7851     | torch.Size([1, 1000, 772]) | 3\n",
-      "7852     | torch.Size([1, 1000, 762]) | 3\n",
-      "7853     | torch.Size([1, 1000, 754]) | 3\n",
-      "7854     | torch.Size([1, 1000, 780]) | 3\n",
-      "7855     | torch.Size([1, 1000, 762]) | 3\n",
-      "7856     | torch.Size([1, 1000, 762]) | 3\n",
-      "7857     | torch.Size([1, 1000, 754]) | 3\n",
-      "7858     | torch.Size([1, 1000, 792]) | 3\n",
-      "7859     | torch.Size([1, 1000, 812]) | 3\n",
-      "7860     | torch.Size([1, 1000, 810]) | 3\n",
-      "7861     | torch.Size([1, 1000, 777]) | 3\n",
-      "7862     | torch.Size([1, 1000, 754]) | 3\n",
-      "7863     | torch.Size([1, 1000, 790]) | 3\n",
-      "7864     | torch.Size([1, 1000, 802]) | 3\n",
-      "7865     | torch.Size([1, 1000, 762]) | 3\n",
-      "7866     | torch.Size([1, 1000, 762]) | 3\n",
-      "7867     | torch.Size([1, 1000, 794]) | 3\n",
-      "7868     | torch.Size([1, 1000, 754]) | 3\n",
-      "7869     | torch.Size([1, 1000, 754]) | 3\n",
-      "7870     | torch.Size([1, 1000, 777]) | 3\n",
-      "7871     | torch.Size([1, 1000, 777]) | 3\n",
-      "7872     | torch.Size([1, 1000, 754]) | 3\n",
-      "7873     | torch.Size([1, 1000, 828]) | 3\n",
-      "7874     | torch.Size([1, 1000, 754]) | 3\n",
-      "7875     | torch.Size([1, 1000, 779]) | 3\n",
-      "7876     | torch.Size([1, 1000, 762]) | 3\n",
-      "7877     | torch.Size([1, 1000, 754]) | 3\n",
-      "7878     | torch.Size([1, 1000, 793]) | 3\n",
-      "7879     | torch.Size([1, 1000, 776]) | 3\n",
-      "7880     | torch.Size([1, 1000, 777]) | 3\n",
-      "7881     | torch.Size([1, 1000, 786]) | 3\n",
-      "7882     | torch.Size([1, 1000, 754]) | 3\n",
-      "7883     | torch.Size([1, 1000, 754]) | 3\n",
-      "7884     | torch.Size([1, 1000, 770]) | 3\n",
-      "7885     | torch.Size([1, 1000, 762]) | 3\n",
-      "7886     | torch.Size([1, 1000, 754]) | 3\n",
-      "7887     | torch.Size([1, 1000, 754]) | 3\n",
-      "7888     | torch.Size([1, 1000, 793]) | 3\n",
-      "7889     | torch.Size([1, 1000, 769]) | 3\n",
-      "7890     | torch.Size([1, 1000, 777]) | 3\n",
-      "7891     | torch.Size([1, 1000, 754]) | 3\n",
-      "7892     | torch.Size([1, 1000, 779]) | 3\n",
-      "7893     | torch.Size([1, 1000, 754]) | 3\n",
-      "7894     | torch.Size([1, 1000, 777]) | 3\n",
-      "7895     | torch.Size([1, 1000, 713]) | 3\n",
-      "7896     | torch.Size([1, 1000, 723]) | 3\n",
-      "7897     | torch.Size([1, 1000, 754]) | 3\n",
-      "7898     | torch.Size([1, 1000, 777]) | 3\n",
-      "7899     | torch.Size([1, 1000, 754]) | 3\n",
-      "7900     | torch.Size([1, 1000, 785]) | 3\n",
-      "7901     | torch.Size([1, 1000, 772]) | 3\n",
-      "7902     | torch.Size([1, 1000, 798]) | 3\n",
-      "7903     | torch.Size([1, 1000, 754]) | 3\n",
-      "7904     | torch.Size([1, 1000, 778]) | 3\n",
-      "7905     | torch.Size([1, 1000, 754]) | 3\n",
-      "7906     | torch.Size([1, 1000, 777]) | 3\n",
-      "7907     | torch.Size([1, 1000, 723]) | 3\n",
-      "7908     | torch.Size([1, 1000, 842]) | 3\n",
-      "7909     | torch.Size([1, 1000, 754]) | 3\n",
-      "7910     | torch.Size([1, 1000, 754]) | 3\n",
-      "7911     | torch.Size([1, 1000, 762]) | 3\n",
-      "7912     | torch.Size([1, 1000, 762]) | 3\n",
-      "7913     | torch.Size([1, 1000, 754]) | 3\n",
-      "7914     | torch.Size([1, 1000, 777]) | 3\n",
-      "7915     | torch.Size([1, 1000, 777]) | 3\n",
-      "7916     | torch.Size([1, 1000, 771]) | 3\n",
-      "7917     | torch.Size([1, 1000, 844]) | 3\n",
-      "7918     | torch.Size([1, 1000, 849]) | 3\n",
-      "7919     | torch.Size([1, 1000, 754]) | 3\n",
-      "7920     | torch.Size([1, 1000, 793]) | 3\n",
-      "7921     | torch.Size([1, 1000, 762]) | 3\n",
-      "7922     | torch.Size([1, 1000, 762]) | 3\n",
-      "7923     | torch.Size([1, 1000, 754]) | 3\n",
-      "7924     | torch.Size([1, 1000, 767]) | 3\n",
-      "7925     | torch.Size([1, 1000, 784]) | 3\n",
-      "7926     | torch.Size([1, 1000, 812]) | 3\n",
-      "7927     | torch.Size([1, 1000, 762]) | 3\n",
-      "7928     | torch.Size([1, 1000, 762]) | 3\n",
-      "7929     | torch.Size([1, 1000, 777]) | 3\n",
-      "7930     | torch.Size([1, 1000, 754]) | 3\n",
-      "7931     | torch.Size([1, 1000, 754]) | 3\n",
-      "7932     | torch.Size([1, 1000, 762]) | 3\n",
-      "7933     | torch.Size([1, 1000, 825]) | 3\n",
-      "7934     | torch.Size([1, 1000, 754]) | 3\n",
-      "7935     | torch.Size([1, 1000, 783]) | 3\n",
-      "7936     | torch.Size([1, 1000, 772]) | 3\n",
-      "7937     | torch.Size([1, 1000, 754]) | 3\n",
-      "7938     | torch.Size([1, 1000, 775]) | 3\n",
-      "7939     | torch.Size([1, 1000, 793]) | 3\n",
-      "7940     | torch.Size([1, 1000, 754]) | 3\n",
-      "7941     | torch.Size([1, 1000, 754]) | 3\n",
-      "7942     | torch.Size([1, 1000, 754]) | 3\n",
-      "7943     | torch.Size([1, 1000, 757]) | 3\n",
-      "7944     | torch.Size([1, 1000, 751]) | 3\n",
-      "7945     | torch.Size([1, 1000, 754]) | 3\n",
-      "7946     | torch.Size([1, 1000, 784]) | 3\n",
-      "7947     | torch.Size([1, 1000, 754]) | 3\n",
-      "7948     | torch.Size([1, 1000, 777]) | 3\n",
-      "7949     | torch.Size([1, 1000, 762]) | 3\n",
-      "7950     | torch.Size([1, 1000, 777]) | 3\n",
-      "7951     | torch.Size([1, 1000, 847]) | 3\n",
-      "7952     | torch.Size([1, 1000, 754]) | 3\n",
-      "7953     | torch.Size([1, 1000, 762]) | 3\n",
-      "7954     | torch.Size([1, 1000, 754]) | 3\n",
-      "7955     | torch.Size([1, 1000, 777]) | 3\n",
-      "7956     | torch.Size([1, 1000, 777]) | 3\n",
-      "7957     | torch.Size([1, 1000, 843]) | 3\n",
-      "7958     | torch.Size([1, 1000, 754]) | 3\n",
-      "7959     | torch.Size([1, 1000, 776]) | 3\n",
-      "7960     | torch.Size([1, 1000, 762]) | 3\n",
-      "7961     | torch.Size([1, 1000, 754]) | 3\n",
-      "7962     | torch.Size([1, 1000, 754]) | 3\n",
-      "7963     | torch.Size([1, 1000, 754]) | 3\n",
-      "7964     | torch.Size([1, 1000, 754]) | 3\n",
-      "7965     | torch.Size([1, 1000, 762]) | 3\n",
-      "7966     | torch.Size([1, 1000, 762]) | 3\n",
-      "7967     | torch.Size([1, 1000, 778]) | 3\n",
-      "7968     | torch.Size([1, 1000, 762]) | 3\n",
-      "7969     | torch.Size([1, 1000, 771]) | 3\n",
-      "7970     | torch.Size([1, 1000, 793]) | 3\n",
-      "7971     | torch.Size([1, 1000, 754]) | 3\n",
-      "7972     | torch.Size([1, 1000, 754]) | 3\n",
-      "7973     | torch.Size([1, 1000, 762]) | 3\n",
-      "7974     | torch.Size([1, 1000, 688]) | 3\n",
-      "7975     | torch.Size([1, 1000, 762]) | 3\n",
-      "7976     | torch.Size([1, 1000, 762]) | 3\n",
-      "7977     | torch.Size([1, 1000, 762]) | 3\n",
-      "7978     | torch.Size([1, 1000, 727]) | 3\n",
-      "7979     | torch.Size([1, 1000, 777]) | 3\n",
-      "7980     | torch.Size([1, 1000, 754]) | 3\n",
-      "7981     | torch.Size([1, 1000, 866]) | 3\n",
-      "7982     | torch.Size([1, 1000, 774]) | 3\n",
-      "7983     | torch.Size([1, 1000, 777]) | 3\n",
-      "7984     | torch.Size([1, 1000, 777]) | 3\n",
-      "7985     | torch.Size([1, 1000, 780]) | 3\n",
-      "7986     | torch.Size([1, 1000, 754]) | 3\n",
-      "7987     | torch.Size([1, 1000, 777]) | 3\n",
-      "7988     | torch.Size([1, 1000, 799]) | 3\n",
-      "7989     | torch.Size([1, 1000, 785]) | 3\n",
-      "7990     | torch.Size([1, 1000, 777]) | 3\n",
-      "7991     | torch.Size([1, 1000, 762]) | 3\n",
-      "7992     | torch.Size([1, 1000, 754]) | 3\n",
-      "7993     | torch.Size([1, 1000, 777]) | 3\n",
-      "7994     | torch.Size([1, 1000, 797]) | 3\n",
-      "7995     | torch.Size([1, 1000, 724]) | 3\n",
-      "7996     | torch.Size([1, 1000, 754]) | 3\n",
-      "7997     | torch.Size([1, 1000, 776]) | 3\n",
-      "7998     | torch.Size([1, 1000, 814]) | 3\n",
-      "7999     | torch.Size([1, 1000, 754]) | 3\n",
-      "8000     | torch.Size([1, 1000, 754]) | 3\n",
-      "8001     | torch.Size([1, 1000, 777]) | 3\n",
-      "8002     | torch.Size([1, 1000, 754]) | 3\n",
-      "8003     | torch.Size([1, 1000, 754]) | 3\n",
-      "8004     | torch.Size([1, 1000, 791]) | 3\n",
-      "8005     | torch.Size([1, 1000, 762]) | 3\n",
-      "8006     | torch.Size([1, 1000, 777]) | 3\n",
-      "8007     | torch.Size([1, 1000, 777]) | 3\n",
-      "8008     | torch.Size([1, 1000, 773]) | 3\n",
-      "8009     | torch.Size([1, 1000, 772]) | 3\n",
-      "8010     | torch.Size([1, 1000, 754]) | 3\n",
-      "8011     | torch.Size([1, 1000, 754]) | 3\n",
-      "8012     | torch.Size([1, 1000, 754]) | 3\n",
-      "8013     | torch.Size([1, 1000, 782]) | 3\n",
-      "8014     | torch.Size([1, 1000, 762]) | 3\n",
-      "8015     | torch.Size([1, 1000, 754]) | 3\n",
-      "8016     | torch.Size([1, 1000, 754]) | 3\n",
-      "8017     | torch.Size([1, 1000, 754]) | 3\n",
-      "8018     | torch.Size([1, 1000, 754]) | 3\n",
-      "8019     | torch.Size([1, 1000, 847]) | 3\n",
-      "8020     | torch.Size([1, 1000, 754]) | 3\n",
-      "8021     | torch.Size([1, 1000, 777]) | 3\n",
-      "8022     | torch.Size([1, 1000, 786]) | 3\n",
-      "8023     | torch.Size([1, 1000, 817]) | 3\n",
-      "8024     | torch.Size([1, 1000, 754]) | 3\n",
-      "8025     | torch.Size([1, 1000, 754]) | 3\n",
-      "8026     | torch.Size([1, 1000, 772]) | 3\n",
-      "8027     | torch.Size([1, 1000, 778]) | 3\n",
-      "8028     | torch.Size([1, 1000, 777]) | 3\n",
-      "8029     | torch.Size([1, 1000, 754]) | 3\n",
-      "8030     | torch.Size([1, 1000, 762]) | 3\n",
-      "8031     | torch.Size([1, 1000, 827]) | 3\n",
-      "8032     | torch.Size([1, 1000, 782]) | 3\n",
-      "8033     | torch.Size([1, 1000, 611]) | 3\n",
-      "8034     | torch.Size([1, 1000, 754]) | 3\n",
-      "8035     | torch.Size([1, 1000, 762]) | 3\n",
-      "8036     | torch.Size([1, 1000, 754]) | 3\n",
-      "8037     | torch.Size([1, 1000, 754]) | 3\n",
-      "8038     | torch.Size([1, 1000, 777]) | 3\n",
-      "8039     | torch.Size([1, 1000, 782]) | 3\n",
-      "8040     | torch.Size([1, 1000, 754]) | 3\n",
-      "8041     | torch.Size([1, 1000, 802]) | 3\n",
-      "8042     | torch.Size([1, 1000, 785]) | 3\n",
-      "8043     | torch.Size([1, 1000, 754]) | 3\n",
-      "8044     | torch.Size([1, 1000, 775]) | 3\n",
-      "8045     | torch.Size([1, 1000, 754]) | 3\n",
-      "8046     | torch.Size([1, 1000, 762]) | 3\n",
-      "8047     | torch.Size([1, 1000, 847]) | 3\n",
-      "8048     | torch.Size([1, 1000, 754]) | 3\n",
-      "8049     | torch.Size([1, 1000, 754]) | 3\n",
-      "8050     | torch.Size([1, 1000, 703]) | 3\n",
-      "8051     | torch.Size([1, 1000, 775]) | 3\n",
-      "8052     | torch.Size([1, 1000, 754]) | 3\n",
-      "8053     | torch.Size([1, 1000, 754]) | 3\n",
-      "8054     | torch.Size([1, 1000, 781]) | 3\n",
-      "8055     | torch.Size([1, 1000, 845]) | 3\n",
-      "8056     | torch.Size([1, 1000, 777]) | 3\n",
-      "8057     | torch.Size([1, 1000, 754]) | 3\n",
-      "8058     | torch.Size([1, 1000, 777]) | 3\n",
-      "8059     | torch.Size([1, 1000, 754]) | 3\n",
-      "8060     | torch.Size([1, 1000, 797]) | 3\n",
-      "8061     | torch.Size([1, 1000, 754]) | 3\n",
-      "8062     | torch.Size([1, 1000, 762]) | 3\n",
-      "8063     | torch.Size([1, 1000, 754]) | 3\n",
-      "8064     | torch.Size([1, 1000, 754]) | 3\n",
-      "8065     | torch.Size([1, 1000, 831]) | 3\n",
-      "8066     | torch.Size([1, 1000, 754]) | 3\n",
-      "8067     | torch.Size([1, 1000, 776]) | 3\n",
-      "8068     | torch.Size([1, 1000, 777]) | 3\n",
-      "8069     | torch.Size([1, 1000, 754]) | 3\n",
-      "8070     | torch.Size([1, 1000, 703]) | 3\n",
-      "8071     | torch.Size([1, 1000, 772]) | 3\n",
-      "8072     | torch.Size([1, 1000, 762]) | 3\n",
-      "8073     | torch.Size([1, 1000, 778]) | 3\n",
-      "8074     | torch.Size([1, 1000, 795]) | 3\n",
-      "8075     | torch.Size([1, 1000, 777]) | 3\n",
-      "8076     | torch.Size([1, 1000, 754]) | 3\n",
-      "8077     | torch.Size([1, 1000, 781]) | 3\n",
-      "8078     | torch.Size([1, 1000, 773]) | 3\n",
-      "8079     | torch.Size([1, 1000, 762]) | 3\n",
-      "8080     | torch.Size([1, 1000, 762]) | 3\n",
-      "8081     | torch.Size([1, 1000, 731]) | 3\n",
-      "8082     | torch.Size([1, 1000, 754]) | 3\n",
-      "8083     | torch.Size([1, 1000, 772]) | 3\n",
-      "8084     | torch.Size([1, 1000, 781]) | 3\n",
-      "8085     | torch.Size([1, 1000, 754]) | 3\n",
-      "8086     | torch.Size([1, 1000, 754]) | 3\n",
-      "8087     | torch.Size([1, 1000, 762]) | 3\n",
-      "8088     | torch.Size([1, 1000, 754]) | 3\n",
-      "8089     | torch.Size([1, 1000, 762]) | 3\n",
-      "8090     | torch.Size([1, 1000, 762]) | 3\n",
-      "8091     | torch.Size([1, 1000, 779]) | 3\n",
-      "8092     | torch.Size([1, 1000, 754]) | 3\n",
-      "8093     | torch.Size([1, 1000, 754]) | 3\n",
-      "8094     | torch.Size([1, 1000, 768]) | 3\n",
-      "8095     | torch.Size([1, 1000, 757]) | 3\n",
-      "8096     | torch.Size([1, 1000, 754]) | 3\n",
-      "8097     | torch.Size([1, 1000, 777]) | 3\n",
-      "8098     | torch.Size([1, 1000, 762]) | 3\n",
-      "8099     | torch.Size([1, 1000, 754]) | 3\n",
-      "8100     | torch.Size([1, 1000, 754]) | 3\n",
-      "8101     | torch.Size([1, 1000, 762]) | 3\n",
-      "8102     | torch.Size([1, 1000, 830]) | 3\n",
-      "8103     | torch.Size([1, 1000, 793]) | 3\n",
-      "8104     | torch.Size([1, 1000, 722]) | 3\n",
-      "8105     | torch.Size([1, 1000, 785]) | 3\n",
-      "8106     | torch.Size([1, 1000, 754]) | 3\n",
-      "8107     | torch.Size([1, 1000, 766]) | 3\n",
-      "8108     | torch.Size([1, 1000, 762]) | 3\n",
-      "8109     | torch.Size([1, 1000, 762]) | 3\n",
-      "8110     | torch.Size([1, 1000, 777]) | 3\n",
-      "8111     | torch.Size([1, 1000, 777]) | 3\n",
-      "8112     | torch.Size([1, 1000, 754]) | 3\n",
-      "8113     | torch.Size([1, 1000, 754]) | 3\n",
-      "8114     | torch.Size([1, 1000, 792]) | 3\n",
-      "8115     | torch.Size([1, 1000, 777]) | 3\n",
-      "8116     | torch.Size([1, 1000, 762]) | 3\n",
-      "8117     | torch.Size([1, 1000, 754]) | 3\n",
-      "8118     | torch.Size([1, 1000, 762]) | 3\n",
-      "8119     | torch.Size([1, 1000, 754]) | 3\n",
-      "8120     | torch.Size([1, 1000, 754]) | 3\n",
-      "8121     | torch.Size([1, 1000, 778]) | 3\n",
-      "8122     | torch.Size([1, 1000, 777]) | 3\n",
-      "8123     | torch.Size([1, 1000, 754]) | 3\n",
-      "8124     | torch.Size([1, 1000, 762]) | 3\n",
-      "8125     | torch.Size([1, 1000, 847]) | 3\n",
-      "8126     | torch.Size([1, 1000, 754]) | 3\n",
-      "8127     | torch.Size([1, 1000, 777]) | 3\n",
-      "8128     | torch.Size([1, 1000, 777]) | 3\n",
-      "8129     | torch.Size([1, 1000, 787]) | 3\n",
-      "8130     | torch.Size([1, 1000, 762]) | 3\n",
-      "8131     | torch.Size([1, 1000, 754]) | 3\n",
-      "8132     | torch.Size([1, 1000, 777]) | 3\n",
-      "8133     | torch.Size([1, 1000, 762]) | 3\n",
-      "8134     | torch.Size([1, 1000, 754]) | 3\n",
-      "8135     | torch.Size([1, 1000, 777]) | 3\n",
-      "8136     | torch.Size([1, 1000, 777]) | 3\n",
-      "8137     | torch.Size([1, 1000, 754]) | 3\n",
-      "8138     | torch.Size([1, 1000, 754]) | 3\n",
-      "8139     | torch.Size([1, 1000, 713]) | 3\n",
-      "8140     | torch.Size([1, 1000, 777]) | 3\n",
-      "8141     | torch.Size([1, 1000, 777]) | 3\n",
-      "8142     | torch.Size([1, 1000, 762]) | 3\n",
-      "8143     | torch.Size([1, 1000, 730]) | 3\n",
-      "8144     | torch.Size([1, 1000, 778]) | 3\n",
-      "8145     | torch.Size([1, 1000, 754]) | 3\n",
-      "8146     | torch.Size([1, 1000, 803]) | 3\n",
-      "8147     | torch.Size([1, 1000, 762]) | 3\n",
-      "8148     | torch.Size([1, 1000, 762]) | 3\n",
-      "8149     | torch.Size([1, 1000, 754]) | 3\n",
-      "8150     | torch.Size([1, 1000, 772]) | 3\n",
-      "8151     | torch.Size([1, 1000, 754]) | 3\n",
-      "8152     | torch.Size([1, 1000, 777]) | 3\n",
-      "8153     | torch.Size([1, 1000, 783]) | 3\n",
-      "8154     | torch.Size([1, 1000, 713]) | 3\n",
-      "8155     | torch.Size([1, 1000, 754]) | 3\n",
-      "8156     | torch.Size([1, 1000, 762]) | 3\n",
-      "8157     | torch.Size([1, 1000, 777]) | 3\n",
-      "8158     | torch.Size([1, 1000, 766]) | 3\n",
-      "8159     | torch.Size([1, 1000, 754]) | 3\n",
-      "8160     | torch.Size([1, 1000, 791]) | 3\n",
-      "8161     | torch.Size([1, 1000, 778]) | 3\n",
-      "8162     | torch.Size([1, 1000, 840]) | 3\n",
-      "8163     | torch.Size([1, 1000, 754]) | 3\n",
-      "8164     | torch.Size([1, 1000, 767]) | 3\n",
-      "8165     | torch.Size([1, 1000, 848]) | 3\n",
-      "8166     | torch.Size([1, 1000, 714]) | 3\n",
-      "8167     | torch.Size([1, 1000, 770]) | 3\n",
-      "8168     | torch.Size([1, 1000, 762]) | 3\n",
-      "8169     | torch.Size([1, 1000, 811]) | 3\n",
-      "8170     | torch.Size([1, 1000, 754]) | 3\n",
-      "8171     | torch.Size([1, 1000, 779]) | 3\n",
-      "8172     | torch.Size([1, 1000, 799]) | 3\n",
-      "8173     | torch.Size([1, 1000, 754]) | 3\n",
-      "8174     | torch.Size([1, 1000, 762]) | 3\n",
-      "8175     | torch.Size([1, 1000, 762]) | 3\n",
-      "8176     | torch.Size([1, 1000, 754]) | 3\n",
-      "8177     | torch.Size([1, 1000, 851]) | 3\n",
-      "8178     | torch.Size([1, 1000, 754]) | 3\n",
-      "8179     | torch.Size([1, 1000, 872]) | 3\n",
-      "8180     | torch.Size([1, 1000, 754]) | 3\n",
-      "8181     | torch.Size([1, 1000, 762]) | 3\n",
-      "8182     | torch.Size([1, 1000, 762]) | 3\n",
-      "8183     | torch.Size([1, 1000, 762]) | 3\n",
-      "8184     | torch.Size([1, 1000, 778]) | 3\n",
-      "8185     | torch.Size([1, 1000, 786]) | 3\n",
-      "8186     | torch.Size([1, 1000, 777]) | 3\n",
-      "8187     | torch.Size([1, 1000, 754]) | 3\n",
-      "8188     | torch.Size([1, 1000, 754]) | 3\n",
-      "8189     | torch.Size([1, 1000, 787]) | 3\n",
-      "8190     | torch.Size([1, 1000, 754]) | 3\n",
-      "8191     | torch.Size([1, 1000, 762]) | 3\n",
-      "8192     | torch.Size([1, 1000, 791]) | 3\n",
-      "8193     | torch.Size([1, 1000, 754]) | 3\n",
-      "8194     | torch.Size([1, 1000, 772]) | 3\n",
-      "8195     | torch.Size([1, 1000, 771]) | 3\n",
-      "8196     | torch.Size([1, 1000, 754]) | 3\n",
-      "8197     | torch.Size([1, 1000, 754]) | 3\n",
-      "8198     | torch.Size([1, 1000, 777]) | 3\n",
-      "8199     | torch.Size([1, 1000, 777]) | 3\n",
-      "8200     | torch.Size([1, 1000, 754]) | 3\n",
-      "8201     | torch.Size([1, 1000, 777]) | 3\n",
-      "8202     | torch.Size([1, 1000, 762]) | 3\n",
-      "8203     | torch.Size([1, 1000, 777]) | 3\n",
-      "8204     | torch.Size([1, 1000, 723]) | 3\n",
-      "8205     | torch.Size([1, 1000, 785]) | 3\n",
-      "8206     | torch.Size([1, 1000, 777]) | 3\n",
-      "8207     | torch.Size([1, 1000, 794]) | 3\n",
-      "8208     | torch.Size([1, 1000, 758]) | 3\n",
-      "8209     | torch.Size([1, 1000, 801]) | 3\n",
-      "8210     | torch.Size([1, 1000, 754]) | 3\n",
-      "8211     | torch.Size([1, 1000, 762]) | 3\n",
-      "8212     | torch.Size([1, 1000, 809]) | 3\n",
-      "8213     | torch.Size([1, 1000, 786]) | 3\n",
-      "8214     | torch.Size([1, 1000, 777]) | 3\n",
-      "8215     | torch.Size([1, 1000, 777]) | 3\n",
-      "8216     | torch.Size([1, 1000, 777]) | 3\n",
-      "8217     | torch.Size([1, 1000, 777]) | 3\n",
-      "8218     | torch.Size([1, 1000, 774]) | 3\n",
-      "8219     | torch.Size([1, 1000, 772]) | 3\n",
-      "8220     | torch.Size([1, 1000, 799]) | 3\n",
-      "8221     | torch.Size([1, 1000, 838]) | 3\n",
-      "8222     | torch.Size([1, 1000, 777]) | 3\n",
-      "8223     | torch.Size([1, 1000, 777]) | 3\n",
-      "8224     | torch.Size([1, 1000, 754]) | 3\n",
-      "8225     | torch.Size([1, 1000, 777]) | 3\n",
-      "8226     | torch.Size([1, 1000, 762]) | 3\n",
-      "8227     | torch.Size([1, 1000, 754]) | 3\n",
-      "8228     | torch.Size([1, 1000, 754]) | 3\n",
-      "8229     | torch.Size([1, 1000, 777]) | 3\n",
-      "8230     | torch.Size([1, 1000, 777]) | 3\n",
-      "8231     | torch.Size([1, 1000, 778]) | 3\n",
-      "8232     | torch.Size([1, 1000, 782]) | 3\n",
-      "8233     | torch.Size([1, 1000, 762]) | 3\n",
-      "8234     | torch.Size([1, 1000, 754]) | 3\n",
-      "8235     | torch.Size([1, 1000, 754]) | 3\n",
-      "8236     | torch.Size([1, 1000, 802]) | 3\n",
-      "8237     | torch.Size([1, 1000, 777]) | 3\n",
-      "8238     | torch.Size([1, 1000, 754]) | 3\n",
-      "8239     | torch.Size([1, 1000, 771]) | 3\n",
-      "8240     | torch.Size([1, 1000, 754]) | 3\n",
-      "8241     | torch.Size([1, 1000, 754]) | 3\n",
-      "8242     | torch.Size([1, 1000, 780]) | 3\n",
-      "8243     | torch.Size([1, 1000, 777]) | 3\n",
-      "8244     | torch.Size([1, 1000, 754]) | 3\n",
-      "8245     | torch.Size([1, 1000, 754]) | 3\n",
-      "8246     | torch.Size([1, 1000, 773]) | 3\n",
-      "8247     | torch.Size([1, 1000, 754]) | 3\n",
-      "8248     | torch.Size([1, 1000, 789]) | 3\n",
-      "8249     | torch.Size([1, 1000, 777]) | 3\n",
-      "8250     | torch.Size([1, 1000, 797]) | 3\n",
-      "8251     | torch.Size([1, 1000, 754]) | 3\n",
-      "8252     | torch.Size([1, 1000, 754]) | 3\n",
-      "8253     | torch.Size([1, 1000, 762]) | 3\n",
-      "8254     | torch.Size([1, 1000, 762]) | 3\n",
-      "8255     | torch.Size([1, 1000, 754]) | 3\n",
-      "8256     | torch.Size([1, 1000, 762]) | 3\n",
-      "8257     | torch.Size([1, 1000, 777]) | 3\n",
-      "8258     | torch.Size([1, 1000, 754]) | 3\n",
-      "8259     | torch.Size([1, 1000, 779]) | 3\n",
-      "8260     | torch.Size([1, 1000, 754]) | 3\n",
-      "8261     | torch.Size([1, 1000, 754]) | 3\n",
-      "8262     | torch.Size([1, 1000, 772]) | 3\n",
-      "8263     | torch.Size([1, 1000, 754]) | 3\n",
-      "8264     | torch.Size([1, 1000, 847]) | 3\n",
-      "8265     | torch.Size([1, 1000, 754]) | 3\n",
-      "8266     | torch.Size([1, 1000, 786]) | 3\n",
-      "8267     | torch.Size([1, 1000, 754]) | 3\n",
-      "8268     | torch.Size([1, 1000, 762]) | 3\n",
-      "8269     | torch.Size([1, 1000, 754]) | 3\n",
-      "8270     | torch.Size([1, 1000, 762]) | 3\n",
-      "8271     | torch.Size([1, 1000, 754]) | 3\n",
-      "8272     | torch.Size([1, 1000, 754]) | 3\n",
-      "8273     | torch.Size([1, 1000, 762]) | 3\n",
-      "8274     | torch.Size([1, 1000, 754]) | 3\n",
-      "8275     | torch.Size([1, 1000, 778]) | 3\n",
-      "8276     | torch.Size([1, 1000, 783]) | 3\n",
-      "8277     | torch.Size([1, 1000, 754]) | 3\n",
-      "8278     | torch.Size([1, 1000, 762]) | 3\n",
-      "8279     | torch.Size([1, 1000, 762]) | 3\n",
-      "8280     | torch.Size([1, 1000, 870]) | 3\n",
-      "8281     | torch.Size([1, 1000, 762]) | 3\n",
-      "8282     | torch.Size([1, 1000, 754]) | 3\n",
-      "8283     | torch.Size([1, 1000, 754]) | 3\n",
-      "8284     | torch.Size([1, 1000, 777]) | 3\n",
-      "8285     | torch.Size([1, 1000, 777]) | 3\n",
-      "8286     | torch.Size([1, 1000, 777]) | 3\n",
-      "8287     | torch.Size([1, 1000, 762]) | 3\n",
-      "8288     | torch.Size([1, 1000, 754]) | 3\n",
-      "8289     | torch.Size([1, 1000, 754]) | 3\n",
-      "8290     | torch.Size([1, 1000, 813]) | 3\n",
-      "8291     | torch.Size([1, 1000, 793]) | 3\n",
-      "8292     | torch.Size([1, 1000, 848]) | 3\n",
-      "8293     | torch.Size([1, 1000, 757]) | 3\n",
-      "8294     | torch.Size([1, 1000, 754]) | 3\n",
-      "8295     | torch.Size([1, 1000, 794]) | 3\n",
-      "8296     | torch.Size([1, 1000, 867]) | 3\n",
-      "8297     | torch.Size([1, 1000, 754]) | 3\n",
-      "8298     | torch.Size([1, 1000, 762]) | 3\n",
-      "8299     | torch.Size([1, 1000, 777]) | 3\n",
-      "8300     | torch.Size([1, 1000, 784]) | 3\n",
-      "8301     | torch.Size([1, 1000, 777]) | 3\n",
-      "8302     | torch.Size([1, 1000, 762]) | 3\n",
-      "8303     | torch.Size([1, 1000, 777]) | 3\n",
-      "8304     | torch.Size([1, 1000, 779]) | 3\n",
-      "8305     | torch.Size([1, 1000, 754]) | 3\n",
-      "8306     | torch.Size([1, 1000, 754]) | 3\n",
-      "8307     | torch.Size([1, 1000, 786]) | 3\n",
-      "8308     | torch.Size([1, 1000, 725]) | 3\n",
-      "8309     | torch.Size([1, 1000, 762]) | 3\n",
-      "8310     | torch.Size([1, 1000, 780]) | 3\n",
-      "8311     | torch.Size([1, 1000, 754]) | 3\n",
-      "8312     | torch.Size([1, 1000, 841]) | 3\n",
-      "8313     | torch.Size([1, 1000, 754]) | 3\n",
-      "8314     | torch.Size([1, 1000, 798]) | 3\n",
-      "8315     | torch.Size([1, 1000, 762]) | 3\n",
-      "8316     | torch.Size([1, 1000, 775]) | 3\n",
-      "8317     | torch.Size([1, 1000, 770]) | 3\n",
-      "8318     | torch.Size([1, 1000, 777]) | 3\n",
-      "8319     | torch.Size([1, 1000, 790]) | 3\n",
-      "8320     | torch.Size([1, 1000, 754]) | 3\n",
-      "8321     | torch.Size([1, 1000, 777]) | 3\n",
-      "8322     | torch.Size([1, 1000, 780]) | 3\n",
-      "8323     | torch.Size([1, 1000, 762]) | 3\n",
-      "8324     | torch.Size([1, 1000, 777]) | 3\n",
-      "8325     | torch.Size([1, 1000, 787]) | 3\n",
-      "8326     | torch.Size([1, 1000, 802]) | 3\n",
-      "8327     | torch.Size([1, 1000, 767]) | 3\n",
-      "8328     | torch.Size([1, 1000, 781]) | 3\n",
-      "8329     | torch.Size([1, 1000, 754]) | 3\n",
-      "8330     | torch.Size([1, 1000, 754]) | 3\n",
-      "8331     | torch.Size([1, 1000, 777]) | 3\n",
-      "8332     | torch.Size([1, 1000, 713]) | 3\n",
-      "8333     | torch.Size([1, 1000, 777]) | 3\n",
-      "8334     | torch.Size([1, 1000, 777]) | 3\n",
-      "8335     | torch.Size([1, 1000, 754]) | 3\n",
-      "8336     | torch.Size([1, 1000, 754]) | 3\n",
-      "8337     | torch.Size([1, 1000, 777]) | 3\n",
-      "8338     | torch.Size([1, 1000, 795]) | 3\n",
-      "8339     | torch.Size([1, 1000, 762]) | 3\n",
-      "8340     | torch.Size([1, 1000, 802]) | 3\n",
-      "8341     | torch.Size([1, 1000, 754]) | 3\n",
-      "8342     | torch.Size([1, 1000, 846]) | 3\n",
-      "8343     | torch.Size([1, 1000, 772]) | 3\n",
-      "8344     | torch.Size([1, 1000, 762]) | 3\n",
-      "8345     | torch.Size([1, 1000, 762]) | 3\n",
-      "8346     | torch.Size([1, 1000, 762]) | 3\n",
-      "8347     | torch.Size([1, 1000, 802]) | 3\n",
-      "8348     | torch.Size([1, 1000, 782]) | 3\n",
-      "8349     | torch.Size([1, 1000, 754]) | 3\n",
-      "8350     | torch.Size([1, 1000, 864]) | 3\n",
-      "8351     | torch.Size([1, 1000, 762]) | 3\n",
-      "8352     | torch.Size([1, 1000, 773]) | 3\n",
-      "8353     | torch.Size([1, 1000, 786]) | 3\n",
-      "8354     | torch.Size([1, 1000, 792]) | 3\n",
-      "8355     | torch.Size([1, 1000, 777]) | 3\n",
-      "8356     | torch.Size([1, 1000, 762]) | 3\n",
-      "8357     | torch.Size([1, 1000, 762]) | 3\n",
-      "8358     | torch.Size([1, 1000, 796]) | 3\n",
-      "8359     | torch.Size([1, 1000, 754]) | 3\n",
-      "8360     | torch.Size([1, 1000, 777]) | 3\n",
-      "8361     | torch.Size([1, 1000, 754]) | 3\n",
-      "8362     | torch.Size([1, 1000, 762]) | 3\n",
-      "8363     | torch.Size([1, 1000, 777]) | 3\n",
-      "8364     | torch.Size([1, 1000, 754]) | 3\n",
-      "8365     | torch.Size([1, 1000, 776]) | 3\n",
-      "8366     | torch.Size([1, 1000, 754]) | 3\n",
-      "8367     | torch.Size([1, 1000, 779]) | 3\n",
-      "8368     | torch.Size([1, 1000, 754]) | 3\n",
-      "8369     | torch.Size([1, 1000, 773]) | 3\n",
-      "8370     | torch.Size([1, 1000, 754]) | 3\n",
-      "8371     | torch.Size([1, 1000, 777]) | 3\n",
-      "8372     | torch.Size([1, 1000, 754]) | 3\n",
-      "8373     | torch.Size([1, 1000, 779]) | 3\n",
-      "8374     | torch.Size([1, 1000, 773]) | 3\n",
-      "8375     | torch.Size([1, 1000, 754]) | 3\n",
-      "8376     | torch.Size([1, 1000, 757]) | 3\n",
-      "8377     | torch.Size([1, 1000, 754]) | 3\n",
-      "8378     | torch.Size([1, 1000, 787]) | 3\n",
-      "8379     | torch.Size([1, 1000, 762]) | 3\n",
-      "8380     | torch.Size([1, 1000, 775]) | 3\n",
-      "8381     | torch.Size([1, 1000, 762]) | 3\n",
-      "8382     | torch.Size([1, 1000, 762]) | 3\n",
-      "8383     | torch.Size([1, 1000, 777]) | 3\n",
-      "8384     | torch.Size([1, 1000, 782]) | 3\n",
-      "8385     | torch.Size([1, 1000, 754]) | 3\n",
-      "8386     | torch.Size([1, 1000, 769]) | 3\n",
-      "8387     | torch.Size([1, 1000, 772]) | 3\n",
-      "8388     | torch.Size([1, 1000, 754]) | 3\n",
-      "8389     | torch.Size([1, 1000, 754]) | 3\n",
-      "8390     | torch.Size([1, 1000, 754]) | 3\n",
-      "8391     | torch.Size([1, 1000, 762]) | 3\n",
-      "8392     | torch.Size([1, 1000, 762]) | 3\n",
-      "8393     | torch.Size([1, 1000, 792]) | 3\n",
-      "8394     | torch.Size([1, 1000, 785]) | 3\n",
-      "8395     | torch.Size([1, 1000, 754]) | 3\n",
-      "8396     | torch.Size([1, 1000, 777]) | 3\n",
-      "8397     | torch.Size([1, 1000, 754]) | 3\n",
-      "8398     | torch.Size([1, 1000, 762]) | 3\n",
-      "8399     | torch.Size([1, 1000, 754]) | 3\n",
-      "8400     | torch.Size([1, 1000, 754]) | 3\n",
-      "8401     | torch.Size([1, 1000, 781]) | 3\n",
-      "8402     | torch.Size([1, 1000, 777]) | 3\n",
-      "8403     | torch.Size([1, 1000, 754]) | 3\n",
-      "8404     | torch.Size([1, 1000, 754]) | 3\n",
-      "8405     | torch.Size([1, 1000, 754]) | 3\n",
-      "8406     | torch.Size([1, 1000, 754]) | 3\n",
-      "8407     | torch.Size([1, 1000, 777]) | 3\n",
-      "8408     | torch.Size([1, 1000, 788]) | 3\n",
-      "8409     | torch.Size([1, 1000, 754]) | 3\n",
-      "8410     | torch.Size([1, 1000, 781]) | 3\n",
-      "8411     | torch.Size([1, 1000, 762]) | 3\n",
-      "8412     | torch.Size([1, 1000, 762]) | 3\n",
-      "8413     | torch.Size([1, 1000, 773]) | 3\n",
-      "8414     | torch.Size([1, 1000, 754]) | 3\n",
-      "8415     | torch.Size([1, 1000, 780]) | 3\n",
-      "8416     | torch.Size([1, 1000, 754]) | 3\n",
-      "8417     | torch.Size([1, 1000, 777]) | 3\n",
-      "8418     | torch.Size([1, 1000, 808]) | 3\n",
-      "8419     | torch.Size([1, 1000, 754]) | 3\n",
-      "8420     | torch.Size([1, 1000, 754]) | 3\n",
-      "8421     | torch.Size([1, 1000, 797]) | 3\n",
-      "8422     | torch.Size([1, 1000, 837]) | 3\n",
-      "8423     | torch.Size([1, 1000, 754]) | 3\n",
-      "8424     | torch.Size([1, 1000, 809]) | 3\n",
-      "8425     | torch.Size([1, 1000, 777]) | 3\n",
-      "8426     | torch.Size([1, 1000, 754]) | 3\n",
-      "8427     | torch.Size([1, 1000, 762]) | 3\n",
-      "8428     | torch.Size([1, 1000, 754]) | 3\n",
-      "8429     | torch.Size([1, 1000, 754]) | 3\n",
-      "8430     | torch.Size([1, 1000, 703]) | 3\n",
-      "8431     | torch.Size([1, 1000, 784]) | 3\n",
-      "8432     | torch.Size([1, 1000, 762]) | 3\n",
-      "8433     | torch.Size([1, 1000, 771]) | 3\n",
-      "8434     | torch.Size([1, 1000, 762]) | 3\n",
-      "8435     | torch.Size([1, 1000, 782]) | 3\n",
-      "8436     | torch.Size([1, 1000, 754]) | 3\n",
-      "8437     | torch.Size([1, 1000, 754]) | 3\n",
-      "8438     | torch.Size([1, 1000, 754]) | 3\n",
-      "8439     | torch.Size([1, 1000, 754]) | 3\n",
-      "8440     | torch.Size([1, 1000, 762]) | 3\n",
-      "8441     | torch.Size([1, 1000, 762]) | 3\n",
-      "8442     | torch.Size([1, 1000, 703]) | 3\n",
-      "8443     | torch.Size([1, 1000, 754]) | 3\n",
-      "8444     | torch.Size([1, 1000, 726]) | 3\n",
-      "8445     | torch.Size([1, 1000, 713]) | 3\n",
-      "8446     | torch.Size([1, 1000, 754]) | 3\n",
-      "8447     | torch.Size([1, 1000, 754]) | 3\n",
-      "8448     | torch.Size([1, 1000, 754]) | 3\n",
-      "8449     | torch.Size([1, 1000, 777]) | 3\n",
-      "8450     | torch.Size([1, 1000, 777]) | 3\n",
-      "8451     | torch.Size([1, 1000, 787]) | 3\n",
-      "8452     | torch.Size([1, 1000, 772]) | 3\n",
-      "8453     | torch.Size([1, 1000, 777]) | 3\n",
-      "8454     | torch.Size([1, 1000, 777]) | 3\n",
-      "8455     | torch.Size([1, 1000, 777]) | 3\n",
-      "8456     | torch.Size([1, 1000, 766]) | 3\n",
-      "8457     | torch.Size([1, 1000, 777]) | 3\n",
-      "8458     | torch.Size([1, 1000, 778]) | 3\n",
-      "8459     | torch.Size([1, 1000, 754]) | 3\n",
-      "8460     | torch.Size([1, 1000, 754]) | 3\n",
-      "8461     | torch.Size([1, 1000, 777]) | 3\n",
-      "8462     | torch.Size([1, 1000, 775]) | 3\n",
-      "8463     | torch.Size([1, 1000, 754]) | 3\n",
-      "8464     | torch.Size([1, 1000, 762]) | 3\n",
-      "8465     | torch.Size([1, 1000, 754]) | 3\n",
-      "8466     | torch.Size([1, 1000, 802]) | 3\n",
-      "8467     | torch.Size([1, 1000, 762]) | 3\n",
-      "8468     | torch.Size([1, 1000, 777]) | 3\n",
-      "8469     | torch.Size([1, 1000, 754]) | 3\n",
-      "8470     | torch.Size([1, 1000, 799]) | 3\n",
-      "8471     | torch.Size([1, 1000, 812]) | 3\n",
-      "8472     | torch.Size([1, 1000, 754]) | 3\n",
-      "8473     | torch.Size([1, 1000, 762]) | 3\n",
-      "8474     | torch.Size([1, 1000, 775]) | 3\n",
-      "8475     | torch.Size([1, 1000, 777]) | 3\n",
-      "8476     | torch.Size([1, 1000, 754]) | 3\n",
-      "8477     | torch.Size([1, 1000, 754]) | 3\n",
-      "8478     | torch.Size([1, 1000, 777]) | 3\n",
-      "8479     | torch.Size([1, 1000, 777]) | 3\n",
-      "8480     | torch.Size([1, 1000, 762]) | 3\n",
-      "8481     | torch.Size([1, 1000, 781]) | 3\n",
-      "8482     | torch.Size([1, 1000, 768]) | 3\n",
-      "8483     | torch.Size([1, 1000, 754]) | 3\n",
-      "8484     | torch.Size([1, 1000, 806]) | 3\n",
-      "8485     | torch.Size([1, 1000, 793]) | 3\n",
-      "8486     | torch.Size([1, 1000, 754]) | 3\n",
-      "8487     | torch.Size([1, 1000, 767]) | 3\n",
-      "8488     | torch.Size([1, 1000, 777]) | 3\n",
-      "8489     | torch.Size([1, 1000, 762]) | 3\n",
-      "8490     | torch.Size([1, 1000, 754]) | 3\n",
-      "8491     | torch.Size([1, 1000, 754]) | 3\n",
-      "8492     | torch.Size([1, 1000, 754]) | 3\n",
-      "8493     | torch.Size([1, 1000, 762]) | 3\n",
-      "8494     | torch.Size([1, 1000, 762]) | 3\n",
-      "8495     | torch.Size([1, 1000, 802]) | 3\n",
-      "8496     | torch.Size([1, 1000, 754]) | 3\n",
-      "8497     | torch.Size([1, 1000, 781]) | 3\n",
-      "8498     | torch.Size([1, 1000, 754]) | 3\n",
-      "8499     | torch.Size([1, 1000, 812]) | 3\n",
-      "8500     | torch.Size([1, 1000, 762]) | 3\n",
-      "8501     | torch.Size([1, 1000, 754]) | 3\n",
-      "8502     | torch.Size([1, 1000, 778]) | 3\n",
-      "8503     | torch.Size([1, 1000, 777]) | 3\n",
-      "8504     | torch.Size([1, 1000, 754]) | 3\n",
-      "8505     | torch.Size([1, 1000, 754]) | 3\n",
-      "8506     | torch.Size([1, 1000, 754]) | 3\n",
-      "8507     | torch.Size([1, 1000, 777]) | 3\n",
-      "8508     | torch.Size([1, 1000, 777]) | 3\n",
-      "8509     | torch.Size([1, 1000, 754]) | 3\n",
-      "8510     | torch.Size([1, 1000, 777]) | 3\n",
-      "8511     | torch.Size([1, 1000, 762]) | 3\n",
-      "8512     | torch.Size([1, 1000, 754]) | 3\n",
-      "8513     | torch.Size([1, 1000, 784]) | 3\n",
-      "8514     | torch.Size([1, 1000, 788]) | 3\n",
-      "8515     | torch.Size([1, 1000, 784]) | 3\n",
-      "8516     | torch.Size([1, 1000, 777]) | 3\n",
-      "8517     | torch.Size([1, 1000, 777]) | 3\n",
-      "8518     | torch.Size([1, 1000, 754]) | 3\n",
-      "8519     | torch.Size([1, 1000, 786]) | 3\n",
-      "8520     | torch.Size([1, 1000, 754]) | 3\n",
-      "8521     | torch.Size([1, 1000, 762]) | 3\n",
-      "8522     | torch.Size([1, 1000, 754]) | 3\n",
-      "8523     | torch.Size([1, 1000, 754]) | 3\n",
-      "8524     | torch.Size([1, 1000, 802]) | 3\n",
-      "8525     | torch.Size([1, 1000, 780]) | 3\n",
-      "8526     | torch.Size([1, 1000, 754]) | 3\n",
-      "8527     | torch.Size([1, 1000, 754]) | 3\n",
-      "8528     | torch.Size([1, 1000, 754]) | 3\n",
-      "8529     | torch.Size([1, 1000, 777]) | 3\n",
-      "8530     | torch.Size([1, 1000, 772]) | 3\n",
-      "8531     | torch.Size([1, 1000, 784]) | 3\n",
-      "8532     | torch.Size([1, 1000, 872]) | 3\n",
-      "8533     | torch.Size([1, 1000, 779]) | 3\n",
-      "8534     | torch.Size([1, 1000, 754]) | 3\n",
-      "8535     | torch.Size([1, 1000, 771]) | 3\n",
-      "8536     | torch.Size([1, 1000, 723]) | 3\n",
-      "8537     | torch.Size([1, 1000, 754]) | 3\n",
-      "8538     | torch.Size([1, 1000, 754]) | 3\n",
-      "8539     | torch.Size([1, 1000, 754]) | 3\n",
-      "8540     | torch.Size([1, 1000, 772]) | 3\n",
-      "8541     | torch.Size([1, 1000, 754]) | 3\n",
-      "8542     | torch.Size([1, 1000, 754]) | 3\n",
-      "8543     | torch.Size([1, 1000, 767]) | 3\n",
-      "8544     | torch.Size([1, 1000, 762]) | 3\n",
-      "8545     | torch.Size([1, 1000, 777]) | 3\n",
-      "8546     | torch.Size([1, 1000, 772]) | 3\n",
-      "8547     | torch.Size([1, 1000, 754]) | 3\n",
-      "8548     | torch.Size([1, 1000, 797]) | 3\n",
-      "8549     | torch.Size([1, 1000, 772]) | 3\n",
-      "8550     | torch.Size([1, 1000, 777]) | 3\n",
-      "8551     | torch.Size([1, 1000, 777]) | 3\n",
-      "8552     | torch.Size([1, 1000, 754]) | 3\n",
-      "8553     | torch.Size([1, 1000, 777]) | 3\n",
-      "8554     | torch.Size([1, 1000, 781]) | 3\n",
-      "8555     | torch.Size([1, 1000, 793]) | 3\n",
-      "8556     | torch.Size([1, 1000, 754]) | 3\n",
-      "8557     | torch.Size([1, 1000, 754]) | 3\n",
-      "8558     | torch.Size([1, 1000, 754]) | 3\n",
-      "8559     | torch.Size([1, 1000, 754]) | 3\n",
-      "8560     | torch.Size([1, 1000, 765]) | 3\n",
-      "8561     | torch.Size([1, 1000, 754]) | 3\n",
-      "8562     | torch.Size([1, 1000, 754]) | 3\n",
-      "8563     | torch.Size([1, 1000, 754]) | 3\n",
-      "8564     | torch.Size([1, 1000, 777]) | 3\n",
-      "8565     | torch.Size([1, 1000, 762]) | 3\n",
-      "8566     | torch.Size([1, 1000, 776]) | 3\n",
-      "8567     | torch.Size([1, 1000, 754]) | 3\n",
-      "8568     | torch.Size([1, 1000, 754]) | 3\n",
-      "8569     | torch.Size([1, 1000, 754]) | 3\n",
-      "8570     | torch.Size([1, 1000, 762]) | 3\n",
-      "8571     | torch.Size([1, 1000, 798]) | 3\n",
-      "8572     | torch.Size([1, 1000, 777]) | 3\n",
-      "8573     | torch.Size([1, 1000, 754]) | 3\n",
-      "8574     | torch.Size([1, 1000, 786]) | 3\n",
-      "8575     | torch.Size([1, 1000, 777]) | 3\n",
-      "8576     | torch.Size([1, 1000, 754]) | 3\n",
-      "8577     | torch.Size([1, 1000, 846]) | 3\n",
-      "8578     | torch.Size([1, 1000, 779]) | 3\n",
-      "8579     | torch.Size([1, 1000, 754]) | 3\n",
-      "8580     | torch.Size([1, 1000, 762]) | 3\n",
-      "8581     | torch.Size([1, 1000, 777]) | 3\n",
-      "8582     | torch.Size([1, 1000, 802]) | 3\n",
-      "8583     | torch.Size([1, 1000, 754]) | 3\n",
-      "8584     | torch.Size([1, 1000, 769]) | 3\n",
-      "8585     | torch.Size([1, 1000, 762]) | 3\n",
-      "8586     | torch.Size([1, 1000, 754]) | 3\n",
-      "8587     | torch.Size([1, 1000, 785]) | 3\n",
-      "8588     | torch.Size([1, 1000, 768]) | 3\n",
-      "8589     | torch.Size([1, 1000, 762]) | 3\n",
-      "8590     | torch.Size([1, 1000, 762]) | 3\n",
-      "8591     | torch.Size([1, 1000, 762]) | 3\n",
-      "8592     | torch.Size([1, 1000, 777]) | 3\n",
-      "8593     | torch.Size([1, 1000, 762]) | 3\n",
-      "8594     | torch.Size([1, 1000, 762]) | 3\n",
-      "8595     | torch.Size([1, 1000, 754]) | 3\n",
-      "8596     | torch.Size([1, 1000, 777]) | 3\n",
-      "8597     | torch.Size([1, 1000, 762]) | 3\n",
-      "8598     | torch.Size([1, 1000, 754]) | 3\n",
-      "8599     | torch.Size([1, 1000, 754]) | 3\n",
-      "8600     | torch.Size([1, 1000, 765]) | 3\n",
-      "8601     | torch.Size([1, 1000, 762]) | 3\n",
-      "8602     | torch.Size([1, 1000, 754]) | 3\n",
-      "8603     | torch.Size([1, 1000, 754]) | 3\n",
-      "8604     | torch.Size([1, 1000, 777]) | 3\n",
-      "8605     | torch.Size([1, 1000, 762]) | 3\n",
-      "8606     | torch.Size([1, 1000, 787]) | 3\n",
-      "8607     | torch.Size([1, 1000, 612]) | 3\n",
-      "8608     | torch.Size([1, 1000, 777]) | 3\n",
-      "8609     | torch.Size([1, 1000, 754]) | 3\n",
-      "8610     | torch.Size([1, 1000, 754]) | 3\n",
-      "8611     | torch.Size([1, 1000, 754]) | 3\n",
-      "8612     | torch.Size([1, 1000, 777]) | 3\n",
-      "8613     | torch.Size([1, 1000, 777]) | 3\n",
-      "8614     | torch.Size([1, 1000, 754]) | 3\n",
-      "8615     | torch.Size([1, 1000, 777]) | 3\n",
-      "8616     | torch.Size([1, 1000, 777]) | 3\n",
-      "8617     | torch.Size([1, 1000, 807]) | 3\n",
-      "8618     | torch.Size([1, 1000, 754]) | 3\n",
-      "8619     | torch.Size([1, 1000, 787]) | 3\n",
-      "8620     | torch.Size([1, 1000, 762]) | 3\n",
-      "8621     | torch.Size([1, 1000, 754]) | 3\n",
-      "8622     | torch.Size([1, 1000, 754]) | 3\n",
-      "8623     | torch.Size([1, 1000, 762]) | 3\n",
-      "8624     | torch.Size([1, 1000, 728]) | 3\n",
-      "8625     | torch.Size([1, 1000, 754]) | 3\n",
-      "8626     | torch.Size([1, 1000, 762]) | 3\n",
-      "8627     | torch.Size([1, 1000, 754]) | 3\n",
-      "8628     | torch.Size([1, 1000, 777]) | 3\n",
-      "8629     | torch.Size([1, 1000, 754]) | 3\n",
-      "8630     | torch.Size([1, 1000, 777]) | 3\n",
-      "8631     | torch.Size([1, 1000, 811]) | 3\n",
-      "8632     | torch.Size([1, 1000, 762]) | 3\n",
-      "8633     | torch.Size([1, 1000, 780]) | 3\n",
-      "8634     | torch.Size([1, 1000, 754]) | 3\n",
-      "8635     | torch.Size([1, 1000, 778]) | 3\n",
-      "8636     | torch.Size([1, 1000, 754]) | 3\n",
-      "8637     | torch.Size([1, 1000, 777]) | 3\n",
-      "8638     | torch.Size([1, 1000, 754]) | 3\n",
-      "8639     | torch.Size([1, 1000, 754]) | 3\n",
-      "8640     | torch.Size([1, 1000, 779]) | 3\n",
-      "8641     | torch.Size([1, 1000, 754]) | 3\n",
-      "8642     | torch.Size([1, 1000, 754]) | 3\n",
-      "8643     | torch.Size([1, 1000, 777]) | 3\n",
-      "8644     | torch.Size([1, 1000, 777]) | 3\n",
-      "8645     | torch.Size([1, 1000, 754]) | 3\n",
-      "8646     | torch.Size([1, 1000, 754]) | 3\n",
-      "8647     | torch.Size([1, 1000, 777]) | 3\n",
-      "8648     | torch.Size([1, 1000, 773]) | 3\n",
-      "8649     | torch.Size([1, 1000, 762]) | 3\n",
-      "8650     | torch.Size([1, 1000, 762]) | 3\n",
-      "8651     | torch.Size([1, 1000, 762]) | 3\n",
-      "8652     | torch.Size([1, 1000, 754]) | 3\n",
-      "8653     | torch.Size([1, 1000, 754]) | 3\n",
-      "8654     | torch.Size([1, 1000, 777]) | 3\n",
-      "8655     | torch.Size([1, 1000, 793]) | 3\n",
-      "8656     | torch.Size([1, 1000, 777]) | 3\n",
-      "8657     | torch.Size([1, 1000, 777]) | 3\n",
-      "8658     | torch.Size([1, 1000, 762]) | 3\n",
-      "8659     | torch.Size([1, 1000, 784]) | 3\n",
-      "8660     | torch.Size([1, 1000, 754]) | 3\n",
-      "8661     | torch.Size([1, 1000, 762]) | 3\n",
-      "8662     | torch.Size([1, 1000, 762]) | 3\n",
-      "8663     | torch.Size([1, 1000, 783]) | 3\n",
-      "8664     | torch.Size([1, 1000, 773]) | 3\n",
-      "8665     | torch.Size([1, 1000, 724]) | 3\n",
-      "8666     | torch.Size([1, 1000, 777]) | 3\n",
-      "8667     | torch.Size([1, 1000, 777]) | 3\n",
-      "8668     | torch.Size([1, 1000, 754]) | 3\n",
-      "8669     | torch.Size([1, 1000, 754]) | 3\n",
-      "8670     | torch.Size([1, 1000, 762]) | 3\n",
-      "8671     | torch.Size([1, 1000, 754]) | 3\n",
-      "8672     | torch.Size([1, 1000, 777]) | 3\n",
-      "8673     | torch.Size([1, 1000, 783]) | 3\n",
-      "8674     | torch.Size([1, 1000, 777]) | 3\n",
-      "8675     | torch.Size([1, 1000, 779]) | 3\n",
-      "8676     | torch.Size([1, 1000, 778]) | 3\n",
-      "8677     | torch.Size([1, 1000, 754]) | 3\n",
-      "8678     | torch.Size([1, 1000, 782]) | 3\n",
-      "8679     | torch.Size([1, 1000, 754]) | 3\n",
-      "8680     | torch.Size([1, 1000, 754]) | 3\n",
-      "8681     | torch.Size([1, 1000, 762]) | 3\n",
-      "8682     | torch.Size([1, 1000, 762]) | 3\n",
-      "8683     | torch.Size([1, 1000, 754]) | 3\n",
-      "8684     | torch.Size([1, 1000, 777]) | 3\n",
-      "8685     | torch.Size([1, 1000, 816]) | 3\n",
-      "8686     | torch.Size([1, 1000, 790]) | 3\n",
-      "8687     | torch.Size([1, 1000, 777]) | 3\n",
-      "8688     | torch.Size([1, 1000, 762]) | 3\n",
-      "8689     | torch.Size([1, 1000, 777]) | 3\n",
-      "8690     | torch.Size([1, 1000, 754]) | 3\n",
-      "8691     | torch.Size([1, 1000, 762]) | 3\n",
-      "8692     | torch.Size([1, 1000, 795]) | 3\n",
-      "8693     | torch.Size([1, 1000, 762]) | 3\n",
-      "8694     | torch.Size([1, 1000, 762]) | 3\n",
-      "8695     | torch.Size([1, 1000, 811]) | 3\n",
-      "8696     | torch.Size([1, 1000, 762]) | 3\n",
-      "8697     | torch.Size([1, 1000, 754]) | 3\n",
-      "8698     | torch.Size([1, 1000, 777]) | 3\n",
-      "8699     | torch.Size([1, 1000, 777]) | 3\n",
-      "8700     | torch.Size([1, 1000, 795]) | 3\n",
-      "8701     | torch.Size([1, 1000, 754]) | 3\n",
-      "8702     | torch.Size([1, 1000, 754]) | 3\n",
-      "8703     | torch.Size([1, 1000, 772]) | 3\n",
-      "8704     | torch.Size([1, 1000, 777]) | 3\n",
-      "8705     | torch.Size([1, 1000, 779]) | 3\n",
-      "8706     | torch.Size([1, 1000, 776]) | 3\n",
-      "8707     | torch.Size([1, 1000, 787]) | 3\n",
-      "8708     | torch.Size([1, 1000, 828]) | 3\n",
-      "8709     | torch.Size([1, 1000, 754]) | 3\n",
-      "8710     | torch.Size([1, 1000, 762]) | 3\n",
-      "8711     | torch.Size([1, 1000, 810]) | 3\n",
-      "8712     | torch.Size([1, 1000, 777]) | 3\n",
-      "8713     | torch.Size([1, 1000, 754]) | 3\n",
-      "8714     | torch.Size([1, 1000, 777]) | 3\n",
-      "8715     | torch.Size([1, 1000, 777]) | 3\n",
-      "8716     | torch.Size([1, 1000, 754]) | 3\n",
-      "8717     | torch.Size([1, 1000, 713]) | 3\n",
-      "8718     | torch.Size([1, 1000, 754]) | 3\n",
-      "8719     | torch.Size([1, 1000, 762]) | 3\n",
-      "8720     | torch.Size([1, 1000, 777]) | 3\n",
-      "8721     | torch.Size([1, 1000, 754]) | 3\n",
-      "8722     | torch.Size([1, 1000, 800]) | 3\n",
-      "8723     | torch.Size([1, 1000, 754]) | 3\n",
-      "8724     | torch.Size([1, 1000, 754]) | 3\n",
-      "8725     | torch.Size([1, 1000, 785]) | 3\n",
-      "8726     | torch.Size([1, 1000, 777]) | 3\n",
-      "8727     | torch.Size([1, 1000, 777]) | 3\n",
-      "8728     | torch.Size([1, 1000, 762]) | 3\n",
-      "8729     | torch.Size([1, 1000, 798]) | 3\n",
-      "8730     | torch.Size([1, 1000, 777]) | 3\n",
-      "8731     | torch.Size([1, 1000, 762]) | 3\n",
-      "8732     | torch.Size([1, 1000, 762]) | 3\n",
-      "8733     | torch.Size([1, 1000, 779]) | 3\n",
-      "8734     | torch.Size([1, 1000, 777]) | 3\n",
-      "8735     | torch.Size([1, 1000, 754]) | 3\n",
-      "8736     | torch.Size([1, 1000, 781]) | 3\n",
-      "8737     | torch.Size([1, 1000, 754]) | 3\n",
-      "8738     | torch.Size([1, 1000, 789]) | 3\n",
-      "8739     | torch.Size([1, 1000, 754]) | 3\n",
-      "8740     | torch.Size([1, 1000, 754]) | 3\n",
-      "8741     | torch.Size([1, 1000, 773]) | 3\n",
-      "8742     | torch.Size([1, 1000, 754]) | 3\n",
-      "8743     | torch.Size([1, 1000, 754]) | 3\n",
-      "8744     | torch.Size([1, 1000, 754]) | 3\n",
-      "8745     | torch.Size([1, 1000, 762]) | 3\n",
-      "8746     | torch.Size([1, 1000, 797]) | 3\n",
-      "8747     | torch.Size([1, 1000, 754]) | 3\n",
-      "8748     | torch.Size([1, 1000, 802]) | 3\n",
-      "8749     | torch.Size([1, 1000, 786]) | 3\n",
-      "8750     | torch.Size([1, 1000, 777]) | 3\n",
-      "8751     | torch.Size([1, 1000, 762]) | 3\n",
-      "8752     | torch.Size([1, 1000, 777]) | 3\n",
-      "8753     | torch.Size([1, 1000, 761]) | 3\n",
-      "8754     | torch.Size([1, 1000, 754]) | 3\n",
-      "8755     | torch.Size([1, 1000, 754]) | 3\n",
-      "8756     | torch.Size([1, 1000, 790]) | 3\n",
-      "8757     | torch.Size([1, 1000, 762]) | 3\n",
-      "8758     | torch.Size([1, 1000, 773]) | 3\n",
-      "8759     | torch.Size([1, 1000, 772]) | 3\n",
-      "8760     | torch.Size([1, 1000, 777]) | 3\n",
-      "8761     | torch.Size([1, 1000, 754]) | 3\n",
-      "8762     | torch.Size([1, 1000, 754]) | 3\n",
-      "8763     | torch.Size([1, 1000, 770]) | 3\n",
-      "8764     | torch.Size([1, 1000, 772]) | 3\n",
-      "8765     | torch.Size([1, 1000, 754]) | 3\n",
-      "8766     | torch.Size([1, 1000, 713]) | 3\n",
-      "8767     | torch.Size([1, 1000, 762]) | 3\n",
-      "8768     | torch.Size([1, 1000, 780]) | 3\n",
-      "8769     | torch.Size([1, 1000, 762]) | 3\n",
-      "8770     | torch.Size([1, 1000, 786]) | 3\n",
-      "8771     | torch.Size([1, 1000, 754]) | 3\n",
-      "8772     | torch.Size([1, 1000, 762]) | 3\n",
-      "8773     | torch.Size([1, 1000, 862]) | 3\n",
-      "8774     | torch.Size([1, 1000, 772]) | 3\n",
-      "8775     | torch.Size([1, 1000, 762]) | 3\n",
-      "8776     | torch.Size([1, 1000, 788]) | 3\n",
-      "8777     | torch.Size([1, 1000, 754]) | 3\n",
-      "8778     | torch.Size([1, 1000, 762]) | 3\n",
-      "8779     | torch.Size([1, 1000, 777]) | 3\n",
-      "8780     | torch.Size([1, 1000, 782]) | 3\n",
-      "8781     | torch.Size([1, 1000, 777]) | 3\n",
-      "8782     | torch.Size([1, 1000, 762]) | 3\n",
-      "8783     | torch.Size([1, 1000, 773]) | 3\n",
-      "8784     | torch.Size([1, 1000, 762]) | 3\n",
-      "8785     | torch.Size([1, 1000, 763]) | 3\n",
-      "8786     | torch.Size([1, 1000, 754]) | 3\n",
-      "8787     | torch.Size([1, 1000, 762]) | 3\n",
-      "8788     | torch.Size([1, 1000, 754]) | 3\n",
-      "8789     | torch.Size([1, 1000, 778]) | 3\n",
-      "8790     | torch.Size([1, 1000, 763]) | 3\n",
-      "8791     | torch.Size([1, 1000, 781]) | 3\n",
-      "8792     | torch.Size([1, 1000, 613]) | 3\n",
-      "8793     | torch.Size([1, 1000, 785]) | 3\n",
-      "8794     | torch.Size([1, 1000, 754]) | 3\n",
-      "8795     | torch.Size([1, 1000, 759]) | 3\n",
-      "8796     | torch.Size([1, 1000, 725]) | 3\n",
-      "8797     | torch.Size([1, 1000, 754]) | 3\n",
-      "8798     | torch.Size([1, 1000, 777]) | 3\n",
-      "8799     | torch.Size([1, 1000, 754]) | 3\n",
-      "8800     | torch.Size([1, 1000, 762]) | 3\n",
-      "8801     | torch.Size([1, 1000, 754]) | 3\n",
-      "8802     | torch.Size([1, 1000, 762]) | 3\n",
-      "8803     | torch.Size([1, 1000, 841]) | 3\n",
-      "8804     | torch.Size([1, 1000, 789]) | 3\n",
-      "8805     | torch.Size([1, 1000, 773]) | 3\n",
-      "8806     | torch.Size([1, 1000, 784]) | 3\n",
-      "8807     | torch.Size([1, 1000, 781]) | 3\n",
-      "8808     | torch.Size([1, 1000, 754]) | 3\n",
-      "8809     | torch.Size([1, 1000, 715]) | 3\n",
-      "8810     | torch.Size([1, 1000, 754]) | 3\n",
-      "8811     | torch.Size([1, 1000, 794]) | 3\n",
-      "8812     | torch.Size([1, 1000, 754]) | 3\n",
-      "8813     | torch.Size([1, 1000, 754]) | 3\n",
-      "8814     | torch.Size([1, 1000, 782]) | 3\n",
-      "8815     | torch.Size([1, 1000, 762]) | 3\n",
-      "8816     | torch.Size([1, 1000, 754]) | 3\n",
-      "8817     | torch.Size([1, 1000, 770]) | 3\n",
-      "8818     | torch.Size([1, 1000, 779]) | 3\n",
-      "8819     | torch.Size([1, 1000, 802]) | 3\n",
-      "8820     | torch.Size([1, 1000, 762]) | 3\n",
-      "8821     | torch.Size([1, 1000, 713]) | 3\n",
-      "8822     | torch.Size([1, 1000, 777]) | 3\n",
-      "8823     | torch.Size([1, 1000, 754]) | 3\n",
-      "8824     | torch.Size([1, 1000, 777]) | 3\n",
-      "8825     | torch.Size([1, 1000, 777]) | 3\n",
-      "8826     | torch.Size([1, 1000, 754]) | 3\n",
-      "8827     | torch.Size([1, 1000, 782]) | 3\n",
-      "8828     | torch.Size([1, 1000, 754]) | 3\n",
-      "8829     | torch.Size([1, 1000, 772]) | 3\n",
-      "8830     | torch.Size([1, 1000, 754]) | 3\n",
-      "8831     | torch.Size([1, 1000, 754]) | 3\n",
-      "8832     | torch.Size([1, 1000, 796]) | 3\n",
-      "8833     | torch.Size([1, 1000, 777]) | 3\n",
-      "8834     | torch.Size([1, 1000, 810]) | 3\n",
-      "8835     | torch.Size([1, 1000, 777]) | 3\n",
-      "8836     | torch.Size([1, 1000, 754]) | 3\n",
-      "8837     | torch.Size([1, 1000, 767]) | 3\n",
-      "8838     | torch.Size([1, 1000, 762]) | 3\n",
-      "8839     | torch.Size([1, 1000, 782]) | 3\n",
-      "8840     | torch.Size([1, 1000, 762]) | 3\n",
-      "8841     | torch.Size([1, 1000, 766]) | 3\n",
-      "8842     | torch.Size([1, 1000, 754]) | 3\n",
-      "8843     | torch.Size([1, 1000, 762]) | 3\n",
-      "8844     | torch.Size([1, 1000, 771]) | 3\n",
-      "8845     | torch.Size([1, 1000, 754]) | 3\n",
-      "8846     | torch.Size([1, 1000, 762]) | 3\n",
-      "8847     | torch.Size([1, 1000, 769]) | 3\n",
-      "8848     | torch.Size([1, 1000, 777]) | 3\n",
-      "8849     | torch.Size([1, 1000, 713]) | 3\n",
-      "8850     | torch.Size([1, 1000, 777]) | 3\n",
-      "8851     | torch.Size([1, 1000, 754]) | 3\n",
-      "8852     | torch.Size([1, 1000, 754]) | 3\n",
-      "8853     | torch.Size([1, 1000, 754]) | 3\n",
-      "8854     | torch.Size([1, 1000, 772]) | 3\n",
-      "8855     | torch.Size([1, 1000, 777]) | 3\n",
-      "8856     | torch.Size([1, 1000, 754]) | 3\n",
-      "8857     | torch.Size([1, 1000, 804]) | 3\n",
-      "8858     | torch.Size([1, 1000, 777]) | 3\n",
-      "8859     | torch.Size([1, 1000, 754]) | 3\n",
-      "8860     | torch.Size([1, 1000, 754]) | 3\n",
-      "8861     | torch.Size([1, 1000, 806]) | 3\n",
-      "8862     | torch.Size([1, 1000, 773]) | 3\n",
-      "8863     | torch.Size([1, 1000, 754]) | 3\n",
-      "8864     | torch.Size([1, 1000, 754]) | 3\n",
-      "8865     | torch.Size([1, 1000, 754]) | 3\n",
-      "8866     | torch.Size([1, 1000, 762]) | 3\n",
-      "8867     | torch.Size([1, 1000, 771]) | 3\n",
-      "8868     | torch.Size([1, 1000, 762]) | 3\n",
-      "8869     | torch.Size([1, 1000, 723]) | 3\n",
-      "8870     | torch.Size([1, 1000, 754]) | 3\n",
-      "8871     | torch.Size([1, 1000, 754]) | 3\n",
-      "8872     | torch.Size([1, 1000, 762]) | 3\n",
-      "8873     | torch.Size([1, 1000, 606]) | 3\n",
-      "8874     | torch.Size([1, 1000, 801]) | 3\n",
-      "8875     | torch.Size([1, 1000, 703]) | 3\n",
-      "8876     | torch.Size([1, 1000, 762]) | 3\n",
-      "8877     | torch.Size([1, 1000, 777]) | 3\n",
-      "8878     | torch.Size([1, 1000, 795]) | 3\n",
-      "8879     | torch.Size([1, 1000, 754]) | 3\n",
-      "8880     | torch.Size([1, 1000, 777]) | 3\n",
-      "8881     | torch.Size([1, 1000, 754]) | 3\n",
-      "8882     | torch.Size([1, 1000, 779]) | 3\n",
-      "8883     | torch.Size([1, 1000, 754]) | 3\n",
-      "8884     | torch.Size([1, 1000, 754]) | 3\n",
-      "8885     | torch.Size([1, 1000, 845]) | 3\n",
-      "8886     | torch.Size([1, 1000, 788]) | 3\n",
-      "8887     | torch.Size([1, 1000, 762]) | 3\n",
-      "8888     | torch.Size([1, 1000, 762]) | 3\n",
-      "8889     | torch.Size([1, 1000, 754]) | 3\n",
-      "8890     | torch.Size([1, 1000, 754]) | 3\n",
-      "8891     | torch.Size([1, 1000, 771]) | 3\n",
-      "8892     | torch.Size([1, 1000, 785]) | 3\n",
-      "8893     | torch.Size([1, 1000, 754]) | 3\n",
-      "8894     | torch.Size([1, 1000, 829]) | 3\n",
-      "8895     | torch.Size([1, 1000, 777]) | 3\n",
-      "8896     | torch.Size([1, 1000, 772]) | 3\n",
-      "8897     | torch.Size([1, 1000, 764]) | 3\n",
-      "8898     | torch.Size([1, 1000, 767]) | 3\n",
-      "8899     | torch.Size([1, 1000, 754]) | 3\n",
-      "8900     | torch.Size([1, 1000, 777]) | 3\n",
-      "8901     | torch.Size([1, 1000, 754]) | 3\n",
-      "8902     | torch.Size([1, 1000, 762]) | 3\n",
-      "8903     | torch.Size([1, 1000, 771]) | 3\n",
-      "8904     | torch.Size([1, 1000, 764]) | 3\n",
-      "8905     | torch.Size([1, 1000, 762]) | 3\n",
-      "8906     | torch.Size([1, 1000, 771]) | 3\n",
-      "8907     | torch.Size([1, 1000, 762]) | 3\n",
-      "8908     | torch.Size([1, 1000, 777]) | 3\n",
-      "8909     | torch.Size([1, 1000, 754]) | 3\n",
-      "8910     | torch.Size([1, 1000, 762]) | 3\n",
-      "8911     | torch.Size([1, 1000, 772]) | 3\n",
-      "8912     | torch.Size([1, 1000, 787]) | 3\n",
-      "8913     | torch.Size([1, 1000, 777]) | 3\n",
-      "8914     | torch.Size([1, 1000, 754]) | 3\n",
-      "8915     | torch.Size([1, 1000, 754]) | 3\n",
-      "8916     | torch.Size([1, 1000, 781]) | 3\n",
-      "8917     | torch.Size([1, 1000, 781]) | 3\n",
-      "8918     | torch.Size([1, 1000, 773]) | 3\n",
-      "8919     | torch.Size([1, 1000, 839]) | 3\n",
-      "8920     | torch.Size([1, 1000, 762]) | 3\n",
-      "8921     | torch.Size([1, 1000, 754]) | 3\n",
-      "8922     | torch.Size([1, 1000, 762]) | 3\n",
-      "8923     | torch.Size([1, 1000, 754]) | 3\n",
-      "8924     | torch.Size([1, 1000, 777]) | 3\n",
-      "8925     | torch.Size([1, 1000, 762]) | 3\n",
-      "8926     | torch.Size([1, 1000, 754]) | 3\n",
-      "8927     | torch.Size([1, 1000, 767]) | 3\n",
-      "8928     | torch.Size([1, 1000, 754]) | 3\n",
-      "8929     | torch.Size([1, 1000, 754]) | 3\n",
-      "8930     | torch.Size([1, 1000, 762]) | 3\n",
-      "8931     | torch.Size([1, 1000, 754]) | 3\n",
-      "8932     | torch.Size([1, 1000, 754]) | 3\n",
-      "8933     | torch.Size([1, 1000, 739]) | 3\n",
-      "8934     | torch.Size([1, 1000, 762]) | 3\n",
-      "8935     | torch.Size([1, 1000, 773]) | 3\n",
-      "8936     | torch.Size([1, 1000, 754]) | 3\n",
-      "8937     | torch.Size([1, 1000, 795]) | 3\n",
-      "8938     | torch.Size([1, 1000, 802]) | 3\n",
-      "8939     | torch.Size([1, 1000, 754]) | 3\n",
-      "8940     | torch.Size([1, 1000, 762]) | 3\n",
-      "8941     | torch.Size([1, 1000, 777]) | 3\n",
-      "8942     | torch.Size([1, 1000, 777]) | 3\n",
-      "8943     | torch.Size([1, 1000, 754]) | 3\n",
-      "8944     | torch.Size([1, 1000, 777]) | 3\n",
-      "8945     | torch.Size([1, 1000, 762]) | 3\n",
-      "8946     | torch.Size([1, 1000, 798]) | 3\n",
-      "8947     | torch.Size([1, 1000, 754]) | 3\n",
-      "8948     | torch.Size([1, 1000, 762]) | 3\n",
-      "8949     | torch.Size([1, 1000, 800]) | 3\n",
-      "8950     | torch.Size([1, 1000, 777]) | 3\n",
-      "8951     | torch.Size([1, 1000, 777]) | 3\n",
-      "8952     | torch.Size([1, 1000, 795]) | 3\n",
-      "8953     | torch.Size([1, 1000, 754]) | 3\n",
-      "8954     | torch.Size([1, 1000, 799]) | 3\n",
-      "8955     | torch.Size([1, 1000, 777]) | 3\n",
-      "8956     | torch.Size([1, 1000, 777]) | 3\n",
-      "8957     | torch.Size([1, 1000, 777]) | 3\n",
-      "8958     | torch.Size([1, 1000, 762]) | 3\n",
-      "8959     | torch.Size([1, 1000, 779]) | 3\n",
-      "8960     | torch.Size([1, 1000, 754]) | 3\n",
-      "8961     | torch.Size([1, 1000, 754]) | 3\n",
-      "8962     | torch.Size([1, 1000, 785]) | 3\n",
-      "8963     | torch.Size([1, 1000, 754]) | 3\n",
-      "8964     | torch.Size([1, 1000, 776]) | 3\n",
-      "8965     | torch.Size([1, 1000, 754]) | 3\n",
-      "8966     | torch.Size([1, 1000, 777]) | 3\n",
-      "8967     | torch.Size([1, 1000, 713]) | 3\n",
-      "8968     | torch.Size([1, 1000, 754]) | 3\n",
-      "8969     | torch.Size([1, 1000, 754]) | 3\n",
-      "8970     | torch.Size([1, 1000, 778]) | 3\n",
-      "8971     | torch.Size([1, 1000, 772]) | 3\n",
-      "8972     | torch.Size([1, 1000, 777]) | 3\n",
-      "8973     | torch.Size([1, 1000, 754]) | 3\n",
-      "8974     | torch.Size([1, 1000, 800]) | 3\n",
-      "8975     | torch.Size([1, 1000, 776]) | 3\n",
-      "8976     | torch.Size([1, 1000, 754]) | 3\n",
-      "8977     | torch.Size([1, 1000, 774]) | 3\n",
-      "8978     | torch.Size([1, 1000, 787]) | 3\n",
-      "8979     | torch.Size([1, 1000, 762]) | 3\n",
-      "8980     | torch.Size([1, 1000, 777]) | 3\n",
-      "8981     | torch.Size([1, 1000, 762]) | 3\n",
-      "8982     | torch.Size([1, 1000, 777]) | 3\n",
-      "8983     | torch.Size([1, 1000, 764]) | 3\n",
-      "8984     | torch.Size([1, 1000, 790]) | 3\n",
-      "8985     | torch.Size([1, 1000, 754]) | 3\n",
-      "8986     | torch.Size([1, 1000, 754]) | 3\n",
-      "8987     | torch.Size([1, 1000, 754]) | 3\n",
-      "8988     | torch.Size([1, 1000, 843]) | 3\n",
-      "8989     | torch.Size([1, 1000, 792]) | 3\n",
-      "8990     | torch.Size([1, 1000, 777]) | 3\n",
-      "8991     | torch.Size([1, 1000, 777]) | 3\n",
-      "8992     | torch.Size([1, 1000, 715]) | 3\n",
-      "8993     | torch.Size([1, 1000, 780]) | 3\n",
-      "8994     | torch.Size([1, 1000, 770]) | 3\n",
-      "8995     | torch.Size([1, 1000, 762]) | 3\n",
-      "8996     | torch.Size([1, 1000, 839]) | 3\n",
-      "8997     | torch.Size([1, 1000, 754]) | 3\n",
-      "8998     | torch.Size([1, 1000, 769]) | 3\n",
-      "8999     | torch.Size([1, 1000, 789]) | 3\n",
-      "9000     | torch.Size([1, 1000, 754]) | 3\n",
-      "9001     | torch.Size([1, 1000, 777]) | 3\n",
-      "9002     | torch.Size([1, 1000, 777]) | 3\n",
-      "9003     | torch.Size([1, 1000, 754]) | 3\n",
-      "9004     | torch.Size([1, 1000, 795]) | 3\n",
-      "9005     | torch.Size([1, 1000, 783]) | 3\n",
-      "9006     | torch.Size([1, 1000, 868]) | 3\n",
-      "9007     | torch.Size([1, 1000, 762]) | 3\n",
-      "9008     | torch.Size([1, 1000, 778]) | 3\n",
-      "9009     | torch.Size([1, 1000, 779]) | 3\n",
-      "9010     | torch.Size([1, 1000, 754]) | 3\n",
-      "9011     | torch.Size([1, 1000, 778]) | 3\n",
-      "9012     | torch.Size([1, 1000, 786]) | 3\n",
-      "9013     | torch.Size([1, 1000, 781]) | 3\n",
-      "9014     | torch.Size([1, 1000, 754]) | 3\n",
-      "9015     | torch.Size([1, 1000, 786]) | 3\n",
-      "9016     | torch.Size([1, 1000, 794]) | 3\n",
-      "9017     | torch.Size([1, 1000, 754]) | 3\n",
-      "9018     | torch.Size([1, 1000, 660]) | 3\n",
-      "9019     | torch.Size([1, 1000, 802]) | 3\n",
-      "9020     | torch.Size([1, 1000, 754]) | 3\n",
-      "9021     | torch.Size([1, 1000, 762]) | 3\n",
-      "9022     | torch.Size([1, 1000, 777]) | 3\n",
-      "9023     | torch.Size([1, 1000, 777]) | 3\n",
-      "9024     | torch.Size([1, 1000, 754]) | 3\n",
-      "9025     | torch.Size([1, 1000, 754]) | 3\n",
-      "9026     | torch.Size([1, 1000, 777]) | 3\n",
-      "9027     | torch.Size([1, 1000, 790]) | 3\n",
-      "9028     | torch.Size([1, 1000, 754]) | 3\n",
-      "9029     | torch.Size([1, 1000, 777]) | 3\n",
-      "9030     | torch.Size([1, 1000, 762]) | 3\n",
-      "9031     | torch.Size([1, 1000, 754]) | 3\n",
-      "9032     | torch.Size([1, 1000, 762]) | 3\n",
-      "9033     | torch.Size([1, 1000, 762]) | 3\n",
-      "9034     | torch.Size([1, 1000, 762]) | 3\n",
-      "9035     | torch.Size([1, 1000, 762]) | 3\n",
-      "9036     | torch.Size([1, 1000, 826]) | 3\n",
-      "9037     | torch.Size([1, 1000, 718]) | 3\n",
-      "9038     | torch.Size([1, 1000, 773]) | 3\n",
-      "9039     | torch.Size([1, 1000, 777]) | 3\n",
-      "9040     | torch.Size([1, 1000, 777]) | 3\n",
-      "9041     | torch.Size([1, 1000, 833]) | 3\n",
-      "9042     | torch.Size([1, 1000, 754]) | 3\n",
-      "9043     | torch.Size([1, 1000, 754]) | 3\n",
-      "9044     | torch.Size([1, 1000, 845]) | 3\n",
-      "9045     | torch.Size([1, 1000, 754]) | 3\n",
-      "9046     | torch.Size([1, 1000, 777]) | 3\n",
-      "9047     | torch.Size([1, 1000, 754]) | 3\n",
-      "9048     | torch.Size([1, 1000, 754]) | 3\n",
-      "9049     | torch.Size([1, 1000, 777]) | 3\n",
-      "9050     | torch.Size([1, 1000, 752]) | 3\n",
-      "9051     | torch.Size([1, 1000, 762]) | 3\n",
-      "9052     | torch.Size([1, 1000, 754]) | 3\n",
-      "9053     | torch.Size([1, 1000, 754]) | 3\n",
-      "9054     | torch.Size([1, 1000, 754]) | 3\n",
-      "9055     | torch.Size([1, 1000, 754]) | 3\n",
-      "9056     | torch.Size([1, 1000, 772]) | 3\n",
-      "9057     | torch.Size([1, 1000, 777]) | 3\n",
-      "9058     | torch.Size([1, 1000, 796]) | 3\n",
-      "9059     | torch.Size([1, 1000, 754]) | 3\n",
-      "9060     | torch.Size([1, 1000, 754]) | 3\n",
-      "9061     | torch.Size([1, 1000, 781]) | 3\n",
-      "9062     | torch.Size([1, 1000, 754]) | 3\n",
-      "9063     | torch.Size([1, 1000, 754]) | 3\n",
-      "9064     | torch.Size([1, 1000, 776]) | 3\n",
-      "9065     | torch.Size([1, 1000, 754]) | 3\n",
-      "9066     | torch.Size([1, 1000, 772]) | 3\n",
-      "9067     | torch.Size([1, 1000, 789]) | 3\n",
-      "9068     | torch.Size([1, 1000, 848]) | 3\n",
-      "9069     | torch.Size([1, 1000, 754]) | 3\n",
-      "9070     | torch.Size([1, 1000, 769]) | 3\n",
-      "9071     | torch.Size([1, 1000, 762]) | 3\n",
-      "9072     | torch.Size([1, 1000, 754]) | 3\n",
-      "9073     | torch.Size([1, 1000, 754]) | 3\n",
-      "9074     | torch.Size([1, 1000, 754]) | 3\n",
-      "9075     | torch.Size([1, 1000, 762]) | 3\n",
-      "9076     | torch.Size([1, 1000, 806]) | 3\n",
-      "9077     | torch.Size([1, 1000, 754]) | 3\n",
-      "9078     | torch.Size([1, 1000, 777]) | 3\n",
-      "9079     | torch.Size([1, 1000, 762]) | 3\n",
-      "9080     | torch.Size([1, 1000, 754]) | 3\n",
-      "9081     | torch.Size([1, 1000, 779]) | 3\n",
-      "9082     | torch.Size([1, 1000, 762]) | 3\n",
-      "9083     | torch.Size([1, 1000, 788]) | 3\n",
-      "9084     | torch.Size([1, 1000, 754]) | 3\n",
-      "9085     | torch.Size([1, 1000, 863]) | 3\n",
-      "9086     | torch.Size([1, 1000, 784]) | 3\n",
-      "9087     | torch.Size([1, 1000, 777]) | 3\n",
-      "9088     | torch.Size([1, 1000, 869]) | 3\n",
-      "9089     | torch.Size([1, 1000, 754]) | 3\n",
-      "9090     | torch.Size([1, 1000, 754]) | 3\n",
-      "9091     | torch.Size([1, 1000, 762]) | 3\n",
-      "9092     | torch.Size([1, 1000, 772]) | 3\n",
-      "9093     | torch.Size([1, 1000, 782]) | 3\n",
-      "9094     | torch.Size([1, 1000, 771]) | 3\n",
-      "9095     | torch.Size([1, 1000, 870]) | 3\n",
-      "9096     | torch.Size([1, 1000, 787]) | 3\n",
-      "9097     | torch.Size([1, 1000, 732]) | 3\n",
-      "9098     | torch.Size([1, 1000, 772]) | 3\n",
-      "9099     | torch.Size([1, 1000, 770]) | 3\n",
-      "9100     | torch.Size([1, 1000, 754]) | 3\n",
-      "9101     | torch.Size([1, 1000, 754]) | 3\n",
-      "9102     | torch.Size([1, 1000, 762]) | 3\n",
-      "9103     | torch.Size([1, 1000, 754]) | 3\n",
-      "9104     | torch.Size([1, 1000, 754]) | 3\n",
-      "9105     | torch.Size([1, 1000, 762]) | 3\n",
-      "9106     | torch.Size([1, 1000, 754]) | 3\n",
-      "9107     | torch.Size([1, 1000, 754]) | 3\n",
-      "9108     | torch.Size([1, 1000, 776]) | 3\n",
-      "9109     | torch.Size([1, 1000, 762]) | 3\n",
-      "9110     | torch.Size([1, 1000, 754]) | 3\n",
-      "9111     | torch.Size([1, 1000, 779]) | 3\n",
-      "9112     | torch.Size([1, 1000, 762]) | 3\n",
-      "9113     | torch.Size([1, 1000, 777]) | 3\n",
-      "9114     | torch.Size([1, 1000, 777]) | 3\n",
-      "9115     | torch.Size([1, 1000, 754]) | 3\n",
-      "9116     | torch.Size([1, 1000, 762]) | 3\n",
-      "9117     | torch.Size([1, 1000, 775]) | 3\n",
-      "9118     | torch.Size([1, 1000, 762]) | 3\n",
-      "9119     | torch.Size([1, 1000, 754]) | 3\n",
-      "9120     | torch.Size([1, 1000, 761]) | 3\n",
-      "9121     | torch.Size([1, 1000, 762]) | 3\n",
-      "9122     | torch.Size([1, 1000, 790]) | 3\n",
-      "9123     | torch.Size([1, 1000, 820]) | 3\n",
-      "9124     | torch.Size([1, 1000, 762]) | 3\n",
-      "9125     | torch.Size([1, 1000, 777]) | 3\n",
-      "9126     | torch.Size([1, 1000, 754]) | 3\n",
-      "9127     | torch.Size([1, 1000, 788]) | 3\n",
-      "9128     | torch.Size([1, 1000, 837]) | 3\n",
-      "9129     | torch.Size([1, 1000, 762]) | 3\n",
-      "9130     | torch.Size([1, 1000, 827]) | 3\n",
-      "9131     | torch.Size([1, 1000, 794]) | 3\n",
-      "9132     | torch.Size([1, 1000, 762]) | 3\n",
-      "9133     | torch.Size([1, 1000, 754]) | 3\n",
-      "9134     | torch.Size([1, 1000, 762]) | 3\n",
-      "9135     | torch.Size([1, 1000, 814]) | 3\n",
-      "9136     | torch.Size([1, 1000, 777]) | 3\n",
-      "9137     | torch.Size([1, 1000, 754]) | 3\n",
-      "9138     | torch.Size([1, 1000, 762]) | 3\n",
-      "9139     | torch.Size([1, 1000, 754]) | 3\n",
-      "9140     | torch.Size([1, 1000, 754]) | 3\n",
-      "9141     | torch.Size([1, 1000, 754]) | 3\n",
-      "9142     | torch.Size([1, 1000, 777]) | 3\n",
-      "9143     | torch.Size([1, 1000, 754]) | 3\n",
-      "9144     | torch.Size([1, 1000, 778]) | 3\n",
-      "9145     | torch.Size([1, 1000, 772]) | 3\n",
-      "9146     | torch.Size([1, 1000, 754]) | 3\n",
-      "9147     | torch.Size([1, 1000, 777]) | 3\n",
-      "9148     | torch.Size([1, 1000, 762]) | 3\n",
-      "9149     | torch.Size([1, 1000, 754]) | 3\n",
-      "9150     | torch.Size([1, 1000, 779]) | 3\n",
-      "9151     | torch.Size([1, 1000, 714]) | 3\n",
-      "9152     | torch.Size([1, 1000, 777]) | 3\n",
-      "9153     | torch.Size([1, 1000, 754]) | 3\n",
-      "9154     | torch.Size([1, 1000, 754]) | 3\n",
-      "9155     | torch.Size([1, 1000, 754]) | 3\n",
-      "9156     | torch.Size([1, 1000, 777]) | 3\n",
-      "9157     | torch.Size([1, 1000, 762]) | 3\n",
-      "9158     | torch.Size([1, 1000, 762]) | 3\n",
-      "9159     | torch.Size([1, 1000, 754]) | 3\n",
-      "9160     | torch.Size([1, 1000, 754]) | 3\n",
-      "9161     | torch.Size([1, 1000, 754]) | 3\n",
-      "9162     | torch.Size([1, 1000, 814]) | 3\n",
-      "9163     | torch.Size([1, 1000, 754]) | 3\n",
-      "9164     | torch.Size([1, 1000, 762]) | 3\n",
-      "9165     | torch.Size([1, 1000, 777]) | 3\n",
-      "9166     | torch.Size([1, 1000, 787]) | 3\n",
-      "9167     | torch.Size([1, 1000, 762]) | 3\n",
-      "9168     | torch.Size([1, 1000, 734]) | 3\n",
-      "9169     | torch.Size([1, 1000, 781]) | 3\n",
-      "9170     | torch.Size([1, 1000, 777]) | 3\n",
-      "9171     | torch.Size([1, 1000, 754]) | 3\n",
-      "9172     | torch.Size([1, 1000, 714]) | 3\n",
-      "9173     | torch.Size([1, 1000, 811]) | 3\n",
-      "9174     | torch.Size([1, 1000, 777]) | 3\n",
-      "9175     | torch.Size([1, 1000, 781]) | 3\n",
-      "9176     | torch.Size([1, 1000, 777]) | 3\n",
-      "9177     | torch.Size([1, 1000, 762]) | 3\n",
-      "9178     | torch.Size([1, 1000, 754]) | 3\n",
-      "9179     | torch.Size([1, 1000, 784]) | 3\n",
-      "9180     | torch.Size([1, 1000, 777]) | 3\n",
-      "9181     | torch.Size([1, 1000, 777]) | 3\n",
-      "9182     | torch.Size([1, 1000, 770]) | 3\n",
-      "9183     | torch.Size([1, 1000, 795]) | 3\n",
-      "9184     | torch.Size([1, 1000, 844]) | 3\n",
-      "9185     | torch.Size([1, 1000, 754]) | 3\n",
-      "9186     | torch.Size([1, 1000, 790]) | 3\n",
-      "9187     | torch.Size([1, 1000, 809]) | 3\n",
-      "9188     | torch.Size([1, 1000, 754]) | 3\n",
-      "9189     | torch.Size([1, 1000, 777]) | 3\n",
-      "9190     | torch.Size([1, 1000, 754]) | 3\n",
-      "9191     | torch.Size([1, 1000, 781]) | 3\n",
-      "9192     | torch.Size([1, 1000, 777]) | 3\n",
-      "9193     | torch.Size([1, 1000, 772]) | 3\n",
-      "9194     | torch.Size([1, 1000, 777]) | 3\n",
-      "9195     | torch.Size([1, 1000, 754]) | 3\n",
-      "9196     | torch.Size([1, 1000, 754]) | 3\n",
-      "9197     | torch.Size([1, 1000, 754]) | 3\n",
-      "9198     | torch.Size([1, 1000, 762]) | 3\n",
-      "9199     | torch.Size([1, 1000, 754]) | 3\n",
-      "9200     | torch.Size([1, 1000, 774]) | 3\n",
-      "9201     | torch.Size([1, 1000, 777]) | 3\n",
-      "9202     | torch.Size([1, 1000, 805]) | 3\n",
-      "9203     | torch.Size([1, 1000, 756]) | 3\n",
-      "9204     | torch.Size([1, 1000, 754]) | 3\n",
-      "9205     | torch.Size([1, 1000, 762]) | 3\n",
-      "9206     | torch.Size([1, 1000, 795]) | 3\n",
-      "9207     | torch.Size([1, 1000, 754]) | 3\n",
-      "9208     | torch.Size([1, 1000, 754]) | 3\n",
-      "9209     | torch.Size([1, 1000, 754]) | 3\n",
-      "9210     | torch.Size([1, 1000, 830]) | 3\n",
-      "9211     | torch.Size([1, 1000, 762]) | 3\n",
-      "9212     | torch.Size([1, 1000, 777]) | 3\n",
-      "9213     | torch.Size([1, 1000, 754]) | 3\n",
-      "9214     | torch.Size([1, 1000, 758]) | 3\n",
-      "9215     | torch.Size([1, 1000, 767]) | 3\n",
-      "9216     | torch.Size([1, 1000, 754]) | 3\n",
-      "9217     | torch.Size([1, 1000, 777]) | 3\n",
-      "9218     | torch.Size([1, 1000, 774]) | 3\n",
-      "9219     | torch.Size([1, 1000, 762]) | 3\n",
-      "9220     | torch.Size([1, 1000, 762]) | 3\n",
-      "9221     | torch.Size([1, 1000, 797]) | 3\n",
-      "9222     | torch.Size([1, 1000, 754]) | 3\n",
-      "9223     | torch.Size([1, 1000, 779]) | 3\n",
-      "9224     | torch.Size([1, 1000, 781]) | 3\n",
-      "9225     | torch.Size([1, 1000, 754]) | 3\n",
-      "9226     | torch.Size([1, 1000, 754]) | 3\n",
-      "9227     | torch.Size([1, 1000, 762]) | 3\n",
-      "9228     | torch.Size([1, 1000, 786]) | 3\n",
-      "9229     | torch.Size([1, 1000, 777]) | 3\n",
-      "9230     | torch.Size([1, 1000, 762]) | 3\n",
-      "9231     | torch.Size([1, 1000, 786]) | 3\n",
-      "9232     | torch.Size([1, 1000, 762]) | 3\n",
-      "9233     | torch.Size([1, 1000, 754]) | 3\n",
-      "9234     | torch.Size([1, 1000, 802]) | 3\n",
-      "9235     | torch.Size([1, 1000, 762]) | 3\n",
-      "9236     | torch.Size([1, 1000, 762]) | 3\n",
-      "9237     | torch.Size([1, 1000, 754]) | 3\n",
-      "9238     | torch.Size([1, 1000, 779]) | 3\n",
-      "9239     | torch.Size([1, 1000, 751]) | 3\n",
-      "9240     | torch.Size([1, 1000, 780]) | 3\n",
-      "9241     | torch.Size([1, 1000, 719]) | 3\n",
-      "9242     | torch.Size([1, 1000, 833]) | 3\n",
-      "9243     | torch.Size([1, 1000, 754]) | 3\n",
-      "9244     | torch.Size([1, 1000, 754]) | 3\n",
-      "9245     | torch.Size([1, 1000, 871]) | 3\n",
-      "9246     | torch.Size([1, 1000, 777]) | 3\n",
-      "9247     | torch.Size([1, 1000, 762]) | 3\n",
-      "9248     | torch.Size([1, 1000, 762]) | 3\n",
-      "9249     | torch.Size([1, 1000, 776]) | 3\n",
-      "9250     | torch.Size([1, 1000, 754]) | 3\n",
-      "9251     | torch.Size([1, 1000, 762]) | 3\n",
-      "9252     | torch.Size([1, 1000, 777]) | 3\n",
-      "9253     | torch.Size([1, 1000, 762]) | 3\n",
-      "9254     | torch.Size([1, 1000, 777]) | 3\n",
-      "9255     | torch.Size([1, 1000, 806]) | 3\n",
-      "9256     | torch.Size([1, 1000, 754]) | 3\n",
-      "9257     | torch.Size([1, 1000, 777]) | 3\n",
-      "9258     | torch.Size([1, 1000, 785]) | 3\n",
-      "9259     | torch.Size([1, 1000, 777]) | 3\n",
-      "9260     | torch.Size([1, 1000, 762]) | 3\n",
-      "9261     | torch.Size([1, 1000, 785]) | 3\n",
-      "9262     | torch.Size([1, 1000, 754]) | 3\n",
-      "9263     | torch.Size([1, 1000, 767]) | 3\n",
-      "9264     | torch.Size([1, 1000, 762]) | 3\n",
-      "9265     | torch.Size([1, 1000, 796]) | 3\n",
-      "9266     | torch.Size([1, 1000, 754]) | 3\n",
-      "9267     | torch.Size([1, 1000, 846]) | 3\n",
-      "9268     | torch.Size([1, 1000, 754]) | 3\n",
-      "9269     | torch.Size([1, 1000, 762]) | 3\n",
-      "9270     | torch.Size([1, 1000, 786]) | 3\n",
-      "9271     | torch.Size([1, 1000, 777]) | 3\n",
-      "9272     | torch.Size([1, 1000, 762]) | 3\n",
-      "9273     | torch.Size([1, 1000, 783]) | 3\n",
-      "9274     | torch.Size([1, 1000, 798]) | 3\n",
-      "9275     | torch.Size([1, 1000, 799]) | 3\n",
-      "9276     | torch.Size([1, 1000, 762]) | 3\n",
-      "9277     | torch.Size([1, 1000, 866]) | 3\n",
-      "9278     | torch.Size([1, 1000, 724]) | 3\n",
-      "9279     | torch.Size([1, 1000, 762]) | 3\n",
-      "9280     | torch.Size([1, 1000, 779]) | 3\n",
-      "9281     | torch.Size([1, 1000, 777]) | 3\n",
-      "9282     | torch.Size([1, 1000, 754]) | 3\n",
-      "9283     | torch.Size([1, 1000, 717]) | 3\n",
-      "9284     | torch.Size([1, 1000, 767]) | 3\n",
-      "9285     | torch.Size([1, 1000, 866]) | 3\n",
-      "9286     | torch.Size([1, 1000, 786]) | 3\n",
-      "9287     | torch.Size([1, 1000, 762]) | 3\n",
-      "9288     | torch.Size([1, 1000, 768]) | 3\n",
-      "9289     | torch.Size([1, 1000, 754]) | 3\n",
-      "9290     | torch.Size([1, 1000, 789]) | 3\n",
-      "9291     | torch.Size([1, 1000, 773]) | 3\n",
-      "9292     | torch.Size([1, 1000, 713]) | 3\n",
-      "9293     | torch.Size([1, 1000, 754]) | 3\n",
-      "9294     | torch.Size([1, 1000, 802]) | 3\n",
-      "9295     | torch.Size([1, 1000, 866]) | 3\n",
-      "9296     | torch.Size([1, 1000, 794]) | 3\n",
-      "9297     | torch.Size([1, 1000, 781]) | 3\n",
-      "9298     | torch.Size([1, 1000, 754]) | 3\n",
-      "9299     | torch.Size([1, 1000, 754]) | 3\n",
-      "9300     | torch.Size([1, 1000, 786]) | 3\n",
-      "9301     | torch.Size([1, 1000, 754]) | 3\n",
-      "9302     | torch.Size([1, 1000, 762]) | 3\n",
-      "9303     | torch.Size([1, 1000, 777]) | 3\n",
-      "9304     | torch.Size([1, 1000, 754]) | 3\n",
-      "9305     | torch.Size([1, 1000, 774]) | 3\n",
-      "9306     | torch.Size([1, 1000, 754]) | 3\n",
-      "9307     | torch.Size([1, 1000, 754]) | 3\n",
-      "9308     | torch.Size([1, 1000, 772]) | 3\n",
-      "9309     | torch.Size([1, 1000, 754]) | 3\n",
-      "9310     | torch.Size([1, 1000, 779]) | 3\n",
-      "9311     | torch.Size([1, 1000, 778]) | 3\n",
-      "9312     | torch.Size([1, 1000, 777]) | 3\n",
-      "9313     | torch.Size([1, 1000, 754]) | 3\n",
-      "9314     | torch.Size([1, 1000, 789]) | 3\n",
-      "9315     | torch.Size([1, 1000, 799]) | 3\n",
-      "9316     | torch.Size([1, 1000, 754]) | 3\n",
-      "9317     | torch.Size([1, 1000, 795]) | 3\n",
-      "9318     | torch.Size([1, 1000, 754]) | 3\n",
-      "9319     | torch.Size([1, 1000, 843]) | 3\n",
-      "9320     | torch.Size([1, 1000, 777]) | 3\n",
-      "9321     | torch.Size([1, 1000, 724]) | 3\n",
-      "9322     | torch.Size([1, 1000, 787]) | 3\n",
-      "9323     | torch.Size([1, 1000, 847]) | 3\n",
-      "9324     | torch.Size([1, 1000, 764]) | 3\n",
-      "9325     | torch.Size([1, 1000, 754]) | 3\n",
-      "9326     | torch.Size([1, 1000, 811]) | 3\n",
-      "9327     | torch.Size([1, 1000, 778]) | 3\n",
-      "9328     | torch.Size([1, 1000, 754]) | 3\n",
-      "9329     | torch.Size([1, 1000, 777]) | 3\n",
-      "9330     | torch.Size([1, 1000, 789]) | 3\n",
-      "9331     | torch.Size([1, 1000, 778]) | 3\n",
-      "9332     | torch.Size([1, 1000, 754]) | 3\n",
-      "9333     | torch.Size([1, 1000, 784]) | 3\n",
-      "9334     | torch.Size([1, 1000, 762]) | 3\n",
-      "9335     | torch.Size([1, 1000, 762]) | 3\n",
-      "9336     | torch.Size([1, 1000, 754]) | 3\n",
-      "9337     | torch.Size([1, 1000, 848]) | 3\n",
-      "9338     | torch.Size([1, 1000, 777]) | 3\n",
-      "9339     | torch.Size([1, 1000, 774]) | 3\n",
-      "9340     | torch.Size([1, 1000, 777]) | 3\n",
-      "9341     | torch.Size([1, 1000, 784]) | 3\n",
-      "9342     | torch.Size([1, 1000, 781]) | 3\n",
-      "9343     | torch.Size([1, 1000, 708]) | 3\n",
-      "9344     | torch.Size([1, 1000, 810]) | 3\n",
-      "9345     | torch.Size([1, 1000, 777]) | 3\n",
-      "9346     | torch.Size([1, 1000, 779]) | 3\n",
-      "9347     | torch.Size([1, 1000, 762]) | 3\n",
-      "9348     | torch.Size([1, 1000, 797]) | 3\n",
-      "9349     | torch.Size([1, 1000, 762]) | 3\n",
-      "9350     | torch.Size([1, 1000, 777]) | 3\n",
-      "9351     | torch.Size([1, 1000, 754]) | 3\n",
-      "9352     | torch.Size([1, 1000, 754]) | 3\n",
-      "9353     | torch.Size([1, 1000, 777]) | 3\n",
-      "9354     | torch.Size([1, 1000, 754]) | 3\n",
-      "9355     | torch.Size([1, 1000, 785]) | 3\n",
-      "9356     | torch.Size([1, 1000, 780]) | 3\n",
-      "9357     | torch.Size([1, 1000, 762]) | 3\n",
-      "9358     | torch.Size([1, 1000, 754]) | 3\n",
-      "9359     | torch.Size([1, 1000, 777]) | 3\n",
-      "9360     | torch.Size([1, 1000, 762]) | 3\n",
-      "9361     | torch.Size([1, 1000, 754]) | 3\n",
-      "9362     | torch.Size([1, 1000, 777]) | 3\n",
-      "9363     | torch.Size([1, 1000, 754]) | 3\n",
-      "9364     | torch.Size([1, 1000, 727]) | 3\n",
-      "9365     | torch.Size([1, 1000, 762]) | 3\n",
-      "9366     | torch.Size([1, 1000, 754]) | 3\n",
-      "9367     | torch.Size([1, 1000, 779]) | 3\n",
-      "9368     | torch.Size([1, 1000, 777]) | 3\n",
-      "9369     | torch.Size([1, 1000, 782]) | 3\n",
-      "9370     | torch.Size([1, 1000, 777]) | 3\n",
-      "9371     | torch.Size([1, 1000, 772]) | 3\n",
-      "9372     | torch.Size([1, 1000, 762]) | 3\n",
-      "9373     | torch.Size([1, 1000, 762]) | 3\n",
-      "9374     | torch.Size([1, 1000, 762]) | 3\n",
-      "9375     | torch.Size([1, 1000, 754]) | 3\n",
-      "9376     | torch.Size([1, 1000, 777]) | 3\n",
-      "9377     | torch.Size([1, 1000, 754]) | 3\n",
-      "9378     | torch.Size([1, 1000, 754]) | 3\n",
-      "9379     | torch.Size([1, 1000, 785]) | 3\n",
-      "9380     | torch.Size([1, 1000, 777]) | 3\n",
-      "9381     | torch.Size([1, 1000, 754]) | 3\n",
-      "9382     | torch.Size([1, 1000, 754]) | 3\n",
-      "9383     | torch.Size([1, 1000, 805]) | 3\n",
-      "9384     | torch.Size([1, 1000, 781]) | 3\n",
-      "9385     | torch.Size([1, 1000, 754]) | 3\n",
-      "9386     | torch.Size([1, 1000, 778]) | 3\n",
-      "9387     | torch.Size([1, 1000, 754]) | 3\n",
-      "9388     | torch.Size([1, 1000, 754]) | 3\n",
-      "9389     | torch.Size([1, 1000, 754]) | 3\n",
-      "9390     | torch.Size([1, 1000, 754]) | 3\n",
-      "9391     | torch.Size([1, 1000, 777]) | 3\n",
-      "9392     | torch.Size([1, 1000, 762]) | 3\n",
-      "9393     | torch.Size([1, 1000, 745]) | 3\n",
-      "9394     | torch.Size([1, 1000, 770]) | 3\n",
-      "9395     | torch.Size([1, 1000, 767]) | 3\n",
-      "9396     | torch.Size([1, 1000, 726]) | 3\n",
-      "9397     | torch.Size([1, 1000, 784]) | 3\n",
-      "9398     | torch.Size([1, 1000, 778]) | 3\n",
-      "9399     | torch.Size([1, 1000, 777]) | 3\n",
-      "9400     | torch.Size([1, 1000, 793]) | 3\n",
-      "9401     | torch.Size([1, 1000, 729]) | 3\n",
-      "9402     | torch.Size([1, 1000, 780]) | 3\n",
-      "9403     | torch.Size([1, 1000, 754]) | 3\n",
-      "9404     | torch.Size([1, 1000, 762]) | 3\n",
-      "9405     | torch.Size([1, 1000, 754]) | 3\n",
-      "9406     | torch.Size([1, 1000, 754]) | 3\n",
-      "9407     | torch.Size([1, 1000, 754]) | 3\n",
-      "9408     | torch.Size([1, 1000, 777]) | 3\n",
-      "9409     | torch.Size([1, 1000, 783]) | 3\n",
-      "9410     | torch.Size([1, 1000, 777]) | 3\n",
-      "9411     | torch.Size([1, 1000, 754]) | 3\n",
-      "9412     | torch.Size([1, 1000, 782]) | 3\n",
-      "9413     | torch.Size([1, 1000, 781]) | 3\n",
-      "9414     | torch.Size([1, 1000, 777]) | 3\n",
-      "9415     | torch.Size([1, 1000, 754]) | 3\n",
-      "9416     | torch.Size([1, 1000, 767]) | 3\n",
-      "9417     | torch.Size([1, 1000, 754]) | 3\n",
-      "9418     | torch.Size([1, 1000, 780]) | 3\n",
-      "9419     | torch.Size([1, 1000, 754]) | 3\n",
-      "9420     | torch.Size([1, 1000, 762]) | 3\n",
-      "9421     | torch.Size([1, 1000, 777]) | 3\n",
-      "9422     | torch.Size([1, 1000, 754]) | 3\n",
-      "9423     | torch.Size([1, 1000, 777]) | 3\n",
-      "9424     | torch.Size([1, 1000, 777]) | 3\n",
-      "9425     | torch.Size([1, 1000, 762]) | 3\n",
-      "9426     | torch.Size([1, 1000, 762]) | 3\n",
-      "9427     | torch.Size([1, 1000, 762]) | 3\n",
-      "9428     | torch.Size([1, 1000, 762]) | 3\n",
-      "9429     | torch.Size([1, 1000, 778]) | 3\n",
-      "9430     | torch.Size([1, 1000, 754]) | 3\n",
-      "9431     | torch.Size([1, 1000, 777]) | 3\n",
-      "9432     | torch.Size([1, 1000, 828]) | 3\n",
-      "9433     | torch.Size([1, 1000, 848]) | 3\n",
-      "9434     | torch.Size([1, 1000, 786]) | 3\n",
-      "9435     | torch.Size([1, 1000, 789]) | 3\n",
-      "9436     | torch.Size([1, 1000, 772]) | 3\n",
-      "9437     | torch.Size([1, 1000, 796]) | 3\n",
-      "9438     | torch.Size([1, 1000, 754]) | 3\n",
-      "9439     | torch.Size([1, 1000, 797]) | 3\n",
-      "9440     | torch.Size([1, 1000, 754]) | 3\n",
-      "9441     | torch.Size([1, 1000, 762]) | 3\n",
-      "9442     | torch.Size([1, 1000, 762]) | 3\n",
-      "9443     | torch.Size([1, 1000, 784]) | 3\n",
-      "9444     | torch.Size([1, 1000, 786]) | 3\n",
-      "9445     | torch.Size([1, 1000, 777]) | 3\n",
-      "9446     | torch.Size([1, 1000, 754]) | 3\n",
-      "9447     | torch.Size([1, 1000, 754]) | 3\n",
-      "9448     | torch.Size([1, 1000, 777]) | 3\n",
-      "9449     | torch.Size([1, 1000, 754]) | 3\n",
-      "9450     | torch.Size([1, 1000, 825]) | 3\n",
-      "9451     | torch.Size([1, 1000, 762]) | 3\n",
-      "9452     | torch.Size([1, 1000, 754]) | 3\n",
-      "9453     | torch.Size([1, 1000, 762]) | 3\n",
-      "9454     | torch.Size([1, 1000, 754]) | 3\n",
-      "9455     | torch.Size([1, 1000, 811]) | 3\n",
-      "9456     | torch.Size([1, 1000, 785]) | 3\n",
-      "9457     | torch.Size([1, 1000, 762]) | 3\n",
-      "9458     | torch.Size([1, 1000, 762]) | 3\n",
-      "9459     | torch.Size([1, 1000, 754]) | 3\n",
-      "9460     | torch.Size([1, 1000, 788]) | 3\n",
-      "9461     | torch.Size([1, 1000, 754]) | 3\n",
-      "9462     | torch.Size([1, 1000, 754]) | 3\n",
-      "9463     | torch.Size([1, 1000, 754]) | 3\n",
-      "9464     | torch.Size([1, 1000, 754]) | 3\n",
-      "9465     | torch.Size([1, 1000, 754]) | 3\n",
-      "9466     | torch.Size([1, 1000, 754]) | 3\n",
-      "9467     | torch.Size([1, 1000, 775]) | 3\n",
-      "9468     | torch.Size([1, 1000, 788]) | 3\n",
-      "9469     | torch.Size([1, 1000, 754]) | 3\n",
-      "9470     | torch.Size([1, 1000, 762]) | 3\n",
-      "9471     | torch.Size([1, 1000, 780]) | 3\n",
-      "9472     | torch.Size([1, 1000, 754]) | 3\n",
-      "9473     | torch.Size([1, 1000, 773]) | 3\n",
-      "9474     | torch.Size([1, 1000, 783]) | 3\n",
-      "9475     | torch.Size([1, 1000, 802]) | 3\n",
-      "9476     | torch.Size([1, 1000, 762]) | 3\n",
-      "9477     | torch.Size([1, 1000, 802]) | 3\n",
-      "9478     | torch.Size([1, 1000, 754]) | 3\n",
-      "9479     | torch.Size([1, 1000, 754]) | 3\n",
-      "9480     | torch.Size([1, 1000, 754]) | 3\n",
-      "9481     | torch.Size([1, 1000, 770]) | 3\n",
-      "9482     | torch.Size([1, 1000, 767]) | 3\n",
-      "9483     | torch.Size([1, 1000, 773]) | 3\n",
-      "9484     | torch.Size([1, 1000, 762]) | 3\n",
-      "9485     | torch.Size([1, 1000, 754]) | 3\n",
-      "9486     | torch.Size([1, 1000, 754]) | 3\n",
-      "9487     | torch.Size([1, 1000, 777]) | 3\n",
-      "9488     | torch.Size([1, 1000, 777]) | 3\n",
-      "9489     | torch.Size([1, 1000, 754]) | 3\n",
-      "9490     | torch.Size([1, 1000, 806]) | 3\n",
-      "9491     | torch.Size([1, 1000, 754]) | 3\n",
-      "9492     | torch.Size([1, 1000, 754]) | 3\n",
-      "9493     | torch.Size([1, 1000, 777]) | 3\n",
-      "9494     | torch.Size([1, 1000, 754]) | 3\n",
-      "9495     | torch.Size([1, 1000, 754]) | 3\n",
-      "9496     | torch.Size([1, 1000, 753]) | 3\n",
-      "9497     | torch.Size([1, 1000, 762]) | 3\n",
-      "9498     | torch.Size([1, 1000, 786]) | 3\n",
-      "9499     | torch.Size([1, 1000, 754]) | 3\n",
-      "9500     | torch.Size([1, 1000, 777]) | 3\n",
-      "9501     | torch.Size([1, 1000, 754]) | 3\n",
-      "9502     | torch.Size([1, 1000, 787]) | 3\n",
-      "9503     | torch.Size([1, 1000, 777]) | 3\n",
-      "9504     | torch.Size([1, 1000, 771]) | 3\n",
-      "9505     | torch.Size([1, 1000, 802]) | 3\n",
-      "9506     | torch.Size([1, 1000, 810]) | 3\n",
-      "9507     | torch.Size([1, 1000, 762]) | 3\n",
-      "9508     | torch.Size([1, 1000, 786]) | 3\n",
-      "9509     | torch.Size([1, 1000, 779]) | 3\n",
-      "9510     | torch.Size([1, 1000, 754]) | 3\n",
-      "9511     | torch.Size([1, 1000, 798]) | 3\n",
-      "9512     | torch.Size([1, 1000, 777]) | 3\n",
-      "9513     | torch.Size([1, 1000, 777]) | 3\n",
-      "9514     | torch.Size([1, 1000, 773]) | 3\n",
-      "9515     | torch.Size([1, 1000, 762]) | 3\n",
-      "9516     | torch.Size([1, 1000, 762]) | 3\n",
-      "9517     | torch.Size([1, 1000, 754]) | 3\n",
-      "9518     | torch.Size([1, 1000, 811]) | 3\n",
-      "9519     | torch.Size([1, 1000, 754]) | 3\n",
-      "9520     | torch.Size([1, 1000, 754]) | 3\n",
-      "9521     | torch.Size([1, 1000, 798]) | 3\n",
-      "9522     | torch.Size([1, 1000, 803]) | 3\n",
-      "9523     | torch.Size([1, 1000, 754]) | 3\n",
-      "9524     | torch.Size([1, 1000, 797]) | 3\n",
-      "9525     | torch.Size([1, 1000, 754]) | 3\n",
-      "9526     | torch.Size([1, 1000, 762]) | 3\n",
-      "9527     | torch.Size([1, 1000, 762]) | 3\n",
-      "9528     | torch.Size([1, 1000, 754]) | 3\n",
-      "9529     | torch.Size([1, 1000, 754]) | 3\n",
-      "9530     | torch.Size([1, 1000, 777]) | 3\n",
-      "9531     | torch.Size([1, 1000, 754]) | 3\n",
-      "9532     | torch.Size([1, 1000, 754]) | 3\n",
-      "9533     | torch.Size([1, 1000, 762]) | 3\n",
-      "9534     | torch.Size([1, 1000, 762]) | 3\n",
-      "9535     | torch.Size([1, 1000, 780]) | 3\n",
-      "9536     | torch.Size([1, 1000, 754]) | 3\n",
-      "9537     | torch.Size([1, 1000, 754]) | 3\n",
-      "9538     | torch.Size([1, 1000, 773]) | 3\n",
-      "9539     | torch.Size([1, 1000, 754]) | 3\n",
-      "9540     | torch.Size([1, 1000, 714]) | 3\n",
-      "9541     | torch.Size([1, 1000, 754]) | 3\n",
-      "9542     | torch.Size([1, 1000, 754]) | 3\n",
-      "9543     | torch.Size([1, 1000, 754]) | 3\n",
-      "9544     | torch.Size([1, 1000, 762]) | 3\n",
-      "9545     | torch.Size([1, 1000, 754]) | 3\n",
-      "9546     | torch.Size([1, 1000, 754]) | 3\n",
-      "9547     | torch.Size([1, 1000, 771]) | 3\n",
-      "9548     | torch.Size([1, 1000, 763]) | 3\n",
-      "9549     | torch.Size([1, 1000, 777]) | 3\n",
-      "9550     | torch.Size([1, 1000, 762]) | 3\n",
-      "9551     | torch.Size([1, 1000, 754]) | 3\n",
-      "9552     | torch.Size([1, 1000, 754]) | 3\n",
-      "9553     | torch.Size([1, 1000, 772]) | 3\n",
-      "9554     | torch.Size([1, 1000, 777]) | 3\n",
-      "9555     | torch.Size([1, 1000, 754]) | 3\n",
-      "9556     | torch.Size([1, 1000, 754]) | 3\n",
-      "9557     | torch.Size([1, 1000, 782]) | 3\n",
-      "9558     | torch.Size([1, 1000, 762]) | 3\n",
-      "9559     | torch.Size([1, 1000, 765]) | 3\n",
-      "9560     | torch.Size([1, 1000, 757]) | 3\n",
-      "9561     | torch.Size([1, 1000, 754]) | 3\n",
-      "9562     | torch.Size([1, 1000, 762]) | 3\n",
-      "9563     | torch.Size([1, 1000, 713]) | 3\n",
-      "9564     | torch.Size([1, 1000, 762]) | 3\n",
-      "9565     | torch.Size([1, 1000, 754]) | 3\n",
-      "9566     | torch.Size([1, 1000, 762]) | 3\n",
-      "9567     | torch.Size([1, 1000, 762]) | 3\n",
-      "9568     | torch.Size([1, 1000, 754]) | 3\n",
-      "9569     | torch.Size([1, 1000, 762]) | 3\n",
-      "9570     | torch.Size([1, 1000, 794]) | 3\n",
-      "9571     | torch.Size([1, 1000, 754]) | 3\n",
-      "9572     | torch.Size([1, 1000, 773]) | 3\n",
-      "9573     | torch.Size([1, 1000, 777]) | 3\n",
-      "9574     | torch.Size([1, 1000, 754]) | 3\n",
-      "9575     | torch.Size([1, 1000, 771]) | 3\n",
-      "9576     | torch.Size([1, 1000, 777]) | 3\n",
-      "9577     | torch.Size([1, 1000, 780]) | 3\n",
-      "9578     | torch.Size([1, 1000, 754]) | 3\n",
-      "9579     | torch.Size([1, 1000, 772]) | 3\n",
-      "9580     | torch.Size([1, 1000, 754]) | 3\n",
-      "9581     | torch.Size([1, 1000, 754]) | 3\n",
-      "9582     | torch.Size([1, 1000, 724]) | 3\n",
-      "9583     | torch.Size([1, 1000, 754]) | 3\n",
-      "9584     | torch.Size([1, 1000, 754]) | 3\n",
-      "9585     | torch.Size([1, 1000, 754]) | 3\n",
-      "9586     | torch.Size([1, 1000, 762]) | 3\n",
-      "9587     | torch.Size([1, 1000, 754]) | 3\n",
-      "9588     | torch.Size([1, 1000, 772]) | 3\n",
-      "9589     | torch.Size([1, 1000, 754]) | 3\n",
-      "9590     | torch.Size([1, 1000, 754]) | 3\n",
-      "9591     | torch.Size([1, 1000, 762]) | 3\n",
-      "9592     | torch.Size([1, 1000, 754]) | 3\n",
-      "9593     | torch.Size([1, 1000, 762]) | 3\n",
-      "9594     | torch.Size([1, 1000, 781]) | 3\n",
-      "9595     | torch.Size([1, 1000, 754]) | 3\n",
-      "9596     | torch.Size([1, 1000, 754]) | 3\n",
-      "9597     | torch.Size([1, 1000, 777]) | 3\n",
-      "9598     | torch.Size([1, 1000, 754]) | 3\n",
-      "9599     | torch.Size([1, 1000, 754]) | 3\n",
-      "9600     | torch.Size([1, 1000, 777]) | 3\n",
-      "9601     | torch.Size([1, 1000, 777]) | 3\n",
-      "9602     | torch.Size([1, 1000, 754]) | 3\n",
-      "9603     | torch.Size([1, 1000, 762]) | 3\n",
-      "9604     | torch.Size([1, 1000, 754]) | 3\n",
-      "9605     | torch.Size([1, 1000, 762]) | 3\n",
-      "9606     | torch.Size([1, 1000, 762]) | 3\n",
-      "9607     | torch.Size([1, 1000, 777]) | 3\n",
-      "9608     | torch.Size([1, 1000, 762]) | 3\n",
-      "9609     | torch.Size([1, 1000, 777]) | 3\n",
-      "9610     | torch.Size([1, 1000, 777]) | 3\n",
-      "9611     | torch.Size([1, 1000, 847]) | 3\n",
-      "9612     | torch.Size([1, 1000, 762]) | 3\n",
-      "9613     | torch.Size([1, 1000, 777]) | 3\n",
-      "9614     | torch.Size([1, 1000, 754]) | 3\n",
-      "9615     | torch.Size([1, 1000, 777]) | 3\n",
-      "9616     | torch.Size([1, 1000, 754]) | 3\n",
-      "9617     | torch.Size([1, 1000, 794]) | 3\n",
-      "9618     | torch.Size([1, 1000, 754]) | 3\n",
-      "9619     | torch.Size([1, 1000, 754]) | 3\n",
-      "9620     | torch.Size([1, 1000, 777]) | 3\n",
-      "9621     | torch.Size([1, 1000, 777]) | 3\n",
-      "9622     | torch.Size([1, 1000, 762]) | 3\n",
-      "9623     | torch.Size([1, 1000, 754]) | 3\n",
-      "9624     | torch.Size([1, 1000, 777]) | 3\n",
-      "9625     | torch.Size([1, 1000, 762]) | 3\n",
-      "9626     | torch.Size([1, 1000, 779]) | 3\n",
-      "9627     | torch.Size([1, 1000, 777]) | 3\n",
-      "9628     | torch.Size([1, 1000, 754]) | 3\n",
-      "9629     | torch.Size([1, 1000, 754]) | 3\n",
-      "9630     | torch.Size([1, 1000, 809]) | 3\n",
-      "9631     | torch.Size([1, 1000, 789]) | 3\n",
-      "9632     | torch.Size([1, 1000, 754]) | 3\n",
-      "9633     | torch.Size([1, 1000, 754]) | 3\n",
-      "9634     | torch.Size([1, 1000, 762]) | 3\n",
-      "9635     | torch.Size([1, 1000, 754]) | 3\n",
-      "9636     | torch.Size([1, 1000, 754]) | 3\n",
-      "9637     | torch.Size([1, 1000, 777]) | 3\n",
-      "9638     | torch.Size([1, 1000, 777]) | 3\n",
-      "9639     | torch.Size([1, 1000, 773]) | 3\n",
-      "9640     | torch.Size([1, 1000, 754]) | 3\n",
-      "9641     | torch.Size([1, 1000, 777]) | 3\n",
-      "9642     | torch.Size([1, 1000, 809]) | 3\n",
-      "9643     | torch.Size([1, 1000, 778]) | 3\n",
-      "9644     | torch.Size([1, 1000, 779]) | 3\n",
-      "9645     | torch.Size([1, 1000, 754]) | 3\n",
-      "9646     | torch.Size([1, 1000, 754]) | 3\n",
-      "9647     | torch.Size([1, 1000, 754]) | 3\n",
-      "9648     | torch.Size([1, 1000, 779]) | 3\n",
-      "9649     | torch.Size([1, 1000, 762]) | 3\n",
-      "9650     | torch.Size([1, 1000, 762]) | 3\n",
-      "9651     | torch.Size([1, 1000, 777]) | 3\n",
-      "9652     | torch.Size([1, 1000, 772]) | 3\n",
-      "9653     | torch.Size([1, 1000, 754]) | 3\n",
-      "9654     | torch.Size([1, 1000, 777]) | 3\n",
-      "9655     | torch.Size([1, 1000, 788]) | 3\n",
-      "9656     | torch.Size([1, 1000, 778]) | 3\n",
-      "9657     | torch.Size([1, 1000, 754]) | 3\n",
-      "9658     | torch.Size([1, 1000, 720]) | 3\n",
-      "9659     | torch.Size([1, 1000, 762]) | 3\n",
-      "9660     | torch.Size([1, 1000, 754]) | 3\n",
-      "9661     | torch.Size([1, 1000, 762]) | 3\n",
-      "9662     | torch.Size([1, 1000, 777]) | 3\n",
-      "9663     | torch.Size([1, 1000, 720]) | 3\n",
-      "9664     | torch.Size([1, 1000, 750]) | 3\n",
-      "9665     | torch.Size([1, 1000, 777]) | 3\n",
-      "9666     | torch.Size([1, 1000, 762]) | 3\n",
-      "9667     | torch.Size([1, 1000, 777]) | 3\n",
-      "9668     | torch.Size([1, 1000, 754]) | 3\n",
-      "9669     | torch.Size([1, 1000, 781]) | 3\n",
-      "9670     | torch.Size([1, 1000, 773]) | 3\n",
-      "9671     | torch.Size([1, 1000, 807]) | 3\n",
-      "9672     | torch.Size([1, 1000, 772]) | 3\n",
-      "9673     | torch.Size([1, 1000, 802]) | 3\n",
-      "9674     | torch.Size([1, 1000, 762]) | 3\n",
-      "9675     | torch.Size([1, 1000, 754]) | 3\n",
-      "9676     | torch.Size([1, 1000, 759]) | 3\n",
-      "9677     | torch.Size([1, 1000, 754]) | 3\n",
-      "9678     | torch.Size([1, 1000, 754]) | 3\n",
-      "9679     | torch.Size([1, 1000, 809]) | 3\n",
-      "9680     | torch.Size([1, 1000, 762]) | 3\n",
-      "9681     | torch.Size([1, 1000, 808]) | 3\n",
-      "9682     | torch.Size([1, 1000, 760]) | 3\n",
-      "9683     | torch.Size([1, 1000, 789]) | 3\n",
-      "9684     | torch.Size([1, 1000, 783]) | 3\n",
-      "9685     | torch.Size([1, 1000, 754]) | 3\n",
-      "9686     | torch.Size([1, 1000, 754]) | 3\n",
-      "9687     | torch.Size([1, 1000, 768]) | 3\n",
-      "9688     | torch.Size([1, 1000, 726]) | 3\n",
-      "9689     | torch.Size([1, 1000, 777]) | 3\n",
-      "9690     | torch.Size([1, 1000, 793]) | 3\n",
-      "9691     | torch.Size([1, 1000, 777]) | 3\n",
-      "9692     | torch.Size([1, 1000, 773]) | 3\n",
-      "9693     | torch.Size([1, 1000, 764]) | 3\n",
-      "9694     | torch.Size([1, 1000, 798]) | 3\n",
-      "9695     | torch.Size([1, 1000, 796]) | 3\n",
-      "9696     | torch.Size([1, 1000, 812]) | 3\n",
-      "9697     | torch.Size([1, 1000, 754]) | 3\n",
-      "9698     | torch.Size([1, 1000, 802]) | 3\n",
-      "9699     | torch.Size([1, 1000, 754]) | 3\n",
-      "9700     | torch.Size([1, 1000, 771]) | 3\n",
-      "9701     | torch.Size([1, 1000, 762]) | 3\n",
-      "9702     | torch.Size([1, 1000, 751]) | 3\n",
-      "9703     | torch.Size([1, 1000, 772]) | 3\n",
-      "9704     | torch.Size([1, 1000, 782]) | 3\n",
-      "9705     | torch.Size([1, 1000, 762]) | 3\n",
-      "9706     | torch.Size([1, 1000, 779]) | 3\n",
-      "9707     | torch.Size([1, 1000, 762]) | 3\n",
-      "9708     | torch.Size([1, 1000, 754]) | 3\n",
-      "9709     | torch.Size([1, 1000, 779]) | 3\n",
-      "9710     | torch.Size([1, 1000, 779]) | 3\n",
-      "9711     | torch.Size([1, 1000, 754]) | 3\n",
-      "9712     | torch.Size([1, 1000, 777]) | 3\n",
-      "9713     | torch.Size([1, 1000, 754]) | 3\n",
-      "9714     | torch.Size([1, 1000, 777]) | 3\n",
-      "9715     | torch.Size([1, 1000, 777]) | 3\n",
-      "9716     | torch.Size([1, 1000, 754]) | 3\n",
-      "9717     | torch.Size([1, 1000, 829]) | 3\n",
-      "9718     | torch.Size([1, 1000, 762]) | 3\n",
-      "9719     | torch.Size([1, 1000, 782]) | 3\n",
-      "9720     | torch.Size([1, 1000, 774]) | 3\n",
-      "9721     | torch.Size([1, 1000, 754]) | 3\n",
-      "9722     | torch.Size([1, 1000, 777]) | 3\n",
-      "9723     | torch.Size([1, 1000, 732]) | 3\n",
-      "9724     | torch.Size([1, 1000, 772]) | 3\n",
-      "9725     | torch.Size([1, 1000, 754]) | 3\n",
-      "9726     | torch.Size([1, 1000, 762]) | 3\n",
-      "9727     | torch.Size([1, 1000, 754]) | 3\n",
-      "9728     | torch.Size([1, 1000, 729]) | 3\n",
-      "9729     | torch.Size([1, 1000, 810]) | 3\n",
-      "9730     | torch.Size([1, 1000, 754]) | 3\n",
-      "9731     | torch.Size([1, 1000, 754]) | 3\n",
-      "9732     | torch.Size([1, 1000, 762]) | 3\n",
-      "9733     | torch.Size([1, 1000, 783]) | 3\n",
-      "9734     | torch.Size([1, 1000, 777]) | 3\n",
-      "9735     | torch.Size([1, 1000, 754]) | 3\n",
-      "9736     | torch.Size([1, 1000, 713]) | 3\n",
-      "9737     | torch.Size([1, 1000, 762]) | 3\n",
-      "9738     | torch.Size([1, 1000, 754]) | 3\n",
-      "9739     | torch.Size([1, 1000, 754]) | 3\n",
-      "9740     | torch.Size([1, 1000, 726]) | 3\n",
-      "9741     | torch.Size([1, 1000, 762]) | 3\n",
-      "9742     | torch.Size([1, 1000, 754]) | 3\n",
-      "9743     | torch.Size([1, 1000, 762]) | 3\n",
-      "9744     | torch.Size([1, 1000, 786]) | 3\n",
-      "9745     | torch.Size([1, 1000, 804]) | 3\n",
-      "9746     | torch.Size([1, 1000, 772]) | 3\n",
-      "9747     | torch.Size([1, 1000, 754]) | 3\n",
-      "9748     | torch.Size([1, 1000, 802]) | 3\n",
-      "9749     | torch.Size([1, 1000, 769]) | 3\n",
-      "9750     | torch.Size([1, 1000, 731]) | 3\n",
-      "9751     | torch.Size([1, 1000, 716]) | 3\n",
-      "9752     | torch.Size([1, 1000, 754]) | 3\n",
-      "9753     | torch.Size([1, 1000, 754]) | 3\n",
-      "9754     | torch.Size([1, 1000, 779]) | 3\n",
-      "9755     | torch.Size([1, 1000, 754]) | 3\n",
-      "9756     | torch.Size([1, 1000, 754]) | 3\n",
-      "9757     | torch.Size([1, 1000, 802]) | 3\n",
-      "9758     | torch.Size([1, 1000, 754]) | 3\n",
-      "9759     | torch.Size([1, 1000, 754]) | 3\n",
-      "9760     | torch.Size([1, 1000, 767]) | 3\n",
-      "9761     | torch.Size([1, 1000, 846]) | 3\n",
-      "9762     | torch.Size([1, 1000, 754]) | 3\n",
-      "9763     | torch.Size([1, 1000, 754]) | 3\n",
-      "9764     | torch.Size([1, 1000, 791]) | 3\n",
-      "9765     | torch.Size([1, 1000, 809]) | 3\n",
-      "9766     | torch.Size([1, 1000, 805]) | 3\n",
-      "9767     | torch.Size([1, 1000, 762]) | 3\n",
-      "9768     | torch.Size([1, 1000, 801]) | 3\n",
-      "9769     | torch.Size([1, 1000, 785]) | 3\n",
-      "9770     | torch.Size([1, 1000, 762]) | 3\n",
-      "9771     | torch.Size([1, 1000, 799]) | 3\n",
-      "9772     | torch.Size([1, 1000, 767]) | 3\n",
-      "9773     | torch.Size([1, 1000, 777]) | 3\n",
-      "9774     | torch.Size([1, 1000, 754]) | 3\n",
-      "9775     | torch.Size([1, 1000, 754]) | 3\n",
-      "9776     | torch.Size([1, 1000, 732]) | 3\n",
-      "9777     | torch.Size([1, 1000, 762]) | 3\n",
-      "9778     | torch.Size([1, 1000, 754]) | 3\n",
-      "9779     | torch.Size([1, 1000, 762]) | 3\n",
-      "9780     | torch.Size([1, 1000, 754]) | 3\n",
-      "9781     | torch.Size([1, 1000, 754]) | 3\n",
-      "9782     | torch.Size([1, 1000, 762]) | 3\n",
-      "9783     | torch.Size([1, 1000, 765]) | 3\n",
-      "9784     | torch.Size([1, 1000, 777]) | 3\n",
-      "9785     | torch.Size([1, 1000, 762]) | 3\n",
-      "9786     | torch.Size([1, 1000, 762]) | 3\n",
-      "9787     | torch.Size([1, 1000, 762]) | 3\n",
-      "9788     | torch.Size([1, 1000, 754]) | 3\n",
-      "9789     | torch.Size([1, 1000, 804]) | 3\n",
-      "9790     | torch.Size([1, 1000, 782]) | 3\n",
-      "9791     | torch.Size([1, 1000, 754]) | 3\n",
-      "9792     | torch.Size([1, 1000, 762]) | 3\n",
-      "9793     | torch.Size([1, 1000, 776]) | 3\n",
-      "9794     | torch.Size([1, 1000, 784]) | 3\n",
-      "9795     | torch.Size([1, 1000, 754]) | 3\n",
-      "9796     | torch.Size([1, 1000, 762]) | 3\n",
-      "9797     | torch.Size([1, 1000, 729]) | 3\n",
-      "9798     | torch.Size([1, 1000, 777]) | 3\n",
-      "9799     | torch.Size([1, 1000, 754]) | 3\n",
-      "9800     | torch.Size([1, 1000, 754]) | 3\n",
-      "9801     | torch.Size([1, 1000, 754]) | 3\n",
-      "9802     | torch.Size([1, 1000, 779]) | 3\n",
-      "9803     | torch.Size([1, 1000, 754]) | 3\n",
-      "9804     | torch.Size([1, 1000, 754]) | 3\n",
-      "9805     | torch.Size([1, 1000, 762]) | 3\n",
-      "9806     | torch.Size([1, 1000, 780]) | 3\n",
-      "9807     | torch.Size([1, 1000, 762]) | 3\n",
-      "9808     | torch.Size([1, 1000, 754]) | 3\n",
-      "9809     | torch.Size([1, 1000, 762]) | 3\n",
-      "9810     | torch.Size([1, 1000, 777]) | 3\n",
-      "9811     | torch.Size([1, 1000, 754]) | 3\n",
-      "9812     | torch.Size([1, 1000, 772]) | 3\n",
-      "9813     | torch.Size([1, 1000, 754]) | 3\n",
-      "9814     | torch.Size([1, 1000, 754]) | 3\n",
-      "9815     | torch.Size([1, 1000, 777]) | 3\n",
-      "9816     | torch.Size([1, 1000, 754]) | 3\n",
-      "9817     | torch.Size([1, 1000, 813]) | 3\n",
-      "9818     | torch.Size([1, 1000, 730]) | 3\n",
-      "9819     | torch.Size([1, 1000, 771]) | 3\n",
-      "9820     | torch.Size([1, 1000, 762]) | 3\n",
-      "9821     | torch.Size([1, 1000, 754]) | 3\n",
-      "9822     | torch.Size([1, 1000, 762]) | 3\n",
-      "9823     | torch.Size([1, 1000, 754]) | 3\n",
-      "9824     | torch.Size([1, 1000, 790]) | 3\n",
-      "9825     | torch.Size([1, 1000, 754]) | 3\n",
-      "9826     | torch.Size([1, 1000, 754]) | 3\n",
-      "9827     | torch.Size([1, 1000, 788]) | 3\n",
-      "9828     | torch.Size([1, 1000, 811]) | 3\n",
-      "9829     | torch.Size([1, 1000, 762]) | 3\n",
-      "9830     | torch.Size([1, 1000, 771]) | 3\n",
-      "9831     | torch.Size([1, 1000, 754]) | 3\n",
-      "9832     | torch.Size([1, 1000, 762]) | 3\n",
-      "9833     | torch.Size([1, 1000, 762]) | 3\n",
-      "9834     | torch.Size([1, 1000, 777]) | 3\n",
-      "9835     | torch.Size([1, 1000, 762]) | 3\n",
-      "9836     | torch.Size([1, 1000, 789]) | 3\n",
-      "9837     | torch.Size([1, 1000, 778]) | 3\n",
-      "9838     | torch.Size([1, 1000, 777]) | 3\n",
-      "9839     | torch.Size([1, 1000, 725]) | 3\n",
-      "9840     | torch.Size([1, 1000, 754]) | 3\n",
-      "9841     | torch.Size([1, 1000, 777]) | 3\n",
-      "9842     | torch.Size([1, 1000, 789]) | 3\n",
-      "9843     | torch.Size([1, 1000, 754]) | 3\n",
-      "9844     | torch.Size([1, 1000, 754]) | 3\n",
-      "9845     | torch.Size([1, 1000, 754]) | 3\n",
-      "9846     | torch.Size([1, 1000, 771]) | 3\n",
-      "9847     | torch.Size([1, 1000, 775]) | 3\n",
-      "9848     | torch.Size([1, 1000, 777]) | 3\n",
-      "9849     | torch.Size([1, 1000, 783]) | 3\n",
-      "9850     | torch.Size([1, 1000, 754]) | 3\n",
-      "9851     | torch.Size([1, 1000, 777]) | 3\n",
-      "9852     | torch.Size([1, 1000, 762]) | 3\n",
-      "9853     | torch.Size([1, 1000, 775]) | 3\n",
-      "9854     | torch.Size([1, 1000, 754]) | 3\n",
-      "9855     | torch.Size([1, 1000, 772]) | 3\n",
-      "9856     | torch.Size([1, 1000, 754]) | 3\n",
-      "9857     | torch.Size([1, 1000, 782]) | 3\n",
-      "9858     | torch.Size([1, 1000, 777]) | 3\n",
-      "9859     | torch.Size([1, 1000, 762]) | 3\n",
-      "9860     | torch.Size([1, 1000, 776]) | 3\n",
-      "9861     | torch.Size([1, 1000, 808]) | 3\n",
-      "9862     | torch.Size([1, 1000, 754]) | 3\n",
-      "9863     | torch.Size([1, 1000, 787]) | 3\n",
-      "9864     | torch.Size([1, 1000, 762]) | 3\n",
-      "9865     | torch.Size([1, 1000, 754]) | 3\n",
-      "9866     | torch.Size([1, 1000, 784]) | 3\n",
-      "9867     | torch.Size([1, 1000, 786]) | 3\n",
-      "9868     | torch.Size([1, 1000, 754]) | 3\n",
-      "9869     | torch.Size([1, 1000, 754]) | 3\n",
-      "9870     | torch.Size([1, 1000, 754]) | 3\n",
-      "9871     | torch.Size([1, 1000, 762]) | 3\n",
-      "9872     | torch.Size([1, 1000, 754]) | 3\n",
-      "9873     | torch.Size([1, 1000, 754]) | 3\n",
-      "9874     | torch.Size([1, 1000, 754]) | 3\n",
-      "9875     | torch.Size([1, 1000, 778]) | 3\n",
-      "9876     | torch.Size([1, 1000, 777]) | 3\n",
-      "9877     | torch.Size([1, 1000, 754]) | 3\n",
-      "9878     | torch.Size([1, 1000, 762]) | 3\n",
-      "9879     | torch.Size([1, 1000, 779]) | 3\n",
-      "9880     | torch.Size([1, 1000, 769]) | 3\n",
-      "9881     | torch.Size([1, 1000, 754]) | 3\n",
-      "9882     | torch.Size([1, 1000, 791]) | 3\n",
-      "9883     | torch.Size([1, 1000, 754]) | 3\n",
-      "9884     | torch.Size([1, 1000, 720]) | 3\n",
-      "9885     | torch.Size([1, 1000, 754]) | 3\n",
-      "9886     | torch.Size([1, 1000, 802]) | 3\n",
-      "9887     | torch.Size([1, 1000, 785]) | 3\n",
-      "9888     | torch.Size([1, 1000, 762]) | 3\n",
-      "9889     | torch.Size([1, 1000, 786]) | 3\n",
-      "9890     | torch.Size([1, 1000, 777]) | 3\n",
-      "9891     | torch.Size([1, 1000, 721]) | 3\n",
-      "9892     | torch.Size([1, 1000, 754]) | 3\n",
-      "9893     | torch.Size([1, 1000, 754]) | 3\n",
-      "9894     | torch.Size([1, 1000, 773]) | 3\n",
-      "9895     | torch.Size([1, 1000, 762]) | 3\n",
-      "9896     | torch.Size([1, 1000, 754]) | 3\n",
-      "9897     | torch.Size([1, 1000, 798]) | 3\n",
-      "9898     | torch.Size([1, 1000, 754]) | 3\n",
-      "9899     | torch.Size([1, 1000, 777]) | 3\n",
-      "9900     | torch.Size([1, 1000, 770]) | 3\n",
-      "9901     | torch.Size([1, 1000, 754]) | 3\n",
-      "9902     | torch.Size([1, 1000, 754]) | 3\n",
-      "9903     | torch.Size([1, 1000, 721]) | 3\n",
-      "9904     | torch.Size([1, 1000, 778]) | 3\n",
-      "9905     | torch.Size([1, 1000, 754]) | 3\n",
-      "9906     | torch.Size([1, 1000, 778]) | 3\n",
-      "9907     | torch.Size([1, 1000, 788]) | 3\n",
-      "9908     | torch.Size([1, 1000, 762]) | 3\n",
-      "9909     | torch.Size([1, 1000, 762]) | 3\n",
-      "9910     | torch.Size([1, 1000, 777]) | 3\n",
-      "9911     | torch.Size([1, 1000, 762]) | 3\n",
-      "9912     | torch.Size([1, 1000, 808]) | 3\n",
-      "9913     | torch.Size([1, 1000, 754]) | 3\n",
-      "9914     | torch.Size([1, 1000, 754]) | 3\n",
-      "9915     | torch.Size([1, 1000, 854]) | 3\n",
-      "9916     | torch.Size([1, 1000, 762]) | 3\n",
-      "9917     | torch.Size([1, 1000, 754]) | 3\n",
-      "9918     | torch.Size([1, 1000, 820]) | 3\n",
-      "9919     | torch.Size([1, 1000, 781]) | 3\n",
-      "9920     | torch.Size([1, 1000, 733]) | 3\n",
-      "9921     | torch.Size([1, 1000, 754]) | 3\n",
-      "9922     | torch.Size([1, 1000, 777]) | 3\n",
-      "9923     | torch.Size([1, 1000, 762]) | 3\n",
-      "9924     | torch.Size([1, 1000, 779]) | 3\n",
-      "9925     | torch.Size([1, 1000, 763]) | 3\n",
-      "9926     | torch.Size([1, 1000, 754]) | 3\n",
-      "9927     | torch.Size([1, 1000, 754]) | 3\n",
-      "9928     | torch.Size([1, 1000, 777]) | 3\n",
-      "9929     | torch.Size([1, 1000, 786]) | 3\n",
-      "9930     | torch.Size([1, 1000, 754]) | 3\n",
-      "9931     | torch.Size([1, 1000, 778]) | 3\n",
-      "9932     | torch.Size([1, 1000, 762]) | 3\n",
-      "9933     | torch.Size([1, 1000, 754]) | 3\n",
-      "9934     | torch.Size([1, 1000, 762]) | 3\n",
-      "9935     | torch.Size([1, 1000, 802]) | 3\n",
-      "9936     | torch.Size([1, 1000, 762]) | 3\n",
-      "9937     | torch.Size([1, 1000, 777]) | 3\n",
-      "9938     | torch.Size([1, 1000, 809]) | 3\n",
-      "9939     | torch.Size([1, 1000, 762]) | 3\n",
-      "9940     | torch.Size([1, 1000, 774]) | 3\n",
-      "9941     | torch.Size([1, 1000, 754]) | 3\n",
-      "9942     | torch.Size([1, 1000, 795]) | 3\n",
-      "9943     | torch.Size([1, 1000, 762]) | 3\n",
-      "9944     | torch.Size([1, 1000, 777]) | 3\n",
-      "9945     | torch.Size([1, 1000, 777]) | 3\n",
-      "9946     | torch.Size([1, 1000, 777]) | 3\n",
-      "9947     | torch.Size([1, 1000, 782]) | 3\n",
-      "9948     | torch.Size([1, 1000, 782]) | 3\n",
-      "9949     | torch.Size([1, 1000, 777]) | 3\n",
-      "9950     | torch.Size([1, 1000, 762]) | 3\n",
-      "9951     | torch.Size([1, 1000, 754]) | 3\n",
-      "9952     | torch.Size([1, 1000, 787]) | 3\n",
-      "9953     | torch.Size([1, 1000, 772]) | 3\n",
-      "9954     | torch.Size([1, 1000, 754]) | 3\n",
-      "9955     | torch.Size([1, 1000, 754]) | 3\n",
-      "9956     | torch.Size([1, 1000, 754]) | 3\n",
-      "9957     | torch.Size([1, 1000, 777]) | 3\n",
-      "9958     | torch.Size([1, 1000, 754]) | 3\n",
-      "9959     | torch.Size([1, 1000, 802]) | 3\n",
-      "9960     | torch.Size([1, 1000, 754]) | 3\n",
-      "9961     | torch.Size([1, 1000, 762]) | 3\n",
-      "9962     | torch.Size([1, 1000, 754]) | 3\n",
-      "9963     | torch.Size([1, 1000, 777]) | 3\n",
-      "9964     | torch.Size([1, 1000, 754]) | 3\n",
-      "9965     | torch.Size([1, 1000, 762]) | 3\n",
-      "9966     | torch.Size([1, 1000, 754]) | 3\n",
-      "9967     | torch.Size([1, 1000, 751]) | 3\n",
-      "9968     | torch.Size([1, 1000, 754]) | 3\n",
-      "9969     | torch.Size([1, 1000, 762]) | 3\n",
-      "9970     | torch.Size([1, 1000, 754]) | 3\n",
-      "9971     | torch.Size([1, 1000, 778]) | 3\n",
-      "9972     | torch.Size([1, 1000, 768]) | 3\n",
-      "9973     | torch.Size([1, 1000, 762]) | 3\n",
-      "9974     | torch.Size([1, 1000, 754]) | 3\n",
-      "9975     | torch.Size([1, 1000, 762]) | 3\n",
-      "9976     | torch.Size([1, 1000, 777]) | 3\n",
-      "9977     | torch.Size([1, 1000, 796]) | 3\n",
-      "9978     | torch.Size([1, 1000, 754]) | 3\n",
-      "9979     | torch.Size([1, 1000, 762]) | 3\n",
-      "9980     | torch.Size([1, 1000, 762]) | 3\n",
-      "9981     | torch.Size([1, 1000, 792]) | 3\n",
-      "9982     | torch.Size([1, 1000, 813]) | 3\n",
-      "9983     | torch.Size([1, 1000, 713]) | 3\n",
-      "9984     | torch.Size([1, 1000, 777]) | 3\n",
-      "9985     | torch.Size([1, 1000, 777]) | 3\n",
-      "9986     | torch.Size([1, 1000, 802]) | 3\n",
-      "9987     | torch.Size([1, 1000, 754]) | 3\n",
-      "9988     | torch.Size([1, 1000, 765]) | 3\n",
-      "9989     | torch.Size([1, 1000, 777]) | 3\n",
-      "9990     | torch.Size([1, 1000, 794]) | 3\n",
-      "9991     | torch.Size([1, 1000, 762]) | 3\n",
-      "9992     | torch.Size([1, 1000, 762]) | 3\n",
-      "9993     | torch.Size([1, 1000, 754]) | 3\n",
-      "9994     | torch.Size([1, 1000, 772]) | 3\n",
-      "9995     | torch.Size([1, 1000, 802]) | 3\n",
-      "9996     | torch.Size([1, 1000, 777]) | 3\n",
-      "9997     | torch.Size([1, 1000, 777]) | 3\n",
-      "9998     | torch.Size([1, 1000, 762]) | 3\n",
-      "9999     | torch.Size([1, 1000, 713]) | 3\n",
-      "10000    | torch.Size([1, 1000, 754]) | 3\n",
-      "10001    | torch.Size([1, 1000, 773]) | 3\n",
-      "10002    | torch.Size([1, 1000, 762]) | 3\n",
-      "10003    | torch.Size([1, 1000, 777]) | 3\n",
-      "10004    | torch.Size([1, 1000, 762]) | 3\n",
-      "10005    | torch.Size([1, 1000, 782]) | 3\n",
-      "10006    | torch.Size([1, 1000, 777]) | 3\n",
-      "10007    | torch.Size([1, 1000, 761]) | 3\n",
-      "10008    | torch.Size([1, 1000, 762]) | 3\n",
-      "10009    | torch.Size([1, 1000, 754]) | 3\n",
-      "10010    | torch.Size([1, 1000, 762]) | 3\n",
-      "10011    | torch.Size([1, 1000, 762]) | 3\n",
-      "10012    | torch.Size([1, 1000, 754]) | 3\n",
-      "10013    | torch.Size([1, 1000, 754]) | 3\n",
-      "10014    | torch.Size([1, 1000, 627]) | 3\n",
-      "10015    | torch.Size([1, 1000, 754]) | 3\n",
-      "10016    | torch.Size([1, 1000, 754]) | 3\n",
-      "10017    | torch.Size([1, 1000, 754]) | 3\n",
-      "10018    | torch.Size([1, 1000, 713]) | 3\n",
-      "10019    | torch.Size([1, 1000, 806]) | 3\n",
-      "10020    | torch.Size([1, 1000, 754]) | 3\n",
-      "10021    | torch.Size([1, 1000, 754]) | 3\n",
-      "10022    | torch.Size([1, 1000, 802]) | 3\n",
-      "10023    | torch.Size([1, 1000, 754]) | 3\n",
-      "10024    | torch.Size([1, 1000, 778]) | 3\n",
-      "10025    | torch.Size([1, 1000, 777]) | 3\n",
-      "10026    | torch.Size([1, 1000, 777]) | 3\n",
-      "10027    | torch.Size([1, 1000, 762]) | 3\n",
-      "10028    | torch.Size([1, 1000, 797]) | 3\n",
-      "10029    | torch.Size([1, 1000, 762]) | 3\n",
-      "10030    | torch.Size([1, 1000, 780]) | 3\n",
-      "10031    | torch.Size([1, 1000, 811]) | 3\n",
-      "10032    | torch.Size([1, 1000, 762]) | 3\n",
-      "10033    | torch.Size([1, 1000, 754]) | 3\n",
-      "10034    | torch.Size([1, 1000, 754]) | 3\n",
-      "10035    | torch.Size([1, 1000, 754]) | 3\n",
-      "10036    | torch.Size([1, 1000, 754]) | 3\n",
-      "10037    | torch.Size([1, 1000, 762]) | 3\n",
-      "10038    | torch.Size([1, 1000, 868]) | 3\n",
-      "10039    | torch.Size([1, 1000, 777]) | 3\n",
-      "10040    | torch.Size([1, 1000, 762]) | 3\n",
-      "10041    | torch.Size([1, 1000, 764]) | 3\n",
-      "10042    | torch.Size([1, 1000, 777]) | 3\n",
-      "10043    | torch.Size([1, 1000, 781]) | 3\n",
-      "10044    | torch.Size([1, 1000, 796]) | 3\n",
-      "10045    | torch.Size([1, 1000, 754]) | 3\n",
-      "10046    | torch.Size([1, 1000, 754]) | 3\n",
-      "10047    | torch.Size([1, 1000, 754]) | 3\n",
-      "10048    | torch.Size([1, 1000, 849]) | 3\n",
-      "10049    | torch.Size([1, 1000, 779]) | 3\n",
-      "10050    | torch.Size([1, 1000, 806]) | 3\n",
-      "10051    | torch.Size([1, 1000, 754]) | 3\n",
-      "10052    | torch.Size([1, 1000, 754]) | 3\n",
-      "10053    | torch.Size([1, 1000, 762]) | 3\n",
-      "10054    | torch.Size([1, 1000, 754]) | 3\n",
-      "10055    | torch.Size([1, 1000, 762]) | 3\n",
-      "10056    | torch.Size([1, 1000, 762]) | 3\n",
-      "10057    | torch.Size([1, 1000, 762]) | 3\n",
-      "10058    | torch.Size([1, 1000, 781]) | 3\n",
-      "10059    | torch.Size([1, 1000, 762]) | 3\n",
-      "10060    | torch.Size([1, 1000, 762]) | 3\n",
-      "10061    | torch.Size([1, 1000, 777]) | 3\n",
-      "10062    | torch.Size([1, 1000, 783]) | 3\n",
-      "10063    | torch.Size([1, 1000, 785]) | 4\n",
-      "10064    | torch.Size([1, 1000, 754]) | 4\n",
-      "10065    | torch.Size([1, 1000, 778]) | 4\n",
-      "10066    | torch.Size([1, 1000, 762]) | 4\n",
-      "10067    | torch.Size([1, 1000, 777]) | 4\n",
-      "10068    | torch.Size([1, 1000, 802]) | 4\n",
-      "10069    | torch.Size([1, 1000, 754]) | 4\n",
-      "10070    | torch.Size([1, 1000, 754]) | 4\n",
-      "10071    | torch.Size([1, 1000, 769]) | 4\n",
-      "10072    | torch.Size([1, 1000, 767]) | 4\n",
-      "10073    | torch.Size([1, 1000, 754]) | 4\n",
-      "10074    | torch.Size([1, 1000, 778]) | 4\n",
-      "10075    | torch.Size([1, 1000, 723]) | 4\n",
-      "10076    | torch.Size([1, 1000, 613]) | 4\n",
-      "10077    | torch.Size([1, 1000, 786]) | 4\n",
-      "10078    | torch.Size([1, 1000, 754]) | 4\n",
-      "10079    | torch.Size([1, 1000, 762]) | 4\n",
-      "10080    | torch.Size([1, 1000, 754]) | 4\n",
-      "10081    | torch.Size([1, 1000, 754]) | 4\n",
-      "10082    | torch.Size([1, 1000, 730]) | 4\n",
-      "10083    | torch.Size([1, 1000, 777]) | 4\n",
-      "10084    | torch.Size([1, 1000, 754]) | 4\n",
-      "10085    | torch.Size([1, 1000, 754]) | 4\n",
-      "10086    | torch.Size([1, 1000, 777]) | 4\n",
-      "10087    | torch.Size([1, 1000, 797]) | 4\n",
-      "10088    | torch.Size([1, 1000, 777]) | 4\n",
-      "10089    | torch.Size([1, 1000, 754]) | 4\n",
-      "10090    | torch.Size([1, 1000, 754]) | 4\n",
-      "10091    | torch.Size([1, 1000, 754]) | 4\n",
-      "10092    | torch.Size([1, 1000, 788]) | 4\n",
-      "10093    | torch.Size([1, 1000, 754]) | 4\n",
-      "10094    | torch.Size([1, 1000, 754]) | 4\n",
-      "10095    | torch.Size([1, 1000, 777]) | 4\n",
-      "10096    | torch.Size([1, 1000, 754]) | 4\n",
-      "10097    | torch.Size([1, 1000, 770]) | 4\n",
-      "10098    | torch.Size([1, 1000, 754]) | 4\n",
-      "10099    | torch.Size([1, 1000, 754]) | 4\n",
-      "10100    | torch.Size([1, 1000, 778]) | 4\n",
-      "10101    | torch.Size([1, 1000, 703]) | 4\n",
-      "10102    | torch.Size([1, 1000, 754]) | 4\n",
-      "10103    | torch.Size([1, 1000, 777]) | 4\n",
-      "10104    | torch.Size([1, 1000, 993]) | 4\n",
-      "10105    | torch.Size([1, 1000, 754]) | 4\n",
-      "10106    | torch.Size([1, 1000, 754]) | 4\n",
-      "10107    | torch.Size([1, 1000, 754]) | 4\n",
-      "10108    | torch.Size([1, 1000, 754]) | 4\n",
-      "10109    | torch.Size([1, 1000, 766]) | 4\n",
-      "10110    | torch.Size([1, 1000, 754]) | 4\n",
-      "10111    | torch.Size([1, 1000, 777]) | 4\n",
-      "10112    | torch.Size([1, 1000, 784]) | 4\n",
-      "10113    | torch.Size([1, 1000, 754]) | 4\n",
-      "10114    | torch.Size([1, 1000, 851]) | 4\n",
-      "10115    | torch.Size([1, 1000, 754]) | 4\n",
-      "10116    | torch.Size([1, 1000, 762]) | 4\n",
-      "10117    | torch.Size([1, 1000, 773]) | 4\n",
-      "10118    | torch.Size([1, 1000, 754]) | 4\n",
-      "10119    | torch.Size([1, 1000, 754]) | 4\n",
-      "10120    | torch.Size([1, 1000, 777]) | 4\n",
-      "10121    | torch.Size([1, 1000, 754]) | 4\n",
-      "10122    | torch.Size([1, 1000, 754]) | 4\n",
-      "10123    | torch.Size([1, 1000, 802]) | 4\n",
-      "10124    | torch.Size([1, 1000, 776]) | 4\n",
-      "10125    | torch.Size([1, 1000, 754]) | 4\n",
-      "10126    | torch.Size([1, 1000, 754]) | 4\n",
-      "10127    | torch.Size([1, 1000, 754]) | 4\n",
-      "10128    | torch.Size([1, 1000, 777]) | 4\n",
-      "10129    | torch.Size([1, 1000, 777]) | 4\n",
-      "10130    | torch.Size([1, 1000, 754]) | 4\n",
-      "10131    | torch.Size([1, 1000, 754]) | 4\n",
-      "10132    | torch.Size([1, 1000, 754]) | 4\n",
-      "10133    | torch.Size([1, 1000, 754]) | 4\n",
-      "10134    | torch.Size([1, 1000, 764]) | 4\n",
-      "10135    | torch.Size([1, 1000, 754]) | 4\n",
-      "10136    | torch.Size([1, 1000, 777]) | 4\n",
-      "10137    | torch.Size([1, 1000, 754]) | 4\n",
-      "10138    | torch.Size([1, 1000, 773]) | 4\n",
-      "10139    | torch.Size([1, 1000, 754]) | 4\n",
-      "10140    | torch.Size([1, 1000, 771]) | 4\n",
-      "10141    | torch.Size([1, 1000, 754]) | 4\n",
-      "10142    | torch.Size([1, 1000, 754]) | 4\n",
-      "10143    | torch.Size([1, 1000, 777]) | 4\n",
-      "10144    | torch.Size([1, 1000, 754]) | 4\n",
-      "10145    | torch.Size([1, 1000, 754]) | 4\n",
-      "10146    | torch.Size([1, 1000, 766]) | 4\n",
-      "10147    | torch.Size([1, 1000, 754]) | 4\n",
-      "10148    | torch.Size([1, 1000, 754]) | 4\n",
-      "10149    | torch.Size([1, 1000, 780]) | 4\n",
-      "10150    | torch.Size([1, 1000, 754]) | 4\n",
-      "10151    | torch.Size([1, 1000, 754]) | 4\n",
-      "10152    | torch.Size([1, 1000, 754]) | 4\n",
-      "10153    | torch.Size([1, 1000, 754]) | 4\n",
-      "10154    | torch.Size([1, 1000, 777]) | 4\n",
-      "10155    | torch.Size([1, 1000, 788]) | 4\n",
-      "10156    | torch.Size([1, 1000, 783]) | 4\n",
-      "10157    | torch.Size([1, 1000, 754]) | 4\n",
-      "10158    | torch.Size([1, 1000, 754]) | 4\n",
-      "10159    | torch.Size([1, 1000, 777]) | 4\n",
-      "10160    | torch.Size([1, 1000, 777]) | 4\n",
-      "10161    | torch.Size([1, 1000, 703]) | 4\n",
-      "10162    | torch.Size([1, 1000, 754]) | 4\n",
-      "10163    | torch.Size([1, 1000, 777]) | 4\n",
-      "10164    | torch.Size([1, 1000, 754]) | 4\n",
-      "10165    | torch.Size([1, 1000, 781]) | 4\n",
-      "10166    | torch.Size([1, 1000, 777]) | 4\n",
-      "10167    | torch.Size([1, 1000, 726]) | 4\n",
-      "10168    | torch.Size([1, 1000, 754]) | 4\n",
-      "10169    | torch.Size([1, 1000, 754]) | 4\n",
-      "10170    | torch.Size([1, 1000, 754]) | 4\n",
-      "10171    | torch.Size([1, 1000, 754]) | 4\n",
-      "10172    | torch.Size([1, 1000, 754]) | 4\n",
-      "10173    | torch.Size([1, 1000, 789]) | 4\n",
-      "10174    | torch.Size([1, 1000, 754]) | 4\n",
-      "10175    | torch.Size([1, 1000, 754]) | 4\n",
-      "10176    | torch.Size([1, 1000, 783]) | 4\n",
-      "10177    | torch.Size([1, 1000, 801]) | 4\n",
-      "10178    | torch.Size([1, 1000, 775]) | 4\n",
-      "10179    | torch.Size([1, 1000, 754]) | 4\n",
-      "10180    | torch.Size([1, 1000, 754]) | 4\n",
-      "10181    | torch.Size([1, 1000, 754]) | 4\n",
-      "10182    | torch.Size([1, 1000, 754]) | 4\n",
-      "10183    | torch.Size([1, 1000, 802]) | 4\n",
-      "10184    | torch.Size([1, 1000, 754]) | 4\n",
-      "10185    | torch.Size([1, 1000, 773]) | 4\n",
-      "10186    | torch.Size([1, 1000, 754]) | 4\n",
-      "10187    | torch.Size([1, 1000, 754]) | 4\n",
-      "10188    | torch.Size([1, 1000, 809]) | 4\n",
-      "10189    | torch.Size([1, 1000, 754]) | 4\n",
-      "10190    | torch.Size([1, 1000, 754]) | 4\n",
-      "10191    | torch.Size([1, 1000, 754]) | 4\n",
-      "10192    | torch.Size([1, 1000, 754]) | 4\n",
-      "10193    | torch.Size([1, 1000, 762]) | 4\n",
-      "10194    | torch.Size([1, 1000, 765]) | 4\n",
-      "10195    | torch.Size([1, 1000, 754]) | 4\n",
-      "10196    | torch.Size([1, 1000, 754]) | 4\n",
-      "10197    | torch.Size([1, 1000, 777]) | 4\n",
-      "10198    | torch.Size([1, 1000, 762]) | 4\n",
-      "10199    | torch.Size([1, 1000, 754]) | 4\n",
-      "10200    | torch.Size([1, 1000, 754]) | 4\n",
-      "10201    | torch.Size([1, 1000, 754]) | 4\n",
-      "10202    | torch.Size([1, 1000, 810]) | 4\n",
-      "10203    | torch.Size([1, 1000, 783]) | 4\n",
-      "10204    | torch.Size([1, 1000, 778]) | 4\n",
-      "10205    | torch.Size([1, 1000, 754]) | 4\n",
-      "10206    | torch.Size([1, 1000, 754]) | 4\n",
-      "10207    | torch.Size([1, 1000, 754]) | 4\n",
-      "10208    | torch.Size([1, 1000, 790]) | 4\n",
-      "10209    | torch.Size([1, 1000, 796]) | 4\n",
-      "10210    | torch.Size([1, 1000, 786]) | 4\n",
-      "10211    | torch.Size([1, 1000, 769]) | 4\n",
-      "10212    | torch.Size([1, 1000, 754]) | 4\n",
-      "10213    | torch.Size([1, 1000, 778]) | 4\n",
-      "10214    | torch.Size([1, 1000, 754]) | 4\n",
-      "10215    | torch.Size([1, 1000, 784]) | 4\n",
-      "10216    | torch.Size([1, 1000, 786]) | 4\n",
-      "10217    | torch.Size([1, 1000, 754]) | 4\n",
-      "10218    | torch.Size([1, 1000, 754]) | 4\n",
-      "10219    | torch.Size([1, 1000, 776]) | 4\n",
-      "10220    | torch.Size([1, 1000, 778]) | 4\n",
-      "10221    | torch.Size([1, 1000, 774]) | 4\n",
-      "10222    | torch.Size([1, 1000, 754]) | 4\n",
-      "10223    | torch.Size([1, 1000, 802]) | 4\n",
-      "10224    | torch.Size([1, 1000, 754]) | 4\n",
-      "10225    | torch.Size([1, 1000, 771]) | 4\n",
-      "10226    | torch.Size([1, 1000, 754]) | 4\n",
-      "10227    | torch.Size([1, 1000, 802]) | 4\n",
-      "10228    | torch.Size([1, 1000, 754]) | 4\n",
-      "10229    | torch.Size([1, 1000, 777]) | 4\n",
-      "10230    | torch.Size([1, 1000, 777]) | 4\n",
-      "10231    | torch.Size([1, 1000, 754]) | 4\n",
-      "10232    | torch.Size([1, 1000, 779]) | 4\n",
-      "10233    | torch.Size([1, 1000, 777]) | 4\n",
-      "10234    | torch.Size([1, 1000, 754]) | 4\n",
-      "10235    | torch.Size([1, 1000, 774]) | 4\n",
-      "10236    | torch.Size([1, 1000, 818]) | 4\n",
-      "10237    | torch.Size([1, 1000, 777]) | 4\n",
-      "10238    | torch.Size([1, 1000, 754]) | 4\n",
-      "10239    | torch.Size([1, 1000, 754]) | 4\n",
-      "10240    | torch.Size([1, 1000, 765]) | 4\n",
-      "10241    | torch.Size([1, 1000, 771]) | 4\n",
-      "10242    | torch.Size([1, 1000, 754]) | 4\n",
-      "10243    | torch.Size([1, 1000, 754]) | 4\n",
-      "10244    | torch.Size([1, 1000, 762]) | 4\n",
-      "10245    | torch.Size([1, 1000, 781]) | 4\n",
-      "10246    | torch.Size([1, 1000, 754]) | 4\n",
-      "10247    | torch.Size([1, 1000, 771]) | 4\n",
-      "10248    | torch.Size([1, 1000, 754]) | 4\n",
-      "10249    | torch.Size([1, 1000, 754]) | 4\n",
-      "10250    | torch.Size([1, 1000, 777]) | 4\n",
-      "10251    | torch.Size([1, 1000, 754]) | 4\n",
-      "10252    | torch.Size([1, 1000, 754]) | 4\n",
-      "10253    | torch.Size([1, 1000, 777]) | 4\n",
-      "10254    | torch.Size([1, 1000, 848]) | 4\n",
-      "10255    | torch.Size([1, 1000, 754]) | 4\n",
-      "10256    | torch.Size([1, 1000, 754]) | 4\n",
-      "10257    | torch.Size([1, 1000, 777]) | 4\n",
-      "10258    | torch.Size([1, 1000, 754]) | 4\n",
-      "10259    | torch.Size([1, 1000, 754]) | 4\n",
-      "10260    | torch.Size([1, 1000, 863]) | 4\n",
-      "10261    | torch.Size([1, 1000, 762]) | 4\n",
-      "10262    | torch.Size([1, 1000, 754]) | 4\n",
-      "10263    | torch.Size([1, 1000, 779]) | 4\n",
-      "10264    | torch.Size([1, 1000, 754]) | 4\n",
-      "10265    | torch.Size([1, 1000, 754]) | 4\n",
-      "10266    | torch.Size([1, 1000, 754]) | 4\n",
-      "10267    | torch.Size([1, 1000, 754]) | 4\n",
-      "10268    | torch.Size([1, 1000, 785]) | 4\n",
-      "10269    | torch.Size([1, 1000, 774]) | 4\n",
-      "10270    | torch.Size([1, 1000, 777]) | 4\n",
-      "10271    | torch.Size([1, 1000, 762]) | 4\n",
-      "10272    | torch.Size([1, 1000, 754]) | 4\n",
-      "10273    | torch.Size([1, 1000, 847]) | 4\n",
-      "10274    | torch.Size([1, 1000, 754]) | 4\n",
-      "10275    | torch.Size([1, 1000, 754]) | 4\n",
-      "10276    | torch.Size([1, 1000, 754]) | 4\n",
-      "10277    | torch.Size([1, 1000, 754]) | 4\n",
-      "10278    | torch.Size([1, 1000, 777]) | 4\n",
-      "10279    | torch.Size([1, 1000, 771]) | 4\n",
-      "10280    | torch.Size([1, 1000, 754]) | 4\n",
-      "10281    | torch.Size([1, 1000, 754]) | 4\n",
-      "10282    | torch.Size([1, 1000, 754]) | 4\n",
-      "10283    | torch.Size([1, 1000, 754]) | 4\n",
-      "10284    | torch.Size([1, 1000, 626]) | 4\n",
-      "10285    | torch.Size([1, 1000, 777]) | 4\n",
-      "10286    | torch.Size([1, 1000, 754]) | 4\n",
-      "10287    | torch.Size([1, 1000, 754]) | 4\n",
-      "10288    | torch.Size([1, 1000, 802]) | 4\n",
-      "10289    | torch.Size([1, 1000, 754]) | 4\n",
-      "10290    | torch.Size([1, 1000, 754]) | 4\n",
-      "10291    | torch.Size([1, 1000, 777]) | 4\n",
-      "10292    | torch.Size([1, 1000, 754]) | 4\n",
-      "10293    | torch.Size([1, 1000, 802]) | 4\n",
-      "10294    | torch.Size([1, 1000, 777]) | 4\n",
-      "10295    | torch.Size([1, 1000, 817]) | 4\n",
-      "10296    | torch.Size([1, 1000, 764]) | 4\n",
-      "10297    | torch.Size([1, 1000, 754]) | 4\n",
-      "10298    | torch.Size([1, 1000, 754]) | 4\n",
-      "10299    | torch.Size([1, 1000, 754]) | 4\n",
-      "10300    | torch.Size([1, 1000, 802]) | 4\n",
-      "10301    | torch.Size([1, 1000, 754]) | 4\n",
-      "10302    | torch.Size([1, 1000, 754]) | 4\n",
-      "10303    | torch.Size([1, 1000, 784]) | 4\n",
-      "10304    | torch.Size([1, 1000, 754]) | 4\n",
-      "10305    | torch.Size([1, 1000, 777]) | 4\n",
-      "10306    | torch.Size([1, 1000, 754]) | 4\n",
-      "10307    | torch.Size([1, 1000, 781]) | 4\n",
-      "10308    | torch.Size([1, 1000, 777]) | 4\n",
-      "10309    | torch.Size([1, 1000, 765]) | 4\n",
-      "10310    | torch.Size([1, 1000, 787]) | 4\n",
-      "10311    | torch.Size([1, 1000, 740]) | 4\n",
-      "10312    | torch.Size([1, 1000, 754]) | 4\n",
-      "10313    | torch.Size([1, 1000, 754]) | 4\n",
-      "10314    | torch.Size([1, 1000, 846]) | 4\n",
-      "10315    | torch.Size([1, 1000, 780]) | 4\n",
-      "10316    | torch.Size([1, 1000, 777]) | 4\n",
-      "10317    | torch.Size([1, 1000, 754]) | 4\n",
-      "10318    | torch.Size([1, 1000, 767]) | 4\n",
-      "10319    | torch.Size([1, 1000, 754]) | 4\n",
-      "10320    | torch.Size([1, 1000, 777]) | 4\n",
-      "10321    | torch.Size([1, 1000, 762]) | 4\n",
-      "10322    | torch.Size([1, 1000, 754]) | 4\n",
-      "10323    | torch.Size([1, 1000, 754]) | 4\n",
-      "10324    | torch.Size([1, 1000, 777]) | 4\n",
-      "10325    | torch.Size([1, 1000, 754]) | 4\n",
-      "10326    | torch.Size([1, 1000, 754]) | 4\n",
-      "10327    | torch.Size([1, 1000, 754]) | 4\n",
-      "10328    | torch.Size([1, 1000, 777]) | 4\n",
-      "10329    | torch.Size([1, 1000, 778]) | 4\n",
-      "10330    | torch.Size([1, 1000, 762]) | 4\n",
-      "10331    | torch.Size([1, 1000, 777]) | 4\n",
-      "10332    | torch.Size([1, 1000, 777]) | 4\n",
-      "10333    | torch.Size([1, 1000, 611]) | 4\n",
-      "10334    | torch.Size([1, 1000, 754]) | 4\n",
-      "10335    | torch.Size([1, 1000, 754]) | 4\n",
-      "10336    | torch.Size([1, 1000, 611]) | 4\n",
-      "10337    | torch.Size([1, 1000, 777]) | 4\n",
-      "10338    | torch.Size([1, 1000, 754]) | 4\n",
-      "10339    | torch.Size([1, 1000, 779]) | 4\n",
-      "10340    | torch.Size([1, 1000, 779]) | 4\n",
-      "10341    | torch.Size([1, 1000, 762]) | 4\n",
-      "10342    | torch.Size([1, 1000, 754]) | 4\n",
-      "10343    | torch.Size([1, 1000, 803]) | 4\n",
-      "10344    | torch.Size([1, 1000, 789]) | 4\n",
-      "10345    | torch.Size([1, 1000, 777]) | 4\n",
-      "10346    | torch.Size([1, 1000, 754]) | 4\n",
-      "10347    | torch.Size([1, 1000, 783]) | 4\n",
-      "10348    | torch.Size([1, 1000, 754]) | 4\n",
-      "10349    | torch.Size([1, 1000, 754]) | 4\n",
-      "10350    | torch.Size([1, 1000, 783]) | 4\n",
-      "10351    | torch.Size([1, 1000, 611]) | 4\n",
-      "10352    | torch.Size([1, 1000, 754]) | 4\n",
-      "10353    | torch.Size([1, 1000, 754]) | 4\n",
-      "10354    | torch.Size([1, 1000, 789]) | 4\n",
-      "10355    | torch.Size([1, 1000, 754]) | 4\n",
-      "10356    | torch.Size([1, 1000, 780]) | 4\n",
-      "10357    | torch.Size([1, 1000, 754]) | 4\n",
-      "10358    | torch.Size([1, 1000, 846]) | 4\n",
-      "10359    | torch.Size([1, 1000, 754]) | 4\n",
-      "10360    | torch.Size([1, 1000, 754]) | 4\n",
-      "10361    | torch.Size([1, 1000, 754]) | 4\n",
-      "10362    | torch.Size([1, 1000, 754]) | 4\n",
-      "10363    | torch.Size([1, 1000, 754]) | 4\n",
-      "10364    | torch.Size([1, 1000, 754]) | 4\n",
-      "10365    | torch.Size([1, 1000, 777]) | 4\n",
-      "10366    | torch.Size([1, 1000, 768]) | 4\n",
-      "10367    | torch.Size([1, 1000, 776]) | 4\n",
-      "10368    | torch.Size([1, 1000, 754]) | 4\n",
-      "10369    | torch.Size([1, 1000, 754]) | 4\n",
-      "10370    | torch.Size([1, 1000, 754]) | 4\n",
-      "10371    | torch.Size([1, 1000, 754]) | 4\n",
-      "10372    | torch.Size([1, 1000, 762]) | 4\n",
-      "10373    | torch.Size([1, 1000, 754]) | 4\n",
-      "10374    | torch.Size([1, 1000, 754]) | 4\n",
-      "10375    | torch.Size([1, 1000, 754]) | 4\n",
-      "10376    | torch.Size([1, 1000, 754]) | 4\n",
-      "10377    | torch.Size([1, 1000, 754]) | 4\n",
-      "10378    | torch.Size([1, 1000, 754]) | 4\n",
-      "10379    | torch.Size([1, 1000, 777]) | 4\n",
-      "10380    | torch.Size([1, 1000, 754]) | 4\n",
-      "10381    | torch.Size([1, 1000, 754]) | 4\n",
-      "10382    | torch.Size([1, 1000, 785]) | 4\n",
-      "10383    | torch.Size([1, 1000, 754]) | 4\n",
-      "10384    | torch.Size([1, 1000, 754]) | 4\n",
-      "10385    | torch.Size([1, 1000, 754]) | 4\n",
-      "10386    | torch.Size([1, 1000, 777]) | 4\n",
-      "10387    | torch.Size([1, 1000, 785]) | 4\n",
-      "10388    | torch.Size([1, 1000, 754]) | 4\n",
-      "10389    | torch.Size([1, 1000, 754]) | 4\n",
-      "10390    | torch.Size([1, 1000, 762]) | 4\n",
-      "10391    | torch.Size([1, 1000, 754]) | 4\n",
-      "10392    | torch.Size([1, 1000, 754]) | 4\n",
-      "10393    | torch.Size([1, 1000, 780]) | 4\n",
-      "10394    | torch.Size([1, 1000, 787]) | 4\n",
-      "10395    | torch.Size([1, 1000, 784]) | 4\n",
-      "10396    | torch.Size([1, 1000, 791]) | 4\n",
-      "10397    | torch.Size([1, 1000, 754]) | 4\n",
-      "10398    | torch.Size([1, 1000, 754]) | 4\n",
-      "10399    | torch.Size([1, 1000, 754]) | 4\n",
-      "10400    | torch.Size([1, 1000, 777]) | 4\n",
-      "10401    | torch.Size([1, 1000, 754]) | 4\n",
-      "10402    | torch.Size([1, 1000, 777]) | 4\n",
-      "10403    | torch.Size([1, 1000, 754]) | 4\n",
-      "10404    | torch.Size([1, 1000, 779]) | 4\n",
-      "10405    | torch.Size([1, 1000, 754]) | 4\n",
-      "10406    | torch.Size([1, 1000, 754]) | 4\n",
-      "10407    | torch.Size([1, 1000, 754]) | 4\n",
-      "10408    | torch.Size([1, 1000, 754]) | 4\n",
-      "10409    | torch.Size([1, 1000, 754]) | 4\n",
-      "10410    | torch.Size([1, 1000, 754]) | 4\n",
-      "10411    | torch.Size([1, 1000, 754]) | 4\n",
-      "10412    | torch.Size([1, 1000, 770]) | 4\n",
-      "10413    | torch.Size([1, 1000, 754]) | 4\n",
-      "10414    | torch.Size([1, 1000, 754]) | 4\n",
-      "10415    | torch.Size([1, 1000, 776]) | 4\n",
-      "10416    | torch.Size([1, 1000, 779]) | 4\n",
-      "10417    | torch.Size([1, 1000, 754]) | 4\n",
-      "10418    | torch.Size([1, 1000, 754]) | 4\n",
-      "10419    | torch.Size([1, 1000, 754]) | 4\n",
-      "10420    | torch.Size([1, 1000, 754]) | 4\n",
-      "10421    | torch.Size([1, 1000, 754]) | 4\n",
-      "10422    | torch.Size([1, 1000, 771]) | 4\n",
-      "10423    | torch.Size([1, 1000, 847]) | 4\n",
-      "10424    | torch.Size([1, 1000, 754]) | 4\n",
-      "10425    | torch.Size([1, 1000, 754]) | 4\n",
-      "10426    | torch.Size([1, 1000, 754]) | 4\n",
-      "10427    | torch.Size([1, 1000, 754]) | 4\n",
-      "10428    | torch.Size([1, 1000, 754]) | 4\n",
-      "10429    | torch.Size([1, 1000, 772]) | 4\n",
-      "10430    | torch.Size([1, 1000, 754]) | 4\n",
-      "10431    | torch.Size([1, 1000, 785]) | 4\n",
-      "10432    | torch.Size([1, 1000, 754]) | 4\n",
-      "10433    | torch.Size([1, 1000, 754]) | 4\n",
-      "10434    | torch.Size([1, 1000, 789]) | 4\n",
-      "10435    | torch.Size([1, 1000, 777]) | 4\n",
-      "10436    | torch.Size([1, 1000, 777]) | 4\n",
-      "10437    | torch.Size([1, 1000, 786]) | 4\n",
-      "10438    | torch.Size([1, 1000, 754]) | 4\n",
-      "10439    | torch.Size([1, 1000, 754]) | 4\n",
-      "10440    | torch.Size([1, 1000, 754]) | 4\n",
-      "10441    | torch.Size([1, 1000, 754]) | 4\n",
-      "10442    | torch.Size([1, 1000, 777]) | 4\n",
-      "10443    | torch.Size([1, 1000, 763]) | 4\n",
-      "10444    | torch.Size([1, 1000, 754]) | 4\n",
-      "10445    | torch.Size([1, 1000, 780]) | 4\n",
-      "10446    | torch.Size([1, 1000, 754]) | 4\n",
-      "10447    | torch.Size([1, 1000, 754]) | 4\n",
-      "10448    | torch.Size([1, 1000, 771]) | 4\n",
-      "10449    | torch.Size([1, 1000, 774]) | 4\n",
-      "10450    | torch.Size([1, 1000, 777]) | 4\n",
-      "10451    | torch.Size([1, 1000, 754]) | 4\n",
-      "10452    | torch.Size([1, 1000, 754]) | 4\n",
-      "10453    | torch.Size([1, 1000, 754]) | 4\n",
-      "10454    | torch.Size([1, 1000, 754]) | 4\n",
-      "10455    | torch.Size([1, 1000, 762]) | 4\n",
-      "10456    | torch.Size([1, 1000, 754]) | 4\n",
-      "10457    | torch.Size([1, 1000, 777]) | 4\n",
-      "10458    | torch.Size([1, 1000, 778]) | 4\n",
-      "10459    | torch.Size([1, 1000, 777]) | 4\n",
-      "10460    | torch.Size([1, 1000, 762]) | 4\n",
-      "10461    | torch.Size([1, 1000, 754]) | 4\n",
-      "10462    | torch.Size([1, 1000, 777]) | 4\n",
-      "10463    | torch.Size([1, 1000, 754]) | 4\n",
-      "10464    | torch.Size([1, 1000, 788]) | 4\n",
-      "10465    | torch.Size([1, 1000, 780]) | 4\n",
-      "10466    | torch.Size([1, 1000, 777]) | 4\n",
-      "10467    | torch.Size([1, 1000, 787]) | 4\n",
-      "10468    | torch.Size([1, 1000, 796]) | 4\n",
-      "10469    | torch.Size([1, 1000, 772]) | 4\n",
-      "10470    | torch.Size([1, 1000, 782]) | 4\n",
-      "10471    | torch.Size([1, 1000, 754]) | 4\n",
-      "10472    | torch.Size([1, 1000, 754]) | 4\n",
-      "10473    | torch.Size([1, 1000, 754]) | 4\n",
-      "10474    | torch.Size([1, 1000, 754]) | 4\n",
-      "10475    | torch.Size([1, 1000, 777]) | 4\n",
-      "10476    | torch.Size([1, 1000, 765]) | 4\n",
-      "10477    | torch.Size([1, 1000, 762]) | 4\n",
-      "10478    | torch.Size([1, 1000, 754]) | 4\n",
-      "10479    | torch.Size([1, 1000, 754]) | 4\n",
-      "10480    | torch.Size([1, 1000, 777]) | 4\n",
-      "10481    | torch.Size([1, 1000, 802]) | 4\n",
-      "10482    | torch.Size([1, 1000, 754]) | 4\n",
-      "10483    | torch.Size([1, 1000, 802]) | 4\n",
-      "10484    | torch.Size([1, 1000, 754]) | 4\n",
-      "10485    | torch.Size([1, 1000, 826]) | 4\n",
-      "10486    | torch.Size([1, 1000, 754]) | 4\n",
-      "10487    | torch.Size([1, 1000, 840]) | 4\n",
-      "10488    | torch.Size([1, 1000, 754]) | 4\n",
-      "10489    | torch.Size([1, 1000, 612]) | 4\n",
-      "10490    | torch.Size([1, 1000, 754]) | 4\n",
-      "10491    | torch.Size([1, 1000, 802]) | 4\n",
-      "10492    | torch.Size([1, 1000, 754]) | 4\n",
-      "10493    | torch.Size([1, 1000, 754]) | 4\n",
-      "10494    | torch.Size([1, 1000, 754]) | 4\n",
-      "10495    | torch.Size([1, 1000, 777]) | 4\n",
-      "10496    | torch.Size([1, 1000, 776]) | 4\n",
-      "10497    | torch.Size([1, 1000, 754]) | 4\n",
-      "10498    | torch.Size([1, 1000, 765]) | 4\n",
-      "10499    | torch.Size([1, 1000, 754]) | 4\n",
-      "10500    | torch.Size([1, 1000, 754]) | 4\n",
-      "10501    | torch.Size([1, 1000, 754]) | 4\n",
-      "10502    | torch.Size([1, 1000, 754]) | 4\n",
-      "10503    | torch.Size([1, 1000, 786]) | 4\n",
-      "10504    | torch.Size([1, 1000, 766]) | 4\n",
-      "10505    | torch.Size([1, 1000, 777]) | 4\n",
-      "10506    | torch.Size([1, 1000, 643]) | 4\n",
-      "10507    | torch.Size([1, 1000, 741]) | 4\n",
-      "10508    | torch.Size([1, 1000, 762]) | 4\n",
-      "10509    | torch.Size([1, 1000, 762]) | 4\n",
-      "10510    | torch.Size([1, 1000, 777]) | 4\n",
-      "10511    | torch.Size([1, 1000, 754]) | 4\n",
-      "10512    | torch.Size([1, 1000, 754]) | 4\n",
-      "10513    | torch.Size([1, 1000, 787]) | 4\n",
-      "10514    | torch.Size([1, 1000, 754]) | 4\n",
-      "10515    | torch.Size([1, 1000, 832]) | 4\n",
-      "10516    | torch.Size([1, 1000, 754]) | 4\n",
-      "10517    | torch.Size([1, 1000, 754]) | 4\n",
-      "10518    | torch.Size([1, 1000, 754]) | 4\n",
-      "10519    | torch.Size([1, 1000, 777]) | 4\n",
-      "10520    | torch.Size([1, 1000, 777]) | 4\n",
-      "10521    | torch.Size([1, 1000, 782]) | 4\n",
-      "10522    | torch.Size([1, 1000, 828]) | 4\n",
-      "10523    | torch.Size([1, 1000, 738]) | 4\n",
-      "10524    | torch.Size([1, 1000, 802]) | 4\n",
-      "10525    | torch.Size([1, 1000, 774]) | 4\n",
-      "10526    | torch.Size([1, 1000, 754]) | 4\n",
-      "10527    | torch.Size([1, 1000, 777]) | 4\n",
-      "10528    | torch.Size([1, 1000, 754]) | 4\n",
-      "10529    | torch.Size([1, 1000, 754]) | 4\n",
-      "10530    | torch.Size([1, 1000, 754]) | 4\n",
-      "10531    | torch.Size([1, 1000, 754]) | 4\n",
-      "10532    | torch.Size([1, 1000, 754]) | 4\n",
-      "10533    | torch.Size([1, 1000, 754]) | 4\n",
-      "10534    | torch.Size([1, 1000, 754]) | 4\n",
-      "10535    | torch.Size([1, 1000, 754]) | 4\n",
-      "10536    | torch.Size([1, 1000, 762]) | 4\n",
-      "10537    | torch.Size([1, 1000, 754]) | 4\n",
-      "10538    | torch.Size([1, 1000, 777]) | 4\n",
-      "10539    | torch.Size([1, 1000, 754]) | 4\n",
-      "10540    | torch.Size([1, 1000, 754]) | 4\n",
-      "10541    | torch.Size([1, 1000, 767]) | 4\n",
-      "10542    | torch.Size([1, 1000, 754]) | 4\n",
-      "10543    | torch.Size([1, 1000, 754]) | 4\n",
-      "10544    | torch.Size([1, 1000, 754]) | 4\n",
-      "10545    | torch.Size([1, 1000, 788]) | 4\n",
-      "10546    | torch.Size([1, 1000, 754]) | 4\n",
-      "10547    | torch.Size([1, 1000, 851]) | 4\n",
-      "10548    | torch.Size([1, 1000, 796]) | 4\n",
-      "10549    | torch.Size([1, 1000, 754]) | 4\n",
-      "10550    | torch.Size([1, 1000, 754]) | 4\n",
-      "10551    | torch.Size([1, 1000, 754]) | 4\n",
-      "10552    | torch.Size([1, 1000, 754]) | 4\n",
-      "10553    | torch.Size([1, 1000, 754]) | 4\n",
-      "10554    | torch.Size([1, 1000, 754]) | 4\n",
-      "10555    | torch.Size([1, 1000, 777]) | 4\n",
-      "10556    | torch.Size([1, 1000, 754]) | 4\n",
-      "10557    | torch.Size([1, 1000, 754]) | 4\n",
-      "10558    | torch.Size([1, 1000, 754]) | 4\n",
-      "10559    | torch.Size([1, 1000, 754]) | 4\n",
-      "10560    | torch.Size([1, 1000, 754]) | 4\n",
-      "10561    | torch.Size([1, 1000, 754]) | 4\n",
-      "10562    | torch.Size([1, 1000, 754]) | 4\n",
-      "10563    | torch.Size([1, 1000, 754]) | 4\n",
-      "10564    | torch.Size([1, 1000, 754]) | 4\n",
-      "10565    | torch.Size([1, 1000, 785]) | 4\n",
-      "10566    | torch.Size([1, 1000, 784]) | 4\n",
-      "10567    | torch.Size([1, 1000, 754]) | 4\n",
-      "10568    | torch.Size([1, 1000, 754]) | 4\n",
-      "10569    | torch.Size([1, 1000, 754]) | 4\n",
-      "10570    | torch.Size([1, 1000, 754]) | 4\n",
-      "10571    | torch.Size([1, 1000, 754]) | 4\n",
-      "10572    | torch.Size([1, 1000, 754]) | 4\n",
-      "10573    | torch.Size([1, 1000, 754]) | 4\n",
-      "10574    | torch.Size([1, 1000, 754]) | 4\n",
-      "10575    | torch.Size([1, 1000, 754]) | 4\n",
-      "10576    | torch.Size([1, 1000, 754]) | 4\n",
-      "10577    | torch.Size([1, 1000, 787]) | 4\n",
-      "10578    | torch.Size([1, 1000, 787]) | 4\n",
-      "10579    | torch.Size([1, 1000, 802]) | 4\n",
-      "10580    | torch.Size([1, 1000, 774]) | 4\n",
-      "10581    | torch.Size([1, 1000, 754]) | 4\n",
-      "10582    | torch.Size([1, 1000, 806]) | 4\n",
-      "10583    | torch.Size([1, 1000, 764]) | 4\n",
-      "10584    | torch.Size([1, 1000, 724]) | 4\n",
-      "10585    | torch.Size([1, 1000, 754]) | 4\n",
-      "10586    | torch.Size([1, 1000, 777]) | 4\n",
-      "10587    | torch.Size([1, 1000, 777]) | 4\n",
-      "10588    | torch.Size([1, 1000, 777]) | 4\n",
-      "10589    | torch.Size([1, 1000, 754]) | 4\n",
-      "10590    | torch.Size([1, 1000, 777]) | 4\n",
-      "10591    | torch.Size([1, 1000, 802]) | 4\n",
-      "10592    | torch.Size([1, 1000, 762]) | 4\n",
-      "10593    | torch.Size([1, 1000, 754]) | 4\n",
-      "10594    | torch.Size([1, 1000, 754]) | 4\n",
-      "10595    | torch.Size([1, 1000, 754]) | 4\n",
-      "10596    | torch.Size([1, 1000, 754]) | 4\n",
-      "10597    | torch.Size([1, 1000, 778]) | 4\n",
-      "10598    | torch.Size([1, 1000, 765]) | 4\n",
-      "10599    | torch.Size([1, 1000, 776]) | 4\n",
-      "10600    | torch.Size([1, 1000, 777]) | 4\n",
-      "10601    | torch.Size([1, 1000, 763]) | 4\n",
-      "10602    | torch.Size([1, 1000, 764]) | 4\n",
-      "10603    | torch.Size([1, 1000, 779]) | 4\n",
-      "10604    | torch.Size([1, 1000, 754]) | 4\n",
-      "10605    | torch.Size([1, 1000, 778]) | 4\n",
-      "10606    | torch.Size([1, 1000, 754]) | 4\n",
-      "10607    | torch.Size([1, 1000, 754]) | 4\n",
-      "10608    | torch.Size([1, 1000, 777]) | 4\n",
-      "10609    | torch.Size([1, 1000, 614]) | 4\n",
-      "10610    | torch.Size([1, 1000, 802]) | 4\n",
-      "10611    | torch.Size([1, 1000, 611]) | 4\n",
-      "10612    | torch.Size([1, 1000, 778]) | 4\n",
-      "10613    | torch.Size([1, 1000, 754]) | 4\n",
-      "10614    | torch.Size([1, 1000, 754]) | 4\n",
-      "10615    | torch.Size([1, 1000, 754]) | 4\n",
-      "10616    | torch.Size([1, 1000, 772]) | 4\n",
-      "10617    | torch.Size([1, 1000, 777]) | 4\n",
-      "10618    | torch.Size([1, 1000, 714]) | 4\n",
-      "10619    | torch.Size([1, 1000, 754]) | 4\n",
-      "10620    | torch.Size([1, 1000, 754]) | 4\n",
-      "10621    | torch.Size([1, 1000, 782]) | 4\n",
-      "10622    | torch.Size([1, 1000, 788]) | 4\n",
-      "10623    | torch.Size([1, 1000, 777]) | 4\n",
-      "10624    | torch.Size([1, 1000, 792]) | 4\n",
-      "10625    | torch.Size([1, 1000, 754]) | 4\n",
-      "10626    | torch.Size([1, 1000, 764]) | 4\n",
-      "10627    | torch.Size([1, 1000, 774]) | 4\n",
-      "10628    | torch.Size([1, 1000, 777]) | 4\n",
-      "10629    | torch.Size([1, 1000, 754]) | 4\n",
-      "10630    | torch.Size([1, 1000, 754]) | 4\n",
-      "10631    | torch.Size([1, 1000, 754]) | 4\n",
-      "10632    | torch.Size([1, 1000, 762]) | 4\n",
-      "10633    | torch.Size([1, 1000, 754]) | 4\n",
-      "10634    | torch.Size([1, 1000, 784]) | 4\n",
-      "10635    | torch.Size([1, 1000, 796]) | 4\n",
-      "10636    | torch.Size([1, 1000, 754]) | 4\n",
-      "10637    | torch.Size([1, 1000, 754]) | 4\n",
-      "10638    | torch.Size([1, 1000, 781]) | 4\n",
-      "10639    | torch.Size([1, 1000, 788]) | 4\n",
-      "10640    | torch.Size([1, 1000, 754]) | 4\n",
-      "10641    | torch.Size([1, 1000, 754]) | 4\n",
-      "10642    | torch.Size([1, 1000, 777]) | 4\n",
-      "10643    | torch.Size([1, 1000, 774]) | 4\n",
-      "10644    | torch.Size([1, 1000, 754]) | 4\n",
-      "10645    | torch.Size([1, 1000, 754]) | 4\n",
-      "10646    | torch.Size([1, 1000, 754]) | 4\n",
-      "10647    | torch.Size([1, 1000, 754]) | 4\n",
-      "10648    | torch.Size([1, 1000, 776]) | 4\n",
-      "10649    | torch.Size([1, 1000, 774]) | 4\n",
-      "10650    | torch.Size([1, 1000, 777]) | 4\n",
-      "10651    | torch.Size([1, 1000, 754]) | 4\n",
-      "10652    | torch.Size([1, 1000, 754]) | 4\n",
-      "10653    | torch.Size([1, 1000, 754]) | 4\n",
-      "10654    | torch.Size([1, 1000, 754]) | 4\n",
-      "10655    | torch.Size([1, 1000, 754]) | 4\n",
-      "10656    | torch.Size([1, 1000, 777]) | 4\n",
-      "10657    | torch.Size([1, 1000, 754]) | 4\n",
-      "10658    | torch.Size([1, 1000, 754]) | 4\n",
-      "10659    | torch.Size([1, 1000, 754]) | 4\n",
-      "10660    | torch.Size([1, 1000, 754]) | 4\n",
-      "10661    | torch.Size([1, 1000, 773]) | 4\n",
-      "10662    | torch.Size([1, 1000, 754]) | 4\n",
-      "10663    | torch.Size([1, 1000, 754]) | 4\n",
-      "10664    | torch.Size([1, 1000, 784]) | 4\n",
-      "10665    | torch.Size([1, 1000, 754]) | 4\n",
-      "10666    | torch.Size([1, 1000, 754]) | 4\n",
-      "10667    | torch.Size([1, 1000, 754]) | 4\n",
-      "10668    | torch.Size([1, 1000, 754]) | 4\n",
-      "10669    | torch.Size([1, 1000, 754]) | 4\n",
-      "10670    | torch.Size([1, 1000, 754]) | 4\n",
-      "10671    | torch.Size([1, 1000, 754]) | 4\n",
-      "10672    | torch.Size([1, 1000, 787]) | 4\n",
-      "10673    | torch.Size([1, 1000, 754]) | 4\n",
-      "10674    | torch.Size([1, 1000, 777]) | 4\n",
-      "10675    | torch.Size([1, 1000, 774]) | 4\n",
-      "10676    | torch.Size([1, 1000, 779]) | 4\n",
-      "10677    | torch.Size([1, 1000, 777]) | 4\n",
-      "10678    | torch.Size([1, 1000, 777]) | 4\n",
-      "10679    | torch.Size([1, 1000, 754]) | 4\n",
-      "10680    | torch.Size([1, 1000, 754]) | 4\n",
-      "10681    | torch.Size([1, 1000, 780]) | 4\n",
-      "10682    | torch.Size([1, 1000, 754]) | 4\n",
-      "10683    | torch.Size([1, 1000, 777]) | 4\n",
-      "10684    | torch.Size([1, 1000, 783]) | 4\n",
-      "10685    | torch.Size([1, 1000, 754]) | 4\n",
-      "10686    | torch.Size([1, 1000, 754]) | 4\n",
-      "10687    | torch.Size([1, 1000, 754]) | 4\n",
-      "10688    | torch.Size([1, 1000, 754]) | 4\n",
-      "10689    | torch.Size([1, 1000, 754]) | 4\n",
-      "10690    | torch.Size([1, 1000, 772]) | 4\n",
-      "10691    | torch.Size([1, 1000, 754]) | 4\n",
-      "10692    | torch.Size([1, 1000, 754]) | 4\n",
-      "10693    | torch.Size([1, 1000, 797]) | 4\n",
-      "10694    | torch.Size([1, 1000, 777]) | 4\n",
-      "10695    | torch.Size([1, 1000, 796]) | 4\n",
-      "10696    | torch.Size([1, 1000, 777]) | 4\n",
-      "10697    | torch.Size([1, 1000, 754]) | 4\n",
-      "10698    | torch.Size([1, 1000, 754]) | 4\n",
-      "10699    | torch.Size([1, 1000, 1473]) | 4\n",
-      "10700    | torch.Size([1, 1000, 776]) | 4\n",
-      "10701    | torch.Size([1, 1000, 786]) | 4\n",
-      "10702    | torch.Size([1, 1000, 754]) | 4\n",
-      "10703    | torch.Size([1, 1000, 775]) | 4\n",
-      "10704    | torch.Size([1, 1000, 754]) | 4\n",
-      "10705    | torch.Size([1, 1000, 754]) | 4\n",
-      "10706    | torch.Size([1, 1000, 754]) | 4\n",
-      "10707    | torch.Size([1, 1000, 754]) | 4\n",
-      "10708    | torch.Size([1, 1000, 754]) | 4\n",
-      "10709    | torch.Size([1, 1000, 777]) | 4\n",
-      "10710    | torch.Size([1, 1000, 790]) | 4\n",
-      "10711    | torch.Size([1, 1000, 703]) | 4\n",
-      "10712    | torch.Size([1, 1000, 782]) | 4\n",
-      "10713    | torch.Size([1, 1000, 777]) | 4\n",
-      "10714    | torch.Size([1, 1000, 754]) | 4\n",
-      "10715    | torch.Size([1, 1000, 754]) | 4\n",
-      "10716    | torch.Size([1, 1000, 779]) | 4\n",
-      "10717    | torch.Size([1, 1000, 754]) | 4\n",
-      "10718    | torch.Size([1, 1000, 777]) | 4\n",
-      "10719    | torch.Size([1, 1000, 754]) | 4\n",
-      "10720    | torch.Size([1, 1000, 802]) | 4\n",
-      "10721    | torch.Size([1, 1000, 781]) | 4\n",
-      "10722    | torch.Size([1, 1000, 777]) | 4\n",
-      "10723    | torch.Size([1, 1000, 754]) | 4\n",
-      "10724    | torch.Size([1, 1000, 796]) | 4\n",
-      "10725    | torch.Size([1, 1000, 754]) | 4\n",
-      "10726    | torch.Size([1, 1000, 754]) | 4\n",
-      "10727    | torch.Size([1, 1000, 754]) | 4\n",
-      "10728    | torch.Size([1, 1000, 754]) | 4\n",
-      "10729    | torch.Size([1, 1000, 786]) | 4\n",
-      "10730    | torch.Size([1, 1000, 754]) | 4\n",
-      "10731    | torch.Size([1, 1000, 754]) | 4\n",
-      "10732    | torch.Size([1, 1000, 777]) | 4\n",
-      "10733    | torch.Size([1, 1000, 754]) | 4\n",
-      "10734    | torch.Size([1, 1000, 777]) | 4\n",
-      "10735    | torch.Size([1, 1000, 754]) | 4\n",
-      "10736    | torch.Size([1, 1000, 783]) | 4\n",
-      "10737    | torch.Size([1, 1000, 762]) | 4\n",
-      "10738    | torch.Size([1, 1000, 754]) | 4\n",
-      "10739    | torch.Size([1, 1000, 777]) | 4\n",
-      "10740    | torch.Size([1, 1000, 777]) | 4\n",
-      "10741    | torch.Size([1, 1000, 754]) | 4\n",
-      "10742    | torch.Size([1, 1000, 726]) | 4\n",
-      "10743    | torch.Size([1, 1000, 777]) | 4\n",
-      "10744    | torch.Size([1, 1000, 754]) | 4\n",
-      "10745    | torch.Size([1, 1000, 762]) | 4\n",
-      "10746    | torch.Size([1, 1000, 777]) | 4\n",
-      "10747    | torch.Size([1, 1000, 766]) | 4\n",
-      "10748    | torch.Size([1, 1000, 786]) | 4\n",
-      "10749    | torch.Size([1, 1000, 778]) | 4\n",
-      "10750    | torch.Size([1, 1000, 792]) | 4\n",
-      "10751    | torch.Size([1, 1000, 775]) | 4\n",
-      "10752    | torch.Size([1, 1000, 646]) | 4\n",
-      "10753    | torch.Size([1, 1000, 777]) | 4\n",
-      "10754    | torch.Size([1, 1000, 781]) | 4\n",
-      "10755    | torch.Size([1, 1000, 754]) | 4\n",
-      "10756    | torch.Size([1, 1000, 765]) | 4\n",
-      "10757    | torch.Size([1, 1000, 774]) | 4\n",
-      "10758    | torch.Size([1, 1000, 754]) | 4\n",
-      "10759    | torch.Size([1, 1000, 788]) | 4\n",
-      "10760    | torch.Size([1, 1000, 754]) | 4\n",
-      "10761    | torch.Size([1, 1000, 777]) | 4\n",
-      "10762    | torch.Size([1, 1000, 762]) | 4\n",
-      "10763    | torch.Size([1, 1000, 754]) | 4\n",
-      "10764    | torch.Size([1, 1000, 789]) | 4\n",
-      "10765    | torch.Size([1, 1000, 777]) | 4\n",
-      "10766    | torch.Size([1, 1000, 782]) | 4\n",
-      "10767    | torch.Size([1, 1000, 762]) | 4\n",
-      "10768    | torch.Size([1, 1000, 754]) | 4\n",
-      "10769    | torch.Size([1, 1000, 777]) | 4\n",
-      "10770    | torch.Size([1, 1000, 769]) | 4\n",
-      "10771    | torch.Size([1, 1000, 754]) | 4\n",
-      "10772    | torch.Size([1, 1000, 766]) | 4\n",
-      "10773    | torch.Size([1, 1000, 777]) | 4\n",
-      "10774    | torch.Size([1, 1000, 754]) | 4\n",
-      "10775    | torch.Size([1, 1000, 754]) | 4\n",
-      "10776    | torch.Size([1, 1000, 754]) | 4\n",
-      "10777    | torch.Size([1, 1000, 754]) | 4\n",
-      "10778    | torch.Size([1, 1000, 754]) | 4\n",
-      "10779    | torch.Size([1, 1000, 754]) | 4\n",
-      "10780    | torch.Size([1, 1000, 754]) | 4\n",
-      "10781    | torch.Size([1, 1000, 754]) | 4\n",
-      "10782    | torch.Size([1, 1000, 846]) | 4\n",
-      "10783    | torch.Size([1, 1000, 776]) | 4\n",
-      "10784    | torch.Size([1, 1000, 754]) | 4\n",
-      "10785    | torch.Size([1, 1000, 754]) | 4\n",
-      "10786    | torch.Size([1, 1000, 777]) | 4\n",
-      "10787    | torch.Size([1, 1000, 754]) | 4\n",
-      "10788    | torch.Size([1, 1000, 808]) | 4\n",
-      "10789    | torch.Size([1, 1000, 754]) | 4\n",
-      "10790    | torch.Size([1, 1000, 782]) | 4\n",
-      "10791    | torch.Size([1, 1000, 754]) | 4\n",
-      "10792    | torch.Size([1, 1000, 780]) | 4\n",
-      "10793    | torch.Size([1, 1000, 780]) | 4\n",
-      "10794    | torch.Size([1, 1000, 754]) | 4\n",
-      "10795    | torch.Size([1, 1000, 798]) | 4\n",
-      "10796    | torch.Size([1, 1000, 754]) | 4\n",
-      "10797    | torch.Size([1, 1000, 777]) | 4\n",
-      "10798    | torch.Size([1, 1000, 754]) | 4\n",
-      "10799    | torch.Size([1, 1000, 754]) | 4\n",
-      "10800    | torch.Size([1, 1000, 754]) | 4\n",
-      "10801    | torch.Size([1, 1000, 754]) | 4\n",
-      "10802    | torch.Size([1, 1000, 768]) | 4\n",
-      "10803    | torch.Size([1, 1000, 754]) | 4\n",
-      "10804    | torch.Size([1, 1000, 783]) | 4\n",
-      "10805    | torch.Size([1, 1000, 754]) | 4\n",
-      "10806    | torch.Size([1, 1000, 780]) | 4\n",
-      "10807    | torch.Size([1, 1000, 754]) | 4\n",
-      "10808    | torch.Size([1, 1000, 754]) | 4\n",
-      "10809    | torch.Size([1, 1000, 790]) | 4\n",
-      "10810    | torch.Size([1, 1000, 777]) | 4\n",
-      "10811    | torch.Size([1, 1000, 754]) | 4\n",
-      "10812    | torch.Size([1, 1000, 777]) | 4\n",
-      "10813    | torch.Size([1, 1000, 762]) | 4\n",
-      "10814    | torch.Size([1, 1000, 754]) | 4\n",
-      "10815    | torch.Size([1, 1000, 754]) | 4\n",
-      "10816    | torch.Size([1, 1000, 762]) | 4\n",
-      "10817    | torch.Size([1, 1000, 778]) | 4\n",
-      "10818    | torch.Size([1, 1000, 773]) | 4\n",
-      "10819    | torch.Size([1, 1000, 777]) | 4\n",
-      "10820    | torch.Size([1, 1000, 777]) | 4\n",
-      "10821    | torch.Size([1, 1000, 777]) | 4\n",
-      "10822    | torch.Size([1, 1000, 754]) | 4\n",
-      "10823    | torch.Size([1, 1000, 777]) | 4\n",
-      "10824    | torch.Size([1, 1000, 781]) | 4\n",
-      "10825    | torch.Size([1, 1000, 754]) | 4\n",
-      "10826    | torch.Size([1, 1000, 786]) | 4\n",
-      "10827    | torch.Size([1, 1000, 754]) | 4\n",
-      "10828    | torch.Size([1, 1000, 754]) | 4\n",
-      "10829    | torch.Size([1, 1000, 754]) | 4\n",
-      "10830    | torch.Size([1, 1000, 754]) | 4\n",
-      "10831    | torch.Size([1, 1000, 778]) | 4\n",
-      "10832    | torch.Size([1, 1000, 754]) | 4\n",
-      "10833    | torch.Size([1, 1000, 762]) | 4\n",
-      "10834    | torch.Size([1, 1000, 777]) | 4\n",
-      "10835    | torch.Size([1, 1000, 778]) | 4\n",
-      "10836    | torch.Size([1, 1000, 771]) | 4\n",
-      "10837    | torch.Size([1, 1000, 754]) | 4\n",
-      "10838    | torch.Size([1, 1000, 754]) | 4\n",
-      "10839    | torch.Size([1, 1000, 754]) | 4\n",
-      "10840    | torch.Size([1, 1000, 754]) | 4\n",
-      "10841    | torch.Size([1, 1000, 754]) | 4\n",
-      "10842    | torch.Size([1, 1000, 762]) | 4\n",
-      "10843    | torch.Size([1, 1000, 754]) | 4\n",
-      "10844    | torch.Size([1, 1000, 762]) | 4\n",
-      "10845    | torch.Size([1, 1000, 754]) | 4\n",
-      "10846    | torch.Size([1, 1000, 777]) | 4\n",
-      "10847    | torch.Size([1, 1000, 754]) | 4\n",
-      "10848    | torch.Size([1, 1000, 754]) | 4\n",
-      "10849    | torch.Size([1, 1000, 754]) | 4\n",
-      "10850    | torch.Size([1, 1000, 775]) | 4\n",
-      "10851    | torch.Size([1, 1000, 754]) | 4\n",
-      "10852    | torch.Size([1, 1000, 865]) | 4\n",
-      "10853    | torch.Size([1, 1000, 754]) | 4\n",
-      "10854    | torch.Size([1, 1000, 754]) | 4\n",
-      "10855    | torch.Size([1, 1000, 777]) | 4\n",
-      "10856    | torch.Size([1, 1000, 754]) | 4\n",
-      "10857    | torch.Size([1, 1000, 791]) | 4\n",
-      "10858    | torch.Size([1, 1000, 777]) | 4\n",
-      "10859    | torch.Size([1, 1000, 754]) | 4\n",
-      "10860    | torch.Size([1, 1000, 777]) | 4\n",
-      "10861    | torch.Size([1, 1000, 787]) | 4\n",
-      "10862    | torch.Size([1, 1000, 754]) | 4\n",
-      "10863    | torch.Size([1, 1000, 754]) | 4\n",
-      "10864    | torch.Size([1, 1000, 783]) | 4\n",
-      "10865    | torch.Size([1, 1000, 754]) | 4\n",
-      "10866    | torch.Size([1, 1000, 754]) | 4\n",
-      "10867    | torch.Size([1, 1000, 754]) | 4\n",
-      "10868    | torch.Size([1, 1000, 754]) | 4\n",
-      "10869    | torch.Size([1, 1000, 754]) | 4\n",
-      "10870    | torch.Size([1, 1000, 777]) | 4\n",
-      "10871    | torch.Size([1, 1000, 754]) | 4\n",
-      "10872    | torch.Size([1, 1000, 754]) | 4\n",
-      "10873    | torch.Size([1, 1000, 762]) | 4\n",
-      "10874    | torch.Size([1, 1000, 754]) | 4\n",
-      "10875    | torch.Size([1, 1000, 785]) | 4\n",
-      "10876    | torch.Size([1, 1000, 868]) | 4\n",
-      "10877    | torch.Size([1, 1000, 754]) | 4\n",
-      "10878    | torch.Size([1, 1000, 783]) | 4\n",
-      "10879    | torch.Size([1, 1000, 754]) | 4\n",
-      "10880    | torch.Size([1, 1000, 754]) | 4\n",
-      "10881    | torch.Size([1, 1000, 754]) | 4\n",
-      "10882    | torch.Size([1, 1000, 754]) | 4\n",
-      "10883    | torch.Size([1, 1000, 777]) | 4\n",
-      "10884    | torch.Size([1, 1000, 754]) | 4\n",
-      "10885    | torch.Size([1, 1000, 771]) | 4\n",
-      "10886    | torch.Size([1, 1000, 777]) | 4\n",
-      "10887    | torch.Size([1, 1000, 754]) | 4\n",
-      "10888    | torch.Size([1, 1000, 770]) | 4\n",
-      "10889    | torch.Size([1, 1000, 754]) | 4\n",
-      "10890    | torch.Size([1, 1000, 754]) | 4\n",
-      "10891    | torch.Size([1, 1000, 777]) | 4\n",
-      "10892    | torch.Size([1, 1000, 793]) | 4\n",
-      "10893    | torch.Size([1, 1000, 754]) | 4\n",
-      "10894    | torch.Size([1, 1000, 785]) | 4\n",
-      "10895    | torch.Size([1, 1000, 762]) | 4\n",
-      "10896    | torch.Size([1, 1000, 754]) | 4\n",
-      "10897    | torch.Size([1, 1000, 754]) | 4\n",
-      "10898    | torch.Size([1, 1000, 754]) | 4\n",
-      "10899    | torch.Size([1, 1000, 754]) | 4\n",
-      "10900    | torch.Size([1, 1000, 780]) | 4\n",
-      "10901    | torch.Size([1, 1000, 777]) | 4\n",
-      "10902    | torch.Size([1, 1000, 864]) | 4\n",
-      "10903    | torch.Size([1, 1000, 745]) | 4\n",
-      "10904    | torch.Size([1, 1000, 754]) | 4\n",
-      "10905    | torch.Size([1, 1000, 784]) | 4\n",
-      "10906    | torch.Size([1, 1000, 754]) | 4\n",
-      "10907    | torch.Size([1, 1000, 754]) | 4\n",
-      "10908    | torch.Size([1, 1000, 777]) | 4\n",
-      "10909    | torch.Size([1, 1000, 754]) | 4\n",
-      "10910    | torch.Size([1, 1000, 777]) | 4\n",
-      "10911    | torch.Size([1, 1000, 777]) | 4\n",
-      "10912    | torch.Size([1, 1000, 789]) | 4\n",
-      "10913    | torch.Size([1, 1000, 759]) | 4\n",
-      "10914    | torch.Size([1, 1000, 777]) | 4\n",
-      "10915    | torch.Size([1, 1000, 783]) | 4\n",
-      "10916    | torch.Size([1, 1000, 754]) | 4\n",
-      "10917    | torch.Size([1, 1000, 762]) | 4\n",
-      "10918    | torch.Size([1, 1000, 782]) | 4\n",
-      "10919    | torch.Size([1, 1000, 754]) | 4\n",
-      "10920    | torch.Size([1, 1000, 777]) | 4\n",
-      "10921    | torch.Size([1, 1000, 754]) | 4\n",
-      "10922    | torch.Size([1, 1000, 754]) | 4\n",
-      "10923    | torch.Size([1, 1000, 762]) | 4\n",
-      "10924    | torch.Size([1, 1000, 754]) | 4\n",
-      "10925    | torch.Size([1, 1000, 777]) | 4\n",
-      "10926    | torch.Size([1, 1000, 754]) | 4\n",
-      "10927    | torch.Size([1, 1000, 777]) | 4\n",
-      "10928    | torch.Size([1, 1000, 778]) | 4\n",
-      "10929    | torch.Size([1, 1000, 786]) | 4\n",
-      "10930    | torch.Size([1, 1000, 777]) | 4\n",
-      "10931    | torch.Size([1, 1000, 754]) | 4\n",
-      "10932    | torch.Size([1, 1000, 762]) | 4\n",
-      "10933    | torch.Size([1, 1000, 795]) | 4\n",
-      "10934    | torch.Size([1, 1000, 754]) | 4\n",
-      "10935    | torch.Size([1, 1000, 762]) | 4\n",
-      "10936    | torch.Size([1, 1000, 780]) | 4\n",
-      "10937    | torch.Size([1, 1000, 754]) | 4\n",
-      "10938    | torch.Size([1, 1000, 625]) | 4\n",
-      "10939    | torch.Size([1, 1000, 754]) | 4\n",
-      "10940    | torch.Size([1, 1000, 777]) | 4\n",
-      "10941    | torch.Size([1, 1000, 754]) | 4\n",
-      "10942    | torch.Size([1, 1000, 754]) | 4\n",
-      "10943    | torch.Size([1, 1000, 754]) | 4\n",
-      "10944    | torch.Size([1, 1000, 762]) | 4\n",
-      "10945    | torch.Size([1, 1000, 754]) | 4\n",
-      "10946    | torch.Size([1, 1000, 868]) | 4\n",
-      "10947    | torch.Size([1, 1000, 754]) | 4\n",
-      "10948    | torch.Size([1, 1000, 770]) | 4\n",
-      "10949    | torch.Size([1, 1000, 754]) | 4\n",
-      "10950    | torch.Size([1, 1000, 754]) | 4\n",
-      "10951    | torch.Size([1, 1000, 611]) | 4\n",
-      "10952    | torch.Size([1, 1000, 754]) | 4\n",
-      "10953    | torch.Size([1, 1000, 811]) | 4\n",
-      "10954    | torch.Size([1, 1000, 802]) | 4\n",
-      "10955    | torch.Size([1, 1000, 754]) | 4\n",
-      "10956    | torch.Size([1, 1000, 754]) | 4\n",
-      "10957    | torch.Size([1, 1000, 754]) | 4\n",
-      "10958    | torch.Size([1, 1000, 754]) | 4\n",
-      "10959    | torch.Size([1, 1000, 772]) | 4\n",
-      "10960    | torch.Size([1, 1000, 794]) | 4\n",
-      "10961    | torch.Size([1, 1000, 777]) | 4\n",
-      "10962    | torch.Size([1, 1000, 798]) | 4\n",
-      "10963    | torch.Size([1, 1000, 754]) | 4\n",
-      "10964    | torch.Size([1, 1000, 777]) | 4\n",
-      "10965    | torch.Size([1, 1000, 754]) | 4\n",
-      "10966    | torch.Size([1, 1000, 754]) | 4\n",
-      "10967    | torch.Size([1, 1000, 754]) | 4\n",
-      "10968    | torch.Size([1, 1000, 754]) | 4\n",
-      "10969    | torch.Size([1, 1000, 777]) | 4\n",
-      "10970    | torch.Size([1, 1000, 754]) | 4\n",
-      "10971    | torch.Size([1, 1000, 779]) | 4\n",
-      "10972    | torch.Size([1, 1000, 754]) | 4\n",
-      "10973    | torch.Size([1, 1000, 754]) | 4\n",
-      "10974    | torch.Size([1, 1000, 780]) | 4\n",
-      "10975    | torch.Size([1, 1000, 754]) | 4\n",
-      "10976    | torch.Size([1, 1000, 777]) | 4\n",
-      "10977    | torch.Size([1, 1000, 754]) | 4\n",
-      "10978    | torch.Size([1, 1000, 769]) | 4\n",
-      "10979    | torch.Size([1, 1000, 733]) | 4\n",
-      "10980    | torch.Size([1, 1000, 762]) | 4\n",
-      "10981    | torch.Size([1, 1000, 784]) | 4\n",
-      "10982    | torch.Size([1, 1000, 778]) | 4\n",
-      "10983    | torch.Size([1, 1000, 754]) | 4\n",
-      "10984    | torch.Size([1, 1000, 754]) | 4\n",
-      "10985    | torch.Size([1, 1000, 754]) | 4\n",
-      "10986    | torch.Size([1, 1000, 777]) | 4\n",
-      "10987    | torch.Size([1, 1000, 754]) | 4\n",
-      "10988    | torch.Size([1, 1000, 785]) | 4\n",
-      "10989    | torch.Size([1, 1000, 754]) | 4\n",
-      "10990    | torch.Size([1, 1000, 778]) | 4\n",
-      "10991    | torch.Size([1, 1000, 754]) | 4\n",
-      "10992    | torch.Size([1, 1000, 789]) | 4\n",
-      "10993    | torch.Size([1, 1000, 754]) | 4\n",
-      "10994    | torch.Size([1, 1000, 754]) | 4\n",
-      "10995    | torch.Size([1, 1000, 762]) | 4\n",
-      "10996    | torch.Size([1, 1000, 754]) | 4\n",
-      "10997    | torch.Size([1, 1000, 754]) | 4\n",
-      "10998    | torch.Size([1, 1000, 754]) | 4\n",
-      "10999    | torch.Size([1, 1000, 754]) | 4\n",
-      "11000    | torch.Size([1, 1000, 762]) | 4\n",
-      "11001    | torch.Size([1, 1000, 777]) | 4\n",
-      "11002    | torch.Size([1, 1000, 754]) | 4\n",
-      "11003    | torch.Size([1, 1000, 754]) | 4\n",
-      "11004    | torch.Size([1, 1000, 777]) | 4\n",
-      "11005    | torch.Size([1, 1000, 754]) | 4\n",
-      "11006    | torch.Size([1, 1000, 754]) | 4\n",
-      "11007    | torch.Size([1, 1000, 754]) | 4\n",
-      "11008    | torch.Size([1, 1000, 754]) | 4\n",
-      "11009    | torch.Size([1, 1000, 754]) | 4\n",
-      "11010    | torch.Size([1, 1000, 788]) | 4\n",
-      "11011    | torch.Size([1, 1000, 795]) | 4\n",
-      "11012    | torch.Size([1, 1000, 754]) | 4\n",
-      "11013    | torch.Size([1, 1000, 777]) | 4\n",
-      "11014    | torch.Size([1, 1000, 754]) | 4\n",
-      "11015    | torch.Size([1, 1000, 754]) | 4\n",
-      "11016    | torch.Size([1, 1000, 754]) | 4\n",
-      "11017    | torch.Size([1, 1000, 754]) | 4\n",
-      "11018    | torch.Size([1, 1000, 754]) | 4\n",
-      "11019    | torch.Size([1, 1000, 754]) | 4\n",
-      "11020    | torch.Size([1, 1000, 754]) | 4\n",
-      "11021    | torch.Size([1, 1000, 754]) | 4\n",
-      "11022    | torch.Size([1, 1000, 754]) | 4\n",
-      "11023    | torch.Size([1, 1000, 789]) | 4\n",
-      "11024    | torch.Size([1, 1000, 777]) | 4\n",
-      "11025    | torch.Size([1, 1000, 730]) | 4\n",
-      "11026    | torch.Size([1, 1000, 754]) | 4\n",
-      "11027    | torch.Size([1, 1000, 777]) | 4\n",
-      "11028    | torch.Size([1, 1000, 795]) | 4\n",
-      "11029    | torch.Size([1, 1000, 777]) | 4\n",
-      "11030    | torch.Size([1, 1000, 754]) | 4\n",
-      "11031    | torch.Size([1, 1000, 754]) | 4\n",
-      "11032    | torch.Size([1, 1000, 781]) | 4\n",
-      "11033    | torch.Size([1, 1000, 754]) | 4\n",
-      "11034    | torch.Size([1, 1000, 777]) | 4\n",
-      "11035    | torch.Size([1, 1000, 781]) | 4\n",
-      "11036    | torch.Size([1, 1000, 754]) | 4\n",
-      "11037    | torch.Size([1, 1000, 754]) | 4\n",
-      "11038    | torch.Size([1, 1000, 777]) | 4\n",
-      "11039    | torch.Size([1, 1000, 782]) | 4\n",
-      "11040    | torch.Size([1, 1000, 754]) | 4\n",
-      "11041    | torch.Size([1, 1000, 771]) | 4\n",
-      "11042    | torch.Size([1, 1000, 754]) | 4\n",
-      "11043    | torch.Size([1, 1000, 777]) | 4\n",
-      "11044    | torch.Size([1, 1000, 754]) | 4\n",
-      "11045    | torch.Size([1, 1000, 777]) | 4\n",
-      "11046    | torch.Size([1, 1000, 754]) | 4\n",
-      "11047    | torch.Size([1, 1000, 751]) | 4\n",
-      "11048    | torch.Size([1, 1000, 754]) | 4\n",
-      "11049    | torch.Size([1, 1000, 802]) | 4\n",
-      "11050    | torch.Size([1, 1000, 777]) | 4\n",
-      "11051    | torch.Size([1, 1000, 754]) | 4\n",
-      "11052    | torch.Size([1, 1000, 777]) | 4\n",
-      "11053    | torch.Size([1, 1000, 777]) | 4\n",
-      "11054    | torch.Size([1, 1000, 762]) | 4\n",
-      "11055    | torch.Size([1, 1000, 836]) | 4\n",
-      "11056    | torch.Size([1, 1000, 754]) | 4\n",
-      "11057    | torch.Size([1, 1000, 754]) | 4\n",
-      "11058    | torch.Size([1, 1000, 780]) | 4\n",
-      "11059    | torch.Size([1, 1000, 754]) | 4\n",
-      "11060    | torch.Size([1, 1000, 754]) | 4\n",
-      "11061    | torch.Size([1, 1000, 762]) | 4\n",
-      "11062    | torch.Size([1, 1000, 754]) | 4\n",
-      "11063    | torch.Size([1, 1000, 783]) | 4\n",
-      "11064    | torch.Size([1, 1000, 754]) | 4\n",
-      "11065    | torch.Size([1, 1000, 754]) | 4\n",
-      "11066    | torch.Size([1, 1000, 779]) | 4\n",
-      "11067    | torch.Size([1, 1000, 754]) | 4\n",
-      "11068    | torch.Size([1, 1000, 754]) | 4\n",
-      "11069    | torch.Size([1, 1000, 754]) | 4\n",
-      "11070    | torch.Size([1, 1000, 777]) | 4\n",
-      "11071    | torch.Size([1, 1000, 766]) | 4\n",
-      "11072    | torch.Size([1, 1000, 782]) | 4\n",
-      "11073    | torch.Size([1, 1000, 773]) | 4\n",
-      "11074    | torch.Size([1, 1000, 754]) | 4\n",
-      "11075    | torch.Size([1, 1000, 754]) | 4\n",
-      "11076    | torch.Size([1, 1000, 754]) | 4\n",
-      "11077    | torch.Size([1, 1000, 754]) | 4\n",
-      "11078    | torch.Size([1, 1000, 754]) | 4\n",
-      "11079    | torch.Size([1, 1000, 754]) | 4\n",
-      "11080    | torch.Size([1, 1000, 781]) | 4\n",
-      "11081    | torch.Size([1, 1000, 777]) | 4\n",
-      "11082    | torch.Size([1, 1000, 754]) | 4\n",
-      "11083    | torch.Size([1, 1000, 754]) | 4\n",
-      "11084    | torch.Size([1, 1000, 754]) | 4\n",
-      "11085    | torch.Size([1, 1000, 754]) | 4\n",
-      "11086    | torch.Size([1, 1000, 754]) | 4\n",
-      "11087    | torch.Size([1, 1000, 754]) | 4\n",
-      "11088    | torch.Size([1, 1000, 777]) | 4\n",
-      "11089    | torch.Size([1, 1000, 754]) | 4\n",
-      "11090    | torch.Size([1, 1000, 754]) | 4\n",
-      "11091    | torch.Size([1, 1000, 755]) | 4\n",
-      "11092    | torch.Size([1, 1000, 754]) | 4\n",
-      "11093    | torch.Size([1, 1000, 754]) | 4\n",
-      "11094    | torch.Size([1, 1000, 794]) | 4\n",
-      "11095    | torch.Size([1, 1000, 754]) | 4\n",
-      "11096    | torch.Size([1, 1000, 754]) | 4\n",
-      "11097    | torch.Size([1, 1000, 788]) | 4\n",
-      "11098    | torch.Size([1, 1000, 754]) | 4\n",
-      "11099    | torch.Size([1, 1000, 777]) | 4\n",
-      "11100    | torch.Size([1, 1000, 779]) | 4\n",
-      "11101    | torch.Size([1, 1000, 754]) | 4\n",
-      "11102    | torch.Size([1, 1000, 777]) | 4\n",
-      "11103    | torch.Size([1, 1000, 754]) | 4\n",
-      "11104    | torch.Size([1, 1000, 778]) | 4\n",
-      "11105    | torch.Size([1, 1000, 754]) | 4\n",
-      "11106    | torch.Size([1, 1000, 754]) | 4\n",
-      "11107    | torch.Size([1, 1000, 792]) | 4\n",
-      "11108    | torch.Size([1, 1000, 754]) | 4\n",
-      "11109    | torch.Size([1, 1000, 777]) | 4\n",
-      "11110    | torch.Size([1, 1000, 754]) | 4\n",
-      "11111    | torch.Size([1, 1000, 771]) | 4\n",
-      "11112    | torch.Size([1, 1000, 788]) | 4\n",
-      "11113    | torch.Size([1, 1000, 757]) | 4\n",
-      "11114    | torch.Size([1, 1000, 762]) | 4\n",
-      "11115    | torch.Size([1, 1000, 779]) | 4\n",
-      "11116    | torch.Size([1, 1000, 802]) | 4\n",
-      "11117    | torch.Size([1, 1000, 754]) | 4\n",
-      "11118    | torch.Size([1, 1000, 802]) | 4\n",
-      "11119    | torch.Size([1, 1000, 754]) | 4\n",
-      "11120    | torch.Size([1, 1000, 783]) | 4\n",
-      "11121    | torch.Size([1, 1000, 754]) | 4\n",
-      "11122    | torch.Size([1, 1000, 779]) | 4\n",
-      "11123    | torch.Size([1, 1000, 774]) | 4\n",
-      "11124    | torch.Size([1, 1000, 754]) | 4\n",
-      "11125    | torch.Size([1, 1000, 754]) | 4\n",
-      "11126    | torch.Size([1, 1000, 771]) | 4\n",
-      "11127    | torch.Size([1, 1000, 777]) | 4\n",
-      "11128    | torch.Size([1, 1000, 754]) | 4\n",
-      "11129    | torch.Size([1, 1000, 754]) | 4\n",
-      "11130    | torch.Size([1, 1000, 782]) | 4\n",
-      "11131    | torch.Size([1, 1000, 754]) | 4\n",
-      "11132    | torch.Size([1, 1000, 788]) | 4\n",
-      "11133    | torch.Size([1, 1000, 754]) | 4\n",
-      "11134    | torch.Size([1, 1000, 777]) | 4\n",
-      "11135    | torch.Size([1, 1000, 774]) | 4\n",
-      "11136    | torch.Size([1, 1000, 754]) | 4\n",
-      "11137    | torch.Size([1, 1000, 754]) | 4\n",
-      "11138    | torch.Size([1, 1000, 762]) | 4\n",
-      "11139    | torch.Size([1, 1000, 774]) | 4\n",
-      "11140    | torch.Size([1, 1000, 785]) | 4\n",
-      "11141    | torch.Size([1, 1000, 754]) | 4\n",
-      "11142    | torch.Size([1, 1000, 754]) | 4\n",
-      "11143    | torch.Size([1, 1000, 754]) | 4\n",
-      "11144    | torch.Size([1, 1000, 751]) | 4\n",
-      "11145    | torch.Size([1, 1000, 754]) | 4\n",
-      "11146    | torch.Size([1, 1000, 778]) | 4\n",
-      "11147    | torch.Size([1, 1000, 794]) | 4\n",
-      "11148    | torch.Size([1, 1000, 778]) | 4\n",
-      "11149    | torch.Size([1, 1000, 754]) | 4\n",
-      "11150    | torch.Size([1, 1000, 754]) | 4\n",
-      "11151    | torch.Size([1, 1000, 846]) | 4\n",
-      "11152    | torch.Size([1, 1000, 779]) | 4\n",
-      "11153    | torch.Size([1, 1000, 754]) | 4\n",
-      "11154    | torch.Size([1, 1000, 808]) | 4\n",
-      "11155    | torch.Size([1, 1000, 754]) | 4\n",
-      "11156    | torch.Size([1, 1000, 780]) | 4\n",
-      "11157    | torch.Size([1, 1000, 754]) | 4\n",
-      "11158    | torch.Size([1, 1000, 754]) | 4\n",
-      "11159    | torch.Size([1, 1000, 777]) | 4\n",
-      "11160    | torch.Size([1, 1000, 777]) | 4\n",
-      "11161    | torch.Size([1, 1000, 792]) | 4\n",
-      "11162    | torch.Size([1, 1000, 754]) | 4\n",
-      "11163    | torch.Size([1, 1000, 754]) | 4\n",
-      "11164    | torch.Size([1, 1000, 754]) | 4\n",
-      "11165    | torch.Size([1, 1000, 754]) | 4\n",
-      "11166    | torch.Size([1, 1000, 754]) | 4\n",
-      "11167    | torch.Size([1, 1000, 754]) | 4\n",
-      "11168    | torch.Size([1, 1000, 782]) | 4\n",
-      "11169    | torch.Size([1, 1000, 754]) | 4\n",
-      "11170    | torch.Size([1, 1000, 775]) | 4\n",
-      "11171    | torch.Size([1, 1000, 777]) | 4\n",
-      "11172    | torch.Size([1, 1000, 789]) | 4\n",
-      "11173    | torch.Size([1, 1000, 754]) | 4\n",
-      "11174    | torch.Size([1, 1000, 780]) | 4\n",
-      "11175    | torch.Size([1, 1000, 805]) | 4\n",
-      "11176    | torch.Size([1, 1000, 754]) | 4\n",
-      "11177    | torch.Size([1, 1000, 754]) | 4\n",
-      "11178    | torch.Size([1, 1000, 754]) | 4\n",
-      "11179    | torch.Size([1, 1000, 754]) | 4\n",
-      "11180    | torch.Size([1, 1000, 777]) | 4\n",
-      "11181    | torch.Size([1, 1000, 764]) | 4\n",
-      "11182    | torch.Size([1, 1000, 764]) | 4\n",
-      "11183    | torch.Size([1, 1000, 786]) | 4\n",
-      "11184    | torch.Size([1, 1000, 777]) | 4\n",
-      "11185    | torch.Size([1, 1000, 754]) | 4\n",
-      "11186    | torch.Size([1, 1000, 786]) | 4\n",
-      "11187    | torch.Size([1, 1000, 777]) | 4\n",
-      "11188    | torch.Size([1, 1000, 754]) | 4\n",
-      "11189    | torch.Size([1, 1000, 806]) | 4\n",
-      "11190    | torch.Size([1, 1000, 754]) | 4\n",
-      "11191    | torch.Size([1, 1000, 754]) | 4\n",
-      "11192    | torch.Size([1, 1000, 790]) | 4\n",
-      "11193    | torch.Size([1, 1000, 754]) | 4\n",
-      "11194    | torch.Size([1, 1000, 852]) | 4\n",
-      "11195    | torch.Size([1, 1000, 767]) | 4\n",
-      "11196    | torch.Size([1, 1000, 717]) | 4\n",
-      "11197    | torch.Size([1, 1000, 754]) | 4\n",
-      "11198    | torch.Size([1, 1000, 762]) | 4\n",
-      "11199    | torch.Size([1, 1000, 754]) | 4\n",
-      "11200    | torch.Size([1, 1000, 754]) | 4\n",
-      "11201    | torch.Size([1, 1000, 777]) | 4\n",
-      "11202    | torch.Size([1, 1000, 787]) | 4\n",
-      "11203    | torch.Size([1, 1000, 754]) | 4\n",
-      "11204    | torch.Size([1, 1000, 754]) | 4\n",
-      "11205    | torch.Size([1, 1000, 754]) | 4\n",
-      "11206    | torch.Size([1, 1000, 781]) | 4\n",
-      "11207    | torch.Size([1, 1000, 754]) | 4\n",
-      "11208    | torch.Size([1, 1000, 784]) | 4\n",
-      "11209    | torch.Size([1, 1000, 803]) | 4\n",
-      "11210    | torch.Size([1, 1000, 802]) | 4\n",
-      "11211    | torch.Size([1, 1000, 754]) | 4\n",
-      "11212    | torch.Size([1, 1000, 777]) | 4\n",
-      "11213    | torch.Size([1, 1000, 754]) | 4\n",
-      "11214    | torch.Size([1, 1000, 754]) | 4\n",
-      "11215    | torch.Size([1, 1000, 754]) | 4\n",
-      "11216    | torch.Size([1, 1000, 777]) | 4\n",
-      "11217    | torch.Size([1, 1000, 754]) | 4\n",
-      "11218    | torch.Size([1, 1000, 777]) | 4\n",
-      "11219    | torch.Size([1, 1000, 754]) | 4\n",
-      "11220    | torch.Size([1, 1000, 754]) | 4\n",
-      "11221    | torch.Size([1, 1000, 754]) | 4\n",
-      "11222    | torch.Size([1, 1000, 754]) | 4\n",
-      "11223    | torch.Size([1, 1000, 754]) | 4\n",
-      "11224    | torch.Size([1, 1000, 754]) | 4\n",
-      "11225    | torch.Size([1, 1000, 777]) | 4\n",
-      "11226    | torch.Size([1, 1000, 779]) | 4\n",
-      "11227    | torch.Size([1, 1000, 754]) | 4\n",
-      "11228    | torch.Size([1, 1000, 754]) | 4\n",
-      "11229    | torch.Size([1, 1000, 754]) | 4\n",
-      "11230    | torch.Size([1, 1000, 759]) | 4\n",
-      "11231    | torch.Size([1, 1000, 779]) | 4\n",
-      "11232    | torch.Size([1, 1000, 754]) | 4\n",
-      "11233    | torch.Size([1, 1000, 786]) | 4\n",
-      "11234    | torch.Size([1, 1000, 789]) | 4\n",
-      "11235    | torch.Size([1, 1000, 754]) | 4\n",
-      "11236    | torch.Size([1, 1000, 754]) | 4\n",
-      "11237    | torch.Size([1, 1000, 802]) | 4\n",
-      "11238    | torch.Size([1, 1000, 754]) | 4\n",
-      "11239    | torch.Size([1, 1000, 754]) | 4\n",
-      "11240    | torch.Size([1, 1000, 779]) | 4\n",
-      "11241    | torch.Size([1, 1000, 754]) | 4\n",
-      "11242    | torch.Size([1, 1000, 754]) | 4\n",
-      "11243    | torch.Size([1, 1000, 754]) | 4\n",
-      "11244    | torch.Size([1, 1000, 754]) | 4\n",
-      "11245    | torch.Size([1, 1000, 754]) | 4\n",
-      "11246    | torch.Size([1, 1000, 754]) | 4\n",
-      "11247    | torch.Size([1, 1000, 802]) | 4\n",
-      "11248    | torch.Size([1, 1000, 777]) | 4\n",
-      "11249    | torch.Size([1, 1000, 802]) | 4\n",
-      "11250    | torch.Size([1, 1000, 754]) | 4\n",
-      "11251    | torch.Size([1, 1000, 754]) | 4\n",
-      "11252    | torch.Size([1, 1000, 778]) | 4\n",
-      "11253    | torch.Size([1, 1000, 778]) | 4\n",
-      "11254    | torch.Size([1, 1000, 792]) | 4\n",
-      "11255    | torch.Size([1, 1000, 754]) | 4\n",
-      "11256    | torch.Size([1, 1000, 754]) | 4\n",
-      "11257    | torch.Size([1, 1000, 754]) | 4\n",
-      "11258    | torch.Size([1, 1000, 777]) | 4\n",
-      "11259    | torch.Size([1, 1000, 762]) | 4\n",
-      "11260    | torch.Size([1, 1000, 754]) | 4\n",
-      "11261    | torch.Size([1, 1000, 754]) | 4\n",
-      "11262    | torch.Size([1, 1000, 795]) | 4\n",
-      "11263    | torch.Size([1, 1000, 762]) | 4\n",
-      "11264    | torch.Size([1, 1000, 754]) | 4\n",
-      "11265    | torch.Size([1, 1000, 777]) | 4\n",
-      "11266    | torch.Size([1, 1000, 777]) | 4\n",
-      "11267    | torch.Size([1, 1000, 754]) | 4\n",
-      "11268    | torch.Size([1, 1000, 779]) | 4\n",
-      "11269    | torch.Size([1, 1000, 754]) | 4\n",
-      "11270    | torch.Size([1, 1000, 754]) | 4\n",
-      "11271    | torch.Size([1, 1000, 754]) | 4\n",
-      "11272    | torch.Size([1, 1000, 785]) | 4\n",
-      "11273    | torch.Size([1, 1000, 783]) | 4\n",
-      "11274    | torch.Size([1, 1000, 754]) | 4\n",
-      "11275    | torch.Size([1, 1000, 774]) | 4\n",
-      "11276    | torch.Size([1, 1000, 777]) | 4\n",
-      "11277    | torch.Size([1, 1000, 754]) | 4\n",
-      "11278    | torch.Size([1, 1000, 754]) | 4\n",
-      "11279    | torch.Size([1, 1000, 754]) | 4\n",
-      "11280    | torch.Size([1, 1000, 754]) | 4\n",
-      "11281    | torch.Size([1, 1000, 777]) | 4\n",
-      "11282    | torch.Size([1, 1000, 777]) | 4\n",
-      "11283    | torch.Size([1, 1000, 843]) | 4\n",
-      "11284    | torch.Size([1, 1000, 762]) | 4\n",
-      "11285    | torch.Size([1, 1000, 754]) | 4\n",
-      "11286    | torch.Size([1, 1000, 770]) | 4\n",
-      "11287    | torch.Size([1, 1000, 777]) | 4\n",
-      "11288    | torch.Size([1, 1000, 754]) | 4\n",
-      "11289    | torch.Size([1, 1000, 754]) | 4\n",
-      "11290    | torch.Size([1, 1000, 777]) | 4\n",
-      "11291    | torch.Size([1, 1000, 783]) | 4\n",
-      "11292    | torch.Size([1, 1000, 777]) | 4\n",
-      "11293    | torch.Size([1, 1000, 754]) | 4\n",
-      "11294    | torch.Size([1, 1000, 754]) | 4\n",
-      "11295    | torch.Size([1, 1000, 796]) | 4\n",
-      "11296    | torch.Size([1, 1000, 754]) | 4\n",
-      "11297    | torch.Size([1, 1000, 781]) | 4\n",
-      "11298    | torch.Size([1, 1000, 754]) | 4\n",
-      "11299    | torch.Size([1, 1000, 754]) | 4\n",
-      "11300    | torch.Size([1, 1000, 754]) | 4\n",
-      "11301    | torch.Size([1, 1000, 754]) | 4\n",
-      "11302    | torch.Size([1, 1000, 762]) | 4\n",
-      "11303    | torch.Size([1, 1000, 777]) | 4\n",
-      "11304    | torch.Size([1, 1000, 754]) | 4\n",
-      "11305    | torch.Size([1, 1000, 754]) | 4\n",
-      "11306    | torch.Size([1, 1000, 754]) | 4\n",
-      "11307    | torch.Size([1, 1000, 754]) | 4\n",
-      "11308    | torch.Size([1, 1000, 754]) | 4\n",
-      "11309    | torch.Size([1, 1000, 754]) | 4\n",
-      "11310    | torch.Size([1, 1000, 754]) | 4\n",
-      "11311    | torch.Size([1, 1000, 725]) | 4\n",
-      "11312    | torch.Size([1, 1000, 754]) | 4\n",
-      "11313    | torch.Size([1, 1000, 785]) | 4\n",
-      "11314    | torch.Size([1, 1000, 774]) | 4\n",
-      "11315    | torch.Size([1, 1000, 754]) | 4\n",
-      "11316    | torch.Size([1, 1000, 754]) | 4\n",
-      "11317    | torch.Size([1, 1000, 754]) | 4\n",
-      "11318    | torch.Size([1, 1000, 611]) | 4\n",
-      "11319    | torch.Size([1, 1000, 783]) | 4\n",
-      "11320    | torch.Size([1, 1000, 754]) | 4\n",
-      "11321    | torch.Size([1, 1000, 781]) | 4\n",
-      "11322    | torch.Size([1, 1000, 754]) | 4\n",
-      "11323    | torch.Size([1, 1000, 754]) | 4\n",
-      "11324    | torch.Size([1, 1000, 754]) | 4\n",
-      "11325    | torch.Size([1, 1000, 754]) | 4\n",
-      "11326    | torch.Size([1, 1000, 754]) | 4\n",
-      "11327    | torch.Size([1, 1000, 777]) | 4\n",
-      "11328    | torch.Size([1, 1000, 754]) | 4\n",
-      "11329    | torch.Size([1, 1000, 777]) | 4\n",
-      "11330    | torch.Size([1, 1000, 754]) | 4\n",
-      "11331    | torch.Size([1, 1000, 792]) | 4\n",
-      "11332    | torch.Size([1, 1000, 658]) | 4\n",
-      "11333    | torch.Size([1, 1000, 754]) | 4\n",
-      "11334    | torch.Size([1, 1000, 787]) | 4\n",
-      "11335    | torch.Size([1, 1000, 802]) | 4\n",
-      "11336    | torch.Size([1, 1000, 754]) | 4\n",
-      "11337    | torch.Size([1, 1000, 796]) | 4\n",
-      "11338    | torch.Size([1, 1000, 773]) | 4\n",
-      "11339    | torch.Size([1, 1000, 754]) | 4\n",
-      "11340    | torch.Size([1, 1000, 754]) | 4\n",
-      "11341    | torch.Size([1, 1000, 777]) | 4\n",
-      "11342    | torch.Size([1, 1000, 754]) | 4\n",
-      "11343    | torch.Size([1, 1000, 754]) | 4\n",
-      "11344    | torch.Size([1, 1000, 754]) | 4\n",
-      "11345    | torch.Size([1, 1000, 777]) | 4\n",
-      "11346    | torch.Size([1, 1000, 754]) | 4\n",
-      "11347    | torch.Size([1, 1000, 812]) | 4\n",
-      "11348    | torch.Size([1, 1000, 771]) | 4\n",
-      "11349    | torch.Size([1, 1000, 754]) | 4\n",
-      "11350    | torch.Size([1, 1000, 777]) | 4\n",
-      "11351    | torch.Size([1, 1000, 754]) | 4\n",
-      "11352    | torch.Size([1, 1000, 754]) | 4\n",
-      "11353    | torch.Size([1, 1000, 776]) | 4\n",
-      "11354    | torch.Size([1, 1000, 754]) | 4\n",
-      "11355    | torch.Size([1, 1000, 754]) | 4\n",
-      "11356    | torch.Size([1, 1000, 754]) | 4\n",
-      "11357    | torch.Size([1, 1000, 754]) | 4\n",
-      "11358    | torch.Size([1, 1000, 754]) | 4\n",
-      "11359    | torch.Size([1, 1000, 777]) | 4\n",
-      "11360    | torch.Size([1, 1000, 754]) | 4\n",
-      "11361    | torch.Size([1, 1000, 773]) | 4\n",
-      "11362    | torch.Size([1, 1000, 754]) | 4\n",
-      "11363    | torch.Size([1, 1000, 754]) | 4\n",
-      "11364    | torch.Size([1, 1000, 754]) | 4\n",
-      "11365    | torch.Size([1, 1000, 781]) | 4\n",
-      "11366    | torch.Size([1, 1000, 754]) | 4\n",
-      "11367    | torch.Size([1, 1000, 781]) | 4\n",
-      "11368    | torch.Size([1, 1000, 754]) | 4\n",
-      "11369    | torch.Size([1, 1000, 762]) | 4\n",
-      "11370    | torch.Size([1, 1000, 777]) | 4\n",
-      "11371    | torch.Size([1, 1000, 802]) | 4\n",
-      "11372    | torch.Size([1, 1000, 754]) | 4\n",
-      "11373    | torch.Size([1, 1000, 771]) | 4\n",
-      "11374    | torch.Size([1, 1000, 754]) | 4\n",
-      "11375    | torch.Size([1, 1000, 754]) | 4\n",
-      "11376    | torch.Size([1, 1000, 740]) | 4\n",
-      "11377    | torch.Size([1, 1000, 754]) | 4\n",
-      "11378    | torch.Size([1, 1000, 754]) | 4\n",
-      "11379    | torch.Size([1, 1000, 754]) | 4\n",
-      "11380    | torch.Size([1, 1000, 754]) | 4\n",
-      "11381    | torch.Size([1, 1000, 754]) | 4\n",
-      "11382    | torch.Size([1, 1000, 754]) | 4\n",
-      "11383    | torch.Size([1, 1000, 777]) | 4\n",
-      "11384    | torch.Size([1, 1000, 787]) | 4\n",
-      "11385    | torch.Size([1, 1000, 777]) | 4\n",
-      "11386    | torch.Size([1, 1000, 793]) | 4\n",
-      "11387    | torch.Size([1, 1000, 754]) | 4\n",
-      "11388    | torch.Size([1, 1000, 777]) | 4\n",
-      "11389    | torch.Size([1, 1000, 784]) | 4\n",
-      "11390    | torch.Size([1, 1000, 754]) | 4\n",
-      "11391    | torch.Size([1, 1000, 754]) | 4\n",
-      "11392    | torch.Size([1, 1000, 788]) | 4\n",
-      "11393    | torch.Size([1, 1000, 782]) | 4\n",
-      "11394    | torch.Size([1, 1000, 754]) | 4\n",
-      "11395    | torch.Size([1, 1000, 781]) | 4\n",
-      "11396    | torch.Size([1, 1000, 754]) | 4\n",
-      "11397    | torch.Size([1, 1000, 754]) | 4\n",
-      "11398    | torch.Size([1, 1000, 775]) | 4\n",
-      "11399    | torch.Size([1, 1000, 754]) | 4\n",
-      "11400    | torch.Size([1, 1000, 764]) | 4\n",
-      "11401    | torch.Size([1, 1000, 754]) | 4\n",
-      "11402    | torch.Size([1, 1000, 754]) | 4\n",
-      "11403    | torch.Size([1, 1000, 754]) | 4\n",
-      "11404    | torch.Size([1, 1000, 786]) | 4\n",
-      "11405    | torch.Size([1, 1000, 754]) | 4\n",
-      "11406    | torch.Size([1, 1000, 754]) | 4\n",
-      "11407    | torch.Size([1, 1000, 754]) | 4\n",
-      "11408    | torch.Size([1, 1000, 777]) | 4\n",
-      "11409    | torch.Size([1, 1000, 754]) | 4\n",
-      "11410    | torch.Size([1, 1000, 795]) | 4\n",
-      "11411    | torch.Size([1, 1000, 754]) | 4\n",
-      "11412    | torch.Size([1, 1000, 754]) | 4\n",
-      "11413    | torch.Size([1, 1000, 754]) | 4\n",
-      "11414    | torch.Size([1, 1000, 754]) | 4\n",
-      "11415    | torch.Size([1, 1000, 754]) | 4\n",
-      "11416    | torch.Size([1, 1000, 754]) | 4\n",
-      "11417    | torch.Size([1, 1000, 713]) | 4\n",
-      "11418    | torch.Size([1, 1000, 754]) | 4\n",
-      "11419    | torch.Size([1, 1000, 794]) | 4\n",
-      "11420    | torch.Size([1, 1000, 762]) | 4\n",
-      "11421    | torch.Size([1, 1000, 754]) | 4\n",
-      "11422    | torch.Size([1, 1000, 778]) | 4\n",
-      "11423    | torch.Size([1, 1000, 762]) | 4\n",
-      "11424    | torch.Size([1, 1000, 781]) | 4\n",
-      "11425    | torch.Size([1, 1000, 754]) | 4\n",
-      "11426    | torch.Size([1, 1000, 754]) | 4\n",
-      "11427    | torch.Size([1, 1000, 754]) | 4\n",
-      "11428    | torch.Size([1, 1000, 781]) | 4\n",
-      "11429    | torch.Size([1, 1000, 754]) | 4\n",
-      "11430    | torch.Size([1, 1000, 782]) | 4\n",
-      "11431    | torch.Size([1, 1000, 764]) | 4\n",
-      "11432    | torch.Size([1, 1000, 754]) | 4\n",
-      "11433    | torch.Size([1, 1000, 754]) | 4\n",
-      "11434    | torch.Size([1, 1000, 802]) | 4\n",
-      "11435    | torch.Size([1, 1000, 754]) | 4\n",
-      "11436    | torch.Size([1, 1000, 762]) | 4\n",
-      "11437    | torch.Size([1, 1000, 754]) | 4\n",
-      "11438    | torch.Size([1, 1000, 754]) | 4\n",
-      "11439    | torch.Size([1, 1000, 771]) | 4\n",
-      "11440    | torch.Size([1, 1000, 754]) | 4\n",
-      "11441    | torch.Size([1, 1000, 754]) | 4\n",
-      "11442    | torch.Size([1, 1000, 762]) | 4\n",
-      "11443    | torch.Size([1, 1000, 779]) | 4\n",
-      "11444    | torch.Size([1, 1000, 754]) | 4\n",
-      "11445    | torch.Size([1, 1000, 754]) | 4\n",
-      "11446    | torch.Size([1, 1000, 754]) | 4\n",
-      "11447    | torch.Size([1, 1000, 777]) | 4\n",
-      "11448    | torch.Size([1, 1000, 777]) | 4\n",
-      "11449    | torch.Size([1, 1000, 754]) | 4\n",
-      "11450    | torch.Size([1, 1000, 754]) | 4\n",
-      "11451    | torch.Size([1, 1000, 802]) | 4\n",
-      "11452    | torch.Size([1, 1000, 754]) | 4\n",
-      "11453    | torch.Size([1, 1000, 754]) | 4\n",
-      "11454    | torch.Size([1, 1000, 754]) | 4\n",
-      "11455    | torch.Size([1, 1000, 794]) | 4\n",
-      "11456    | torch.Size([1, 1000, 754]) | 4\n",
-      "11457    | torch.Size([1, 1000, 786]) | 4\n",
-      "11458    | torch.Size([1, 1000, 754]) | 4\n",
-      "11459    | torch.Size([1, 1000, 771]) | 4\n",
-      "11460    | torch.Size([1, 1000, 783]) | 4\n",
-      "11461    | torch.Size([1, 1000, 803]) | 4\n",
-      "11462    | torch.Size([1, 1000, 782]) | 4\n",
-      "11463    | torch.Size([1, 1000, 762]) | 4\n",
-      "11464    | torch.Size([1, 1000, 760]) | 4\n",
-      "11465    | torch.Size([1, 1000, 754]) | 4\n",
-      "11466    | torch.Size([1, 1000, 802]) | 4\n",
-      "11467    | torch.Size([1, 1000, 780]) | 4\n",
-      "11468    | torch.Size([1, 1000, 754]) | 4\n",
-      "11469    | torch.Size([1, 1000, 754]) | 4\n",
-      "11470    | torch.Size([1, 1000, 754]) | 4\n",
-      "11471    | torch.Size([1, 1000, 754]) | 4\n",
-      "11472    | torch.Size([1, 1000, 771]) | 4\n",
-      "11473    | torch.Size([1, 1000, 754]) | 4\n",
-      "11474    | torch.Size([1, 1000, 754]) | 4\n",
-      "11475    | torch.Size([1, 1000, 754]) | 4\n",
-      "11476    | torch.Size([1, 1000, 777]) | 4\n",
-      "11477    | torch.Size([1, 1000, 754]) | 4\n",
-      "11478    | torch.Size([1, 1000, 754]) | 4\n",
-      "11479    | torch.Size([1, 1000, 754]) | 4\n",
-      "11480    | torch.Size([1, 1000, 754]) | 4\n",
-      "11481    | torch.Size([1, 1000, 754]) | 4\n",
-      "11482    | torch.Size([1, 1000, 754]) | 4\n",
-      "11483    | torch.Size([1, 1000, 776]) | 4\n",
-      "11484    | torch.Size([1, 1000, 778]) | 4\n",
-      "11485    | torch.Size([1, 1000, 754]) | 4\n",
-      "11486    | torch.Size([1, 1000, 754]) | 4\n",
-      "11487    | torch.Size([1, 1000, 754]) | 4\n",
-      "11488    | torch.Size([1, 1000, 645]) | 4\n",
-      "11489    | torch.Size([1, 1000, 776]) | 4\n",
-      "11490    | torch.Size([1, 1000, 754]) | 4\n",
-      "11491    | torch.Size([1, 1000, 762]) | 4\n",
-      "11492    | torch.Size([1, 1000, 754]) | 4\n",
-      "11493    | torch.Size([1, 1000, 777]) | 4\n",
-      "11494    | torch.Size([1, 1000, 794]) | 4\n",
-      "11495    | torch.Size([1, 1000, 754]) | 4\n",
-      "11496    | torch.Size([1, 1000, 777]) | 4\n",
-      "11497    | torch.Size([1, 1000, 762]) | 4\n",
-      "11498    | torch.Size([1, 1000, 765]) | 4\n",
-      "11499    | torch.Size([1, 1000, 754]) | 4\n",
-      "11500    | torch.Size([1, 1000, 775]) | 4\n",
-      "11501    | torch.Size([1, 1000, 754]) | 4\n",
-      "11502    | torch.Size([1, 1000, 754]) | 4\n",
-      "11503    | torch.Size([1, 1000, 754]) | 4\n",
-      "11504    | torch.Size([1, 1000, 754]) | 4\n",
-      "11505    | torch.Size([1, 1000, 782]) | 4\n",
-      "11506    | torch.Size([1, 1000, 777]) | 4\n",
-      "11507    | torch.Size([1, 1000, 754]) | 4\n",
-      "11508    | torch.Size([1, 1000, 754]) | 4\n",
-      "11509    | torch.Size([1, 1000, 779]) | 4\n",
-      "11510    | torch.Size([1, 1000, 754]) | 4\n",
-      "11511    | torch.Size([1, 1000, 781]) | 4\n",
-      "11512    | torch.Size([1, 1000, 782]) | 4\n",
-      "11513    | torch.Size([1, 1000, 754]) | 4\n",
-      "11514    | torch.Size([1, 1000, 771]) | 4\n",
-      "11515    | torch.Size([1, 1000, 787]) | 4\n",
-      "11516    | torch.Size([1, 1000, 754]) | 4\n",
-      "11517    | torch.Size([1, 1000, 754]) | 4\n",
-      "11518    | torch.Size([1, 1000, 777]) | 4\n",
-      "11519    | torch.Size([1, 1000, 754]) | 4\n",
-      "11520    | torch.Size([1, 1000, 754]) | 4\n",
-      "11521    | torch.Size([1, 1000, 754]) | 4\n",
-      "11522    | torch.Size([1, 1000, 754]) | 4\n",
-      "11523    | torch.Size([1, 1000, 754]) | 4\n",
-      "11524    | torch.Size([1, 1000, 754]) | 4\n",
-      "11525    | torch.Size([1, 1000, 867]) | 4\n",
-      "11526    | torch.Size([1, 1000, 754]) | 4\n",
-      "11527    | torch.Size([1, 1000, 754]) | 4\n",
-      "11528    | torch.Size([1, 1000, 798]) | 4\n",
-      "11529    | torch.Size([1, 1000, 754]) | 4\n",
-      "11530    | torch.Size([1, 1000, 754]) | 4\n",
-      "11531    | torch.Size([1, 1000, 777]) | 4\n",
-      "11532    | torch.Size([1, 1000, 754]) | 4\n",
-      "11533    | torch.Size([1, 1000, 754]) | 4\n",
-      "11534    | torch.Size([1, 1000, 782]) | 4\n",
-      "11535    | torch.Size([1, 1000, 772]) | 4\n",
-      "11536    | torch.Size([1, 1000, 754]) | 4\n",
-      "11537    | torch.Size([1, 1000, 754]) | 4\n",
-      "11538    | torch.Size([1, 1000, 777]) | 4\n",
-      "11539    | torch.Size([1, 1000, 754]) | 4\n",
-      "11540    | torch.Size([1, 1000, 754]) | 4\n",
-      "11541    | torch.Size([1, 1000, 754]) | 4\n",
-      "11542    | torch.Size([1, 1000, 754]) | 4\n",
-      "11543    | torch.Size([1, 1000, 812]) | 4\n",
-      "11544    | torch.Size([1, 1000, 762]) | 4\n",
-      "11545    | torch.Size([1, 1000, 754]) | 4\n",
-      "11546    | torch.Size([1, 1000, 754]) | 4\n",
-      "11547    | torch.Size([1, 1000, 754]) | 4\n",
-      "11548    | torch.Size([1, 1000, 843]) | 4\n",
-      "11549    | torch.Size([1, 1000, 754]) | 4\n",
-      "11550    | torch.Size([1, 1000, 762]) | 4\n",
-      "11551    | torch.Size([1, 1000, 725]) | 4\n",
-      "11552    | torch.Size([1, 1000, 791]) | 4\n",
-      "11553    | torch.Size([1, 1000, 754]) | 4\n",
-      "11554    | torch.Size([1, 1000, 754]) | 4\n",
-      "11555    | torch.Size([1, 1000, 754]) | 4\n",
-      "11556    | torch.Size([1, 1000, 802]) | 4\n",
-      "11557    | torch.Size([1, 1000, 785]) | 4\n",
-      "11558    | torch.Size([1, 1000, 754]) | 4\n",
-      "11559    | torch.Size([1, 1000, 754]) | 4\n",
-      "11560    | torch.Size([1, 1000, 754]) | 4\n",
-      "11561    | torch.Size([1, 1000, 754]) | 4\n",
-      "11562    | torch.Size([1, 1000, 754]) | 4\n",
-      "11563    | torch.Size([1, 1000, 754]) | 4\n",
-      "11564    | torch.Size([1, 1000, 754]) | 4\n",
-      "11565    | torch.Size([1, 1000, 754]) | 4\n",
-      "11566    | torch.Size([1, 1000, 754]) | 4\n",
-      "11567    | torch.Size([1, 1000, 754]) | 4\n",
-      "11568    | torch.Size([1, 1000, 754]) | 4\n",
-      "11569    | torch.Size([1, 1000, 754]) | 4\n",
-      "11570    | torch.Size([1, 1000, 777]) | 4\n",
-      "11571    | torch.Size([1, 1000, 754]) | 4\n",
-      "11572    | torch.Size([1, 1000, 792]) | 4\n",
-      "11573    | torch.Size([1, 1000, 794]) | 4\n",
-      "11574    | torch.Size([1, 1000, 754]) | 4\n",
-      "11575    | torch.Size([1, 1000, 754]) | 4\n",
-      "11576    | torch.Size([1, 1000, 754]) | 4\n",
-      "11577    | torch.Size([1, 1000, 754]) | 4\n",
-      "11578    | torch.Size([1, 1000, 754]) | 4\n",
-      "11579    | torch.Size([1, 1000, 754]) | 4\n",
-      "11580    | torch.Size([1, 1000, 773]) | 4\n",
-      "11581    | torch.Size([1, 1000, 789]) | 4\n",
-      "11582    | torch.Size([1, 1000, 754]) | 4\n",
-      "11583    | torch.Size([1, 1000, 777]) | 4\n",
-      "11584    | torch.Size([1, 1000, 754]) | 4\n",
-      "11585    | torch.Size([1, 1000, 779]) | 4\n",
-      "11586    | torch.Size([1, 1000, 793]) | 4\n",
-      "11587    | torch.Size([1, 1000, 754]) | 4\n",
-      "11588    | torch.Size([1, 1000, 754]) | 4\n",
-      "11589    | torch.Size([1, 1000, 754]) | 4\n",
-      "11590    | torch.Size([1, 1000, 844]) | 4\n",
-      "11591    | torch.Size([1, 1000, 762]) | 4\n",
-      "11592    | torch.Size([1, 1000, 754]) | 4\n",
-      "11593    | torch.Size([1, 1000, 754]) | 4\n",
-      "11594    | torch.Size([1, 1000, 754]) | 4\n",
-      "11595    | torch.Size([1, 1000, 777]) | 4\n",
-      "11596    | torch.Size([1, 1000, 754]) | 4\n",
-      "11597    | torch.Size([1, 1000, 754]) | 4\n",
-      "11598    | torch.Size([1, 1000, 754]) | 4\n",
-      "11599    | torch.Size([1, 1000, 754]) | 4\n",
-      "11600    | torch.Size([1, 1000, 777]) | 4\n",
-      "11601    | torch.Size([1, 1000, 754]) | 4\n",
-      "11602    | torch.Size([1, 1000, 762]) | 4\n",
-      "11603    | torch.Size([1, 1000, 754]) | 4\n",
-      "11604    | torch.Size([1, 1000, 778]) | 4\n",
-      "11605    | torch.Size([1, 1000, 754]) | 4\n",
-      "11606    | torch.Size([1, 1000, 754]) | 4\n",
-      "11607    | torch.Size([1, 1000, 783]) | 4\n",
-      "11608    | torch.Size([1, 1000, 754]) | 4\n",
-      "11609    | torch.Size([1, 1000, 777]) | 4\n",
-      "11610    | torch.Size([1, 1000, 754]) | 4\n",
-      "11611    | torch.Size([1, 1000, 754]) | 4\n",
-      "11612    | torch.Size([1, 1000, 713]) | 4\n",
-      "11613    | torch.Size([1, 1000, 754]) | 4\n",
-      "11614    | torch.Size([1, 1000, 754]) | 4\n",
-      "11615    | torch.Size([1, 1000, 777]) | 4\n",
-      "11616    | torch.Size([1, 1000, 770]) | 4\n",
-      "11617    | torch.Size([1, 1000, 777]) | 4\n",
-      "11618    | torch.Size([1, 1000, 754]) | 4\n",
-      "11619    | torch.Size([1, 1000, 780]) | 4\n",
-      "11620    | torch.Size([1, 1000, 762]) | 4\n",
-      "11621    | torch.Size([1, 1000, 771]) | 4\n",
-      "11622    | torch.Size([1, 1000, 754]) | 4\n",
-      "11623    | torch.Size([1, 1000, 773]) | 4\n",
-      "11624    | torch.Size([1, 1000, 754]) | 4\n",
-      "11625    | torch.Size([1, 1000, 754]) | 4\n",
-      "11626    | torch.Size([1, 1000, 754]) | 4\n",
-      "11627    | torch.Size([1, 1000, 754]) | 4\n",
-      "11628    | torch.Size([1, 1000, 754]) | 4\n",
-      "11629    | torch.Size([1, 1000, 754]) | 4\n",
-      "11630    | torch.Size([1, 1000, 754]) | 4\n",
-      "11631    | torch.Size([1, 1000, 754]) | 4\n",
-      "11632    | torch.Size([1, 1000, 780]) | 4\n",
-      "11633    | torch.Size([1, 1000, 773]) | 4\n",
-      "11634    | torch.Size([1, 1000, 754]) | 4\n",
-      "11635    | torch.Size([1, 1000, 762]) | 4\n",
-      "11636    | torch.Size([1, 1000, 777]) | 4\n",
-      "11637    | torch.Size([1, 1000, 718]) | 4\n",
-      "11638    | torch.Size([1, 1000, 754]) | 4\n",
-      "11639    | torch.Size([1, 1000, 754]) | 4\n",
-      "11640    | torch.Size([1, 1000, 754]) | 4\n",
-      "11641    | torch.Size([1, 1000, 754]) | 4\n",
-      "11642    | torch.Size([1, 1000, 754]) | 4\n",
-      "11643    | torch.Size([1, 1000, 754]) | 4\n",
-      "11644    | torch.Size([1, 1000, 788]) | 4\n",
-      "11645    | torch.Size([1, 1000, 777]) | 4\n",
-      "11646    | torch.Size([1, 1000, 754]) | 4\n",
-      "11647    | torch.Size([1, 1000, 754]) | 4\n",
-      "11648    | torch.Size([1, 1000, 773]) | 4\n",
-      "11649    | torch.Size([1, 1000, 754]) | 4\n",
-      "11650    | torch.Size([1, 1000, 802]) | 4\n",
-      "11651    | torch.Size([1, 1000, 754]) | 4\n",
-      "11652    | torch.Size([1, 1000, 777]) | 4\n",
-      "11653    | torch.Size([1, 1000, 824]) | 4\n",
-      "11654    | torch.Size([1, 1000, 754]) | 4\n",
-      "11655    | torch.Size([1, 1000, 754]) | 4\n",
-      "11656    | torch.Size([1, 1000, 611]) | 4\n",
-      "11657    | torch.Size([1, 1000, 762]) | 4\n",
-      "11658    | torch.Size([1, 1000, 754]) | 4\n",
-      "11659    | torch.Size([1, 1000, 777]) | 4\n",
-      "11660    | torch.Size([1, 1000, 754]) | 4\n",
-      "11661    | torch.Size([1, 1000, 754]) | 4\n",
-      "11662    | torch.Size([1, 1000, 754]) | 4\n",
-      "11663    | torch.Size([1, 1000, 777]) | 4\n",
-      "11664    | torch.Size([1, 1000, 754]) | 4\n",
-      "11665    | torch.Size([1, 1000, 754]) | 4\n",
-      "11666    | torch.Size([1, 1000, 756]) | 4\n",
-      "11667    | torch.Size([1, 1000, 754]) | 4\n",
-      "11668    | torch.Size([1, 1000, 778]) | 4\n",
-      "11669    | torch.Size([1, 1000, 723]) | 4\n",
-      "11670    | torch.Size([1, 1000, 754]) | 4\n",
-      "11671    | torch.Size([1, 1000, 754]) | 4\n",
-      "11672    | torch.Size([1, 1000, 757]) | 4\n",
-      "11673    | torch.Size([1, 1000, 802]) | 4\n",
-      "11674    | torch.Size([1, 1000, 754]) | 4\n",
-      "11675    | torch.Size([1, 1000, 754]) | 4\n",
-      "11676    | torch.Size([1, 1000, 864]) | 4\n",
-      "11677    | torch.Size([1, 1000, 777]) | 4\n",
-      "11678    | torch.Size([1, 1000, 783]) | 4\n",
-      "11679    | torch.Size([1, 1000, 754]) | 4\n",
-      "11680    | torch.Size([1, 1000, 754]) | 4\n",
-      "11681    | torch.Size([1, 1000, 773]) | 4\n",
-      "11682    | torch.Size([1, 1000, 802]) | 4\n",
-      "11683    | torch.Size([1, 1000, 754]) | 4\n",
-      "11684    | torch.Size([1, 1000, 781]) | 4\n",
-      "11685    | torch.Size([1, 1000, 776]) | 4\n",
-      "11686    | torch.Size([1, 1000, 789]) | 4\n",
-      "11687    | torch.Size([1, 1000, 754]) | 4\n",
-      "11688    | torch.Size([1, 1000, 769]) | 4\n",
-      "11689    | torch.Size([1, 1000, 802]) | 4\n",
-      "11690    | torch.Size([1, 1000, 754]) | 4\n",
-      "11691    | torch.Size([1, 1000, 754]) | 4\n",
-      "11692    | torch.Size([1, 1000, 777]) | 4\n",
-      "11693    | torch.Size([1, 1000, 713]) | 4\n",
-      "11694    | torch.Size([1, 1000, 777]) | 4\n",
-      "11695    | torch.Size([1, 1000, 754]) | 4\n",
-      "11696    | torch.Size([1, 1000, 754]) | 4\n",
-      "11697    | torch.Size([1, 1000, 754]) | 4\n",
-      "11698    | torch.Size([1, 1000, 754]) | 4\n",
-      "11699    | torch.Size([1, 1000, 754]) | 4\n",
-      "11700    | torch.Size([1, 1000, 754]) | 4\n",
-      "11701    | torch.Size([1, 1000, 754]) | 4\n",
-      "11702    | torch.Size([1, 1000, 754]) | 4\n",
-      "11703    | torch.Size([1, 1000, 754]) | 4\n",
-      "11704    | torch.Size([1, 1000, 754]) | 4\n",
-      "11705    | torch.Size([1, 1000, 783]) | 4\n",
-      "11706    | torch.Size([1, 1000, 766]) | 4\n",
-      "11707    | torch.Size([1, 1000, 754]) | 4\n",
-      "11708    | torch.Size([1, 1000, 754]) | 4\n",
-      "11709    | torch.Size([1, 1000, 754]) | 4\n",
-      "11710    | torch.Size([1, 1000, 779]) | 4\n",
-      "11711    | torch.Size([1, 1000, 754]) | 4\n",
-      "11712    | torch.Size([1, 1000, 778]) | 4\n",
-      "11713    | torch.Size([1, 1000, 754]) | 4\n",
-      "11714    | torch.Size([1, 1000, 754]) | 4\n",
-      "11715    | torch.Size([1, 1000, 777]) | 4\n",
-      "11716    | torch.Size([1, 1000, 754]) | 4\n",
-      "11717    | torch.Size([1, 1000, 767]) | 4\n",
-      "11718    | torch.Size([1, 1000, 754]) | 4\n",
-      "11719    | torch.Size([1, 1000, 777]) | 4\n",
-      "11720    | torch.Size([1, 1000, 611]) | 4\n",
-      "11721    | torch.Size([1, 1000, 754]) | 4\n",
-      "11722    | torch.Size([1, 1000, 754]) | 4\n",
-      "11723    | torch.Size([1, 1000, 769]) | 4\n",
-      "11724    | torch.Size([1, 1000, 789]) | 4\n",
-      "11725    | torch.Size([1, 1000, 772]) | 4\n",
-      "11726    | torch.Size([1, 1000, 611]) | 4\n",
-      "11727    | torch.Size([1, 1000, 802]) | 4\n",
-      "11728    | torch.Size([1, 1000, 754]) | 4\n",
-      "11729    | torch.Size([1, 1000, 785]) | 4\n",
-      "11730    | torch.Size([1, 1000, 762]) | 4\n",
-      "11731    | torch.Size([1, 1000, 754]) | 4\n",
-      "11732    | torch.Size([1, 1000, 754]) | 4\n",
-      "11733    | torch.Size([1, 1000, 777]) | 4\n",
-      "11734    | torch.Size([1, 1000, 777]) | 4\n",
-      "11735    | torch.Size([1, 1000, 754]) | 4\n",
-      "11736    | torch.Size([1, 1000, 754]) | 4\n",
-      "11737    | torch.Size([1, 1000, 754]) | 4\n",
-      "11738    | torch.Size([1, 1000, 754]) | 4\n",
-      "11739    | torch.Size([1, 1000, 754]) | 4\n",
-      "11740    | torch.Size([1, 1000, 762]) | 4\n",
-      "11741    | torch.Size([1, 1000, 777]) | 4\n",
-      "11742    | torch.Size([1, 1000, 754]) | 4\n",
-      "11743    | torch.Size([1, 1000, 754]) | 4\n",
-      "11744    | torch.Size([1, 1000, 754]) | 4\n",
-      "11745    | torch.Size([1, 1000, 780]) | 4\n",
-      "11746    | torch.Size([1, 1000, 777]) | 4\n",
-      "11747    | torch.Size([1, 1000, 754]) | 4\n",
-      "11748    | torch.Size([1, 1000, 805]) | 4\n",
-      "11749    | torch.Size([1, 1000, 778]) | 4\n",
-      "11750    | torch.Size([1, 1000, 777]) | 4\n",
-      "11751    | torch.Size([1, 1000, 754]) | 4\n",
-      "11752    | torch.Size([1, 1000, 754]) | 4\n",
-      "11753    | torch.Size([1, 1000, 754]) | 4\n",
-      "11754    | torch.Size([1, 1000, 754]) | 4\n",
-      "11755    | torch.Size([1, 1000, 754]) | 4\n",
-      "11756    | torch.Size([1, 1000, 771]) | 4\n",
-      "11757    | torch.Size([1, 1000, 803]) | 4\n",
-      "11758    | torch.Size([1, 1000, 754]) | 4\n",
-      "11759    | torch.Size([1, 1000, 754]) | 4\n",
-      "11760    | torch.Size([1, 1000, 777]) | 4\n",
-      "11761    | torch.Size([1, 1000, 781]) | 4\n",
-      "11762    | torch.Size([1, 1000, 732]) | 4\n",
-      "11763    | torch.Size([1, 1000, 754]) | 4\n",
-      "11764    | torch.Size([1, 1000, 754]) | 4\n",
-      "11765    | torch.Size([1, 1000, 754]) | 4\n",
-      "11766    | torch.Size([1, 1000, 762]) | 4\n",
-      "11767    | torch.Size([1, 1000, 754]) | 4\n",
-      "11768    | torch.Size([1, 1000, 777]) | 4\n",
-      "11769    | torch.Size([1, 1000, 754]) | 4\n",
-      "11770    | torch.Size([1, 1000, 843]) | 4\n",
-      "11771    | torch.Size([1, 1000, 754]) | 4\n",
-      "11772    | torch.Size([1, 1000, 809]) | 4\n",
-      "11773    | torch.Size([1, 1000, 754]) | 4\n",
-      "11774    | torch.Size([1, 1000, 766]) | 4\n",
-      "11775    | torch.Size([1, 1000, 754]) | 4\n",
-      "11776    | torch.Size([1, 1000, 773]) | 4\n",
-      "11777    | torch.Size([1, 1000, 754]) | 4\n",
-      "11778    | torch.Size([1, 1000, 782]) | 4\n",
-      "11779    | torch.Size([1, 1000, 777]) | 4\n",
-      "11780    | torch.Size([1, 1000, 754]) | 4\n",
-      "11781    | torch.Size([1, 1000, 777]) | 4\n",
-      "11782    | torch.Size([1, 1000, 754]) | 4\n",
-      "11783    | torch.Size([1, 1000, 762]) | 4\n",
-      "11784    | torch.Size([1, 1000, 754]) | 4\n",
-      "11785    | torch.Size([1, 1000, 777]) | 4\n",
-      "11786    | torch.Size([1, 1000, 764]) | 4\n",
-      "11787    | torch.Size([1, 1000, 754]) | 4\n",
-      "11788    | torch.Size([1, 1000, 754]) | 4\n",
-      "11789    | torch.Size([1, 1000, 777]) | 4\n",
-      "11790    | torch.Size([1, 1000, 754]) | 4\n",
-      "11791    | torch.Size([1, 1000, 769]) | 4\n",
-      "11792    | torch.Size([1, 1000, 762]) | 4\n",
-      "11793    | torch.Size([1, 1000, 754]) | 4\n",
-      "11794    | torch.Size([1, 1000, 754]) | 4\n",
-      "11795    | torch.Size([1, 1000, 777]) | 4\n",
-      "11796    | torch.Size([1, 1000, 779]) | 4\n",
-      "11797    | torch.Size([1, 1000, 754]) | 4\n",
-      "11798    | torch.Size([1, 1000, 754]) | 4\n",
-      "11799    | torch.Size([1, 1000, 754]) | 4\n",
-      "11800    | torch.Size([1, 1000, 754]) | 4\n",
-      "11801    | torch.Size([1, 1000, 777]) | 4\n",
-      "11802    | torch.Size([1, 1000, 754]) | 4\n",
-      "11803    | torch.Size([1, 1000, 783]) | 4\n",
-      "11804    | torch.Size([1, 1000, 754]) | 4\n",
-      "11805    | torch.Size([1, 1000, 773]) | 4\n",
-      "11806    | torch.Size([1, 1000, 840]) | 4\n",
-      "11807    | torch.Size([1, 1000, 754]) | 4\n",
-      "11808    | torch.Size([1, 1000, 754]) | 4\n",
-      "11809    | torch.Size([1, 1000, 762]) | 4\n",
-      "11810    | torch.Size([1, 1000, 754]) | 4\n",
-      "11811    | torch.Size([1, 1000, 777]) | 4\n",
-      "11812    | torch.Size([1, 1000, 754]) | 4\n",
-      "11813    | torch.Size([1, 1000, 783]) | 4\n",
-      "11814    | torch.Size([1, 1000, 777]) | 4\n",
-      "11815    | torch.Size([1, 1000, 777]) | 4\n",
-      "11816    | torch.Size([1, 1000, 754]) | 4\n",
-      "11817    | torch.Size([1, 1000, 754]) | 4\n",
-      "11818    | torch.Size([1, 1000, 778]) | 4\n",
-      "11819    | torch.Size([1, 1000, 754]) | 4\n",
-      "11820    | torch.Size([1, 1000, 777]) | 4\n",
-      "11821    | torch.Size([1, 1000, 794]) | 4\n",
-      "11822    | torch.Size([1, 1000, 754]) | 4\n",
-      "11823    | torch.Size([1, 1000, 777]) | 4\n",
-      "11824    | torch.Size([1, 1000, 754]) | 4\n",
-      "11825    | torch.Size([1, 1000, 777]) | 4\n",
-      "11826    | torch.Size([1, 1000, 785]) | 4\n",
-      "11827    | torch.Size([1, 1000, 754]) | 4\n",
-      "11828    | torch.Size([1, 1000, 754]) | 4\n",
-      "11829    | torch.Size([1, 1000, 754]) | 4\n",
-      "11830    | torch.Size([1, 1000, 754]) | 4\n",
-      "11831    | torch.Size([1, 1000, 703]) | 4\n",
-      "11832    | torch.Size([1, 1000, 802]) | 4\n",
-      "11833    | torch.Size([1, 1000, 754]) | 4\n",
-      "11834    | torch.Size([1, 1000, 754]) | 4\n",
-      "11835    | torch.Size([1, 1000, 781]) | 4\n",
-      "11836    | torch.Size([1, 1000, 780]) | 4\n",
-      "11837    | torch.Size([1, 1000, 754]) | 4\n",
-      "11838    | torch.Size([1, 1000, 754]) | 4\n",
-      "11839    | torch.Size([1, 1000, 783]) | 4\n",
-      "11840    | torch.Size([1, 1000, 754]) | 4\n",
-      "11841    | torch.Size([1, 1000, 754]) | 4\n",
-      "11842    | torch.Size([1, 1000, 785]) | 4\n",
-      "11843    | torch.Size([1, 1000, 754]) | 4\n",
-      "11844    | torch.Size([1, 1000, 795]) | 4\n",
-      "11845    | torch.Size([1, 1000, 754]) | 4\n",
-      "11846    | torch.Size([1, 1000, 754]) | 4\n",
-      "11847    | torch.Size([1, 1000, 680]) | 4\n",
-      "11848    | torch.Size([1, 1000, 754]) | 4\n",
-      "11849    | torch.Size([1, 1000, 754]) | 4\n",
-      "11850    | torch.Size([1, 1000, 754]) | 4\n",
-      "11851    | torch.Size([1, 1000, 754]) | 4\n",
-      "11852    | torch.Size([1, 1000, 754]) | 4\n",
-      "11853    | torch.Size([1, 1000, 754]) | 4\n",
-      "11854    | torch.Size([1, 1000, 754]) | 4\n",
-      "11855    | torch.Size([1, 1000, 777]) | 4\n",
-      "11856    | torch.Size([1, 1000, 778]) | 4\n",
-      "11857    | torch.Size([1, 1000, 754]) | 4\n",
-      "11858    | torch.Size([1, 1000, 754]) | 4\n",
-      "11859    | torch.Size([1, 1000, 771]) | 4\n",
-      "11860    | torch.Size([1, 1000, 762]) | 4\n",
-      "11861    | torch.Size([1, 1000, 754]) | 4\n",
-      "11862    | torch.Size([1, 1000, 754]) | 4\n",
-      "11863    | torch.Size([1, 1000, 754]) | 4\n",
-      "11864    | torch.Size([1, 1000, 754]) | 4\n",
-      "11865    | torch.Size([1, 1000, 777]) | 4\n",
-      "11866    | torch.Size([1, 1000, 754]) | 4\n",
-      "11867    | torch.Size([1, 1000, 801]) | 4\n",
-      "11868    | torch.Size([1, 1000, 754]) | 4\n",
-      "11869    | torch.Size([1, 1000, 777]) | 4\n",
-      "11870    | torch.Size([1, 1000, 771]) | 4\n",
-      "11871    | torch.Size([1, 1000, 754]) | 4\n",
-      "11872    | torch.Size([1, 1000, 777]) | 4\n",
-      "11873    | torch.Size([1, 1000, 774]) | 4\n",
-      "11874    | torch.Size([1, 1000, 754]) | 4\n",
-      "11875    | torch.Size([1, 1000, 777]) | 4\n",
-      "11876    | torch.Size([1, 1000, 754]) | 4\n",
-      "11877    | torch.Size([1, 1000, 1207]) | 4\n",
-      "11878    | torch.Size([1, 1000, 777]) | 4\n",
-      "11879    | torch.Size([1, 1000, 761]) | 4\n",
-      "11880    | torch.Size([1, 1000, 771]) | 4\n",
-      "11881    | torch.Size([1, 1000, 787]) | 4\n",
-      "11882    | torch.Size([1, 1000, 760]) | 4\n",
-      "11883    | torch.Size([1, 1000, 754]) | 4\n",
-      "11884    | torch.Size([1, 1000, 754]) | 4\n",
-      "11885    | torch.Size([1, 1000, 754]) | 4\n",
-      "11886    | torch.Size([1, 1000, 754]) | 4\n",
-      "11887    | torch.Size([1, 1000, 754]) | 4\n",
-      "11888    | torch.Size([1, 1000, 797]) | 4\n",
-      "11889    | torch.Size([1, 1000, 754]) | 4\n",
-      "11890    | torch.Size([1, 1000, 772]) | 4\n",
-      "11891    | torch.Size([1, 1000, 785]) | 4\n",
-      "11892    | torch.Size([1, 1000, 790]) | 4\n",
-      "11893    | torch.Size([1, 1000, 853]) | 4\n",
-      "11894    | torch.Size([1, 1000, 754]) | 4\n",
-      "11895    | torch.Size([1, 1000, 782]) | 4\n",
-      "11896    | torch.Size([1, 1000, 773]) | 4\n",
-      "11897    | torch.Size([1, 1000, 754]) | 4\n",
-      "11898    | torch.Size([1, 1000, 754]) | 4\n",
-      "11899    | torch.Size([1, 1000, 754]) | 4\n",
-      "11900    | torch.Size([1, 1000, 762]) | 4\n",
-      "11901    | torch.Size([1, 1000, 784]) | 4\n",
-      "11902    | torch.Size([1, 1000, 783]) | 4\n",
-      "11903    | torch.Size([1, 1000, 754]) | 4\n",
-      "11904    | torch.Size([1, 1000, 777]) | 4\n",
-      "11905    | torch.Size([1, 1000, 777]) | 4\n",
-      "11906    | torch.Size([1, 1000, 762]) | 4\n",
-      "11907    | torch.Size([1, 1000, 802]) | 4\n",
-      "11908    | torch.Size([1, 1000, 754]) | 4\n",
-      "11909    | torch.Size([1, 1000, 784]) | 4\n",
-      "11910    | torch.Size([1, 1000, 754]) | 4\n",
-      "11911    | torch.Size([1, 1000, 762]) | 4\n",
-      "11912    | torch.Size([1, 1000, 775]) | 4\n",
-      "11913    | torch.Size([1, 1000, 777]) | 4\n",
-      "11914    | torch.Size([1, 1000, 754]) | 4\n",
-      "11915    | torch.Size([1, 1000, 754]) | 4\n",
-      "11916    | torch.Size([1, 1000, 789]) | 4\n",
-      "11917    | torch.Size([1, 1000, 845]) | 4\n",
-      "11918    | torch.Size([1, 1000, 804]) | 4\n",
-      "11919    | torch.Size([1, 1000, 754]) | 4\n",
-      "11920    | torch.Size([1, 1000, 777]) | 4\n",
-      "11921    | torch.Size([1, 1000, 773]) | 4\n",
-      "11922    | torch.Size([1, 1000, 754]) | 4\n",
-      "11923    | torch.Size([1, 1000, 784]) | 4\n",
-      "11924    | torch.Size([1, 1000, 762]) | 4\n",
-      "11925    | torch.Size([1, 1000, 777]) | 4\n",
-      "11926    | torch.Size([1, 1000, 773]) | 4\n",
-      "11927    | torch.Size([1, 1000, 754]) | 4\n",
-      "11928    | torch.Size([1, 1000, 771]) | 4\n",
-      "11929    | torch.Size([1, 1000, 791]) | 4\n",
-      "11930    | torch.Size([1, 1000, 794]) | 4\n",
-      "11931    | torch.Size([1, 1000, 776]) | 4\n",
-      "11932    | torch.Size([1, 1000, 777]) | 4\n",
-      "11933    | torch.Size([1, 1000, 754]) | 4\n",
-      "11934    | torch.Size([1, 1000, 754]) | 4\n",
-      "11935    | torch.Size([1, 1000, 754]) | 4\n",
-      "11936    | torch.Size([1, 1000, 781]) | 4\n",
-      "11937    | torch.Size([1, 1000, 754]) | 4\n",
-      "11938    | torch.Size([1, 1000, 798]) | 4\n",
-      "11939    | torch.Size([1, 1000, 802]) | 4\n",
-      "11940    | torch.Size([1, 1000, 802]) | 4\n",
-      "11941    | torch.Size([1, 1000, 754]) | 4\n",
-      "11942    | torch.Size([1, 1000, 754]) | 4\n",
-      "11943    | torch.Size([1, 1000, 764]) | 4\n",
-      "11944    | torch.Size([1, 1000, 806]) | 4\n",
-      "11945    | torch.Size([1, 1000, 762]) | 4\n",
-      "11946    | torch.Size([1, 1000, 754]) | 4\n",
-      "11947    | torch.Size([1, 1000, 754]) | 4\n",
-      "11948    | torch.Size([1, 1000, 754]) | 4\n",
-      "11949    | torch.Size([1, 1000, 754]) | 4\n",
-      "11950    | torch.Size([1, 1000, 777]) | 4\n",
-      "11951    | torch.Size([1, 1000, 775]) | 4\n",
-      "11952    | torch.Size([1, 1000, 781]) | 4\n",
-      "11953    | torch.Size([1, 1000, 759]) | 4\n",
-      "11954    | torch.Size([1, 1000, 777]) | 4\n",
-      "11955    | torch.Size([1, 1000, 777]) | 4\n",
-      "11956    | torch.Size([1, 1000, 754]) | 4\n",
-      "11957    | torch.Size([1, 1000, 754]) | 4\n",
-      "11958    | torch.Size([1, 1000, 788]) | 4\n",
-      "11959    | torch.Size([1, 1000, 754]) | 4\n",
-      "11960    | torch.Size([1, 1000, 754]) | 4\n",
-      "11961    | torch.Size([1, 1000, 775]) | 4\n",
-      "11962    | torch.Size([1, 1000, 754]) | 4\n",
-      "11963    | torch.Size([1, 1000, 754]) | 4\n",
-      "11964    | torch.Size([1, 1000, 778]) | 4\n",
-      "11965    | torch.Size([1, 1000, 788]) | 4\n",
-      "11966    | torch.Size([1, 1000, 754]) | 4\n",
-      "11967    | torch.Size([1, 1000, 785]) | 4\n",
-      "11968    | torch.Size([1, 1000, 802]) | 4\n",
-      "11969    | torch.Size([1, 1000, 780]) | 4\n",
-      "11970    | torch.Size([1, 1000, 754]) | 4\n",
-      "11971    | torch.Size([1, 1000, 802]) | 4\n",
-      "11972    | torch.Size([1, 1000, 781]) | 4\n",
-      "11973    | torch.Size([1, 1000, 754]) | 4\n",
-      "11974    | torch.Size([1, 1000, 789]) | 4\n",
-      "11975    | torch.Size([1, 1000, 754]) | 4\n",
-      "11976    | torch.Size([1, 1000, 754]) | 4\n",
-      "11977    | torch.Size([1, 1000, 754]) | 4\n",
-      "11978    | torch.Size([1, 1000, 771]) | 4\n",
-      "11979    | torch.Size([1, 1000, 754]) | 4\n",
-      "11980    | torch.Size([1, 1000, 754]) | 4\n",
-      "11981    | torch.Size([1, 1000, 802]) | 4\n",
-      "11982    | torch.Size([1, 1000, 789]) | 4\n",
-      "11983    | torch.Size([1, 1000, 761]) | 4\n",
-      "11984    | torch.Size([1, 1000, 754]) | 4\n",
-      "11985    | torch.Size([1, 1000, 754]) | 4\n",
-      "11986    | torch.Size([1, 1000, 777]) | 4\n",
-      "11987    | torch.Size([1, 1000, 796]) | 4\n",
-      "11988    | torch.Size([1, 1000, 754]) | 4\n",
-      "11989    | torch.Size([1, 1000, 783]) | 4\n",
-      "11990    | torch.Size([1, 1000, 754]) | 4\n",
-      "11991    | torch.Size([1, 1000, 754]) | 4\n",
-      "11992    | torch.Size([1, 1000, 754]) | 4\n",
-      "11993    | torch.Size([1, 1000, 754]) | 4\n",
-      "11994    | torch.Size([1, 1000, 762]) | 4\n",
-      "11995    | torch.Size([1, 1000, 777]) | 4\n",
-      "11996    | torch.Size([1, 1000, 754]) | 4\n",
-      "11997    | torch.Size([1, 1000, 762]) | 4\n",
-      "11998    | torch.Size([1, 1000, 754]) | 4\n",
-      "11999    | torch.Size([1, 1000, 802]) | 4\n",
-      "12000    | torch.Size([1, 1000, 770]) | 4\n",
-      "12001    | torch.Size([1, 1000, 754]) | 4\n",
-      "12002    | torch.Size([1, 1000, 777]) | 4\n",
-      "12003    | torch.Size([1, 1000, 754]) | 4\n",
-      "12004    | torch.Size([1, 1000, 761]) | 4\n",
-      "12005    | torch.Size([1, 1000, 777]) | 4\n",
-      "12006    | torch.Size([1, 1000, 754]) | 4\n",
-      "12007    | torch.Size([1, 1000, 785]) | 4\n",
-      "12008    | torch.Size([1, 1000, 754]) | 4\n",
-      "12009    | torch.Size([1, 1000, 754]) | 4\n",
-      "12010    | torch.Size([1, 1000, 777]) | 4\n",
-      "12011    | torch.Size([1, 1000, 762]) | 4\n",
-      "12012    | torch.Size([1, 1000, 775]) | 4\n",
-      "12013    | torch.Size([1, 1000, 802]) | 4\n",
-      "12014    | torch.Size([1, 1000, 787]) | 4\n",
-      "12015    | torch.Size([1, 1000, 754]) | 4\n",
-      "12016    | torch.Size([1, 1000, 754]) | 4\n",
-      "12017    | torch.Size([1, 1000, 754]) | 4\n",
-      "12018    | torch.Size([1, 1000, 777]) | 4\n",
-      "12019    | torch.Size([1, 1000, 754]) | 4\n",
-      "12020    | torch.Size([1, 1000, 754]) | 4\n",
-      "12021    | torch.Size([1, 1000, 767]) | 4\n",
-      "12022    | torch.Size([1, 1000, 789]) | 4\n",
-      "12023    | torch.Size([1, 1000, 770]) | 4\n",
-      "12024    | torch.Size([1, 1000, 754]) | 4\n",
-      "12025    | torch.Size([1, 1000, 754]) | 4\n",
-      "12026    | torch.Size([1, 1000, 754]) | 4\n",
-      "12027    | torch.Size([1, 1000, 754]) | 4\n",
-      "12028    | torch.Size([1, 1000, 754]) | 4\n",
-      "12029    | torch.Size([1, 1000, 778]) | 4\n",
-      "12030    | torch.Size([1, 1000, 777]) | 4\n",
-      "12031    | torch.Size([1, 1000, 783]) | 4\n",
-      "12032    | torch.Size([1, 1000, 754]) | 4\n",
-      "12033    | torch.Size([1, 1000, 762]) | 4\n",
-      "12034    | torch.Size([1, 1000, 777]) | 4\n",
-      "12035    | torch.Size([1, 1000, 754]) | 4\n",
-      "12036    | torch.Size([1, 1000, 754]) | 4\n",
-      "12037    | torch.Size([1, 1000, 754]) | 4\n",
-      "12038    | torch.Size([1, 1000, 764]) | 4\n",
-      "12039    | torch.Size([1, 1000, 754]) | 4\n",
-      "12040    | torch.Size([1, 1000, 770]) | 4\n",
-      "12041    | torch.Size([1, 1000, 775]) | 4\n",
-      "12042    | torch.Size([1, 1000, 843]) | 4\n",
-      "12043    | torch.Size([1, 1000, 927]) | 4\n",
-      "12044    | torch.Size([1, 1000, 754]) | 4\n",
-      "12045    | torch.Size([1, 1000, 796]) | 4\n",
-      "12046    | torch.Size([1, 1000, 754]) | 4\n",
-      "12047    | torch.Size([1, 1000, 783]) | 4\n",
-      "12048    | torch.Size([1, 1000, 754]) | 4\n",
-      "12049    | torch.Size([1, 1000, 754]) | 4\n",
-      "12050    | torch.Size([1, 1000, 802]) | 4\n",
-      "12051    | torch.Size([1, 1000, 754]) | 4\n",
-      "12052    | torch.Size([1, 1000, 777]) | 4\n",
-      "12053    | torch.Size([1, 1000, 754]) | 4\n",
-      "12054    | torch.Size([1, 1000, 799]) | 4\n",
-      "12055    | torch.Size([1, 1000, 754]) | 4\n",
-      "12056    | torch.Size([1, 1000, 766]) | 4\n",
-      "12057    | torch.Size([1, 1000, 771]) | 4\n",
-      "12058    | torch.Size([1, 1000, 767]) | 4\n",
-      "12059    | torch.Size([1, 1000, 777]) | 4\n",
-      "12060    | torch.Size([1, 1000, 754]) | 4\n",
-      "12061    | torch.Size([1, 1000, 800]) | 4\n",
-      "12062    | torch.Size([1, 1000, 754]) | 4\n",
-      "12063    | torch.Size([1, 1000, 754]) | 4\n",
-      "12064    | torch.Size([1, 1000, 754]) | 4\n",
-      "12065    | torch.Size([1, 1000, 762]) | 4\n",
-      "12066    | torch.Size([1, 1000, 754]) | 4\n",
-      "12067    | torch.Size([1, 1000, 754]) | 4\n",
-      "12068    | torch.Size([1, 1000, 847]) | 4\n",
-      "12069    | torch.Size([1, 1000, 762]) | 4\n",
-      "12070    | torch.Size([1, 1000, 754]) | 4\n",
-      "12071    | torch.Size([1, 1000, 754]) | 4\n",
-      "12072    | torch.Size([1, 1000, 777]) | 4\n",
-      "12073    | torch.Size([1, 1000, 754]) | 4\n",
-      "12074    | torch.Size([1, 1000, 754]) | 4\n",
-      "12075    | torch.Size([1, 1000, 754]) | 4\n",
-      "12076    | torch.Size([1, 1000, 754]) | 4\n",
-      "12077    | torch.Size([1, 1000, 754]) | 4\n",
-      "12078    | torch.Size([1, 1000, 754]) | 4\n",
-      "12079    | torch.Size([1, 1000, 802]) | 4\n",
-      "12080    | torch.Size([1, 1000, 775]) | 4\n",
-      "12081    | torch.Size([1, 1000, 781]) | 4\n",
-      "12082    | torch.Size([1, 1000, 754]) | 4\n",
-      "12083    | torch.Size([1, 1000, 777]) | 4\n",
-      "12084    | torch.Size([1, 1000, 786]) | 4\n",
-      "12085    | torch.Size([1, 1000, 776]) | 4\n",
-      "12086    | torch.Size([1, 1000, 754]) | 4\n",
-      "12087    | torch.Size([1, 1000, 777]) | 4\n",
-      "12088    | torch.Size([1, 1000, 777]) | 4\n",
-      "12089    | torch.Size([1, 1000, 754]) | 4\n",
-      "12090    | torch.Size([1, 1000, 754]) | 4\n",
-      "12091    | torch.Size([1, 1000, 754]) | 4\n",
-      "12092    | torch.Size([1, 1000, 777]) | 4\n",
-      "12093    | torch.Size([1, 1000, 777]) | 4\n",
-      "12094    | torch.Size([1, 1000, 661]) | 4\n",
-      "12095    | torch.Size([1, 1000, 754]) | 4\n",
-      "12096    | torch.Size([1, 1000, 754]) | 4\n",
-      "12097    | torch.Size([1, 1000, 782]) | 4\n",
-      "12098    | torch.Size([1, 1000, 767]) | 4\n",
-      "12099    | torch.Size([1, 1000, 754]) | 4\n",
-      "12100    | torch.Size([1, 1000, 754]) | 4\n",
-      "12101    | torch.Size([1, 1000, 754]) | 4\n",
-      "12102    | torch.Size([1, 1000, 786]) | 4\n",
-      "12103    | torch.Size([1, 1000, 777]) | 4\n",
-      "12104    | torch.Size([1, 1000, 754]) | 4\n",
-      "12105    | torch.Size([1, 1000, 779]) | 4\n",
-      "12106    | torch.Size([1, 1000, 762]) | 4\n",
-      "12107    | torch.Size([1, 1000, 754]) | 4\n",
-      "12108    | torch.Size([1, 1000, 777]) | 4\n",
-      "12109    | torch.Size([1, 1000, 754]) | 4\n",
-      "12110    | torch.Size([1, 1000, 754]) | 4\n",
-      "12111    | torch.Size([1, 1000, 767]) | 4\n",
-      "12112    | torch.Size([1, 1000, 783]) | 4\n",
-      "12113    | torch.Size([1, 1000, 754]) | 4\n",
-      "12114    | torch.Size([1, 1000, 754]) | 4\n",
-      "12115    | torch.Size([1, 1000, 777]) | 4\n",
-      "12116    | torch.Size([1, 1000, 777]) | 4\n",
-      "12117    | torch.Size([1, 1000, 802]) | 4\n",
-      "12118    | torch.Size([1, 1000, 785]) | 4\n",
-      "12119    | torch.Size([1, 1000, 754]) | 4\n",
-      "12120    | torch.Size([1, 1000, 754]) | 4\n",
-      "12121    | torch.Size([1, 1000, 754]) | 4\n",
-      "12122    | torch.Size([1, 1000, 754]) | 4\n",
-      "12123    | torch.Size([1, 1000, 777]) | 4\n",
-      "12124    | torch.Size([1, 1000, 754]) | 4\n",
-      "12125    | torch.Size([1, 1000, 784]) | 4\n",
-      "12126    | torch.Size([1, 1000, 777]) | 4\n",
-      "12127    | torch.Size([1, 1000, 754]) | 4\n",
-      "12128    | torch.Size([1, 1000, 754]) | 4\n",
-      "12129    | torch.Size([1, 1000, 754]) | 4\n",
-      "12130    | torch.Size([1, 1000, 754]) | 4\n",
-      "12131    | torch.Size([1, 1000, 754]) | 4\n",
-      "12132    | torch.Size([1, 1000, 754]) | 4\n",
-      "12133    | torch.Size([1, 1000, 777]) | 4\n",
-      "12134    | torch.Size([1, 1000, 762]) | 4\n",
-      "12135    | torch.Size([1, 1000, 754]) | 4\n",
-      "12136    | torch.Size([1, 1000, 713]) | 4\n",
-      "12137    | torch.Size([1, 1000, 754]) | 4\n",
-      "12138    | torch.Size([1, 1000, 754]) | 4\n",
-      "12139    | torch.Size([1, 1000, 754]) | 4\n",
-      "12140    | torch.Size([1, 1000, 754]) | 4\n",
-      "12141    | torch.Size([1, 1000, 754]) | 4\n",
-      "12142    | torch.Size([1, 1000, 777]) | 4\n",
-      "12143    | torch.Size([1, 1000, 770]) | 4\n",
-      "12144    | torch.Size([1, 1000, 757]) | 4\n",
-      "12145    | torch.Size([1, 1000, 754]) | 4\n",
-      "12146    | torch.Size([1, 1000, 754]) | 4\n",
-      "12147    | torch.Size([1, 1000, 754]) | 4\n",
-      "12148    | torch.Size([1, 1000, 754]) | 4\n",
-      "12149    | torch.Size([1, 1000, 762]) | 4\n",
-      "12150    | torch.Size([1, 1000, 754]) | 4\n",
-      "12151    | torch.Size([1, 1000, 786]) | 4\n",
-      "12152    | torch.Size([1, 1000, 754]) | 4\n",
-      "12153    | torch.Size([1, 1000, 754]) | 4\n",
-      "12154    | torch.Size([1, 1000, 754]) | 4\n",
-      "12155    | torch.Size([1, 1000, 754]) | 4\n",
-      "12156    | torch.Size([1, 1000, 754]) | 4\n",
-      "12157    | torch.Size([1, 1000, 771]) | 4\n",
-      "12158    | torch.Size([1, 1000, 754]) | 4\n",
-      "12159    | torch.Size([1, 1000, 754]) | 4\n",
-      "12160    | torch.Size([1, 1000, 754]) | 4\n",
-      "12161    | torch.Size([1, 1000, 754]) | 4\n",
-      "12162    | torch.Size([1, 1000, 754]) | 4\n",
-      "12163    | torch.Size([1, 1000, 754]) | 4\n",
-      "12164    | torch.Size([1, 1000, 754]) | 4\n",
-      "12165    | torch.Size([1, 1000, 754]) | 4\n",
-      "12166    | torch.Size([1, 1000, 789]) | 4\n",
-      "12167    | torch.Size([1, 1000, 779]) | 4\n",
-      "12168    | torch.Size([1, 1000, 781]) | 4\n",
-      "12169    | torch.Size([1, 1000, 848]) | 4\n",
-      "12170    | torch.Size([1, 1000, 802]) | 4\n",
-      "12171    | torch.Size([1, 1000, 777]) | 4\n",
-      "12172    | torch.Size([1, 1000, 767]) | 4\n",
-      "12173    | torch.Size([1, 1000, 754]) | 4\n",
-      "12174    | torch.Size([1, 1000, 777]) | 4\n",
-      "12175    | torch.Size([1, 1000, 848]) | 4\n",
-      "12176    | torch.Size([1, 1000, 789]) | 4\n",
-      "12177    | torch.Size([1, 1000, 782]) | 4\n",
-      "12178    | torch.Size([1, 1000, 754]) | 4\n",
-      "12179    | torch.Size([1, 1000, 754]) | 4\n",
-      "12180    | torch.Size([1, 1000, 754]) | 4\n",
-      "12181    | torch.Size([1, 1000, 777]) | 4\n",
-      "12182    | torch.Size([1, 1000, 754]) | 4\n",
-      "12183    | torch.Size([1, 1000, 762]) | 4\n",
-      "12184    | torch.Size([1, 1000, 777]) | 4\n",
-      "12185    | torch.Size([1, 1000, 846]) | 4\n",
-      "12186    | torch.Size([1, 1000, 754]) | 4\n",
-      "12187    | torch.Size([1, 1000, 777]) | 4\n",
-      "12188    | torch.Size([1, 1000, 754]) | 4\n",
-      "12189    | torch.Size([1, 1000, 754]) | 4\n",
-      "12190    | torch.Size([1, 1000, 771]) | 4\n",
-      "12191    | torch.Size([1, 1000, 754]) | 4\n",
-      "12192    | torch.Size([1, 1000, 664]) | 4\n",
-      "12193    | torch.Size([1, 1000, 802]) | 4\n",
-      "12194    | torch.Size([1, 1000, 754]) | 4\n",
-      "12195    | torch.Size([1, 1000, 754]) | 4\n",
-      "12196    | torch.Size([1, 1000, 778]) | 4\n",
-      "12197    | torch.Size([1, 1000, 778]) | 4\n",
-      "12198    | torch.Size([1, 1000, 754]) | 4\n",
-      "12199    | torch.Size([1, 1000, 775]) | 4\n",
-      "12200    | torch.Size([1, 1000, 816]) | 4\n",
-      "12201    | torch.Size([1, 1000, 754]) | 4\n",
-      "12202    | torch.Size([1, 1000, 783]) | 4\n",
-      "12203    | torch.Size([1, 1000, 754]) | 4\n",
-      "12204    | torch.Size([1, 1000, 786]) | 4\n",
-      "12205    | torch.Size([1, 1000, 754]) | 4\n",
-      "12206    | torch.Size([1, 1000, 754]) | 4\n",
-      "12207    | torch.Size([1, 1000, 754]) | 4\n",
-      "12208    | torch.Size([1, 1000, 1095]) | 4\n",
-      "12209    | torch.Size([1, 1000, 754]) | 4\n",
-      "12210    | torch.Size([1, 1000, 754]) | 4\n",
-      "12211    | torch.Size([1, 1000, 777]) | 4\n",
-      "12212    | torch.Size([1, 1000, 802]) | 4\n",
-      "12213    | torch.Size([1, 1000, 754]) | 4\n",
-      "12214    | torch.Size([1, 1000, 754]) | 4\n",
-      "12215    | torch.Size([1, 1000, 754]) | 4\n",
-      "12216    | torch.Size([1, 1000, 754]) | 4\n",
-      "12217    | torch.Size([1, 1000, 775]) | 4\n",
-      "12218    | torch.Size([1, 1000, 754]) | 4\n",
-      "12219    | torch.Size([1, 1000, 764]) | 4\n",
-      "12220    | torch.Size([1, 1000, 754]) | 4\n",
-      "12221    | torch.Size([1, 1000, 754]) | 4\n",
-      "12222    | torch.Size([1, 1000, 754]) | 4\n",
-      "12223    | torch.Size([1, 1000, 777]) | 4\n",
-      "12224    | torch.Size([1, 1000, 802]) | 4\n",
-      "12225    | torch.Size([1, 1000, 754]) | 4\n",
-      "12226    | torch.Size([1, 1000, 754]) | 4\n",
-      "12227    | torch.Size([1, 1000, 777]) | 4\n",
-      "12228    | torch.Size([1, 1000, 754]) | 4\n",
-      "12229    | torch.Size([1, 1000, 762]) | 4\n",
-      "12230    | torch.Size([1, 1000, 754]) | 4\n",
-      "12231    | torch.Size([1, 1000, 777]) | 4\n",
-      "12232    | torch.Size([1, 1000, 777]) | 4\n",
-      "12233    | torch.Size([1, 1000, 810]) | 4\n",
-      "12234    | torch.Size([1, 1000, 754]) | 4\n",
-      "12235    | torch.Size([1, 1000, 786]) | 4\n",
-      "12236    | torch.Size([1, 1000, 770]) | 4\n",
-      "12237    | torch.Size([1, 1000, 754]) | 4\n",
-      "12238    | torch.Size([1, 1000, 802]) | 4\n",
-      "12239    | torch.Size([1, 1000, 777]) | 4\n",
-      "12240    | torch.Size([1, 1000, 762]) | 4\n",
-      "12241    | torch.Size([1, 1000, 783]) | 4\n",
-      "12242    | torch.Size([1, 1000, 754]) | 4\n",
-      "12243    | torch.Size([1, 1000, 754]) | 4\n",
-      "12244    | torch.Size([1, 1000, 754]) | 4\n",
-      "12245    | torch.Size([1, 1000, 778]) | 4\n",
-      "12246    | torch.Size([1, 1000, 777]) | 4\n",
-      "12247    | torch.Size([1, 1000, 754]) | 4\n",
-      "12248    | torch.Size([1, 1000, 754]) | 4\n",
-      "12249    | torch.Size([1, 1000, 754]) | 4\n",
-      "12250    | torch.Size([1, 1000, 754]) | 4\n",
-      "12251    | torch.Size([1, 1000, 754]) | 4\n",
-      "12252    | torch.Size([1, 1000, 754]) | 4\n",
-      "12253    | torch.Size([1, 1000, 754]) | 4\n",
-      "12254    | torch.Size([1, 1000, 783]) | 4\n",
-      "12255    | torch.Size([1, 1000, 779]) | 4\n",
-      "12256    | torch.Size([1, 1000, 754]) | 4\n",
-      "12257    | torch.Size([1, 1000, 754]) | 4\n",
-      "12258    | torch.Size([1, 1000, 754]) | 4\n",
-      "12259    | torch.Size([1, 1000, 777]) | 4\n",
-      "12260    | torch.Size([1, 1000, 754]) | 4\n",
-      "12261    | torch.Size([1, 1000, 777]) | 4\n",
-      "12262    | torch.Size([1, 1000, 777]) | 4\n",
-      "12263    | torch.Size([1, 1000, 754]) | 4\n",
-      "12264    | torch.Size([1, 1000, 776]) | 4\n",
-      "12265    | torch.Size([1, 1000, 754]) | 4\n",
-      "12266    | torch.Size([1, 1000, 764]) | 4\n",
-      "12267    | torch.Size([1, 1000, 777]) | 4\n",
-      "12268    | torch.Size([1, 1000, 613]) | 4\n",
-      "12269    | torch.Size([1, 1000, 754]) | 4\n",
-      "12270    | torch.Size([1, 1000, 785]) | 4\n",
-      "12271    | torch.Size([1, 1000, 754]) | 4\n",
-      "12272    | torch.Size([1, 1000, 764]) | 4\n",
-      "12273    | torch.Size([1, 1000, 754]) | 4\n",
-      "12274    | torch.Size([1, 1000, 754]) | 4\n",
-      "12275    | torch.Size([1, 1000, 754]) | 4\n",
-      "12276    | torch.Size([1, 1000, 754]) | 4\n",
-      "12277    | torch.Size([1, 1000, 754]) | 4\n",
-      "12278    | torch.Size([1, 1000, 767]) | 4\n",
-      "12279    | torch.Size([1, 1000, 754]) | 4\n",
-      "12280    | torch.Size([1, 1000, 754]) | 4\n",
-      "12281    | torch.Size([1, 1000, 762]) | 4\n",
-      "12282    | torch.Size([1, 1000, 762]) | 4\n",
-      "12283    | torch.Size([1, 1000, 778]) | 4\n",
-      "12284    | torch.Size([1, 1000, 754]) | 4\n",
-      "12285    | torch.Size([1, 1000, 754]) | 4\n",
-      "12286    | torch.Size([1, 1000, 754]) | 4\n",
-      "12287    | torch.Size([1, 1000, 754]) | 4\n",
-      "12288    | torch.Size([1, 1000, 765]) | 4\n",
-      "12289    | torch.Size([1, 1000, 781]) | 4\n",
-      "12290    | torch.Size([1, 1000, 754]) | 4\n",
-      "12291    | torch.Size([1, 1000, 754]) | 4\n",
-      "12292    | torch.Size([1, 1000, 754]) | 4\n",
-      "12293    | torch.Size([1, 1000, 754]) | 4\n",
-      "12294    | torch.Size([1, 1000, 754]) | 4\n",
-      "12295    | torch.Size([1, 1000, 754]) | 4\n",
-      "12296    | torch.Size([1, 1000, 762]) | 4\n",
-      "12297    | torch.Size([1, 1000, 754]) | 4\n",
-      "12298    | torch.Size([1, 1000, 754]) | 4\n",
-      "12299    | torch.Size([1, 1000, 754]) | 4\n",
-      "12300    | torch.Size([1, 1000, 754]) | 4\n",
-      "12301    | torch.Size([1, 1000, 754]) | 4\n",
-      "12302    | torch.Size([1, 1000, 762]) | 4\n",
-      "12303    | torch.Size([1, 1000, 754]) | 4\n",
-      "12304    | torch.Size([1, 1000, 754]) | 4\n",
-      "12305    | torch.Size([1, 1000, 754]) | 4\n",
-      "12306    | torch.Size([1, 1000, 777]) | 4\n",
-      "12307    | torch.Size([1, 1000, 777]) | 4\n",
-      "12308    | torch.Size([1, 1000, 754]) | 4\n",
-      "12309    | torch.Size([1, 1000, 754]) | 4\n",
-      "12310    | torch.Size([1, 1000, 803]) | 4\n",
-      "12311    | torch.Size([1, 1000, 777]) | 4\n",
-      "12312    | torch.Size([1, 1000, 802]) | 4\n",
-      "12313    | torch.Size([1, 1000, 754]) | 4\n",
-      "12314    | torch.Size([1, 1000, 777]) | 4\n",
-      "12315    | torch.Size([1, 1000, 754]) | 4\n",
-      "12316    | torch.Size([1, 1000, 771]) | 4\n",
-      "12317    | torch.Size([1, 1000, 754]) | 4\n",
-      "12318    | torch.Size([1, 1000, 754]) | 4\n",
-      "12319    | torch.Size([1, 1000, 754]) | 4\n",
-      "12320    | torch.Size([1, 1000, 754]) | 4\n",
-      "12321    | torch.Size([1, 1000, 777]) | 4\n",
-      "12322    | torch.Size([1, 1000, 754]) | 4\n",
-      "12323    | torch.Size([1, 1000, 754]) | 4\n",
-      "12324    | torch.Size([1, 1000, 845]) | 4\n",
-      "12325    | torch.Size([1, 1000, 754]) | 4\n",
-      "12326    | torch.Size([1, 1000, 778]) | 4\n",
-      "12327    | torch.Size([1, 1000, 802]) | 4\n",
-      "12328    | torch.Size([1, 1000, 777]) | 4\n",
-      "12329    | torch.Size([1, 1000, 754]) | 4\n",
-      "12330    | torch.Size([1, 1000, 786]) | 4\n",
-      "12331    | torch.Size([1, 1000, 754]) | 4\n",
-      "12332    | torch.Size([1, 1000, 786]) | 4\n",
-      "12333    | torch.Size([1, 1000, 778]) | 4\n",
-      "12334    | torch.Size([1, 1000, 754]) | 4\n",
-      "12335    | torch.Size([1, 1000, 754]) | 4\n",
-      "12336    | torch.Size([1, 1000, 766]) | 4\n",
-      "12337    | torch.Size([1, 1000, 777]) | 4\n",
-      "12338    | torch.Size([1, 1000, 779]) | 4\n",
-      "12339    | torch.Size([1, 1000, 754]) | 4\n",
-      "12340    | torch.Size([1, 1000, 775]) | 4\n",
-      "12341    | torch.Size([1, 1000, 777]) | 4\n",
-      "12342    | torch.Size([1, 1000, 754]) | 4\n",
-      "12343    | torch.Size([1, 1000, 754]) | 4\n",
-      "12344    | torch.Size([1, 1000, 754]) | 4\n",
-      "12345    | torch.Size([1, 1000, 754]) | 4\n",
-      "12346    | torch.Size([1, 1000, 802]) | 4\n",
-      "12347    | torch.Size([1, 1000, 754]) | 4\n",
-      "12348    | torch.Size([1, 1000, 777]) | 4\n",
-      "12349    | torch.Size([1, 1000, 777]) | 4\n",
-      "12350    | torch.Size([1, 1000, 762]) | 4\n",
-      "12351    | torch.Size([1, 1000, 762]) | 4\n",
-      "12352    | torch.Size([1, 1000, 754]) | 4\n",
-      "12353    | torch.Size([1, 1000, 777]) | 4\n",
-      "12354    | torch.Size([1, 1000, 785]) | 4\n",
-      "12355    | torch.Size([1, 1000, 802]) | 4\n",
-      "12356    | torch.Size([1, 1000, 754]) | 4\n",
-      "12357    | torch.Size([1, 1000, 754]) | 4\n",
-      "12358    | torch.Size([1, 1000, 778]) | 4\n",
-      "12359    | torch.Size([1, 1000, 769]) | 4\n",
-      "12360    | torch.Size([1, 1000, 790]) | 4\n",
-      "12361    | torch.Size([1, 1000, 777]) | 4\n",
-      "12362    | torch.Size([1, 1000, 754]) | 4\n",
-      "12363    | torch.Size([1, 1000, 754]) | 4\n",
-      "12364    | torch.Size([1, 1000, 777]) | 4\n",
-      "12365    | torch.Size([1, 1000, 754]) | 4\n",
-      "12366    | torch.Size([1, 1000, 754]) | 4\n",
-      "12367    | torch.Size([1, 1000, 771]) | 4\n",
-      "12368    | torch.Size([1, 1000, 754]) | 4\n",
-      "12369    | torch.Size([1, 1000, 847]) | 4\n",
-      "12370    | torch.Size([1, 1000, 754]) | 4\n",
-      "12371    | torch.Size([1, 1000, 775]) | 4\n",
-      "12372    | torch.Size([1, 1000, 754]) | 4\n",
-      "12373    | torch.Size([1, 1000, 778]) | 4\n",
-      "12374    | torch.Size([1, 1000, 772]) | 4\n",
-      "12375    | torch.Size([1, 1000, 754]) | 4\n",
-      "12376    | torch.Size([1, 1000, 762]) | 4\n",
-      "12377    | torch.Size([1, 1000, 754]) | 4\n",
-      "12378    | torch.Size([1, 1000, 782]) | 4\n",
-      "12379    | torch.Size([1, 1000, 754]) | 4\n",
-      "12380    | torch.Size([1, 1000, 754]) | 4\n",
-      "12381    | torch.Size([1, 1000, 782]) | 4\n",
-      "12382    | torch.Size([1, 1000, 754]) | 4\n",
-      "12383    | torch.Size([1, 1000, 754]) | 4\n",
-      "12384    | torch.Size([1, 1000, 762]) | 4\n",
-      "12385    | torch.Size([1, 1000, 777]) | 4\n",
-      "12386    | torch.Size([1, 1000, 762]) | 4\n",
-      "12387    | torch.Size([1, 1000, 754]) | 4\n",
-      "12388    | torch.Size([1, 1000, 777]) | 4\n",
-      "12389    | torch.Size([1, 1000, 754]) | 4\n",
-      "12390    | torch.Size([1, 1000, 754]) | 4\n",
-      "12391    | torch.Size([1, 1000, 765]) | 4\n",
-      "12392    | torch.Size([1, 1000, 859]) | 4\n",
-      "12393    | torch.Size([1, 1000, 781]) | 4\n",
-      "12394    | torch.Size([1, 1000, 754]) | 4\n",
-      "12395    | torch.Size([1, 1000, 777]) | 4\n",
-      "12396    | torch.Size([1, 1000, 754]) | 4\n",
-      "12397    | torch.Size([1, 1000, 754]) | 4\n",
-      "12398    | torch.Size([1, 1000, 780]) | 4\n",
-      "12399    | torch.Size([1, 1000, 768]) | 4\n",
-      "12400    | torch.Size([1, 1000, 754]) | 4\n",
-      "12401    | torch.Size([1, 1000, 754]) | 4\n",
-      "12402    | torch.Size([1, 1000, 802]) | 4\n",
-      "12403    | torch.Size([1, 1000, 754]) | 4\n",
-      "12404    | torch.Size([1, 1000, 789]) | 4\n",
-      "12405    | torch.Size([1, 1000, 780]) | 4\n",
-      "12406    | torch.Size([1, 1000, 766]) | 4\n",
-      "12407    | torch.Size([1, 1000, 768]) | 4\n",
-      "12408    | torch.Size([1, 1000, 754]) | 4\n",
-      "12409    | torch.Size([1, 1000, 754]) | 4\n",
-      "12410    | torch.Size([1, 1000, 754]) | 4\n",
-      "12411    | torch.Size([1, 1000, 754]) | 4\n",
-      "12412    | torch.Size([1, 1000, 794]) | 4\n",
-      "12413    | torch.Size([1, 1000, 762]) | 4\n",
-      "12414    | torch.Size([1, 1000, 754]) | 4\n",
-      "12415    | torch.Size([1, 1000, 754]) | 4\n",
-      "12416    | torch.Size([1, 1000, 754]) | 4\n",
-      "12417    | torch.Size([1, 1000, 713]) | 4\n",
-      "12418    | torch.Size([1, 1000, 802]) | 4\n",
-      "12419    | torch.Size([1, 1000, 754]) | 4\n",
-      "12420    | torch.Size([1, 1000, 754]) | 4\n",
-      "12421    | torch.Size([1, 1000, 754]) | 4\n",
-      "12422    | torch.Size([1, 1000, 762]) | 4\n",
-      "12423    | torch.Size([1, 1000, 798]) | 4\n",
-      "12424    | torch.Size([1, 1000, 762]) | 4\n",
-      "12425    | torch.Size([1, 1000, 777]) | 4\n",
-      "12426    | torch.Size([1, 1000, 777]) | 4\n",
-      "12427    | torch.Size([1, 1000, 802]) | 4\n",
-      "12428    | torch.Size([1, 1000, 754]) | 4\n",
-      "12429    | torch.Size([1, 1000, 777]) | 4\n",
-      "12430    | torch.Size([1, 1000, 781]) | 4\n",
-      "12431    | torch.Size([1, 1000, 754]) | 4\n",
-      "12432    | torch.Size([1, 1000, 777]) | 4\n",
-      "12433    | torch.Size([1, 1000, 777]) | 4\n",
-      "12434    | torch.Size([1, 1000, 754]) | 4\n",
-      "12435    | torch.Size([1, 1000, 778]) | 4\n",
-      "12436    | torch.Size([1, 1000, 812]) | 4\n",
-      "12437    | torch.Size([1, 1000, 773]) | 4\n",
-      "12438    | torch.Size([1, 1000, 802]) | 4\n",
-      "12439    | torch.Size([1, 1000, 794]) | 4\n",
-      "12440    | torch.Size([1, 1000, 754]) | 4\n",
-      "12441    | torch.Size([1, 1000, 754]) | 4\n",
-      "12442    | torch.Size([1, 1000, 754]) | 4\n",
-      "12443    | torch.Size([1, 1000, 802]) | 4\n",
-      "12444    | torch.Size([1, 1000, 762]) | 4\n",
-      "12445    | torch.Size([1, 1000, 754]) | 4\n",
-      "12446    | torch.Size([1, 1000, 754]) | 4\n",
-      "12447    | torch.Size([1, 1000, 754]) | 4\n",
-      "12448    | torch.Size([1, 1000, 777]) | 4\n",
-      "12449    | torch.Size([1, 1000, 713]) | 4\n",
-      "12450    | torch.Size([1, 1000, 777]) | 4\n",
-      "12451    | torch.Size([1, 1000, 762]) | 4\n",
-      "12452    | torch.Size([1, 1000, 754]) | 4\n",
-      "12453    | torch.Size([1, 1000, 754]) | 4\n",
-      "12454    | torch.Size([1, 1000, 754]) | 4\n",
-      "12455    | torch.Size([1, 1000, 777]) | 4\n",
-      "12456    | torch.Size([1, 1000, 754]) | 4\n",
-      "12457    | torch.Size([1, 1000, 762]) | 4\n",
-      "12458    | torch.Size([1, 1000, 754]) | 4\n",
-      "12459    | torch.Size([1, 1000, 754]) | 4\n",
-      "12460    | torch.Size([1, 1000, 754]) | 4\n",
-      "12461    | torch.Size([1, 1000, 754]) | 4\n",
-      "12462    | torch.Size([1, 1000, 1316]) | 4\n",
-      "12463    | torch.Size([1, 1000, 756]) | 4\n",
-      "12464    | torch.Size([1, 1000, 768]) | 4\n",
-      "12465    | torch.Size([1, 1000, 761]) | 4\n",
-      "12466    | torch.Size([1, 1000, 754]) | 4\n",
-      "12467    | torch.Size([1, 1000, 754]) | 4\n",
-      "12468    | torch.Size([1, 1000, 754]) | 4\n",
-      "12469    | torch.Size([1, 1000, 754]) | 4\n",
-      "12470    | torch.Size([1, 1000, 754]) | 4\n",
-      "12471    | torch.Size([1, 1000, 851]) | 4\n",
-      "12472    | torch.Size([1, 1000, 754]) | 4\n",
-      "12473    | torch.Size([1, 1000, 777]) | 4\n",
-      "12474    | torch.Size([1, 1000, 754]) | 4\n",
-      "12475    | torch.Size([1, 1000, 754]) | 4\n",
-      "12476    | torch.Size([1, 1000, 777]) | 4\n",
-      "12477    | torch.Size([1, 1000, 754]) | 4\n",
-      "12478    | torch.Size([1, 1000, 762]) | 4\n",
-      "12479    | torch.Size([1, 1000, 754]) | 4\n",
-      "12480    | torch.Size([1, 1000, 772]) | 4\n",
-      "12481    | torch.Size([1, 1000, 754]) | 4\n",
-      "12482    | torch.Size([1, 1000, 648]) | 4\n",
-      "12483    | torch.Size([1, 1000, 754]) | 4\n",
-      "12484    | torch.Size([1, 1000, 788]) | 4\n",
-      "12485    | torch.Size([1, 1000, 762]) | 4\n",
-      "12486    | torch.Size([1, 1000, 777]) | 4\n",
-      "12487    | torch.Size([1, 1000, 754]) | 4\n",
-      "12488    | torch.Size([1, 1000, 754]) | 4\n",
-      "12489    | torch.Size([1, 1000, 754]) | 4\n",
-      "12490    | torch.Size([1, 1000, 754]) | 4\n",
-      "12491    | torch.Size([1, 1000, 762]) | 4\n",
-      "12492    | torch.Size([1, 1000, 754]) | 4\n",
-      "12493    | torch.Size([1, 1000, 754]) | 4\n",
-      "12494    | torch.Size([1, 1000, 777]) | 4\n",
-      "12495    | torch.Size([1, 1000, 754]) | 4\n",
-      "12496    | torch.Size([1, 1000, 775]) | 4\n",
-      "12497    | torch.Size([1, 1000, 777]) | 4\n",
-      "12498    | torch.Size([1, 1000, 754]) | 4\n",
-      "12499    | torch.Size([1, 1000, 754]) | 4\n",
-      "12500    | torch.Size([1, 1000, 780]) | 4\n",
-      "12501    | torch.Size([1, 1000, 754]) | 4\n",
-      "12502    | torch.Size([1, 1000, 812]) | 4\n",
-      "12503    | torch.Size([1, 1000, 754]) | 4\n",
-      "12504    | torch.Size([1, 1000, 769]) | 4\n",
-      "12505    | torch.Size([1, 1000, 754]) | 4\n",
-      "12506    | torch.Size([1, 1000, 754]) | 4\n",
-      "12507    | torch.Size([1, 1000, 754]) | 4\n",
-      "12508    | torch.Size([1, 1000, 786]) | 4\n",
-      "12509    | torch.Size([1, 1000, 754]) | 4\n",
-      "12510    | torch.Size([1, 1000, 754]) | 4\n",
-      "12511    | torch.Size([1, 1000, 782]) | 4\n",
-      "12512    | torch.Size([1, 1000, 754]) | 4\n",
-      "12513    | torch.Size([1, 1000, 779]) | 4\n",
-      "12514    | torch.Size([1, 1000, 754]) | 4\n",
-      "12515    | torch.Size([1, 1000, 754]) | 4\n",
-      "12516    | torch.Size([1, 1000, 778]) | 4\n",
-      "12517    | torch.Size([1, 1000, 777]) | 4\n",
-      "12518    | torch.Size([1, 1000, 754]) | 4\n",
-      "12519    | torch.Size([1, 1000, 762]) | 4\n",
-      "12520    | torch.Size([1, 1000, 754]) | 4\n",
-      "12521    | torch.Size([1, 1000, 766]) | 4\n",
-      "12522    | torch.Size([1, 1000, 754]) | 4\n",
-      "12523    | torch.Size([1, 1000, 780]) | 4\n",
-      "12524    | torch.Size([1, 1000, 754]) | 4\n",
-      "12525    | torch.Size([1, 1000, 754]) | 4\n",
-      "12526    | torch.Size([1, 1000, 762]) | 4\n",
-      "12527    | torch.Size([1, 1000, 754]) | 4\n",
-      "12528    | torch.Size([1, 1000, 770]) | 4\n",
-      "12529    | torch.Size([1, 1000, 754]) | 4\n",
-      "12530    | torch.Size([1, 1000, 762]) | 4\n",
-      "12531    | torch.Size([1, 1000, 778]) | 4\n",
-      "12532    | torch.Size([1, 1000, 772]) | 4\n",
-      "12533    | torch.Size([1, 1000, 802]) | 4\n",
-      "12534    | torch.Size([1, 1000, 611]) | 4\n",
-      "12535    | torch.Size([1, 1000, 754]) | 4\n",
-      "12536    | torch.Size([1, 1000, 754]) | 4\n",
-      "12537    | torch.Size([1, 1000, 754]) | 4\n",
-      "12538    | torch.Size([1, 1000, 754]) | 4\n",
-      "12539    | torch.Size([1, 1000, 766]) | 4\n",
-      "12540    | torch.Size([1, 1000, 754]) | 4\n",
-      "12541    | torch.Size([1, 1000, 754]) | 4\n",
-      "12542    | torch.Size([1, 1000, 754]) | 4\n",
-      "12543    | torch.Size([1, 1000, 754]) | 4\n",
-      "12544    | torch.Size([1, 1000, 754]) | 4\n",
-      "12545    | torch.Size([1, 1000, 777]) | 4\n",
-      "12546    | torch.Size([1, 1000, 778]) | 4\n",
-      "12547    | torch.Size([1, 1000, 754]) | 4\n",
-      "12548    | torch.Size([1, 1000, 754]) | 4\n",
-      "12549    | torch.Size([1, 1000, 764]) | 4\n",
-      "12550    | torch.Size([1, 1000, 778]) | 4\n",
-      "12551    | torch.Size([1, 1000, 783]) | 4\n",
-      "12552    | torch.Size([1, 1000, 802]) | 4\n",
-      "12553    | torch.Size([1, 1000, 777]) | 4\n",
-      "12554    | torch.Size([1, 1000, 772]) | 4\n",
-      "12555    | torch.Size([1, 1000, 754]) | 4\n",
-      "12556    | torch.Size([1, 1000, 775]) | 4\n",
-      "12557    | torch.Size([1, 1000, 809]) | 4\n",
-      "12558    | torch.Size([1, 1000, 779]) | 4\n",
-      "12559    | torch.Size([1, 1000, 754]) | 4\n",
-      "12560    | torch.Size([1, 1000, 777]) | 4\n",
-      "12561    | torch.Size([1, 1000, 762]) | 4\n",
-      "12562    | torch.Size([1, 1000, 775]) | 4\n",
-      "12563    | torch.Size([1, 1000, 786]) | 4\n",
-      "12564    | torch.Size([1, 1000, 779]) | 4\n",
-      "12565    | torch.Size([1, 1000, 754]) | 4\n",
-      "12566    | torch.Size([1, 1000, 847]) | 4\n",
-      "12567    | torch.Size([1, 1000, 754]) | 4\n",
-      "12568    | torch.Size([1, 1000, 773]) | 4\n",
-      "12569    | torch.Size([1, 1000, 754]) | 5\n",
-      "12570    | torch.Size([1, 1000, 802]) | 5\n",
-      "12571    | torch.Size([1, 1000, 754]) | 5\n",
-      "12572    | torch.Size([1, 1000, 789]) | 5\n",
-      "12573    | torch.Size([1, 1000, 754]) | 5\n",
-      "12574    | torch.Size([1, 1000, 754]) | 5\n",
-      "12575    | torch.Size([1, 1000, 754]) | 5\n",
-      "12576    | torch.Size([1, 1000, 754]) | 5\n",
-      "12577    | torch.Size([1, 1000, 754]) | 5\n",
-      "12578    | torch.Size([1, 1000, 754]) | 5\n",
-      "12579    | torch.Size([1, 1000, 754]) | 5\n",
-      "12580    | torch.Size([1, 1000, 785]) | 5\n",
-      "12581    | torch.Size([1, 1000, 754]) | 5\n",
-      "12582    | torch.Size([1, 1000, 754]) | 5\n",
-      "12583    | torch.Size([1, 1000, 754]) | 5\n",
-      "12584    | torch.Size([1, 1000, 754]) | 5\n",
-      "12585    | torch.Size([1, 1000, 786]) | 5\n",
-      "12586    | torch.Size([1, 1000, 754]) | 5\n",
-      "12587    | torch.Size([1, 1000, 778]) | 5\n",
-      "12588    | torch.Size([1, 1000, 754]) | 5\n",
-      "12589    | torch.Size([1, 1000, 790]) | 5\n",
-      "12590    | torch.Size([1, 1000, 754]) | 5\n",
-      "12591    | torch.Size([1, 1000, 754]) | 5\n",
-      "12592    | torch.Size([1, 1000, 754]) | 5\n",
-      "12593    | torch.Size([1, 1000, 754]) | 5\n",
-      "12594    | torch.Size([1, 1000, 754]) | 5\n",
-      "12595    | torch.Size([1, 1000, 754]) | 5\n",
-      "12596    | torch.Size([1, 1000, 754]) | 5\n",
-      "12597    | torch.Size([1, 1000, 754]) | 5\n",
-      "12598    | torch.Size([1, 1000, 782]) | 5\n",
-      "12599    | torch.Size([1, 1000, 754]) | 5\n",
-      "12600    | torch.Size([1, 1000, 754]) | 5\n",
-      "12601    | torch.Size([1, 1000, 771]) | 5\n",
-      "12602    | torch.Size([1, 1000, 754]) | 5\n",
-      "12603    | torch.Size([1, 1000, 781]) | 5\n",
-      "12604    | torch.Size([1, 1000, 730]) | 5\n",
-      "12605    | torch.Size([1, 1000, 785]) | 5\n",
-      "12606    | torch.Size([1, 1000, 754]) | 5\n",
-      "12607    | torch.Size([1, 1000, 754]) | 5\n",
-      "12608    | torch.Size([1, 1000, 789]) | 5\n",
-      "12609    | torch.Size([1, 1000, 782]) | 5\n",
-      "12610    | torch.Size([1, 1000, 754]) | 5\n",
-      "12611    | torch.Size([1, 1000, 789]) | 5\n",
-      "12612    | torch.Size([1, 1000, 754]) | 5\n",
-      "12613    | torch.Size([1, 1000, 771]) | 5\n",
-      "12614    | torch.Size([1, 1000, 754]) | 5\n",
-      "12615    | torch.Size([1, 1000, 754]) | 5\n",
-      "12616    | torch.Size([1, 1000, 754]) | 5\n",
-      "12617    | torch.Size([1, 1000, 783]) | 5\n",
-      "12618    | torch.Size([1, 1000, 754]) | 5\n",
-      "12619    | torch.Size([1, 1000, 754]) | 5\n",
-      "12620    | torch.Size([1, 1000, 754]) | 5\n",
-      "12621    | torch.Size([1, 1000, 786]) | 5\n",
-      "12622    | torch.Size([1, 1000, 754]) | 5\n",
-      "12623    | torch.Size([1, 1000, 780]) | 5\n",
-      "12624    | torch.Size([1, 1000, 754]) | 5\n",
-      "12625    | torch.Size([1, 1000, 754]) | 5\n",
-      "12626    | torch.Size([1, 1000, 754]) | 5\n",
-      "12627    | torch.Size([1, 1000, 754]) | 5\n",
-      "12628    | torch.Size([1, 1000, 754]) | 5\n",
-      "12629    | torch.Size([1, 1000, 754]) | 5\n",
-      "12630    | torch.Size([1, 1000, 754]) | 5\n",
-      "12631    | torch.Size([1, 1000, 754]) | 5\n",
-      "12632    | torch.Size([1, 1000, 754]) | 5\n",
-      "12633    | torch.Size([1, 1000, 754]) | 5\n",
-      "12634    | torch.Size([1, 1000, 754]) | 5\n",
-      "12635    | torch.Size([1, 1000, 754]) | 5\n",
-      "12636    | torch.Size([1, 1000, 754]) | 5\n",
-      "12637    | torch.Size([1, 1000, 754]) | 5\n",
-      "12638    | torch.Size([1, 1000, 754]) | 5\n",
-      "12639    | torch.Size([1, 1000, 770]) | 5\n",
-      "12640    | torch.Size([1, 1000, 754]) | 5\n",
-      "12641    | torch.Size([1, 1000, 771]) | 5\n",
-      "12642    | torch.Size([1, 1000, 754]) | 5\n",
-      "12643    | torch.Size([1, 1000, 754]) | 5\n",
-      "12644    | torch.Size([1, 1000, 754]) | 5\n",
-      "12645    | torch.Size([1, 1000, 783]) | 5\n",
-      "12646    | torch.Size([1, 1000, 754]) | 5\n",
-      "12647    | torch.Size([1, 1000, 784]) | 5\n",
-      "12648    | torch.Size([1, 1000, 754]) | 5\n",
-      "12649    | torch.Size([1, 1000, 754]) | 5\n",
-      "12650    | torch.Size([1, 1000, 754]) | 5\n",
-      "12651    | torch.Size([1, 1000, 754]) | 5\n",
-      "12652    | torch.Size([1, 1000, 754]) | 5\n",
-      "12653    | torch.Size([1, 1000, 777]) | 5\n",
-      "12654    | torch.Size([1, 1000, 780]) | 5\n",
-      "12655    | torch.Size([1, 1000, 754]) | 5\n",
-      "12656    | torch.Size([1, 1000, 754]) | 5\n",
-      "12657    | torch.Size([1, 1000, 754]) | 5\n",
-      "12658    | torch.Size([1, 1000, 754]) | 5\n",
-      "12659    | torch.Size([1, 1000, 754]) | 5\n",
-      "12660    | torch.Size([1, 1000, 754]) | 5\n",
-      "12661    | torch.Size([1, 1000, 754]) | 5\n",
-      "12662    | torch.Size([1, 1000, 754]) | 5\n",
-      "12663    | torch.Size([1, 1000, 754]) | 5\n",
-      "12664    | torch.Size([1, 1000, 786]) | 5\n",
-      "12665    | torch.Size([1, 1000, 754]) | 5\n",
-      "12666    | torch.Size([1, 1000, 754]) | 5\n",
-      "12667    | torch.Size([1, 1000, 774]) | 5\n",
-      "12668    | torch.Size([1, 1000, 754]) | 5\n",
-      "12669    | torch.Size([1, 1000, 776]) | 5\n",
-      "12670    | torch.Size([1, 1000, 754]) | 5\n",
-      "12671    | torch.Size([1, 1000, 754]) | 5\n",
-      "12672    | torch.Size([1, 1000, 754]) | 5\n",
-      "12673    | torch.Size([1, 1000, 754]) | 5\n",
-      "12674    | torch.Size([1, 1000, 754]) | 5\n",
-      "12675    | torch.Size([1, 1000, 754]) | 5\n",
-      "12676    | torch.Size([1, 1000, 754]) | 5\n",
-      "12677    | torch.Size([1, 1000, 771]) | 5\n",
-      "12678    | torch.Size([1, 1000, 774]) | 5\n",
-      "12679    | torch.Size([1, 1000, 754]) | 5\n",
-      "12680    | torch.Size([1, 1000, 787]) | 5\n",
-      "12681    | torch.Size([1, 1000, 754]) | 5\n",
-      "12682    | torch.Size([1, 1000, 754]) | 5\n",
-      "12683    | torch.Size([1, 1000, 754]) | 5\n",
-      "12684    | torch.Size([1, 1000, 754]) | 5\n",
-      "12685    | torch.Size([1, 1000, 754]) | 5\n",
-      "12686    | torch.Size([1, 1000, 754]) | 5\n",
-      "12687    | torch.Size([1, 1000, 754]) | 5\n",
-      "12688    | torch.Size([1, 1000, 754]) | 5\n",
-      "12689    | torch.Size([1, 1000, 754]) | 5\n",
-      "12690    | torch.Size([1, 1000, 811]) | 5\n",
-      "12691    | torch.Size([1, 1000, 754]) | 5\n",
-      "12692    | torch.Size([1, 1000, 754]) | 5\n",
-      "12693    | torch.Size([1, 1000, 771]) | 5\n",
-      "12694    | torch.Size([1, 1000, 754]) | 5\n",
-      "12695    | torch.Size([1, 1000, 754]) | 5\n",
-      "12696    | torch.Size([1, 1000, 754]) | 5\n",
-      "12697    | torch.Size([1, 1000, 754]) | 5\n",
-      "12698    | torch.Size([1, 1000, 754]) | 5\n",
-      "12699    | torch.Size([1, 1000, 781]) | 5\n",
-      "12700    | torch.Size([1, 1000, 754]) | 5\n",
-      "12701    | torch.Size([1, 1000, 754]) | 5\n",
-      "12702    | torch.Size([1, 1000, 754]) | 5\n",
-      "12703    | torch.Size([1, 1000, 754]) | 5\n",
-      "12704    | torch.Size([1, 1000, 754]) | 5\n",
-      "12705    | torch.Size([1, 1000, 754]) | 5\n",
-      "12706    | torch.Size([1, 1000, 754]) | 5\n",
-      "12707    | torch.Size([1, 1000, 771]) | 5\n",
-      "12708    | torch.Size([1, 1000, 754]) | 5\n",
-      "12709    | torch.Size([1, 1000, 776]) | 5\n",
-      "12710    | torch.Size([1, 1000, 754]) | 5\n",
-      "12711    | torch.Size([1, 1000, 754]) | 5\n",
-      "12712    | torch.Size([1, 1000, 754]) | 5\n",
-      "12713    | torch.Size([1, 1000, 754]) | 5\n",
-      "12714    | torch.Size([1, 1000, 754]) | 5\n",
-      "12715    | torch.Size([1, 1000, 754]) | 5\n",
-      "12716    | torch.Size([1, 1000, 754]) | 5\n",
-      "12717    | torch.Size([1, 1000, 794]) | 5\n",
-      "12718    | torch.Size([1, 1000, 754]) | 5\n",
-      "12719    | torch.Size([1, 1000, 754]) | 5\n",
-      "12720    | torch.Size([1, 1000, 754]) | 5\n",
-      "12721    | torch.Size([1, 1000, 754]) | 5\n",
-      "12722    | torch.Size([1, 1000, 754]) | 5\n",
-      "12723    | torch.Size([1, 1000, 754]) | 5\n",
-      "12724    | torch.Size([1, 1000, 754]) | 5\n",
-      "12725    | torch.Size([1, 1000, 754]) | 5\n",
-      "12726    | torch.Size([1, 1000, 754]) | 5\n",
-      "12727    | torch.Size([1, 1000, 787]) | 5\n",
-      "12728    | torch.Size([1, 1000, 754]) | 5\n",
-      "12729    | torch.Size([1, 1000, 754]) | 5\n",
-      "12730    | torch.Size([1, 1000, 754]) | 5\n",
-      "12731    | torch.Size([1, 1000, 754]) | 5\n",
-      "12732    | torch.Size([1, 1000, 754]) | 5\n",
-      "12733    | torch.Size([1, 1000, 781]) | 5\n",
-      "12734    | torch.Size([1, 1000, 763]) | 5\n",
-      "12735    | torch.Size([1, 1000, 788]) | 5\n",
-      "12736    | torch.Size([1, 1000, 754]) | 5\n",
-      "12737    | torch.Size([1, 1000, 754]) | 5\n",
-      "12738    | torch.Size([1, 1000, 776]) | 5\n",
-      "12739    | torch.Size([1, 1000, 754]) | 5\n",
-      "12740    | torch.Size([1, 1000, 754]) | 5\n",
-      "12741    | torch.Size([1, 1000, 754]) | 5\n",
-      "12742    | torch.Size([1, 1000, 754]) | 5\n",
-      "12743    | torch.Size([1, 1000, 754]) | 5\n",
-      "12744    | torch.Size([1, 1000, 771]) | 5\n",
-      "12745    | torch.Size([1, 1000, 2542]) | 5\n",
-      "12746    | torch.Size([1, 1000, 754]) | 5\n",
-      "12747    | torch.Size([1, 1000, 780]) | 5\n",
-      "12748    | torch.Size([1, 1000, 754]) | 5\n",
-      "12749    | torch.Size([1, 1000, 754]) | 5\n",
-      "12750    | torch.Size([1, 1000, 781]) | 5\n",
-      "12751    | torch.Size([1, 1000, 754]) | 5\n",
-      "12752    | torch.Size([1, 1000, 754]) | 5\n",
-      "12753    | torch.Size([1, 1000, 754]) | 5\n",
-      "12754    | torch.Size([1, 1000, 754]) | 5\n",
-      "12755    | torch.Size([1, 1000, 754]) | 5\n",
-      "12756    | torch.Size([1, 1000, 754]) | 5\n",
-      "12757    | torch.Size([1, 1000, 770]) | 5\n",
-      "12758    | torch.Size([1, 1000, 754]) | 5\n",
-      "12759    | torch.Size([1, 1000, 754]) | 5\n",
-      "12760    | torch.Size([1, 1000, 771]) | 5\n",
-      "12761    | torch.Size([1, 1000, 754]) | 5\n",
-      "12762    | torch.Size([1, 1000, 754]) | 5\n",
-      "12763    | torch.Size([1, 1000, 754]) | 5\n",
-      "12764    | torch.Size([1, 1000, 754]) | 5\n",
-      "12765    | torch.Size([1, 1000, 768]) | 5\n",
-      "12766    | torch.Size([1, 1000, 754]) | 5\n",
-      "12767    | torch.Size([1, 1000, 754]) | 5\n",
-      "12768    | torch.Size([1, 1000, 754]) | 5\n",
-      "12769    | torch.Size([1, 1000, 754]) | 5\n",
-      "12770    | torch.Size([1, 1000, 754]) | 5\n",
-      "12771    | torch.Size([1, 1000, 754]) | 5\n",
-      "12772    | torch.Size([1, 1000, 754]) | 5\n",
-      "12773    | torch.Size([1, 1000, 775]) | 5\n",
-      "12774    | torch.Size([1, 1000, 754]) | 5\n",
-      "12775    | torch.Size([1, 1000, 776]) | 5\n",
-      "12776    | torch.Size([1, 1000, 754]) | 5\n",
-      "12777    | torch.Size([1, 1000, 754]) | 5\n",
-      "12778    | torch.Size([1, 1000, 754]) | 5\n",
-      "12779    | torch.Size([1, 1000, 754]) | 5\n",
-      "12780    | torch.Size([1, 1000, 754]) | 5\n",
-      "12781    | torch.Size([1, 1000, 779]) | 5\n",
-      "12782    | torch.Size([1, 1000, 802]) | 5\n",
-      "12783    | torch.Size([1, 1000, 754]) | 5\n",
-      "12784    | torch.Size([1, 1000, 754]) | 5\n",
-      "12785    | torch.Size([1, 1000, 754]) | 5\n",
-      "12786    | torch.Size([1, 1000, 778]) | 5\n",
-      "12787    | torch.Size([1, 1000, 754]) | 5\n",
-      "12788    | torch.Size([1, 1000, 787]) | 5\n",
-      "12789    | torch.Size([1, 1000, 754]) | 5\n",
-      "12790    | torch.Size([1, 1000, 754]) | 5\n",
-      "12791    | torch.Size([1, 1000, 778]) | 5\n",
-      "12792    | torch.Size([1, 1000, 754]) | 5\n",
-      "12793    | torch.Size([1, 1000, 754]) | 5\n",
-      "12794    | torch.Size([1, 1000, 754]) | 5\n",
-      "12795    | torch.Size([1, 1000, 754]) | 5\n",
-      "12796    | torch.Size([1, 1000, 754]) | 5\n",
-      "12797    | torch.Size([1, 1000, 754]) | 5\n",
-      "12798    | torch.Size([1, 1000, 782]) | 5\n",
-      "12799    | torch.Size([1, 1000, 754]) | 5\n",
-      "12800    | torch.Size([1, 1000, 754]) | 5\n",
-      "12801    | torch.Size([1, 1000, 754]) | 5\n",
-      "12802    | torch.Size([1, 1000, 754]) | 5\n",
-      "12803    | torch.Size([1, 1000, 754]) | 5\n",
-      "12804    | torch.Size([1, 1000, 754]) | 5\n",
-      "12805    | torch.Size([1, 1000, 765]) | 5\n",
-      "12806    | torch.Size([1, 1000, 754]) | 5\n",
-      "12807    | torch.Size([1, 1000, 771]) | 5\n",
-      "12808    | torch.Size([1, 1000, 754]) | 5\n",
-      "12809    | torch.Size([1, 1000, 754]) | 5\n",
-      "12810    | torch.Size([1, 1000, 754]) | 5\n",
-      "12811    | torch.Size([1, 1000, 754]) | 5\n",
-      "12812    | torch.Size([1, 1000, 754]) | 5\n",
-      "12813    | torch.Size([1, 1000, 754]) | 5\n",
-      "12814    | torch.Size([1, 1000, 754]) | 5\n",
-      "12815    | torch.Size([1, 1000, 754]) | 5\n",
-      "12816    | torch.Size([1, 1000, 754]) | 5\n",
-      "12817    | torch.Size([1, 1000, 754]) | 5\n",
-      "12818    | torch.Size([1, 1000, 754]) | 5\n",
-      "12819    | torch.Size([1, 1000, 754]) | 5\n",
-      "12820    | torch.Size([1, 1000, 820]) | 5\n",
-      "12821    | torch.Size([1, 1000, 754]) | 5\n",
-      "12822    | torch.Size([1, 1000, 754]) | 5\n",
-      "12823    | torch.Size([1, 1000, 754]) | 5\n",
-      "12824    | torch.Size([1, 1000, 777]) | 5\n",
-      "12825    | torch.Size([1, 1000, 754]) | 5\n",
-      "12826    | torch.Size([1, 1000, 754]) | 5\n",
-      "12827    | torch.Size([1, 1000, 754]) | 5\n",
-      "12828    | torch.Size([1, 1000, 754]) | 5\n",
-      "12829    | torch.Size([1, 1000, 771]) | 5\n",
-      "12830    | torch.Size([1, 1000, 754]) | 5\n",
-      "12831    | torch.Size([1, 1000, 754]) | 5\n",
-      "12832    | torch.Size([1, 1000, 788]) | 5\n",
-      "12833    | torch.Size([1, 1000, 777]) | 5\n",
-      "12834    | torch.Size([1, 1000, 774]) | 5\n",
-      "12835    | torch.Size([1, 1000, 754]) | 5\n",
-      "12836    | torch.Size([1, 1000, 754]) | 5\n",
-      "12837    | torch.Size([1, 1000, 754]) | 5\n",
-      "12838    | torch.Size([1, 1000, 771]) | 5\n",
-      "12839    | torch.Size([1, 1000, 754]) | 5\n",
-      "12840    | torch.Size([1, 1000, 754]) | 5\n",
-      "12841    | torch.Size([1, 1000, 754]) | 5\n",
-      "12842    | torch.Size([1, 1000, 754]) | 5\n",
-      "12843    | torch.Size([1, 1000, 754]) | 5\n",
-      "12844    | torch.Size([1, 1000, 754]) | 5\n",
-      "12845    | torch.Size([1, 1000, 754]) | 5\n",
-      "12846    | torch.Size([1, 1000, 775]) | 5\n",
-      "12847    | torch.Size([1, 1000, 754]) | 5\n",
-      "12848    | torch.Size([1, 1000, 779]) | 5\n",
-      "12849    | torch.Size([1, 1000, 783]) | 5\n",
-      "12850    | torch.Size([1, 1000, 754]) | 5\n",
-      "12851    | torch.Size([1, 1000, 754]) | 5\n",
-      "12852    | torch.Size([1, 1000, 754]) | 5\n",
-      "12853    | torch.Size([1, 1000, 754]) | 5\n",
-      "12854    | torch.Size([1, 1000, 754]) | 5\n",
-      "12855    | torch.Size([1, 1000, 754]) | 5\n",
-      "12856    | torch.Size([1, 1000, 754]) | 5\n",
-      "12857    | torch.Size([1, 1000, 754]) | 5\n",
-      "12858    | torch.Size([1, 1000, 771]) | 5\n",
-      "12859    | torch.Size([1, 1000, 754]) | 5\n",
-      "12860    | torch.Size([1, 1000, 794]) | 5\n",
-      "12861    | torch.Size([1, 1000, 754]) | 5\n",
-      "12862    | torch.Size([1, 1000, 779]) | 5\n",
-      "12863    | torch.Size([1, 1000, 754]) | 5\n",
-      "12864    | torch.Size([1, 1000, 754]) | 5\n",
-      "12865    | torch.Size([1, 1000, 754]) | 5\n",
-      "12866    | torch.Size([1, 1000, 754]) | 5\n",
-      "12867    | torch.Size([1, 1000, 754]) | 5\n",
-      "12868    | torch.Size([1, 1000, 795]) | 5\n",
-      "12869    | torch.Size([1, 1000, 754]) | 5\n",
-      "12870    | torch.Size([1, 1000, 754]) | 5\n",
-      "12871    | torch.Size([1, 1000, 754]) | 5\n",
-      "12872    | torch.Size([1, 1000, 754]) | 5\n",
-      "12873    | torch.Size([1, 1000, 754]) | 5\n",
-      "12874    | torch.Size([1, 1000, 754]) | 5\n",
-      "12875    | torch.Size([1, 1000, 754]) | 5\n",
-      "12876    | torch.Size([1, 1000, 754]) | 5\n",
-      "12877    | torch.Size([1, 1000, 754]) | 5\n",
-      "12878    | torch.Size([1, 1000, 742]) | 5\n",
-      "12879    | torch.Size([1, 1000, 754]) | 5\n",
-      "12880    | torch.Size([1, 1000, 754]) | 5\n",
-      "12881    | torch.Size([1, 1000, 754]) | 5\n",
-      "12882    | torch.Size([1, 1000, 771]) | 5\n",
-      "12883    | torch.Size([1, 1000, 754]) | 5\n",
-      "12884    | torch.Size([1, 1000, 781]) | 5\n",
-      "12885    | torch.Size([1, 1000, 754]) | 5\n",
-      "12886    | torch.Size([1, 1000, 754]) | 5\n",
-      "12887    | torch.Size([1, 1000, 754]) | 5\n",
-      "12888    | torch.Size([1, 1000, 754]) | 5\n",
-      "12889    | torch.Size([1, 1000, 754]) | 5\n",
-      "12890    | torch.Size([1, 1000, 754]) | 5\n",
-      "12891    | torch.Size([1, 1000, 754]) | 5\n",
-      "12892    | torch.Size([1, 1000, 754]) | 5\n",
-      "12893    | torch.Size([1, 1000, 780]) | 5\n",
-      "12894    | torch.Size([1, 1000, 754]) | 5\n",
-      "12895    | torch.Size([1, 1000, 771]) | 5\n",
-      "12896    | torch.Size([1, 1000, 754]) | 5\n",
-      "12897    | torch.Size([1, 1000, 806]) | 5\n",
-      "12898    | torch.Size([1, 1000, 754]) | 5\n",
-      "12899    | torch.Size([1, 1000, 778]) | 5\n",
-      "12900    | torch.Size([1, 1000, 754]) | 5\n",
-      "12901    | torch.Size([1, 1000, 776]) | 5\n",
-      "12902    | torch.Size([1, 1000, 800]) | 5\n",
-      "12903    | torch.Size([1, 1000, 754]) | 5\n",
-      "12904    | torch.Size([1, 1000, 754]) | 5\n",
-      "12905    | torch.Size([1, 1000, 754]) | 5\n",
-      "12906    | torch.Size([1, 1000, 782]) | 5\n",
-      "12907    | torch.Size([1, 1000, 787]) | 5\n",
-      "12908    | torch.Size([1, 1000, 754]) | 5\n",
-      "12909    | torch.Size([1, 1000, 754]) | 5\n",
-      "12910    | torch.Size([1, 1000, 754]) | 5\n",
-      "12911    | torch.Size([1, 1000, 775]) | 5\n",
-      "12912    | torch.Size([1, 1000, 754]) | 5\n",
-      "12913    | torch.Size([1, 1000, 771]) | 5\n",
-      "12914    | torch.Size([1, 1000, 754]) | 5\n",
-      "12915    | torch.Size([1, 1000, 754]) | 5\n",
-      "12916    | torch.Size([1, 1000, 808]) | 5\n",
-      "12917    | torch.Size([1, 1000, 754]) | 5\n",
-      "12918    | torch.Size([1, 1000, 754]) | 5\n",
-      "12919    | torch.Size([1, 1000, 783]) | 5\n",
-      "12920    | torch.Size([1, 1000, 754]) | 5\n",
-      "12921    | torch.Size([1, 1000, 774]) | 5\n",
-      "12922    | torch.Size([1, 1000, 754]) | 5\n",
-      "12923    | torch.Size([1, 1000, 768]) | 5\n",
-      "12924    | torch.Size([1, 1000, 754]) | 5\n",
-      "12925    | torch.Size([1, 1000, 754]) | 5\n",
-      "12926    | torch.Size([1, 1000, 754]) | 5\n",
-      "12927    | torch.Size([1, 1000, 754]) | 5\n",
-      "12928    | torch.Size([1, 1000, 754]) | 5\n",
-      "12929    | torch.Size([1, 1000, 785]) | 5\n",
-      "12930    | torch.Size([1, 1000, 754]) | 5\n",
-      "12931    | torch.Size([1, 1000, 767]) | 5\n",
-      "12932    | torch.Size([1, 1000, 754]) | 5\n",
-      "12933    | torch.Size([1, 1000, 754]) | 5\n",
-      "12934    | torch.Size([1, 1000, 754]) | 5\n",
-      "12935    | torch.Size([1, 1000, 783]) | 5\n",
-      "12936    | torch.Size([1, 1000, 754]) | 5\n",
-      "12937    | torch.Size([1, 1000, 754]) | 5\n",
-      "12938    | torch.Size([1, 1000, 754]) | 5\n",
-      "12939    | torch.Size([1, 1000, 754]) | 5\n",
-      "12940    | torch.Size([1, 1000, 776]) | 5\n",
-      "12941    | torch.Size([1, 1000, 754]) | 5\n",
-      "12942    | torch.Size([1, 1000, 754]) | 5\n",
-      "12943    | torch.Size([1, 1000, 754]) | 5\n",
-      "12944    | torch.Size([1, 1000, 786]) | 5\n",
-      "12945    | torch.Size([1, 1000, 754]) | 5\n",
-      "12946    | torch.Size([1, 1000, 776]) | 5\n",
-      "12947    | torch.Size([1, 1000, 754]) | 5\n",
-      "12948    | torch.Size([1, 1000, 789]) | 5\n",
-      "12949    | torch.Size([1, 1000, 754]) | 5\n",
-      "12950    | torch.Size([1, 1000, 754]) | 5\n",
-      "12951    | torch.Size([1, 1000, 754]) | 5\n",
-      "12952    | torch.Size([1, 1000, 754]) | 5\n",
-      "12953    | torch.Size([1, 1000, 754]) | 5\n",
-      "12954    | torch.Size([1, 1000, 754]) | 5\n",
-      "12955    | torch.Size([1, 1000, 754]) | 5\n",
-      "12956    | torch.Size([1, 1000, 792]) | 5\n",
-      "12957    | torch.Size([1, 1000, 754]) | 5\n",
-      "12958    | torch.Size([1, 1000, 754]) | 5\n",
-      "12959    | torch.Size([1, 1000, 754]) | 5\n",
-      "12960    | torch.Size([1, 1000, 754]) | 5\n",
-      "12961    | torch.Size([1, 1000, 754]) | 5\n",
-      "12962    | torch.Size([1, 1000, 754]) | 5\n",
-      "12963    | torch.Size([1, 1000, 754]) | 5\n",
-      "12964    | torch.Size([1, 1000, 754]) | 5\n",
-      "12965    | torch.Size([1, 1000, 771]) | 5\n",
-      "12966    | torch.Size([1, 1000, 754]) | 5\n",
-      "12967    | torch.Size([1, 1000, 782]) | 5\n",
-      "12968    | torch.Size([1, 1000, 754]) | 5\n",
-      "12969    | torch.Size([1, 1000, 754]) | 5\n",
-      "12970    | torch.Size([1, 1000, 765]) | 5\n",
-      "12971    | torch.Size([1, 1000, 754]) | 5\n",
-      "12972    | torch.Size([1, 1000, 754]) | 5\n",
-      "12973    | torch.Size([1, 1000, 754]) | 5\n",
-      "12974    | torch.Size([1, 1000, 754]) | 5\n",
-      "12975    | torch.Size([1, 1000, 776]) | 5\n",
-      "12976    | torch.Size([1, 1000, 754]) | 5\n",
-      "12977    | torch.Size([1, 1000, 790]) | 5\n",
-      "12978    | torch.Size([1, 1000, 754]) | 5\n",
-      "12979    | torch.Size([1, 1000, 779]) | 5\n",
-      "12980    | torch.Size([1, 1000, 754]) | 5\n",
-      "12981    | torch.Size([1, 1000, 754]) | 5\n",
-      "12982    | torch.Size([1, 1000, 806]) | 5\n",
-      "12983    | torch.Size([1, 1000, 784]) | 5\n",
-      "12984    | torch.Size([1, 1000, 754]) | 5\n",
-      "12985    | torch.Size([1, 1000, 771]) | 5\n",
-      "12986    | torch.Size([1, 1000, 786]) | 5\n",
-      "12987    | torch.Size([1, 1000, 754]) | 5\n",
-      "12988    | torch.Size([1, 1000, 754]) | 5\n",
-      "12989    | torch.Size([1, 1000, 754]) | 5\n",
-      "12990    | torch.Size([1, 1000, 754]) | 5\n",
-      "12991    | torch.Size([1, 1000, 754]) | 5\n",
-      "12992    | torch.Size([1, 1000, 786]) | 5\n",
-      "12993    | torch.Size([1, 1000, 754]) | 5\n",
-      "12994    | torch.Size([1, 1000, 754]) | 5\n",
-      "12995    | torch.Size([1, 1000, 754]) | 5\n",
-      "12996    | torch.Size([1, 1000, 754]) | 5\n",
-      "12997    | torch.Size([1, 1000, 754]) | 5\n",
-      "12998    | torch.Size([1, 1000, 771]) | 5\n",
-      "12999    | torch.Size([1, 1000, 754]) | 5\n",
-      "13000    | torch.Size([1, 1000, 754]) | 5\n",
-      "13001    | torch.Size([1, 1000, 754]) | 5\n",
-      "13002    | torch.Size([1, 1000, 754]) | 5\n",
-      "13003    | torch.Size([1, 1000, 781]) | 5\n",
-      "13004    | torch.Size([1, 1000, 754]) | 5\n",
-      "13005    | torch.Size([1, 1000, 754]) | 5\n",
-      "13006    | torch.Size([1, 1000, 754]) | 5\n",
-      "13007    | torch.Size([1, 1000, 754]) | 5\n",
-      "13008    | torch.Size([1, 1000, 754]) | 5\n",
-      "13009    | torch.Size([1, 1000, 781]) | 5\n",
-      "13010    | torch.Size([1, 1000, 754]) | 5\n",
-      "13011    | torch.Size([1, 1000, 776]) | 5\n",
-      "13012    | torch.Size([1, 1000, 754]) | 5\n",
-      "13013    | torch.Size([1, 1000, 754]) | 5\n",
-      "13014    | torch.Size([1, 1000, 754]) | 5\n",
-      "13015    | torch.Size([1, 1000, 784]) | 5\n",
-      "13016    | torch.Size([1, 1000, 754]) | 5\n",
-      "13017    | torch.Size([1, 1000, 754]) | 5\n",
-      "13018    | torch.Size([1, 1000, 782]) | 5\n",
-      "13019    | torch.Size([1, 1000, 754]) | 5\n",
-      "13020    | torch.Size([1, 1000, 754]) | 5\n",
-      "13021    | torch.Size([1, 1000, 769]) | 5\n",
-      "13022    | torch.Size([1, 1000, 754]) | 5\n",
-      "13023    | torch.Size([1, 1000, 754]) | 5\n",
-      "13024    | torch.Size([1, 1000, 786]) | 5\n",
-      "13025    | torch.Size([1, 1000, 758]) | 5\n",
-      "13026    | torch.Size([1, 1000, 754]) | 5\n",
-      "13027    | torch.Size([1, 1000, 754]) | 5\n",
-      "13028    | torch.Size([1, 1000, 754]) | 5\n",
-      "13029    | torch.Size([1, 1000, 754]) | 5\n",
-      "13030    | torch.Size([1, 1000, 754]) | 5\n",
-      "13031    | torch.Size([1, 1000, 787]) | 5\n",
-      "13032    | torch.Size([1, 1000, 754]) | 5\n",
-      "13033    | torch.Size([1, 1000, 754]) | 5\n",
-      "13034    | torch.Size([1, 1000, 754]) | 5\n",
-      "13035    | torch.Size([1, 1000, 754]) | 5\n",
-      "13036    | torch.Size([1, 1000, 754]) | 5\n",
-      "13037    | torch.Size([1, 1000, 788]) | 5\n",
-      "13038    | torch.Size([1, 1000, 754]) | 5\n",
-      "13039    | torch.Size([1, 1000, 789]) | 5\n",
-      "13040    | torch.Size([1, 1000, 754]) | 5\n",
-      "13041    | torch.Size([1, 1000, 754]) | 5\n",
-      "13042    | torch.Size([1, 1000, 754]) | 5\n",
-      "13043    | torch.Size([1, 1000, 783]) | 5\n",
-      "13044    | torch.Size([1, 1000, 754]) | 5\n",
-      "13045    | torch.Size([1, 1000, 754]) | 5\n",
-      "13046    | torch.Size([1, 1000, 754]) | 5\n",
-      "13047    | torch.Size([1, 1000, 754]) | 5\n",
-      "13048    | torch.Size([1, 1000, 799]) | 5\n",
-      "13049    | torch.Size([1, 1000, 754]) | 5\n",
-      "13050    | torch.Size([1, 1000, 754]) | 5\n",
-      "13051    | torch.Size([1, 1000, 766]) | 5\n",
-      "13052    | torch.Size([1, 1000, 754]) | 5\n",
-      "13053    | torch.Size([1, 1000, 754]) | 5\n",
-      "13054    | torch.Size([1, 1000, 754]) | 5\n",
-      "13055    | torch.Size([1, 1000, 775]) | 5\n",
-      "13056    | torch.Size([1, 1000, 754]) | 5\n",
-      "13057    | torch.Size([1, 1000, 754]) | 5\n",
-      "13058    | torch.Size([1, 1000, 754]) | 5\n",
-      "13059    | torch.Size([1, 1000, 788]) | 5\n",
-      "13060    | torch.Size([1, 1000, 754]) | 5\n",
-      "13061    | torch.Size([1, 1000, 754]) | 5\n",
-      "13062    | torch.Size([1, 1000, 781]) | 5\n",
-      "13063    | torch.Size([1, 1000, 774]) | 5\n",
-      "13064    | torch.Size([1, 1000, 754]) | 5\n",
-      "13065    | torch.Size([1, 1000, 754]) | 5\n",
-      "13066    | torch.Size([1, 1000, 754]) | 5\n",
-      "13067    | torch.Size([1, 1000, 781]) | 5\n",
-      "13068    | torch.Size([1, 1000, 754]) | 5\n",
-      "13069    | torch.Size([1, 1000, 781]) | 5\n",
-      "13070    | torch.Size([1, 1000, 754]) | 5\n",
-      "13071    | torch.Size([1, 1000, 754]) | 5\n",
-      "13072    | torch.Size([1, 1000, 754]) | 5\n",
-      "13073    | torch.Size([1, 1000, 754]) | 5\n",
-      "13074    | torch.Size([1, 1000, 762]) | 5\n",
-      "13075    | torch.Size([1, 1000, 754]) | 5\n",
-      "13076    | torch.Size([1, 1000, 754]) | 5\n",
-      "13077    | torch.Size([1, 1000, 771]) | 5\n",
-      "13078    | torch.Size([1, 1000, 754]) | 5\n",
-      "13079    | torch.Size([1, 1000, 788]) | 5\n",
-      "13080    | torch.Size([1, 1000, 788]) | 5\n",
-      "13081    | torch.Size([1, 1000, 754]) | 5\n",
-      "13082    | torch.Size([1, 1000, 754]) | 5\n",
-      "13083    | torch.Size([1, 1000, 773]) | 5\n",
-      "13084    | torch.Size([1, 1000, 789]) | 5\n",
-      "13085    | torch.Size([1, 1000, 754]) | 5\n",
-      "13086    | torch.Size([1, 1000, 754]) | 5\n",
-      "13087    | torch.Size([1, 1000, 754]) | 5\n",
-      "13088    | torch.Size([1, 1000, 754]) | 5\n",
-      "13089    | torch.Size([1, 1000, 754]) | 5\n",
-      "13090    | torch.Size([1, 1000, 754]) | 5\n",
-      "13091    | torch.Size([1, 1000, 754]) | 5\n",
-      "13092    | torch.Size([1, 1000, 779]) | 5\n",
-      "13093    | torch.Size([1, 1000, 805]) | 5\n",
-      "13094    | torch.Size([1, 1000, 766]) | 5\n",
-      "13095    | torch.Size([1, 1000, 754]) | 5\n",
-      "13096    | torch.Size([1, 1000, 754]) | 5\n",
-      "13097    | torch.Size([1, 1000, 754]) | 5\n",
-      "13098    | torch.Size([1, 1000, 789]) | 5\n",
-      "13099    | torch.Size([1, 1000, 754]) | 5\n",
-      "13100    | torch.Size([1, 1000, 754]) | 5\n",
-      "13101    | torch.Size([1, 1000, 754]) | 5\n",
-      "13102    | torch.Size([1, 1000, 781]) | 5\n",
-      "13103    | torch.Size([1, 1000, 754]) | 5\n",
-      "13104    | torch.Size([1, 1000, 754]) | 5\n",
-      "13105    | torch.Size([1, 1000, 797]) | 5\n",
-      "13106    | torch.Size([1, 1000, 754]) | 5\n",
-      "13107    | torch.Size([1, 1000, 754]) | 5\n",
-      "13108    | torch.Size([1, 1000, 754]) | 5\n",
-      "13109    | torch.Size([1, 1000, 786]) | 5\n",
-      "13110    | torch.Size([1, 1000, 754]) | 5\n",
-      "13111    | torch.Size([1, 1000, 754]) | 5\n",
-      "13112    | torch.Size([1, 1000, 754]) | 5\n",
-      "13113    | torch.Size([1, 1000, 754]) | 5\n",
-      "13114    | torch.Size([1, 1000, 754]) | 5\n",
-      "13115    | torch.Size([1, 1000, 754]) | 5\n",
-      "13116    | torch.Size([1, 1000, 754]) | 5\n",
-      "13117    | torch.Size([1, 1000, 771]) | 5\n",
-      "13118    | torch.Size([1, 1000, 754]) | 5\n",
-      "13119    | torch.Size([1, 1000, 754]) | 5\n",
-      "13120    | torch.Size([1, 1000, 754]) | 5\n",
-      "13121    | torch.Size([1, 1000, 754]) | 5\n",
-      "13122    | torch.Size([1, 1000, 754]) | 5\n",
-      "13123    | torch.Size([1, 1000, 754]) | 5\n",
-      "13124    | torch.Size([1, 1000, 754]) | 5\n",
-      "13125    | torch.Size([1, 1000, 754]) | 5\n",
-      "13126    | torch.Size([1, 1000, 754]) | 5\n",
-      "13127    | torch.Size([1, 1000, 754]) | 5\n",
-      "13128    | torch.Size([1, 1000, 754]) | 5\n",
-      "13129    | torch.Size([1, 1000, 771]) | 5\n",
-      "13130    | torch.Size([1, 1000, 754]) | 5\n",
-      "13131    | torch.Size([1, 1000, 754]) | 5\n",
-      "13132    | torch.Size([1, 1000, 754]) | 5\n",
-      "13133    | torch.Size([1, 1000, 754]) | 5\n",
-      "13134    | torch.Size([1, 1000, 754]) | 5\n",
-      "13135    | torch.Size([1, 1000, 754]) | 5\n",
-      "13136    | torch.Size([1, 1000, 754]) | 5\n",
-      "13137    | torch.Size([1, 1000, 786]) | 5\n",
-      "13138    | torch.Size([1, 1000, 754]) | 5\n",
-      "13139    | torch.Size([1, 1000, 754]) | 5\n",
-      "13140    | torch.Size([1, 1000, 785]) | 5\n",
-      "13141    | torch.Size([1, 1000, 777]) | 5\n",
-      "13142    | torch.Size([1, 1000, 754]) | 5\n",
-      "13143    | torch.Size([1, 1000, 771]) | 5\n",
-      "13144    | torch.Size([1, 1000, 754]) | 5\n",
-      "13145    | torch.Size([1, 1000, 754]) | 5\n",
-      "13146    | torch.Size([1, 1000, 754]) | 5\n",
-      "13147    | torch.Size([1, 1000, 754]) | 5\n",
-      "13148    | torch.Size([1, 1000, 754]) | 5\n",
-      "13149    | torch.Size([1, 1000, 754]) | 5\n",
-      "13150    | torch.Size([1, 1000, 786]) | 5\n",
-      "13151    | torch.Size([1, 1000, 754]) | 5\n",
-      "13152    | torch.Size([1, 1000, 754]) | 5\n",
-      "13153    | torch.Size([1, 1000, 754]) | 5\n",
-      "13154    | torch.Size([1, 1000, 754]) | 5\n",
-      "13155    | torch.Size([1, 1000, 781]) | 5\n",
-      "13156    | torch.Size([1, 1000, 754]) | 5\n",
-      "13157    | torch.Size([1, 1000, 788]) | 5\n",
-      "13158    | torch.Size([1, 1000, 754]) | 5\n",
-      "13159    | torch.Size([1, 1000, 754]) | 5\n",
-      "13160    | torch.Size([1, 1000, 779]) | 5\n",
-      "13161    | torch.Size([1, 1000, 787]) | 5\n",
-      "13162    | torch.Size([1, 1000, 754]) | 5\n",
-      "13163    | torch.Size([1, 1000, 784]) | 5\n",
-      "13164    | torch.Size([1, 1000, 754]) | 5\n",
-      "13165    | torch.Size([1, 1000, 754]) | 5\n",
-      "13166    | torch.Size([1, 1000, 754]) | 5\n",
-      "13167    | torch.Size([1, 1000, 754]) | 5\n",
-      "13168    | torch.Size([1, 1000, 754]) | 5\n",
-      "13169    | torch.Size([1, 1000, 754]) | 5\n",
-      "13170    | torch.Size([1, 1000, 754]) | 5\n",
-      "13171    | torch.Size([1, 1000, 754]) | 5\n",
-      "13172    | torch.Size([1, 1000, 754]) | 5\n",
-      "13173    | torch.Size([1, 1000, 754]) | 5\n",
-      "13174    | torch.Size([1, 1000, 754]) | 5\n",
-      "13175    | torch.Size([1, 1000, 754]) | 5\n",
-      "13176    | torch.Size([1, 1000, 789]) | 5\n",
-      "13177    | torch.Size([1, 1000, 754]) | 5\n",
-      "13178    | torch.Size([1, 1000, 754]) | 5\n",
-      "13179    | torch.Size([1, 1000, 754]) | 5\n",
-      "13180    | torch.Size([1, 1000, 754]) | 5\n",
-      "13181    | torch.Size([1, 1000, 754]) | 5\n",
-      "13182    | torch.Size([1, 1000, 754]) | 5\n",
-      "13183    | torch.Size([1, 1000, 754]) | 5\n",
-      "13184    | torch.Size([1, 1000, 771]) | 5\n",
-      "13185    | torch.Size([1, 1000, 754]) | 5\n",
-      "13186    | torch.Size([1, 1000, 754]) | 5\n",
-      "13187    | torch.Size([1, 1000, 773]) | 5\n",
-      "13188    | torch.Size([1, 1000, 775]) | 5\n",
-      "13189    | torch.Size([1, 1000, 754]) | 5\n",
-      "13190    | torch.Size([1, 1000, 754]) | 5\n",
-      "13191    | torch.Size([1, 1000, 754]) | 5\n",
-      "13192    | torch.Size([1, 1000, 754]) | 5\n",
-      "13193    | torch.Size([1, 1000, 778]) | 5\n",
-      "13194    | torch.Size([1, 1000, 754]) | 5\n",
-      "13195    | torch.Size([1, 1000, 788]) | 5\n",
-      "13196    | torch.Size([1, 1000, 754]) | 5\n",
-      "13197    | torch.Size([1, 1000, 754]) | 5\n",
-      "13198    | torch.Size([1, 1000, 754]) | 5\n",
-      "13199    | torch.Size([1, 1000, 754]) | 5\n",
-      "13200    | torch.Size([1, 1000, 754]) | 5\n",
-      "13201    | torch.Size([1, 1000, 754]) | 5\n",
-      "13202    | torch.Size([1, 1000, 754]) | 5\n",
-      "13203    | torch.Size([1, 1000, 754]) | 5\n",
-      "13204    | torch.Size([1, 1000, 754]) | 5\n",
-      "13205    | torch.Size([1, 1000, 754]) | 5\n",
-      "13206    | torch.Size([1, 1000, 754]) | 5\n",
-      "13207    | torch.Size([1, 1000, 754]) | 5\n",
-      "13208    | torch.Size([1, 1000, 822]) | 5\n",
-      "13209    | torch.Size([1, 1000, 779]) | 5\n",
-      "13210    | torch.Size([1, 1000, 754]) | 5\n",
-      "13211    | torch.Size([1, 1000, 784]) | 5\n",
-      "13212    | torch.Size([1, 1000, 754]) | 5\n",
-      "13213    | torch.Size([1, 1000, 754]) | 5\n",
-      "13214    | torch.Size([1, 1000, 779]) | 5\n",
-      "13215    | torch.Size([1, 1000, 754]) | 5\n",
-      "13216    | torch.Size([1, 1000, 754]) | 5\n",
-      "13217    | torch.Size([1, 1000, 754]) | 5\n",
-      "13218    | torch.Size([1, 1000, 782]) | 5\n",
-      "13219    | torch.Size([1, 1000, 754]) | 5\n",
-      "13220    | torch.Size([1, 1000, 754]) | 5\n",
-      "13221    | torch.Size([1, 1000, 754]) | 5\n",
-      "13222    | torch.Size([1, 1000, 754]) | 5\n",
-      "13223    | torch.Size([1, 1000, 778]) | 5\n",
-      "13224    | torch.Size([1, 1000, 788]) | 5\n",
-      "13225    | torch.Size([1, 1000, 754]) | 5\n",
-      "13226    | torch.Size([1, 1000, 754]) | 5\n",
-      "13227    | torch.Size([1, 1000, 754]) | 5\n",
-      "13228    | torch.Size([1, 1000, 754]) | 5\n",
-      "13229    | torch.Size([1, 1000, 754]) | 5\n",
-      "13230    | torch.Size([1, 1000, 754]) | 5\n",
-      "13231    | torch.Size([1, 1000, 754]) | 5\n",
-      "13232    | torch.Size([1, 1000, 754]) | 5\n",
-      "13233    | torch.Size([1, 1000, 771]) | 5\n",
-      "13234    | torch.Size([1, 1000, 780]) | 5\n",
-      "13235    | torch.Size([1, 1000, 771]) | 5\n",
-      "13236    | torch.Size([1, 1000, 780]) | 5\n",
-      "13237    | torch.Size([1, 1000, 754]) | 5\n",
-      "13238    | torch.Size([1, 1000, 786]) | 5\n",
-      "13239    | torch.Size([1, 1000, 754]) | 5\n",
-      "13240    | torch.Size([1, 1000, 754]) | 5\n",
-      "13241    | torch.Size([1, 1000, 754]) | 5\n",
-      "13242    | torch.Size([1, 1000, 754]) | 5\n",
-      "13243    | torch.Size([1, 1000, 775]) | 5\n",
-      "13244    | torch.Size([1, 1000, 754]) | 5\n",
-      "13245    | torch.Size([1, 1000, 779]) | 5\n",
-      "13246    | torch.Size([1, 1000, 754]) | 5\n",
-      "13247    | torch.Size([1, 1000, 754]) | 5\n",
-      "13248    | torch.Size([1, 1000, 754]) | 5\n",
-      "13249    | torch.Size([1, 1000, 754]) | 5\n",
-      "13250    | torch.Size([1, 1000, 754]) | 5\n",
-      "13251    | torch.Size([1, 1000, 754]) | 5\n",
-      "13252    | torch.Size([1, 1000, 806]) | 5\n",
-      "13253    | torch.Size([1, 1000, 754]) | 5\n",
-      "13254    | torch.Size([1, 1000, 771]) | 5\n",
-      "13255    | torch.Size([1, 1000, 754]) | 5\n",
-      "13256    | torch.Size([1, 1000, 754]) | 5\n",
-      "13257    | torch.Size([1, 1000, 781]) | 5\n",
-      "13258    | torch.Size([1, 1000, 754]) | 5\n",
-      "13259    | torch.Size([1, 1000, 774]) | 5\n",
-      "13260    | torch.Size([1, 1000, 754]) | 5\n",
-      "13261    | torch.Size([1, 1000, 754]) | 5\n",
-      "13262    | torch.Size([1, 1000, 754]) | 5\n",
-      "13263    | torch.Size([1, 1000, 754]) | 5\n",
-      "13264    | torch.Size([1, 1000, 754]) | 5\n",
-      "13265    | torch.Size([1, 1000, 754]) | 5\n",
-      "13266    | torch.Size([1, 1000, 754]) | 5\n",
-      "13267    | torch.Size([1, 1000, 754]) | 5\n",
-      "13268    | torch.Size([1, 1000, 777]) | 5\n",
-      "13269    | torch.Size([1, 1000, 754]) | 5\n",
-      "13270    | torch.Size([1, 1000, 754]) | 5\n",
-      "13271    | torch.Size([1, 1000, 754]) | 5\n",
-      "13272    | torch.Size([1, 1000, 754]) | 5\n",
-      "13273    | torch.Size([1, 1000, 780]) | 5\n",
-      "13274    | torch.Size([1, 1000, 754]) | 5\n",
-      "13275    | torch.Size([1, 1000, 782]) | 5\n",
-      "13276    | torch.Size([1, 1000, 754]) | 5\n",
-      "13277    | torch.Size([1, 1000, 754]) | 5\n",
-      "13278    | torch.Size([1, 1000, 754]) | 5\n",
-      "13279    | torch.Size([1, 1000, 754]) | 5\n",
-      "13280    | torch.Size([1, 1000, 754]) | 5\n",
-      "13281    | torch.Size([1, 1000, 754]) | 5\n",
-      "13282    | torch.Size([1, 1000, 754]) | 5\n",
-      "13283    | torch.Size([1, 1000, 754]) | 5\n",
-      "13284    | torch.Size([1, 1000, 754]) | 5\n",
-      "13285    | torch.Size([1, 1000, 773]) | 5\n",
-      "13286    | torch.Size([1, 1000, 754]) | 5\n",
-      "13287    | torch.Size([1, 1000, 754]) | 5\n",
-      "13288    | torch.Size([1, 1000, 754]) | 5\n",
-      "13289    | torch.Size([1, 1000, 754]) | 5\n",
-      "13290    | torch.Size([1, 1000, 754]) | 5\n",
-      "13291    | torch.Size([1, 1000, 783]) | 5\n",
-      "13292    | torch.Size([1, 1000, 754]) | 5\n",
-      "13293    | torch.Size([1, 1000, 754]) | 5\n",
-      "13294    | torch.Size([1, 1000, 754]) | 5\n",
-      "13295    | torch.Size([1, 1000, 754]) | 5\n",
-      "13296    | torch.Size([1, 1000, 754]) | 5\n",
-      "13297    | torch.Size([1, 1000, 771]) | 5\n",
-      "13298    | torch.Size([1, 1000, 754]) | 5\n",
-      "13299    | torch.Size([1, 1000, 754]) | 5\n",
-      "13300    | torch.Size([1, 1000, 788]) | 5\n",
-      "13301    | torch.Size([1, 1000, 754]) | 5\n",
-      "13302    | torch.Size([1, 1000, 775]) | 5\n",
-      "13303    | torch.Size([1, 1000, 754]) | 5\n",
-      "13304    | torch.Size([1, 1000, 754]) | 5\n",
-      "13305    | torch.Size([1, 1000, 754]) | 5\n",
-      "13306    | torch.Size([1, 1000, 754]) | 5\n",
-      "13307    | torch.Size([1, 1000, 771]) | 5\n",
-      "13308    | torch.Size([1, 1000, 754]) | 5\n",
-      "13309    | torch.Size([1, 1000, 754]) | 5\n",
-      "13310    | torch.Size([1, 1000, 754]) | 5\n",
-      "13311    | torch.Size([1, 1000, 754]) | 5\n",
-      "13312    | torch.Size([1, 1000, 783]) | 5\n",
-      "13313    | torch.Size([1, 1000, 784]) | 5\n",
-      "13314    | torch.Size([1, 1000, 754]) | 5\n",
-      "13315    | torch.Size([1, 1000, 754]) | 5\n",
-      "13316    | torch.Size([1, 1000, 754]) | 5\n",
-      "13317    | torch.Size([1, 1000, 779]) | 5\n",
-      "13318    | torch.Size([1, 1000, 754]) | 5\n",
-      "13319    | torch.Size([1, 1000, 754]) | 5\n",
-      "13320    | torch.Size([1, 1000, 754]) | 5\n",
-      "13321    | torch.Size([1, 1000, 754]) | 5\n",
-      "13322    | torch.Size([1, 1000, 754]) | 5\n",
-      "13323    | torch.Size([1, 1000, 754]) | 5\n",
-      "13324    | torch.Size([1, 1000, 754]) | 5\n",
-      "13325    | torch.Size([1, 1000, 754]) | 5\n",
-      "13326    | torch.Size([1, 1000, 790]) | 5\n",
-      "13327    | torch.Size([1, 1000, 754]) | 5\n",
-      "13328    | torch.Size([1, 1000, 754]) | 5\n",
-      "13329    | torch.Size([1, 1000, 754]) | 5\n",
-      "13330    | torch.Size([1, 1000, 754]) | 5\n",
-      "13331    | torch.Size([1, 1000, 771]) | 5\n",
-      "13332    | torch.Size([1, 1000, 785]) | 5\n",
-      "13333    | torch.Size([1, 1000, 754]) | 5\n",
-      "13334    | torch.Size([1, 1000, 781]) | 5\n",
-      "13335    | torch.Size([1, 1000, 766]) | 5\n",
-      "13336    | torch.Size([1, 1000, 776]) | 5\n",
-      "13337    | torch.Size([1, 1000, 754]) | 5\n",
-      "13338    | torch.Size([1, 1000, 788]) | 5\n",
-      "13339    | torch.Size([1, 1000, 787]) | 5\n",
-      "13340    | torch.Size([1, 1000, 771]) | 5\n",
-      "13341    | torch.Size([1, 1000, 754]) | 5\n",
-      "13342    | torch.Size([1, 1000, 754]) | 5\n",
-      "13343    | torch.Size([1, 1000, 754]) | 5\n",
-      "13344    | torch.Size([1, 1000, 754]) | 5\n",
-      "13345    | torch.Size([1, 1000, 754]) | 5\n",
-      "13346    | torch.Size([1, 1000, 787]) | 5\n",
-      "13347    | torch.Size([1, 1000, 754]) | 5\n",
-      "13348    | torch.Size([1, 1000, 790]) | 5\n",
-      "13349    | torch.Size([1, 1000, 754]) | 5\n",
-      "13350    | torch.Size([1, 1000, 754]) | 5\n",
-      "13351    | torch.Size([1, 1000, 754]) | 5\n",
-      "13352    | torch.Size([1, 1000, 754]) | 5\n",
-      "13353    | torch.Size([1, 1000, 823]) | 5\n",
-      "13354    | torch.Size([1, 1000, 754]) | 5\n",
-      "13355    | torch.Size([1, 1000, 754]) | 5\n",
-      "13356    | torch.Size([1, 1000, 785]) | 5\n",
-      "13357    | torch.Size([1, 1000, 754]) | 5\n",
-      "13358    | torch.Size([1, 1000, 788]) | 5\n",
-      "13359    | torch.Size([1, 1000, 754]) | 5\n",
-      "13360    | torch.Size([1, 1000, 787]) | 5\n",
-      "13361    | torch.Size([1, 1000, 754]) | 5\n",
-      "13362    | torch.Size([1, 1000, 754]) | 5\n",
-      "13363    | torch.Size([1, 1000, 754]) | 5\n",
-      "13364    | torch.Size([1, 1000, 754]) | 5\n",
-      "13365    | torch.Size([1, 1000, 771]) | 5\n",
-      "13366    | torch.Size([1, 1000, 789]) | 5\n",
-      "13367    | torch.Size([1, 1000, 754]) | 5\n",
-      "13368    | torch.Size([1, 1000, 778]) | 5\n",
-      "13369    | torch.Size([1, 1000, 754]) | 5\n",
-      "13370    | torch.Size([1, 1000, 754]) | 5\n",
-      "13371    | torch.Size([1, 1000, 790]) | 5\n",
-      "13372    | torch.Size([1, 1000, 754]) | 5\n",
-      "13373    | torch.Size([1, 1000, 776]) | 5\n",
-      "13374    | torch.Size([1, 1000, 754]) | 5\n",
-      "13375    | torch.Size([1, 1000, 783]) | 5\n",
-      "13376    | torch.Size([1, 1000, 765]) | 5\n",
-      "13377    | torch.Size([1, 1000, 754]) | 5\n",
-      "13378    | torch.Size([1, 1000, 786]) | 5\n",
-      "13379    | torch.Size([1, 1000, 754]) | 5\n",
-      "13380    | torch.Size([1, 1000, 754]) | 5\n",
-      "13381    | torch.Size([1, 1000, 783]) | 5\n",
-      "13382    | torch.Size([1, 1000, 786]) | 5\n",
-      "13383    | torch.Size([1, 1000, 754]) | 5\n",
-      "13384    | torch.Size([1, 1000, 754]) | 5\n",
-      "13385    | torch.Size([1, 1000, 754]) | 5\n",
-      "13386    | torch.Size([1, 1000, 754]) | 5\n",
-      "13387    | torch.Size([1, 1000, 754]) | 5\n",
-      "13388    | torch.Size([1, 1000, 754]) | 5\n",
-      "13389    | torch.Size([1, 1000, 771]) | 5\n",
-      "13390    | torch.Size([1, 1000, 754]) | 5\n",
-      "13391    | torch.Size([1, 1000, 754]) | 5\n",
-      "13392    | torch.Size([1, 1000, 754]) | 5\n",
-      "13393    | torch.Size([1, 1000, 754]) | 5\n",
-      "13394    | torch.Size([1, 1000, 754]) | 5\n",
-      "13395    | torch.Size([1, 1000, 767]) | 5\n",
-      "13396    | torch.Size([1, 1000, 754]) | 5\n",
-      "13397    | torch.Size([1, 1000, 754]) | 5\n",
-      "13398    | torch.Size([1, 1000, 754]) | 5\n",
-      "13399    | torch.Size([1, 1000, 754]) | 5\n",
-      "13400    | torch.Size([1, 1000, 754]) | 5\n",
-      "13401    | torch.Size([1, 1000, 754]) | 5\n",
-      "13402    | torch.Size([1, 1000, 754]) | 5\n",
-      "13403    | torch.Size([1, 1000, 754]) | 5\n",
-      "13404    | torch.Size([1, 1000, 754]) | 5\n",
-      "13405    | torch.Size([1, 1000, 787]) | 5\n",
-      "13406    | torch.Size([1, 1000, 773]) | 5\n",
-      "13407    | torch.Size([1, 1000, 754]) | 5\n",
-      "13408    | torch.Size([1, 1000, 754]) | 5\n",
-      "13409    | torch.Size([1, 1000, 754]) | 5\n",
-      "13410    | torch.Size([1, 1000, 754]) | 5\n",
-      "13411    | torch.Size([1, 1000, 773]) | 5\n",
-      "13412    | torch.Size([1, 1000, 781]) | 5\n",
-      "13413    | torch.Size([1, 1000, 802]) | 5\n",
-      "13414    | torch.Size([1, 1000, 771]) | 5\n",
-      "13415    | torch.Size([1, 1000, 783]) | 5\n",
-      "13416    | torch.Size([1, 1000, 754]) | 5\n",
-      "13417    | torch.Size([1, 1000, 754]) | 5\n",
-      "13418    | torch.Size([1, 1000, 754]) | 5\n",
-      "13419    | torch.Size([1, 1000, 754]) | 5\n",
-      "13420    | torch.Size([1, 1000, 754]) | 5\n",
-      "13421    | torch.Size([1, 1000, 785]) | 5\n",
-      "13422    | torch.Size([1, 1000, 754]) | 5\n",
-      "13423    | torch.Size([1, 1000, 754]) | 5\n",
-      "13424    | torch.Size([1, 1000, 754]) | 5\n",
-      "13425    | torch.Size([1, 1000, 754]) | 5\n",
-      "13426    | torch.Size([1, 1000, 754]) | 5\n",
-      "13427    | torch.Size([1, 1000, 754]) | 5\n",
-      "13428    | torch.Size([1, 1000, 754]) | 5\n",
-      "13429    | torch.Size([1, 1000, 775]) | 5\n",
-      "13430    | torch.Size([1, 1000, 754]) | 5\n",
-      "13431    | torch.Size([1, 1000, 754]) | 5\n",
-      "13432    | torch.Size([1, 1000, 754]) | 5\n",
-      "13433    | torch.Size([1, 1000, 754]) | 5\n",
-      "13434    | torch.Size([1, 1000, 775]) | 5\n",
-      "13435    | torch.Size([1, 1000, 775]) | 5\n",
-      "13436    | torch.Size([1, 1000, 754]) | 5\n",
-      "13437    | torch.Size([1, 1000, 754]) | 5\n",
-      "13438    | torch.Size([1, 1000, 754]) | 5\n",
-      "13439    | torch.Size([1, 1000, 754]) | 5\n",
-      "13440    | torch.Size([1, 1000, 772]) | 5\n",
-      "13441    | torch.Size([1, 1000, 771]) | 5\n",
-      "13442    | torch.Size([1, 1000, 754]) | 5\n",
-      "13443    | torch.Size([1, 1000, 754]) | 5\n",
-      "13444    | torch.Size([1, 1000, 754]) | 5\n",
-      "13445    | torch.Size([1, 1000, 754]) | 5\n",
-      "13446    | torch.Size([1, 1000, 754]) | 5\n",
-      "13447    | torch.Size([1, 1000, 754]) | 5\n",
-      "13448    | torch.Size([1, 1000, 780]) | 5\n",
-      "13449    | torch.Size([1, 1000, 778]) | 5\n",
-      "13450    | torch.Size([1, 1000, 754]) | 5\n",
-      "13451    | torch.Size([1, 1000, 754]) | 5\n",
-      "13452    | torch.Size([1, 1000, 754]) | 5\n",
-      "13453    | torch.Size([1, 1000, 785]) | 5\n",
-      "13454    | torch.Size([1, 1000, 754]) | 5\n",
-      "13455    | torch.Size([1, 1000, 787]) | 5\n",
-      "13456    | torch.Size([1, 1000, 779]) | 5\n",
-      "13457    | torch.Size([1, 1000, 778]) | 5\n",
-      "13458    | torch.Size([1, 1000, 754]) | 5\n",
-      "13459    | torch.Size([1, 1000, 785]) | 5\n",
-      "13460    | torch.Size([1, 1000, 774]) | 5\n",
-      "13461    | torch.Size([1, 1000, 771]) | 5\n",
-      "13462    | torch.Size([1, 1000, 754]) | 5\n",
-      "13463    | torch.Size([1, 1000, 791]) | 5\n",
-      "13464    | torch.Size([1, 1000, 754]) | 5\n",
-      "13465    | torch.Size([1, 1000, 754]) | 5\n",
-      "13466    | torch.Size([1, 1000, 754]) | 5\n",
-      "13467    | torch.Size([1, 1000, 778]) | 5\n",
-      "13468    | torch.Size([1, 1000, 754]) | 5\n",
-      "13469    | torch.Size([1, 1000, 754]) | 5\n",
-      "13470    | torch.Size([1, 1000, 754]) | 5\n",
-      "13471    | torch.Size([1, 1000, 754]) | 5\n",
-      "13472    | torch.Size([1, 1000, 754]) | 5\n",
-      "13473    | torch.Size([1, 1000, 777]) | 5\n",
-      "13474    | torch.Size([1, 1000, 754]) | 5\n",
-      "13475    | torch.Size([1, 1000, 784]) | 5\n",
-      "13476    | torch.Size([1, 1000, 754]) | 5\n",
-      "13477    | torch.Size([1, 1000, 754]) | 5\n",
-      "13478    | torch.Size([1, 1000, 754]) | 5\n",
-      "13479    | torch.Size([1, 1000, 754]) | 5\n",
-      "13480    | torch.Size([1, 1000, 786]) | 5\n",
-      "13481    | torch.Size([1, 1000, 754]) | 5\n",
-      "13482    | torch.Size([1, 1000, 754]) | 5\n",
-      "13483    | torch.Size([1, 1000, 754]) | 5\n",
-      "13484    | torch.Size([1, 1000, 754]) | 5\n",
-      "13485    | torch.Size([1, 1000, 754]) | 5\n",
-      "13486    | torch.Size([1, 1000, 784]) | 5\n",
-      "13487    | torch.Size([1, 1000, 754]) | 5\n",
-      "13488    | torch.Size([1, 1000, 754]) | 5\n",
-      "13489    | torch.Size([1, 1000, 754]) | 5\n",
-      "13490    | torch.Size([1, 1000, 754]) | 5\n",
-      "13491    | torch.Size([1, 1000, 788]) | 5\n",
-      "13492    | torch.Size([1, 1000, 774]) | 5\n",
-      "13493    | torch.Size([1, 1000, 774]) | 5\n",
-      "13494    | torch.Size([1, 1000, 754]) | 5\n",
-      "13495    | torch.Size([1, 1000, 754]) | 5\n",
-      "13496    | torch.Size([1, 1000, 754]) | 5\n",
-      "13497    | torch.Size([1, 1000, 781]) | 5\n",
-      "13498    | torch.Size([1, 1000, 754]) | 5\n",
-      "13499    | torch.Size([1, 1000, 754]) | 5\n",
-      "13500    | torch.Size([1, 1000, 754]) | 5\n",
-      "13501    | torch.Size([1, 1000, 754]) | 5\n",
-      "13502    | torch.Size([1, 1000, 754]) | 5\n",
-      "13503    | torch.Size([1, 1000, 754]) | 5\n",
-      "13504    | torch.Size([1, 1000, 779]) | 5\n",
-      "13505    | torch.Size([1, 1000, 754]) | 5\n",
-      "13506    | torch.Size([1, 1000, 754]) | 5\n",
-      "13507    | torch.Size([1, 1000, 754]) | 5\n",
-      "13508    | torch.Size([1, 1000, 754]) | 5\n",
-      "13509    | torch.Size([1, 1000, 754]) | 5\n",
-      "13510    | torch.Size([1, 1000, 775]) | 5\n",
-      "13511    | torch.Size([1, 1000, 774]) | 5\n",
-      "13512    | torch.Size([1, 1000, 754]) | 5\n",
-      "13513    | torch.Size([1, 1000, 740]) | 5\n",
-      "13514    | torch.Size([1, 1000, 754]) | 5\n",
-      "13515    | torch.Size([1, 1000, 754]) | 5\n",
-      "13516    | torch.Size([1, 1000, 754]) | 5\n",
-      "13517    | torch.Size([1, 1000, 781]) | 5\n",
-      "13518    | torch.Size([1, 1000, 754]) | 5\n",
-      "13519    | torch.Size([1, 1000, 754]) | 5\n",
-      "13520    | torch.Size([1, 1000, 754]) | 5\n",
-      "13521    | torch.Size([1, 1000, 754]) | 5\n",
-      "13522    | torch.Size([1, 1000, 754]) | 5\n",
-      "13523    | torch.Size([1, 1000, 754]) | 5\n",
-      "13524    | torch.Size([1, 1000, 754]) | 5\n",
-      "13525    | torch.Size([1, 1000, 754]) | 5\n",
-      "13526    | torch.Size([1, 1000, 754]) | 5\n",
-      "13527    | torch.Size([1, 1000, 789]) | 5\n",
-      "13528    | torch.Size([1, 1000, 754]) | 5\n",
-      "13529    | torch.Size([1, 1000, 787]) | 5\n",
-      "13530    | torch.Size([1, 1000, 754]) | 5\n",
-      "13531    | torch.Size([1, 1000, 754]) | 5\n",
-      "13532    | torch.Size([1, 1000, 754]) | 5\n",
-      "13533    | torch.Size([1, 1000, 771]) | 5\n",
-      "13534    | torch.Size([1, 1000, 754]) | 5\n",
-      "13535    | torch.Size([1, 1000, 754]) | 5\n",
-      "13536    | torch.Size([1, 1000, 771]) | 5\n",
-      "13537    | torch.Size([1, 1000, 754]) | 5\n",
-      "13538    | torch.Size([1, 1000, 802]) | 5\n",
-      "13539    | torch.Size([1, 1000, 754]) | 5\n",
-      "13540    | torch.Size([1, 1000, 782]) | 5\n",
-      "13541    | torch.Size([1, 1000, 780]) | 5\n",
-      "13542    | torch.Size([1, 1000, 754]) | 5\n",
-      "13543    | torch.Size([1, 1000, 754]) | 5\n",
-      "13544    | torch.Size([1, 1000, 754]) | 5\n",
-      "13545    | torch.Size([1, 1000, 754]) | 5\n",
-      "13546    | torch.Size([1, 1000, 793]) | 5\n",
-      "13547    | torch.Size([1, 1000, 777]) | 5\n",
-      "13548    | torch.Size([1, 1000, 754]) | 5\n",
-      "13549    | torch.Size([1, 1000, 771]) | 5\n",
-      "13550    | torch.Size([1, 1000, 754]) | 5\n",
-      "13551    | torch.Size([1, 1000, 754]) | 5\n",
-      "13552    | torch.Size([1, 1000, 754]) | 5\n",
-      "13553    | torch.Size([1, 1000, 754]) | 5\n",
-      "13554    | torch.Size([1, 1000, 754]) | 5\n",
-      "13555    | torch.Size([1, 1000, 810]) | 5\n",
-      "13556    | torch.Size([1, 1000, 787]) | 5\n",
-      "13557    | torch.Size([1, 1000, 754]) | 5\n",
-      "13558    | torch.Size([1, 1000, 784]) | 5\n",
-      "13559    | torch.Size([1, 1000, 764]) | 5\n",
-      "13560    | torch.Size([1, 1000, 754]) | 5\n",
-      "13561    | torch.Size([1, 1000, 754]) | 5\n",
-      "13562    | torch.Size([1, 1000, 754]) | 5\n",
-      "13563    | torch.Size([1, 1000, 754]) | 5\n",
-      "13564    | torch.Size([1, 1000, 789]) | 5\n",
-      "13565    | torch.Size([1, 1000, 777]) | 5\n",
-      "13566    | torch.Size([1, 1000, 754]) | 5\n",
-      "13567    | torch.Size([1, 1000, 754]) | 5\n",
-      "13568    | torch.Size([1, 1000, 780]) | 5\n",
-      "13569    | torch.Size([1, 1000, 781]) | 5\n",
-      "13570    | torch.Size([1, 1000, 754]) | 5\n",
-      "13571    | torch.Size([1, 1000, 754]) | 5\n",
-      "13572    | torch.Size([1, 1000, 771]) | 5\n",
-      "13573    | torch.Size([1, 1000, 754]) | 5\n",
-      "13574    | torch.Size([1, 1000, 754]) | 5\n",
-      "13575    | torch.Size([1, 1000, 776]) | 5\n",
-      "13576    | torch.Size([1, 1000, 754]) | 5\n",
-      "13577    | torch.Size([1, 1000, 754]) | 5\n",
-      "13578    | torch.Size([1, 1000, 787]) | 5\n",
-      "13579    | torch.Size([1, 1000, 777]) | 5\n",
-      "13580    | torch.Size([1, 1000, 774]) | 5\n",
-      "13581    | torch.Size([1, 1000, 754]) | 5\n",
-      "13582    | torch.Size([1, 1000, 771]) | 5\n",
-      "13583    | torch.Size([1, 1000, 754]) | 5\n",
-      "13584    | torch.Size([1, 1000, 754]) | 5\n",
-      "13585    | torch.Size([1, 1000, 754]) | 5\n",
-      "13586    | torch.Size([1, 1000, 781]) | 5\n",
-      "13587    | torch.Size([1, 1000, 771]) | 5\n",
-      "13588    | torch.Size([1, 1000, 754]) | 5\n",
-      "13589    | torch.Size([1, 1000, 754]) | 5\n",
-      "13590    | torch.Size([1, 1000, 754]) | 5\n",
-      "13591    | torch.Size([1, 1000, 754]) | 5\n",
-      "13592    | torch.Size([1, 1000, 754]) | 5\n",
-      "13593    | torch.Size([1, 1000, 754]) | 5\n",
-      "13594    | torch.Size([1, 1000, 754]) | 5\n",
-      "13595    | torch.Size([1, 1000, 754]) | 5\n",
-      "13596    | torch.Size([1, 1000, 754]) | 5\n",
-      "13597    | torch.Size([1, 1000, 754]) | 5\n",
-      "13598    | torch.Size([1, 1000, 754]) | 5\n",
-      "13599    | torch.Size([1, 1000, 754]) | 5\n",
-      "13600    | torch.Size([1, 1000, 775]) | 5\n",
-      "13601    | torch.Size([1, 1000, 754]) | 5\n",
-      "13602    | torch.Size([1, 1000, 754]) | 5\n",
-      "13603    | torch.Size([1, 1000, 754]) | 5\n",
-      "13604    | torch.Size([1, 1000, 754]) | 5\n",
-      "13605    | torch.Size([1, 1000, 754]) | 5\n",
-      "13606    | torch.Size([1, 1000, 754]) | 5\n",
-      "13607    | torch.Size([1, 1000, 773]) | 5\n",
-      "13608    | torch.Size([1, 1000, 754]) | 5\n",
-      "13609    | torch.Size([1, 1000, 754]) | 5\n",
-      "13610    | torch.Size([1, 1000, 786]) | 5\n",
-      "13611    | torch.Size([1, 1000, 754]) | 5\n",
-      "13612    | torch.Size([1, 1000, 754]) | 5\n",
-      "13613    | torch.Size([1, 1000, 754]) | 5\n",
-      "13614    | torch.Size([1, 1000, 754]) | 5\n",
-      "13615    | torch.Size([1, 1000, 754]) | 5\n",
-      "13616    | torch.Size([1, 1000, 786]) | 5\n",
-      "13617    | torch.Size([1, 1000, 754]) | 5\n",
-      "13618    | torch.Size([1, 1000, 785]) | 5\n",
-      "13619    | torch.Size([1, 1000, 771]) | 5\n",
-      "13620    | torch.Size([1, 1000, 754]) | 5\n",
-      "13621    | torch.Size([1, 1000, 754]) | 5\n",
-      "13622    | torch.Size([1, 1000, 754]) | 5\n",
-      "13623    | torch.Size([1, 1000, 774]) | 5\n",
-      "13624    | torch.Size([1, 1000, 754]) | 5\n",
-      "13625    | torch.Size([1, 1000, 754]) | 5\n",
-      "13626    | torch.Size([1, 1000, 754]) | 5\n",
-      "13627    | torch.Size([1, 1000, 754]) | 5\n",
-      "13628    | torch.Size([1, 1000, 754]) | 5\n",
-      "13629    | torch.Size([1, 1000, 784]) | 5\n",
-      "13630    | torch.Size([1, 1000, 754]) | 5\n",
-      "13631    | torch.Size([1, 1000, 754]) | 5\n",
-      "13632    | torch.Size([1, 1000, 754]) | 5\n",
-      "13633    | torch.Size([1, 1000, 790]) | 5\n",
-      "13634    | torch.Size([1, 1000, 785]) | 5\n",
-      "13635    | torch.Size([1, 1000, 754]) | 5\n",
-      "13636    | torch.Size([1, 1000, 787]) | 5\n",
-      "13637    | torch.Size([1, 1000, 754]) | 5\n",
-      "13638    | torch.Size([1, 1000, 784]) | 5\n",
-      "13639    | torch.Size([1, 1000, 771]) | 5\n",
-      "13640    | torch.Size([1, 1000, 776]) | 5\n",
-      "13641    | torch.Size([1, 1000, 773]) | 5\n",
-      "13642    | torch.Size([1, 1000, 782]) | 5\n",
-      "13643    | torch.Size([1, 1000, 754]) | 5\n",
-      "13644    | torch.Size([1, 1000, 775]) | 5\n",
-      "13645    | torch.Size([1, 1000, 786]) | 5\n",
-      "13646    | torch.Size([1, 1000, 788]) | 5\n",
-      "13647    | torch.Size([1, 1000, 754]) | 5\n",
-      "13648    | torch.Size([1, 1000, 754]) | 5\n",
-      "13649    | torch.Size([1, 1000, 754]) | 5\n",
-      "13650    | torch.Size([1, 1000, 802]) | 5\n",
-      "13651    | torch.Size([1, 1000, 771]) | 5\n",
-      "13652    | torch.Size([1, 1000, 754]) | 5\n",
-      "13653    | torch.Size([1, 1000, 754]) | 5\n",
-      "13654    | torch.Size([1, 1000, 754]) | 5\n",
-      "13655    | torch.Size([1, 1000, 754]) | 5\n",
-      "13656    | torch.Size([1, 1000, 754]) | 5\n",
-      "13657    | torch.Size([1, 1000, 754]) | 5\n",
-      "13658    | torch.Size([1, 1000, 782]) | 5\n",
-      "13659    | torch.Size([1, 1000, 781]) | 5\n",
-      "13660    | torch.Size([1, 1000, 771]) | 5\n",
-      "13661    | torch.Size([1, 1000, 754]) | 5\n",
-      "13662    | torch.Size([1, 1000, 754]) | 5\n",
-      "13663    | torch.Size([1, 1000, 778]) | 5\n",
-      "13664    | torch.Size([1, 1000, 754]) | 5\n",
-      "13665    | torch.Size([1, 1000, 754]) | 5\n",
-      "13666    | torch.Size([1, 1000, 754]) | 5\n",
-      "13667    | torch.Size([1, 1000, 790]) | 5\n",
-      "13668    | torch.Size([1, 1000, 754]) | 5\n",
-      "13669    | torch.Size([1, 1000, 754]) | 5\n",
-      "13670    | torch.Size([1, 1000, 805]) | 5\n",
-      "13671    | torch.Size([1, 1000, 754]) | 5\n",
-      "13672    | torch.Size([1, 1000, 754]) | 5\n",
-      "13673    | torch.Size([1, 1000, 754]) | 5\n",
-      "13674    | torch.Size([1, 1000, 754]) | 5\n",
-      "13675    | torch.Size([1, 1000, 790]) | 5\n",
-      "13676    | torch.Size([1, 1000, 771]) | 5\n",
-      "13677    | torch.Size([1, 1000, 754]) | 5\n",
-      "13678    | torch.Size([1, 1000, 754]) | 5\n",
-      "13679    | torch.Size([1, 1000, 754]) | 5\n",
-      "13680    | torch.Size([1, 1000, 754]) | 5\n",
-      "13681    | torch.Size([1, 1000, 754]) | 5\n",
-      "13682    | torch.Size([1, 1000, 754]) | 5\n",
-      "13683    | torch.Size([1, 1000, 754]) | 5\n",
-      "13684    | torch.Size([1, 1000, 754]) | 5\n",
-      "13685    | torch.Size([1, 1000, 754]) | 5\n",
-      "13686    | torch.Size([1, 1000, 754]) | 5\n",
-      "13687    | torch.Size([1, 1000, 754]) | 5\n",
-      "13688    | torch.Size([1, 1000, 754]) | 5\n",
-      "13689    | torch.Size([1, 1000, 754]) | 5\n",
-      "13690    | torch.Size([1, 1000, 754]) | 5\n",
-      "13691    | torch.Size([1, 1000, 754]) | 5\n",
-      "13692    | torch.Size([1, 1000, 754]) | 5\n",
-      "13693    | torch.Size([1, 1000, 754]) | 5\n",
-      "13694    | torch.Size([1, 1000, 754]) | 5\n",
-      "13695    | torch.Size([1, 1000, 754]) | 5\n",
-      "13696    | torch.Size([1, 1000, 754]) | 5\n",
-      "13697    | torch.Size([1, 1000, 754]) | 5\n",
-      "13698    | torch.Size([1, 1000, 754]) | 5\n",
-      "13699    | torch.Size([1, 1000, 754]) | 5\n",
-      "13700    | torch.Size([1, 1000, 784]) | 5\n",
-      "13701    | torch.Size([1, 1000, 785]) | 5\n",
-      "13702    | torch.Size([1, 1000, 754]) | 5\n",
-      "13703    | torch.Size([1, 1000, 754]) | 5\n",
-      "13704    | torch.Size([1, 1000, 769]) | 5\n",
-      "13705    | torch.Size([1, 1000, 754]) | 5\n",
-      "13706    | torch.Size([1, 1000, 754]) | 5\n",
-      "13707    | torch.Size([1, 1000, 754]) | 5\n",
-      "13708    | torch.Size([1, 1000, 788]) | 5\n",
-      "13709    | torch.Size([1, 1000, 754]) | 5\n",
-      "13710    | torch.Size([1, 1000, 775]) | 5\n",
-      "13711    | torch.Size([1, 1000, 771]) | 5\n",
-      "13712    | torch.Size([1, 1000, 754]) | 5\n",
-      "13713    | torch.Size([1, 1000, 754]) | 5\n",
-      "13714    | torch.Size([1, 1000, 754]) | 5\n",
-      "13715    | torch.Size([1, 1000, 754]) | 5\n",
-      "13716    | torch.Size([1, 1000, 754]) | 5\n",
-      "13717    | torch.Size([1, 1000, 791]) | 5\n",
-      "13718    | torch.Size([1, 1000, 754]) | 5\n",
-      "13719    | torch.Size([1, 1000, 754]) | 5\n",
-      "13720    | torch.Size([1, 1000, 754]) | 5\n",
-      "13721    | torch.Size([1, 1000, 754]) | 5\n",
-      "13722    | torch.Size([1, 1000, 785]) | 5\n",
-      "13723    | torch.Size([1, 1000, 774]) | 5\n",
-      "13724    | torch.Size([1, 1000, 754]) | 5\n",
-      "13725    | torch.Size([1, 1000, 754]) | 5\n",
-      "13726    | torch.Size([1, 1000, 754]) | 5\n",
-      "13727    | torch.Size([1, 1000, 754]) | 5\n",
-      "13728    | torch.Size([1, 1000, 754]) | 5\n",
-      "13729    | torch.Size([1, 1000, 754]) | 5\n",
-      "13730    | torch.Size([1, 1000, 754]) | 5\n",
-      "13731    | torch.Size([1, 1000, 776]) | 5\n",
-      "13732    | torch.Size([1, 1000, 754]) | 5\n",
-      "13733    | torch.Size([1, 1000, 778]) | 5\n",
-      "13734    | torch.Size([1, 1000, 754]) | 5\n",
-      "13735    | torch.Size([1, 1000, 754]) | 5\n",
-      "13736    | torch.Size([1, 1000, 754]) | 5\n",
-      "13737    | torch.Size([1, 1000, 785]) | 5\n",
-      "13738    | torch.Size([1, 1000, 787]) | 5\n",
-      "13739    | torch.Size([1, 1000, 754]) | 5\n",
-      "13740    | torch.Size([1, 1000, 788]) | 5\n",
-      "13741    | torch.Size([1, 1000, 754]) | 5\n",
-      "13742    | torch.Size([1, 1000, 754]) | 5\n",
-      "13743    | torch.Size([1, 1000, 798]) | 5\n",
-      "13744    | torch.Size([1, 1000, 754]) | 5\n",
-      "13745    | torch.Size([1, 1000, 784]) | 5\n",
-      "13746    | torch.Size([1, 1000, 754]) | 5\n",
-      "13747    | torch.Size([1, 1000, 754]) | 5\n",
-      "13748    | torch.Size([1, 1000, 754]) | 5\n",
-      "13749    | torch.Size([1, 1000, 754]) | 5\n",
-      "13750    | torch.Size([1, 1000, 778]) | 5\n",
-      "13751    | torch.Size([1, 1000, 754]) | 5\n",
-      "13752    | torch.Size([1, 1000, 754]) | 5\n",
-      "13753    | torch.Size([1, 1000, 754]) | 5\n",
-      "13754    | torch.Size([1, 1000, 786]) | 5\n",
-      "13755    | torch.Size([1, 1000, 754]) | 5\n",
-      "13756    | torch.Size([1, 1000, 754]) | 5\n",
-      "13757    | torch.Size([1, 1000, 754]) | 5\n",
-      "13758    | torch.Size([1, 1000, 754]) | 5\n",
-      "13759    | torch.Size([1, 1000, 754]) | 5\n",
-      "13760    | torch.Size([1, 1000, 754]) | 5\n",
-      "13761    | torch.Size([1, 1000, 787]) | 5\n",
-      "13762    | torch.Size([1, 1000, 754]) | 5\n",
-      "13763    | torch.Size([1, 1000, 754]) | 5\n",
-      "13764    | torch.Size([1, 1000, 776]) | 5\n",
-      "13765    | torch.Size([1, 1000, 754]) | 5\n",
-      "13766    | torch.Size([1, 1000, 754]) | 5\n",
-      "13767    | torch.Size([1, 1000, 754]) | 5\n",
-      "13768    | torch.Size([1, 1000, 781]) | 5\n",
-      "13769    | torch.Size([1, 1000, 783]) | 5\n",
-      "13770    | torch.Size([1, 1000, 754]) | 5\n",
-      "13771    | torch.Size([1, 1000, 754]) | 5\n",
-      "13772    | torch.Size([1, 1000, 754]) | 5\n",
-      "13773    | torch.Size([1, 1000, 787]) | 5\n",
-      "13774    | torch.Size([1, 1000, 778]) | 5\n",
-      "13775    | torch.Size([1, 1000, 754]) | 5\n",
-      "13776    | torch.Size([1, 1000, 782]) | 5\n",
-      "13777    | torch.Size([1, 1000, 754]) | 5\n",
-      "13778    | torch.Size([1, 1000, 771]) | 5\n",
-      "13779    | torch.Size([1, 1000, 786]) | 5\n",
-      "13780    | torch.Size([1, 1000, 754]) | 5\n",
-      "13781    | torch.Size([1, 1000, 754]) | 5\n",
-      "13782    | torch.Size([1, 1000, 754]) | 5\n",
-      "13783    | torch.Size([1, 1000, 781]) | 5\n",
-      "13784    | torch.Size([1, 1000, 754]) | 5\n",
-      "13785    | torch.Size([1, 1000, 780]) | 5\n",
-      "13786    | torch.Size([1, 1000, 788]) | 5\n",
-      "13787    | torch.Size([1, 1000, 754]) | 5\n",
-      "13788    | torch.Size([1, 1000, 780]) | 5\n",
-      "13789    | torch.Size([1, 1000, 754]) | 5\n",
-      "13790    | torch.Size([1, 1000, 777]) | 5\n",
-      "13791    | torch.Size([1, 1000, 778]) | 5\n",
-      "13792    | torch.Size([1, 1000, 754]) | 5\n",
-      "13793    | torch.Size([1, 1000, 754]) | 5\n",
-      "13794    | torch.Size([1, 1000, 754]) | 5\n",
-      "13795    | torch.Size([1, 1000, 779]) | 5\n",
-      "13796    | torch.Size([1, 1000, 774]) | 5\n",
-      "13797    | torch.Size([1, 1000, 754]) | 5\n",
-      "13798    | torch.Size([1, 1000, 772]) | 5\n",
-      "13799    | torch.Size([1, 1000, 754]) | 5\n",
-      "13800    | torch.Size([1, 1000, 783]) | 5\n",
-      "13801    | torch.Size([1, 1000, 754]) | 5\n",
-      "13802    | torch.Size([1, 1000, 754]) | 5\n",
-      "13803    | torch.Size([1, 1000, 775]) | 5\n",
-      "13804    | torch.Size([1, 1000, 754]) | 5\n",
-      "13805    | torch.Size([1, 1000, 771]) | 5\n",
-      "13806    | torch.Size([1, 1000, 823]) | 5\n",
-      "13807    | torch.Size([1, 1000, 754]) | 5\n",
-      "13808    | torch.Size([1, 1000, 754]) | 5\n",
-      "13809    | torch.Size([1, 1000, 754]) | 5\n",
-      "13810    | torch.Size([1, 1000, 754]) | 5\n",
-      "13811    | torch.Size([1, 1000, 754]) | 5\n",
-      "13812    | torch.Size([1, 1000, 754]) | 5\n",
-      "13813    | torch.Size([1, 1000, 767]) | 5\n",
-      "13814    | torch.Size([1, 1000, 754]) | 5\n",
-      "13815    | torch.Size([1, 1000, 754]) | 5\n",
-      "13816    | torch.Size([1, 1000, 754]) | 5\n",
-      "13817    | torch.Size([1, 1000, 778]) | 5\n",
-      "13818    | torch.Size([1, 1000, 788]) | 5\n",
-      "13819    | torch.Size([1, 1000, 754]) | 5\n",
-      "13820    | torch.Size([1, 1000, 754]) | 5\n",
-      "13821    | torch.Size([1, 1000, 754]) | 5\n",
-      "13822    | torch.Size([1, 1000, 754]) | 5\n",
-      "13823    | torch.Size([1, 1000, 754]) | 5\n",
-      "13824    | torch.Size([1, 1000, 754]) | 5\n",
-      "13825    | torch.Size([1, 1000, 783]) | 5\n",
-      "13826    | torch.Size([1, 1000, 754]) | 5\n",
-      "13827    | torch.Size([1, 1000, 754]) | 5\n",
-      "13828    | torch.Size([1, 1000, 754]) | 5\n",
-      "13829    | torch.Size([1, 1000, 742]) | 5\n",
-      "13830    | torch.Size([1, 1000, 754]) | 5\n",
-      "13831    | torch.Size([1, 1000, 754]) | 5\n",
-      "13832    | torch.Size([1, 1000, 754]) | 5\n",
-      "13833    | torch.Size([1, 1000, 754]) | 5\n",
-      "13834    | torch.Size([1, 1000, 816]) | 5\n",
-      "13835    | torch.Size([1, 1000, 754]) | 5\n",
-      "13836    | torch.Size([1, 1000, 754]) | 5\n",
-      "13837    | torch.Size([1, 1000, 754]) | 5\n",
-      "13838    | torch.Size([1, 1000, 754]) | 5\n",
-      "13839    | torch.Size([1, 1000, 754]) | 5\n",
-      "13840    | torch.Size([1, 1000, 754]) | 5\n",
-      "13841    | torch.Size([1, 1000, 754]) | 5\n",
-      "13842    | torch.Size([1, 1000, 754]) | 5\n",
-      "13843    | torch.Size([1, 1000, 754]) | 5\n",
-      "13844    | torch.Size([1, 1000, 754]) | 5\n",
-      "13845    | torch.Size([1, 1000, 754]) | 5\n",
-      "13846    | torch.Size([1, 1000, 787]) | 5\n",
-      "13847    | torch.Size([1, 1000, 754]) | 5\n",
-      "13848    | torch.Size([1, 1000, 779]) | 5\n",
-      "13849    | torch.Size([1, 1000, 754]) | 5\n",
-      "13850    | torch.Size([1, 1000, 754]) | 5\n",
-      "13851    | torch.Size([1, 1000, 754]) | 5\n",
-      "13852    | torch.Size([1, 1000, 754]) | 5\n",
-      "13853    | torch.Size([1, 1000, 754]) | 5\n",
-      "13854    | torch.Size([1, 1000, 754]) | 5\n",
-      "13855    | torch.Size([1, 1000, 754]) | 5\n",
-      "13856    | torch.Size([1, 1000, 754]) | 5\n",
-      "13857    | torch.Size([1, 1000, 784]) | 5\n",
-      "13858    | torch.Size([1, 1000, 754]) | 5\n",
-      "13859    | torch.Size([1, 1000, 754]) | 5\n",
-      "13860    | torch.Size([1, 1000, 754]) | 5\n",
-      "13861    | torch.Size([1, 1000, 754]) | 5\n",
-      "13862    | torch.Size([1, 1000, 754]) | 5\n",
-      "13863    | torch.Size([1, 1000, 793]) | 5\n",
-      "13864    | torch.Size([1, 1000, 776]) | 5\n",
-      "13865    | torch.Size([1, 1000, 781]) | 5\n",
-      "13866    | torch.Size([1, 1000, 754]) | 5\n",
-      "13867    | torch.Size([1, 1000, 754]) | 5\n",
-      "13868    | torch.Size([1, 1000, 754]) | 5\n",
-      "13869    | torch.Size([1, 1000, 801]) | 5\n",
-      "13870    | torch.Size([1, 1000, 754]) | 5\n",
-      "13871    | torch.Size([1, 1000, 775]) | 5\n",
-      "13872    | torch.Size([1, 1000, 754]) | 5\n",
-      "13873    | torch.Size([1, 1000, 754]) | 5\n",
-      "13874    | torch.Size([1, 1000, 754]) | 5\n",
-      "13875    | torch.Size([1, 1000, 754]) | 5\n",
-      "13876    | torch.Size([1, 1000, 754]) | 5\n",
-      "13877    | torch.Size([1, 1000, 754]) | 5\n",
-      "13878    | torch.Size([1, 1000, 754]) | 5\n",
-      "13879    | torch.Size([1, 1000, 754]) | 5\n",
-      "13880    | torch.Size([1, 1000, 771]) | 5\n",
-      "13881    | torch.Size([1, 1000, 788]) | 5\n",
-      "13882    | torch.Size([1, 1000, 754]) | 5\n",
-      "13883    | torch.Size([1, 1000, 754]) | 5\n",
-      "13884    | torch.Size([1, 1000, 774]) | 5\n",
-      "13885    | torch.Size([1, 1000, 773]) | 5\n",
-      "13886    | torch.Size([1, 1000, 783]) | 5\n",
-      "13887    | torch.Size([1, 1000, 754]) | 5\n",
-      "13888    | torch.Size([1, 1000, 754]) | 5\n",
-      "13889    | torch.Size([1, 1000, 754]) | 5\n",
-      "13890    | torch.Size([1, 1000, 754]) | 5\n",
-      "13891    | torch.Size([1, 1000, 754]) | 5\n",
-      "13892    | torch.Size([1, 1000, 754]) | 5\n",
-      "13893    | torch.Size([1, 1000, 754]) | 5\n",
-      "13894    | torch.Size([1, 1000, 754]) | 5\n",
-      "13895    | torch.Size([1, 1000, 784]) | 5\n",
-      "13896    | torch.Size([1, 1000, 754]) | 5\n",
-      "13897    | torch.Size([1, 1000, 754]) | 5\n",
-      "13898    | torch.Size([1, 1000, 754]) | 5\n",
-      "13899    | torch.Size([1, 1000, 754]) | 5\n",
-      "13900    | torch.Size([1, 1000, 754]) | 5\n",
-      "13901    | torch.Size([1, 1000, 754]) | 5\n",
-      "13902    | torch.Size([1, 1000, 794]) | 5\n",
-      "13903    | torch.Size([1, 1000, 754]) | 5\n",
-      "13904    | torch.Size([1, 1000, 786]) | 5\n",
-      "13905    | torch.Size([1, 1000, 754]) | 5\n",
-      "13906    | torch.Size([1, 1000, 754]) | 5\n",
-      "13907    | torch.Size([1, 1000, 778]) | 5\n",
-      "13908    | torch.Size([1, 1000, 754]) | 5\n",
-      "13909    | torch.Size([1, 1000, 754]) | 5\n",
-      "13910    | torch.Size([1, 1000, 754]) | 5\n",
-      "13911    | torch.Size([1, 1000, 754]) | 5\n",
-      "13912    | torch.Size([1, 1000, 754]) | 5\n",
-      "13913    | torch.Size([1, 1000, 771]) | 5\n",
-      "13914    | torch.Size([1, 1000, 754]) | 5\n",
-      "13915    | torch.Size([1, 1000, 754]) | 5\n",
-      "13916    | torch.Size([1, 1000, 754]) | 5\n",
-      "13917    | torch.Size([1, 1000, 785]) | 5\n",
-      "13918    | torch.Size([1, 1000, 754]) | 5\n",
-      "13919    | torch.Size([1, 1000, 754]) | 5\n",
-      "13920    | torch.Size([1, 1000, 772]) | 5\n",
-      "13921    | torch.Size([1, 1000, 754]) | 5\n",
-      "13922    | torch.Size([1, 1000, 754]) | 5\n",
-      "13923    | torch.Size([1, 1000, 741]) | 5\n",
-      "13924    | torch.Size([1, 1000, 754]) | 5\n",
-      "13925    | torch.Size([1, 1000, 754]) | 5\n",
-      "13926    | torch.Size([1, 1000, 772]) | 5\n",
-      "13927    | torch.Size([1, 1000, 754]) | 5\n",
-      "13928    | torch.Size([1, 1000, 754]) | 5\n",
-      "13929    | torch.Size([1, 1000, 779]) | 5\n",
-      "13930    | torch.Size([1, 1000, 754]) | 5\n",
-      "13931    | torch.Size([1, 1000, 754]) | 5\n",
-      "13932    | torch.Size([1, 1000, 754]) | 5\n",
-      "13933    | torch.Size([1, 1000, 754]) | 5\n",
-      "13934    | torch.Size([1, 1000, 785]) | 5\n",
-      "13935    | torch.Size([1, 1000, 792]) | 5\n",
-      "13936    | torch.Size([1, 1000, 807]) | 5\n",
-      "13937    | torch.Size([1, 1000, 754]) | 5\n",
-      "13938    | torch.Size([1, 1000, 773]) | 5\n",
-      "13939    | torch.Size([1, 1000, 754]) | 5\n",
-      "13940    | torch.Size([1, 1000, 754]) | 5\n",
-      "13941    | torch.Size([1, 1000, 776]) | 5\n",
-      "13942    | torch.Size([1, 1000, 754]) | 5\n",
-      "13943    | torch.Size([1, 1000, 754]) | 5\n",
-      "13944    | torch.Size([1, 1000, 798]) | 5\n",
-      "13945    | torch.Size([1, 1000, 754]) | 5\n",
-      "13946    | torch.Size([1, 1000, 754]) | 5\n",
-      "13947    | torch.Size([1, 1000, 754]) | 5\n",
-      "13948    | torch.Size([1, 1000, 778]) | 5\n",
-      "13949    | torch.Size([1, 1000, 754]) | 5\n",
-      "13950    | torch.Size([1, 1000, 754]) | 5\n",
-      "13951    | torch.Size([1, 1000, 754]) | 5\n",
-      "13952    | torch.Size([1, 1000, 754]) | 5\n",
-      "13953    | torch.Size([1, 1000, 754]) | 5\n",
-      "13954    | torch.Size([1, 1000, 773]) | 5\n",
-      "13955    | torch.Size([1, 1000, 754]) | 5\n",
-      "13956    | torch.Size([1, 1000, 754]) | 5\n",
-      "13957    | torch.Size([1, 1000, 754]) | 5\n",
-      "13958    | torch.Size([1, 1000, 779]) | 5\n",
-      "13959    | torch.Size([1, 1000, 786]) | 5\n",
-      "13960    | torch.Size([1, 1000, 784]) | 5\n",
-      "13961    | torch.Size([1, 1000, 754]) | 5\n",
-      "13962    | torch.Size([1, 1000, 779]) | 5\n",
-      "13963    | torch.Size([1, 1000, 754]) | 5\n",
-      "13964    | torch.Size([1, 1000, 754]) | 5\n",
-      "13965    | torch.Size([1, 1000, 754]) | 5\n",
-      "13966    | torch.Size([1, 1000, 754]) | 5\n",
-      "13967    | torch.Size([1, 1000, 769]) | 5\n",
-      "13968    | torch.Size([1, 1000, 754]) | 5\n",
-      "13969    | torch.Size([1, 1000, 754]) | 5\n",
-      "13970    | torch.Size([1, 1000, 771]) | 5\n",
-      "13971    | torch.Size([1, 1000, 754]) | 5\n",
-      "13972    | torch.Size([1, 1000, 776]) | 5\n",
-      "13973    | torch.Size([1, 1000, 789]) | 5\n",
-      "13974    | torch.Size([1, 1000, 754]) | 5\n",
-      "13975    | torch.Size([1, 1000, 754]) | 5\n",
-      "13976    | torch.Size([1, 1000, 754]) | 5\n",
-      "13977    | torch.Size([1, 1000, 754]) | 5\n",
-      "13978    | torch.Size([1, 1000, 754]) | 5\n",
-      "13979    | torch.Size([1, 1000, 754]) | 5\n",
-      "13980    | torch.Size([1, 1000, 754]) | 5\n",
-      "13981    | torch.Size([1, 1000, 754]) | 5\n",
-      "13982    | torch.Size([1, 1000, 754]) | 5\n",
-      "13983    | torch.Size([1, 1000, 754]) | 5\n",
-      "13984    | torch.Size([1, 1000, 754]) | 5\n",
-      "13985    | torch.Size([1, 1000, 754]) | 5\n",
-      "13986    | torch.Size([1, 1000, 754]) | 5\n",
-      "13987    | torch.Size([1, 1000, 754]) | 5\n",
-      "13988    | torch.Size([1, 1000, 754]) | 5\n",
-      "13989    | torch.Size([1, 1000, 773]) | 5\n",
-      "13990    | torch.Size([1, 1000, 782]) | 5\n",
-      "13991    | torch.Size([1, 1000, 785]) | 5\n",
-      "13992    | torch.Size([1, 1000, 740]) | 5\n",
-      "13993    | torch.Size([1, 1000, 789]) | 5\n",
-      "13994    | torch.Size([1, 1000, 754]) | 5\n",
-      "13995    | torch.Size([1, 1000, 754]) | 5\n",
-      "13996    | torch.Size([1, 1000, 754]) | 5\n",
-      "13997    | torch.Size([1, 1000, 771]) | 5\n",
-      "13998    | torch.Size([1, 1000, 754]) | 5\n",
-      "13999    | torch.Size([1, 1000, 754]) | 5\n",
-      "14000    | torch.Size([1, 1000, 771]) | 5\n",
-      "14001    | torch.Size([1, 1000, 771]) | 5\n",
-      "14002    | torch.Size([1, 1000, 754]) | 5\n",
-      "14003    | torch.Size([1, 1000, 791]) | 5\n",
-      "14004    | torch.Size([1, 1000, 754]) | 5\n",
-      "14005    | torch.Size([1, 1000, 754]) | 5\n",
-      "14006    | torch.Size([1, 1000, 777]) | 5\n",
-      "14007    | torch.Size([1, 1000, 754]) | 5\n",
-      "14008    | torch.Size([1, 1000, 754]) | 5\n",
-      "14009    | torch.Size([1, 1000, 754]) | 5\n",
-      "14010    | torch.Size([1, 1000, 754]) | 5\n",
-      "14011    | torch.Size([1, 1000, 754]) | 5\n",
-      "14012    | torch.Size([1, 1000, 754]) | 5\n",
-      "14013    | torch.Size([1, 1000, 754]) | 5\n",
-      "14014    | torch.Size([1, 1000, 754]) | 5\n",
-      "14015    | torch.Size([1, 1000, 788]) | 5\n",
-      "14016    | torch.Size([1, 1000, 754]) | 5\n",
-      "14017    | torch.Size([1, 1000, 784]) | 5\n",
-      "14018    | torch.Size([1, 1000, 754]) | 5\n",
-      "14019    | torch.Size([1, 1000, 754]) | 5\n",
-      "14020    | torch.Size([1, 1000, 771]) | 5\n",
-      "14021    | torch.Size([1, 1000, 754]) | 5\n",
-      "14022    | torch.Size([1, 1000, 754]) | 5\n",
-      "14023    | torch.Size([1, 1000, 754]) | 5\n",
-      "14024    | torch.Size([1, 1000, 754]) | 5\n",
-      "14025    | torch.Size([1, 1000, 754]) | 5\n",
-      "14026    | torch.Size([1, 1000, 754]) | 5\n",
-      "14027    | torch.Size([1, 1000, 754]) | 5\n",
-      "14028    | torch.Size([1, 1000, 754]) | 5\n",
-      "14029    | torch.Size([1, 1000, 754]) | 5\n",
-      "14030    | torch.Size([1, 1000, 754]) | 5\n",
-      "14031    | torch.Size([1, 1000, 754]) | 5\n",
-      "14032    | torch.Size([1, 1000, 754]) | 5\n",
-      "14033    | torch.Size([1, 1000, 783]) | 5\n",
-      "14034    | torch.Size([1, 1000, 754]) | 5\n",
-      "14035    | torch.Size([1, 1000, 754]) | 5\n",
-      "14036    | torch.Size([1, 1000, 754]) | 5\n",
-      "14037    | torch.Size([1, 1000, 779]) | 5\n",
-      "14038    | torch.Size([1, 1000, 754]) | 5\n",
-      "14039    | torch.Size([1, 1000, 754]) | 5\n",
-      "14040    | torch.Size([1, 1000, 754]) | 5\n",
-      "14041    | torch.Size([1, 1000, 754]) | 5\n",
-      "14042    | torch.Size([1, 1000, 780]) | 5\n",
-      "14043    | torch.Size([1, 1000, 754]) | 5\n",
-      "14044    | torch.Size([1, 1000, 754]) | 5\n",
-      "14045    | torch.Size([1, 1000, 754]) | 5\n",
-      "14046    | torch.Size([1, 1000, 754]) | 5\n",
-      "14047    | torch.Size([1, 1000, 754]) | 5\n",
-      "14048    | torch.Size([1, 1000, 754]) | 5\n",
-      "14049    | torch.Size([1, 1000, 754]) | 5\n",
-      "14050    | torch.Size([1, 1000, 784]) | 5\n",
-      "14051    | torch.Size([1, 1000, 754]) | 5\n",
-      "14052    | torch.Size([1, 1000, 754]) | 5\n",
-      "14053    | torch.Size([1, 1000, 754]) | 5\n",
-      "14054    | torch.Size([1, 1000, 754]) | 5\n",
-      "14055    | torch.Size([1, 1000, 754]) | 5\n",
-      "14056    | torch.Size([1, 1000, 773]) | 5\n",
-      "14057    | torch.Size([1, 1000, 754]) | 5\n",
-      "14058    | torch.Size([1, 1000, 773]) | 5\n",
-      "14059    | torch.Size([1, 1000, 754]) | 5\n",
-      "14060    | torch.Size([1, 1000, 754]) | 5\n",
-      "14061    | torch.Size([1, 1000, 779]) | 5\n",
-      "14062    | torch.Size([1, 1000, 754]) | 5\n",
-      "14063    | torch.Size([1, 1000, 771]) | 5\n",
-      "14064    | torch.Size([1, 1000, 786]) | 5\n",
-      "14065    | torch.Size([1, 1000, 754]) | 5\n",
-      "14066    | torch.Size([1, 1000, 784]) | 5\n",
-      "14067    | torch.Size([1, 1000, 771]) | 5\n",
-      "14068    | torch.Size([1, 1000, 754]) | 5\n",
-      "14069    | torch.Size([1, 1000, 754]) | 5\n",
-      "14070    | torch.Size([1, 1000, 754]) | 5\n",
-      "14071    | torch.Size([1, 1000, 754]) | 5\n",
-      "14072    | torch.Size([1, 1000, 754]) | 5\n",
-      "14073    | torch.Size([1, 1000, 808]) | 5\n",
-      "14074    | torch.Size([1, 1000, 754]) | 5\n",
-      "14075    | torch.Size([1, 1000, 754]) | 5\n",
-      "14076    | torch.Size([1, 1000, 754]) | 5\n",
-      "14077    | torch.Size([1, 1000, 754]) | 5\n",
-      "14078    | torch.Size([1, 1000, 778]) | 5\n",
-      "14079    | torch.Size([1, 1000, 776]) | 5\n",
-      "14080    | torch.Size([1, 1000, 799]) | 5\n",
-      "14081    | torch.Size([1, 1000, 810]) | 5\n",
-      "14082    | torch.Size([1, 1000, 787]) | 5\n",
-      "14083    | torch.Size([1, 1000, 789]) | 5\n",
-      "14084    | torch.Size([1, 1000, 754]) | 5\n",
-      "14085    | torch.Size([1, 1000, 754]) | 5\n",
-      "14086    | torch.Size([1, 1000, 754]) | 5\n",
-      "14087    | torch.Size([1, 1000, 754]) | 5\n",
-      "14088    | torch.Size([1, 1000, 754]) | 5\n",
-      "14089    | torch.Size([1, 1000, 754]) | 5\n",
-      "14090    | torch.Size([1, 1000, 754]) | 5\n",
-      "14091    | torch.Size([1, 1000, 754]) | 5\n",
-      "14092    | torch.Size([1, 1000, 754]) | 5\n",
-      "14093    | torch.Size([1, 1000, 782]) | 5\n",
-      "14094    | torch.Size([1, 1000, 754]) | 5\n",
-      "14095    | torch.Size([1, 1000, 754]) | 5\n",
-      "14096    | torch.Size([1, 1000, 754]) | 5\n",
-      "14097    | torch.Size([1, 1000, 773]) | 5\n",
-      "14098    | torch.Size([1, 1000, 754]) | 5\n",
-      "14099    | torch.Size([1, 1000, 787]) | 5\n",
-      "14100    | torch.Size([1, 1000, 787]) | 5\n",
-      "14101    | torch.Size([1, 1000, 764]) | 5\n",
-      "14102    | torch.Size([1, 1000, 818]) | 5\n",
-      "14103    | torch.Size([1, 1000, 754]) | 5\n",
-      "14104    | torch.Size([1, 1000, 754]) | 5\n",
-      "14105    | torch.Size([1, 1000, 805]) | 5\n",
-      "14106    | torch.Size([1, 1000, 754]) | 5\n",
-      "14107    | torch.Size([1, 1000, 788]) | 5\n",
-      "14108    | torch.Size([1, 1000, 754]) | 5\n",
-      "14109    | torch.Size([1, 1000, 754]) | 5\n",
-      "14110    | torch.Size([1, 1000, 754]) | 5\n",
-      "14111    | torch.Size([1, 1000, 786]) | 5\n",
-      "14112    | torch.Size([1, 1000, 775]) | 5\n",
-      "14113    | torch.Size([1, 1000, 775]) | 5\n",
-      "14114    | torch.Size([1, 1000, 754]) | 5\n",
-      "14115    | torch.Size([1, 1000, 754]) | 5\n",
-      "14116    | torch.Size([1, 1000, 754]) | 5\n",
-      "14117    | torch.Size([1, 1000, 778]) | 5\n",
-      "14118    | torch.Size([1, 1000, 771]) | 5\n",
-      "14119    | torch.Size([1, 1000, 754]) | 5\n",
-      "14120    | torch.Size([1, 1000, 780]) | 5\n",
-      "14121    | torch.Size([1, 1000, 780]) | 5\n",
-      "14122    | torch.Size([1, 1000, 780]) | 5\n",
-      "14123    | torch.Size([1, 1000, 771]) | 5\n",
-      "14124    | torch.Size([1, 1000, 780]) | 5\n",
-      "14125    | torch.Size([1, 1000, 754]) | 5\n",
-      "14126    | torch.Size([1, 1000, 779]) | 5\n",
-      "14127    | torch.Size([1, 1000, 754]) | 5\n",
-      "14128    | torch.Size([1, 1000, 754]) | 5\n",
-      "14129    | torch.Size([1, 1000, 754]) | 5\n",
-      "14130    | torch.Size([1, 1000, 754]) | 5\n",
-      "14131    | torch.Size([1, 1000, 754]) | 5\n",
-      "14132    | torch.Size([1, 1000, 771]) | 5\n",
-      "14133    | torch.Size([1, 1000, 754]) | 5\n",
-      "14134    | torch.Size([1, 1000, 754]) | 5\n",
-      "14135    | torch.Size([1, 1000, 754]) | 5\n",
-      "14136    | torch.Size([1, 1000, 754]) | 5\n",
-      "14137    | torch.Size([1, 1000, 754]) | 5\n",
-      "14138    | torch.Size([1, 1000, 786]) | 5\n",
-      "14139    | torch.Size([1, 1000, 782]) | 5\n",
-      "14140    | torch.Size([1, 1000, 754]) | 5\n",
-      "14141    | torch.Size([1, 1000, 783]) | 5\n",
-      "14142    | torch.Size([1, 1000, 754]) | 5\n",
-      "14143    | torch.Size([1, 1000, 754]) | 5\n",
-      "14144    | torch.Size([1, 1000, 754]) | 5\n",
-      "14145    | torch.Size([1, 1000, 754]) | 5\n",
-      "14146    | torch.Size([1, 1000, 754]) | 5\n",
-      "14147    | torch.Size([1, 1000, 754]) | 5\n",
-      "14148    | torch.Size([1, 1000, 754]) | 5\n",
-      "14149    | torch.Size([1, 1000, 788]) | 5\n",
-      "14150    | torch.Size([1, 1000, 754]) | 5\n",
-      "14151    | torch.Size([1, 1000, 754]) | 5\n",
-      "14152    | torch.Size([1, 1000, 785]) | 5\n",
-      "14153    | torch.Size([1, 1000, 781]) | 5\n",
-      "14154    | torch.Size([1, 1000, 754]) | 5\n",
-      "14155    | torch.Size([1, 1000, 787]) | 5\n",
-      "14156    | torch.Size([1, 1000, 754]) | 5\n",
-      "14157    | torch.Size([1, 1000, 754]) | 5\n",
-      "14158    | torch.Size([1, 1000, 785]) | 5\n",
-      "14159    | torch.Size([1, 1000, 754]) | 5\n",
-      "14160    | torch.Size([1, 1000, 754]) | 5\n",
-      "14161    | torch.Size([1, 1000, 754]) | 5\n",
-      "14162    | torch.Size([1, 1000, 754]) | 5\n",
-      "14163    | torch.Size([1, 1000, 784]) | 5\n",
-      "14164    | torch.Size([1, 1000, 754]) | 5\n",
-      "14165    | torch.Size([1, 1000, 754]) | 5\n",
-      "14166    | torch.Size([1, 1000, 754]) | 5\n",
-      "14167    | torch.Size([1, 1000, 754]) | 5\n",
-      "14168    | torch.Size([1, 1000, 754]) | 5\n",
-      "14169    | torch.Size([1, 1000, 754]) | 5\n",
-      "14170    | torch.Size([1, 1000, 754]) | 5\n",
-      "14171    | torch.Size([1, 1000, 754]) | 5\n",
-      "14172    | torch.Size([1, 1000, 774]) | 5\n",
-      "14173    | torch.Size([1, 1000, 770]) | 5\n",
-      "14174    | torch.Size([1, 1000, 754]) | 5\n",
-      "14175    | torch.Size([1, 1000, 783]) | 5\n",
-      "14176    | torch.Size([1, 1000, 746]) | 5\n",
-      "14177    | torch.Size([1, 1000, 754]) | 5\n",
-      "14178    | torch.Size([1, 1000, 754]) | 5\n",
-      "14179    | torch.Size([1, 1000, 754]) | 5\n",
-      "14180    | torch.Size([1, 1000, 754]) | 5\n",
-      "14181    | torch.Size([1, 1000, 754]) | 5\n",
-      "14182    | torch.Size([1, 1000, 754]) | 5\n",
-      "14183    | torch.Size([1, 1000, 754]) | 5\n",
-      "14184    | torch.Size([1, 1000, 754]) | 5\n",
-      "14185    | torch.Size([1, 1000, 754]) | 5\n",
-      "14186    | torch.Size([1, 1000, 754]) | 5\n",
-      "14187    | torch.Size([1, 1000, 754]) | 5\n",
-      "14188    | torch.Size([1, 1000, 754]) | 5\n",
-      "14189    | torch.Size([1, 1000, 775]) | 5\n",
-      "14190    | torch.Size([1, 1000, 754]) | 5\n",
-      "14191    | torch.Size([1, 1000, 754]) | 5\n",
-      "14192    | torch.Size([1, 1000, 754]) | 5\n",
-      "14193    | torch.Size([1, 1000, 754]) | 5\n",
-      "14194    | torch.Size([1, 1000, 754]) | 5\n",
-      "14195    | torch.Size([1, 1000, 754]) | 5\n",
-      "14196    | torch.Size([1, 1000, 754]) | 5\n",
-      "14197    | torch.Size([1, 1000, 754]) | 5\n",
-      "14198    | torch.Size([1, 1000, 779]) | 5\n",
-      "14199    | torch.Size([1, 1000, 754]) | 5\n",
-      "14200    | torch.Size([1, 1000, 754]) | 5\n",
-      "14201    | torch.Size([1, 1000, 754]) | 5\n",
-      "14202    | torch.Size([1, 1000, 754]) | 5\n",
-      "14203    | torch.Size([1, 1000, 803]) | 5\n",
-      "14204    | torch.Size([1, 1000, 819]) | 5\n",
-      "14205    | torch.Size([1, 1000, 754]) | 5\n",
-      "14206    | torch.Size([1, 1000, 754]) | 5\n",
-      "14207    | torch.Size([1, 1000, 771]) | 5\n",
-      "14208    | torch.Size([1, 1000, 754]) | 5\n",
-      "14209    | torch.Size([1, 1000, 754]) | 5\n",
-      "14210    | torch.Size([1, 1000, 784]) | 5\n",
-      "14211    | torch.Size([1, 1000, 754]) | 5\n",
-      "14212    | torch.Size([1, 1000, 754]) | 5\n",
-      "14213    | torch.Size([1, 1000, 754]) | 5\n",
-      "14214    | torch.Size([1, 1000, 754]) | 5\n",
-      "14215    | torch.Size([1, 1000, 754]) | 5\n",
-      "14216    | torch.Size([1, 1000, 790]) | 5\n",
-      "14217    | torch.Size([1, 1000, 754]) | 5\n",
-      "14218    | torch.Size([1, 1000, 754]) | 5\n",
-      "14219    | torch.Size([1, 1000, 771]) | 5\n",
-      "14220    | torch.Size([1, 1000, 771]) | 5\n",
-      "14221    | torch.Size([1, 1000, 786]) | 5\n",
-      "14222    | torch.Size([1, 1000, 754]) | 5\n",
-      "14223    | torch.Size([1, 1000, 773]) | 5\n",
-      "14224    | torch.Size([1, 1000, 776]) | 5\n",
-      "14225    | torch.Size([1, 1000, 754]) | 5\n",
-      "14226    | torch.Size([1, 1000, 754]) | 5\n",
-      "14227    | torch.Size([1, 1000, 754]) | 5\n",
-      "14228    | torch.Size([1, 1000, 754]) | 5\n",
-      "14229    | torch.Size([1, 1000, 754]) | 5\n",
-      "14230    | torch.Size([1, 1000, 799]) | 5\n",
-      "14231    | torch.Size([1, 1000, 754]) | 5\n",
-      "14232    | torch.Size([1, 1000, 785]) | 5\n",
-      "14233    | torch.Size([1, 1000, 754]) | 5\n",
-      "14234    | torch.Size([1, 1000, 754]) | 5\n",
-      "14235    | torch.Size([1, 1000, 754]) | 5\n",
-      "14236    | torch.Size([1, 1000, 754]) | 5\n",
-      "14237    | torch.Size([1, 1000, 787]) | 5\n",
-      "14238    | torch.Size([1, 1000, 754]) | 5\n",
-      "14239    | torch.Size([1, 1000, 754]) | 5\n",
-      "14240    | torch.Size([1, 1000, 754]) | 5\n",
-      "14241    | torch.Size([1, 1000, 754]) | 5\n",
-      "14242    | torch.Size([1, 1000, 754]) | 5\n",
-      "14243    | torch.Size([1, 1000, 778]) | 5\n",
-      "14244    | torch.Size([1, 1000, 754]) | 5\n",
-      "14245    | torch.Size([1, 1000, 754]) | 5\n",
-      "14246    | torch.Size([1, 1000, 754]) | 5\n",
-      "14247    | torch.Size([1, 1000, 784]) | 5\n",
-      "14248    | torch.Size([1, 1000, 754]) | 5\n",
-      "14249    | torch.Size([1, 1000, 754]) | 5\n",
-      "14250    | torch.Size([1, 1000, 754]) | 5\n",
-      "14251    | torch.Size([1, 1000, 754]) | 5\n",
-      "14252    | torch.Size([1, 1000, 805]) | 5\n",
-      "14253    | torch.Size([1, 1000, 754]) | 5\n",
-      "14254    | torch.Size([1, 1000, 754]) | 5\n",
-      "14255    | torch.Size([1, 1000, 754]) | 5\n",
-      "14256    | torch.Size([1, 1000, 754]) | 5\n",
-      "14257    | torch.Size([1, 1000, 754]) | 5\n",
-      "14258    | torch.Size([1, 1000, 754]) | 5\n",
-      "14259    | torch.Size([1, 1000, 754]) | 5\n",
-      "14260    | torch.Size([1, 1000, 754]) | 5\n",
-      "14261    | torch.Size([1, 1000, 754]) | 5\n",
-      "14262    | torch.Size([1, 1000, 754]) | 5\n",
-      "14263    | torch.Size([1, 1000, 754]) | 5\n",
-      "14264    | torch.Size([1, 1000, 754]) | 5\n",
-      "14265    | torch.Size([1, 1000, 754]) | 5\n",
-      "14266    | torch.Size([1, 1000, 768]) | 5\n",
-      "14267    | torch.Size([1, 1000, 754]) | 5\n",
-      "14268    | torch.Size([1, 1000, 800]) | 5\n",
-      "14269    | torch.Size([1, 1000, 781]) | 5\n",
-      "14270    | torch.Size([1, 1000, 781]) | 5\n",
-      "14271    | torch.Size([1, 1000, 754]) | 5\n",
-      "14272    | torch.Size([1, 1000, 754]) | 5\n",
-      "14273    | torch.Size([1, 1000, 788]) | 5\n",
-      "14274    | torch.Size([1, 1000, 791]) | 5\n",
-      "14275    | torch.Size([1, 1000, 754]) | 5\n",
-      "14276    | torch.Size([1, 1000, 805]) | 5\n",
-      "14277    | torch.Size([1, 1000, 754]) | 5\n",
-      "14278    | torch.Size([1, 1000, 754]) | 5\n",
-      "14279    | torch.Size([1, 1000, 774]) | 5\n",
-      "14280    | torch.Size([1, 1000, 754]) | 5\n",
-      "14281    | torch.Size([1, 1000, 754]) | 5\n",
-      "14282    | torch.Size([1, 1000, 754]) | 5\n",
-      "14283    | torch.Size([1, 1000, 754]) | 5\n",
-      "14284    | torch.Size([1, 1000, 754]) | 5\n",
-      "14285    | torch.Size([1, 1000, 789]) | 5\n",
-      "14286    | torch.Size([1, 1000, 754]) | 5\n",
-      "14287    | torch.Size([1, 1000, 754]) | 5\n",
-      "14288    | torch.Size([1, 1000, 780]) | 5\n",
-      "14289    | torch.Size([1, 1000, 775]) | 5\n",
-      "14290    | torch.Size([1, 1000, 754]) | 5\n",
-      "14291    | torch.Size([1, 1000, 754]) | 5\n",
-      "14292    | torch.Size([1, 1000, 754]) | 5\n",
-      "14293    | torch.Size([1, 1000, 754]) | 5\n",
-      "14294    | torch.Size([1, 1000, 754]) | 5\n",
-      "14295    | torch.Size([1, 1000, 766]) | 5\n",
-      "14296    | torch.Size([1, 1000, 754]) | 5\n",
-      "14297    | torch.Size([1, 1000, 754]) | 5\n",
-      "14298    | torch.Size([1, 1000, 781]) | 5\n",
-      "14299    | torch.Size([1, 1000, 754]) | 5\n",
-      "14300    | torch.Size([1, 1000, 754]) | 5\n",
-      "14301    | torch.Size([1, 1000, 779]) | 5\n",
-      "14302    | torch.Size([1, 1000, 784]) | 5\n",
-      "14303    | torch.Size([1, 1000, 791]) | 5\n",
-      "14304    | torch.Size([1, 1000, 754]) | 5\n",
-      "14305    | torch.Size([1, 1000, 754]) | 5\n",
-      "14306    | torch.Size([1, 1000, 754]) | 5\n",
-      "14307    | torch.Size([1, 1000, 754]) | 5\n",
-      "14308    | torch.Size([1, 1000, 754]) | 5\n",
-      "14309    | torch.Size([1, 1000, 754]) | 5\n",
-      "14310    | torch.Size([1, 1000, 754]) | 5\n",
-      "14311    | torch.Size([1, 1000, 774]) | 5\n",
-      "14312    | torch.Size([1, 1000, 754]) | 5\n",
-      "14313    | torch.Size([1, 1000, 754]) | 5\n",
-      "14314    | torch.Size([1, 1000, 754]) | 5\n",
-      "14315    | torch.Size([1, 1000, 754]) | 5\n",
-      "14316    | torch.Size([1, 1000, 774]) | 5\n",
-      "14317    | torch.Size([1, 1000, 754]) | 5\n",
-      "14318    | torch.Size([1, 1000, 780]) | 5\n",
-      "14319    | torch.Size([1, 1000, 754]) | 5\n",
-      "14320    | torch.Size([1, 1000, 754]) | 5\n",
-      "14321    | torch.Size([1, 1000, 754]) | 5\n",
-      "14322    | torch.Size([1, 1000, 783]) | 5\n",
-      "14323    | torch.Size([1, 1000, 785]) | 5\n",
-      "14324    | torch.Size([1, 1000, 754]) | 5\n",
-      "14325    | torch.Size([1, 1000, 754]) | 5\n",
-      "14326    | torch.Size([1, 1000, 754]) | 5\n",
-      "14327    | torch.Size([1, 1000, 754]) | 5\n",
-      "14328    | torch.Size([1, 1000, 754]) | 5\n",
-      "14329    | torch.Size([1, 1000, 754]) | 5\n",
-      "14330    | torch.Size([1, 1000, 754]) | 5\n",
-      "14331    | torch.Size([1, 1000, 754]) | 5\n",
-      "14332    | torch.Size([1, 1000, 754]) | 5\n",
-      "14333    | torch.Size([1, 1000, 754]) | 5\n",
-      "14334    | torch.Size([1, 1000, 787]) | 5\n",
-      "14335    | torch.Size([1, 1000, 754]) | 5\n",
-      "14336    | torch.Size([1, 1000, 754]) | 5\n",
-      "14337    | torch.Size([1, 1000, 778]) | 5\n",
-      "14338    | torch.Size([1, 1000, 798]) | 5\n",
-      "14339    | torch.Size([1, 1000, 754]) | 5\n",
-      "14340    | torch.Size([1, 1000, 754]) | 5\n",
-      "14341    | torch.Size([1, 1000, 780]) | 5\n",
-      "14342    | torch.Size([1, 1000, 754]) | 5\n",
-      "14343    | torch.Size([1, 1000, 774]) | 5\n",
-      "14344    | torch.Size([1, 1000, 754]) | 5\n",
-      "14345    | torch.Size([1, 1000, 754]) | 5\n",
-      "14346    | torch.Size([1, 1000, 789]) | 5\n",
-      "14347    | torch.Size([1, 1000, 754]) | 5\n",
-      "14348    | torch.Size([1, 1000, 754]) | 5\n",
-      "14349    | torch.Size([1, 1000, 789]) | 5\n",
-      "14350    | torch.Size([1, 1000, 754]) | 5\n",
-      "14351    | torch.Size([1, 1000, 754]) | 5\n",
-      "14352    | torch.Size([1, 1000, 824]) | 5\n",
-      "14353    | torch.Size([1, 1000, 754]) | 5\n",
-      "14354    | torch.Size([1, 1000, 775]) | 5\n",
-      "14355    | torch.Size([1, 1000, 754]) | 5\n",
-      "14356    | torch.Size([1, 1000, 754]) | 5\n",
-      "14357    | torch.Size([1, 1000, 809]) | 5\n",
-      "14358    | torch.Size([1, 1000, 754]) | 5\n",
-      "14359    | torch.Size([1, 1000, 754]) | 5\n",
-      "14360    | torch.Size([1, 1000, 754]) | 5\n",
-      "14361    | torch.Size([1, 1000, 754]) | 5\n",
-      "14362    | torch.Size([1, 1000, 754]) | 5\n",
-      "14363    | torch.Size([1, 1000, 754]) | 5\n",
-      "14364    | torch.Size([1, 1000, 754]) | 5\n",
-      "14365    | torch.Size([1, 1000, 754]) | 5\n",
-      "14366    | torch.Size([1, 1000, 754]) | 5\n",
-      "14367    | torch.Size([1, 1000, 754]) | 5\n",
-      "14368    | torch.Size([1, 1000, 754]) | 5\n",
-      "14369    | torch.Size([1, 1000, 754]) | 5\n",
-      "14370    | torch.Size([1, 1000, 754]) | 5\n",
-      "14371    | torch.Size([1, 1000, 754]) | 5\n",
-      "14372    | torch.Size([1, 1000, 754]) | 5\n",
-      "14373    | torch.Size([1, 1000, 754]) | 5\n",
-      "14374    | torch.Size([1, 1000, 754]) | 5\n",
-      "14375    | torch.Size([1, 1000, 754]) | 5\n",
-      "14376    | torch.Size([1, 1000, 754]) | 5\n",
-      "14377    | torch.Size([1, 1000, 754]) | 5\n",
-      "14378    | torch.Size([1, 1000, 787]) | 5\n",
-      "14379    | torch.Size([1, 1000, 754]) | 5\n",
-      "14380    | torch.Size([1, 1000, 754]) | 5\n",
-      "14381    | torch.Size([1, 1000, 754]) | 5\n",
-      "14382    | torch.Size([1, 1000, 754]) | 5\n",
-      "14383    | torch.Size([1, 1000, 754]) | 5\n",
-      "14384    | torch.Size([1, 1000, 782]) | 5\n",
-      "14385    | torch.Size([1, 1000, 754]) | 5\n",
-      "14386    | torch.Size([1, 1000, 754]) | 5\n",
-      "14387    | torch.Size([1, 1000, 754]) | 5\n",
-      "14388    | torch.Size([1, 1000, 771]) | 5\n",
-      "14389    | torch.Size([1, 1000, 754]) | 5\n",
-      "14390    | torch.Size([1, 1000, 754]) | 5\n",
-      "14391    | torch.Size([1, 1000, 754]) | 5\n",
-      "14392    | torch.Size([1, 1000, 785]) | 5\n",
-      "14393    | torch.Size([1, 1000, 754]) | 5\n",
-      "14394    | torch.Size([1, 1000, 754]) | 5\n",
-      "14395    | torch.Size([1, 1000, 771]) | 5\n",
-      "14396    | torch.Size([1, 1000, 784]) | 5\n",
-      "14397    | torch.Size([1, 1000, 788]) | 5\n",
-      "14398    | torch.Size([1, 1000, 754]) | 5\n",
-      "14399    | torch.Size([1, 1000, 771]) | 5\n",
-      "14400    | torch.Size([1, 1000, 790]) | 5\n",
-      "14401    | torch.Size([1, 1000, 754]) | 5\n",
-      "14402    | torch.Size([1, 1000, 754]) | 5\n",
-      "14403    | torch.Size([1, 1000, 754]) | 5\n",
-      "14404    | torch.Size([1, 1000, 754]) | 5\n",
-      "14405    | torch.Size([1, 1000, 754]) | 5\n",
-      "14406    | torch.Size([1, 1000, 823]) | 5\n",
-      "14407    | torch.Size([1, 1000, 754]) | 5\n",
-      "14408    | torch.Size([1, 1000, 788]) | 5\n",
-      "14409    | torch.Size([1, 1000, 754]) | 5\n",
-      "14410    | torch.Size([1, 1000, 754]) | 5\n",
-      "14411    | torch.Size([1, 1000, 754]) | 5\n",
-      "14412    | torch.Size([1, 1000, 754]) | 5\n",
-      "14413    | torch.Size([1, 1000, 754]) | 5\n",
-      "14414    | torch.Size([1, 1000, 754]) | 5\n",
-      "14415    | torch.Size([1, 1000, 754]) | 5\n",
-      "14416    | torch.Size([1, 1000, 754]) | 5\n",
-      "14417    | torch.Size([1, 1000, 804]) | 5\n",
-      "14418    | torch.Size([1, 1000, 782]) | 5\n",
-      "14419    | torch.Size([1, 1000, 754]) | 5\n",
-      "14420    | torch.Size([1, 1000, 754]) | 5\n",
-      "14421    | torch.Size([1, 1000, 754]) | 5\n",
-      "14422    | torch.Size([1, 1000, 754]) | 5\n",
-      "14423    | torch.Size([1, 1000, 754]) | 5\n",
-      "14424    | torch.Size([1, 1000, 754]) | 5\n",
-      "14425    | torch.Size([1, 1000, 788]) | 5\n",
-      "14426    | torch.Size([1, 1000, 772]) | 5\n",
-      "14427    | torch.Size([1, 1000, 783]) | 5\n",
-      "14428    | torch.Size([1, 1000, 754]) | 5\n",
-      "14429    | torch.Size([1, 1000, 754]) | 5\n",
-      "14430    | torch.Size([1, 1000, 781]) | 5\n",
-      "14431    | torch.Size([1, 1000, 754]) | 5\n",
-      "14432    | torch.Size([1, 1000, 754]) | 5\n",
-      "14433    | torch.Size([1, 1000, 754]) | 5\n",
-      "14434    | torch.Size([1, 1000, 754]) | 5\n",
-      "14435    | torch.Size([1, 1000, 754]) | 5\n",
-      "14436    | torch.Size([1, 1000, 754]) | 5\n",
-      "14437    | torch.Size([1, 1000, 754]) | 5\n",
-      "14438    | torch.Size([1, 1000, 754]) | 5\n",
-      "14439    | torch.Size([1, 1000, 754]) | 5\n",
-      "14440    | torch.Size([1, 1000, 754]) | 5\n",
-      "14441    | torch.Size([1, 1000, 789]) | 5\n",
-      "14442    | torch.Size([1, 1000, 754]) | 5\n",
-      "14443    | torch.Size([1, 1000, 787]) | 5\n",
-      "14444    | torch.Size([1, 1000, 754]) | 5\n",
-      "14445    | torch.Size([1, 1000, 754]) | 5\n",
-      "14446    | torch.Size([1, 1000, 778]) | 5\n",
-      "14447    | torch.Size([1, 1000, 779]) | 5\n",
-      "14448    | torch.Size([1, 1000, 754]) | 5\n",
-      "14449    | torch.Size([1, 1000, 777]) | 5\n",
-      "14450    | torch.Size([1, 1000, 754]) | 5\n",
-      "14451    | torch.Size([1, 1000, 754]) | 5\n",
-      "14452    | torch.Size([1, 1000, 754]) | 5\n",
-      "14453    | torch.Size([1, 1000, 754]) | 5\n",
-      "14454    | torch.Size([1, 1000, 806]) | 5\n",
-      "14455    | torch.Size([1, 1000, 775]) | 5\n",
-      "14456    | torch.Size([1, 1000, 754]) | 5\n",
-      "14457    | torch.Size([1, 1000, 754]) | 5\n",
-      "14458    | torch.Size([1, 1000, 754]) | 5\n",
-      "14459    | torch.Size([1, 1000, 797]) | 5\n",
-      "14460    | torch.Size([1, 1000, 781]) | 5\n",
-      "14461    | torch.Size([1, 1000, 754]) | 5\n",
-      "14462    | torch.Size([1, 1000, 754]) | 5\n",
-      "14463    | torch.Size([1, 1000, 754]) | 5\n",
-      "14464    | torch.Size([1, 1000, 784]) | 5\n",
-      "14465    | torch.Size([1, 1000, 789]) | 5\n",
-      "14466    | torch.Size([1, 1000, 780]) | 5\n",
-      "14467    | torch.Size([1, 1000, 796]) | 5\n",
-      "14468    | torch.Size([1, 1000, 806]) | 5\n",
-      "14469    | torch.Size([1, 1000, 783]) | 5\n",
-      "14470    | torch.Size([1, 1000, 775]) | 5\n",
-      "14471    | torch.Size([1, 1000, 754]) | 5\n",
-      "14472    | torch.Size([1, 1000, 754]) | 5\n",
-      "14473    | torch.Size([1, 1000, 754]) | 5\n",
-      "14474    | torch.Size([1, 1000, 754]) | 5\n",
-      "14475    | torch.Size([1, 1000, 754]) | 5\n",
-      "14476    | torch.Size([1, 1000, 764]) | 5\n",
-      "14477    | torch.Size([1, 1000, 754]) | 5\n",
-      "14478    | torch.Size([1, 1000, 754]) | 5\n",
-      "14479    | torch.Size([1, 1000, 754]) | 5\n",
-      "14480    | torch.Size([1, 1000, 754]) | 5\n",
-      "14481    | torch.Size([1, 1000, 754]) | 5\n",
-      "14482    | torch.Size([1, 1000, 754]) | 5\n",
-      "14483    | torch.Size([1, 1000, 754]) | 5\n",
-      "14484    | torch.Size([1, 1000, 763]) | 5\n",
-      "14485    | torch.Size([1, 1000, 778]) | 5\n",
-      "14486    | torch.Size([1, 1000, 754]) | 5\n",
-      "14487    | torch.Size([1, 1000, 754]) | 5\n",
-      "14488    | torch.Size([1, 1000, 773]) | 5\n",
-      "14489    | torch.Size([1, 1000, 785]) | 5\n",
-      "14490    | torch.Size([1, 1000, 754]) | 5\n",
-      "14491    | torch.Size([1, 1000, 754]) | 5\n",
-      "14492    | torch.Size([1, 1000, 772]) | 5\n",
-      "14493    | torch.Size([1, 1000, 754]) | 5\n",
-      "14494    | torch.Size([1, 1000, 773]) | 5\n",
-      "14495    | torch.Size([1, 1000, 754]) | 5\n",
-      "14496    | torch.Size([1, 1000, 773]) | 5\n",
-      "14497    | torch.Size([1, 1000, 754]) | 5\n",
-      "14498    | torch.Size([1, 1000, 797]) | 5\n",
-      "14499    | torch.Size([1, 1000, 754]) | 5\n",
-      "14500    | torch.Size([1, 1000, 774]) | 5\n",
-      "14501    | torch.Size([1, 1000, 754]) | 5\n",
-      "14502    | torch.Size([1, 1000, 754]) | 5\n",
-      "14503    | torch.Size([1, 1000, 786]) | 5\n",
-      "14504    | torch.Size([1, 1000, 754]) | 5\n",
-      "14505    | torch.Size([1, 1000, 785]) | 5\n",
-      "14506    | torch.Size([1, 1000, 754]) | 5\n",
-      "14507    | torch.Size([1, 1000, 791]) | 5\n",
-      "14508    | torch.Size([1, 1000, 754]) | 5\n",
-      "14509    | torch.Size([1, 1000, 790]) | 5\n",
-      "14510    | torch.Size([1, 1000, 754]) | 5\n",
-      "14511    | torch.Size([1, 1000, 754]) | 5\n",
-      "14512    | torch.Size([1, 1000, 804]) | 5\n",
-      "14513    | torch.Size([1, 1000, 754]) | 5\n",
-      "14514    | torch.Size([1, 1000, 754]) | 5\n",
-      "14515    | torch.Size([1, 1000, 791]) | 5\n",
-      "14516    | torch.Size([1, 1000, 754]) | 5\n",
-      "14517    | torch.Size([1, 1000, 754]) | 5\n",
-      "14518    | torch.Size([1, 1000, 770]) | 5\n",
-      "14519    | torch.Size([1, 1000, 754]) | 5\n",
-      "14520    | torch.Size([1, 1000, 754]) | 5\n",
-      "14521    | torch.Size([1, 1000, 754]) | 5\n",
-      "14522    | torch.Size([1, 1000, 754]) | 5\n",
-      "14523    | torch.Size([1, 1000, 779]) | 5\n",
-      "14524    | torch.Size([1, 1000, 754]) | 5\n",
-      "14525    | torch.Size([1, 1000, 754]) | 5\n",
-      "14526    | torch.Size([1, 1000, 754]) | 5\n",
-      "14527    | torch.Size([1, 1000, 754]) | 5\n",
-      "14528    | torch.Size([1, 1000, 754]) | 5\n",
-      "14529    | torch.Size([1, 1000, 754]) | 5\n",
-      "14530    | torch.Size([1, 1000, 754]) | 5\n",
-      "14531    | torch.Size([1, 1000, 754]) | 5\n",
-      "14532    | torch.Size([1, 1000, 738]) | 5\n",
-      "14533    | torch.Size([1, 1000, 771]) | 5\n",
-      "14534    | torch.Size([1, 1000, 754]) | 5\n",
-      "14535    | torch.Size([1, 1000, 754]) | 5\n",
-      "14536    | torch.Size([1, 1000, 754]) | 5\n",
-      "14537    | torch.Size([1, 1000, 790]) | 5\n",
-      "14538    | torch.Size([1, 1000, 754]) | 5\n",
-      "14539    | torch.Size([1, 1000, 754]) | 5\n",
-      "14540    | torch.Size([1, 1000, 754]) | 5\n",
-      "14541    | torch.Size([1, 1000, 754]) | 5\n",
-      "14542    | torch.Size([1, 1000, 754]) | 5\n",
-      "14543    | torch.Size([1, 1000, 754]) | 5\n",
-      "14544    | torch.Size([1, 1000, 754]) | 5\n",
-      "14545    | torch.Size([1, 1000, 778]) | 5\n",
-      "14546    | torch.Size([1, 1000, 754]) | 5\n",
-      "14547    | torch.Size([1, 1000, 754]) | 5\n",
-      "14548    | torch.Size([1, 1000, 754]) | 5\n",
-      "14549    | torch.Size([1, 1000, 754]) | 5\n",
-      "14550    | torch.Size([1, 1000, 781]) | 5\n",
-      "14551    | torch.Size([1, 1000, 754]) | 5\n",
-      "14552    | torch.Size([1, 1000, 754]) | 5\n",
-      "14553    | torch.Size([1, 1000, 754]) | 5\n",
-      "14554    | torch.Size([1, 1000, 754]) | 5\n",
-      "14555    | torch.Size([1, 1000, 754]) | 5\n",
-      "14556    | torch.Size([1, 1000, 754]) | 5\n",
-      "14557    | torch.Size([1, 1000, 754]) | 5\n",
-      "14558    | torch.Size([1, 1000, 808]) | 5\n",
-      "14559    | torch.Size([1, 1000, 754]) | 5\n",
-      "14560    | torch.Size([1, 1000, 771]) | 5\n",
-      "14561    | torch.Size([1, 1000, 754]) | 5\n",
-      "14562    | torch.Size([1, 1000, 780]) | 5\n",
-      "14563    | torch.Size([1, 1000, 775]) | 5\n",
-      "14564    | torch.Size([1, 1000, 754]) | 5\n",
-      "14565    | torch.Size([1, 1000, 754]) | 5\n",
-      "14566    | torch.Size([1, 1000, 754]) | 5\n",
-      "14567    | torch.Size([1, 1000, 780]) | 5\n",
-      "14568    | torch.Size([1, 1000, 788]) | 5\n",
-      "14569    | torch.Size([1, 1000, 771]) | 5\n",
-      "14570    | torch.Size([1, 1000, 754]) | 5\n",
-      "14571    | torch.Size([1, 1000, 754]) | 5\n",
-      "14572    | torch.Size([1, 1000, 754]) | 5\n",
-      "14573    | torch.Size([1, 1000, 754]) | 5\n",
-      "14574    | torch.Size([1, 1000, 754]) | 5\n",
-      "14575    | torch.Size([1, 1000, 754]) | 5\n",
-      "14576    | torch.Size([1, 1000, 771]) | 5\n",
-      "14577    | torch.Size([1, 1000, 805]) | 5\n",
-      "14578    | torch.Size([1, 1000, 754]) | 5\n",
-      "14579    | torch.Size([1, 1000, 754]) | 5\n",
-      "14580    | torch.Size([1, 1000, 754]) | 5\n",
-      "14581    | torch.Size([1, 1000, 773]) | 5\n",
-      "14582    | torch.Size([1, 1000, 754]) | 5\n",
-      "14583    | torch.Size([1, 1000, 754]) | 5\n",
-      "14584    | torch.Size([1, 1000, 754]) | 5\n",
-      "14585    | torch.Size([1, 1000, 771]) | 5\n",
-      "14586    | torch.Size([1, 1000, 754]) | 5\n",
-      "14587    | torch.Size([1, 1000, 785]) | 5\n",
-      "14588    | torch.Size([1, 1000, 754]) | 5\n",
-      "14589    | torch.Size([1, 1000, 754]) | 5\n",
-      "14590    | torch.Size([1, 1000, 754]) | 5\n",
-      "14591    | torch.Size([1, 1000, 789]) | 5\n",
-      "14592    | torch.Size([1, 1000, 754]) | 5\n",
-      "14593    | torch.Size([1, 1000, 754]) | 5\n",
-      "14594    | torch.Size([1, 1000, 754]) | 5\n",
-      "14595    | torch.Size([1, 1000, 797]) | 5\n",
-      "14596    | torch.Size([1, 1000, 754]) | 5\n",
-      "14597    | torch.Size([1, 1000, 754]) | 5\n",
-      "14598    | torch.Size([1, 1000, 754]) | 5\n",
-      "14599    | torch.Size([1, 1000, 754]) | 5\n",
-      "14600    | torch.Size([1, 1000, 754]) | 5\n",
-      "14601    | torch.Size([1, 1000, 754]) | 5\n",
-      "14602    | torch.Size([1, 1000, 754]) | 5\n",
-      "14603    | torch.Size([1, 1000, 754]) | 5\n",
-      "14604    | torch.Size([1, 1000, 754]) | 5\n",
-      "14605    | torch.Size([1, 1000, 754]) | 5\n",
-      "14606    | torch.Size([1, 1000, 771]) | 5\n",
-      "14607    | torch.Size([1, 1000, 754]) | 5\n",
-      "14608    | torch.Size([1, 1000, 754]) | 5\n",
-      "14609    | torch.Size([1, 1000, 785]) | 5\n",
-      "14610    | torch.Size([1, 1000, 754]) | 5\n",
-      "14611    | torch.Size([1, 1000, 754]) | 5\n",
-      "14612    | torch.Size([1, 1000, 754]) | 5\n",
-      "14613    | torch.Size([1, 1000, 754]) | 5\n",
-      "14614    | torch.Size([1, 1000, 754]) | 5\n",
-      "14615    | torch.Size([1, 1000, 771]) | 5\n",
-      "14616    | torch.Size([1, 1000, 771]) | 5\n",
-      "14617    | torch.Size([1, 1000, 754]) | 5\n",
-      "14618    | torch.Size([1, 1000, 754]) | 5\n",
-      "14619    | torch.Size([1, 1000, 754]) | 5\n",
-      "14620    | torch.Size([1, 1000, 754]) | 5\n",
-      "14621    | torch.Size([1, 1000, 754]) | 5\n",
-      "14622    | torch.Size([1, 1000, 754]) | 5\n",
-      "14623    | torch.Size([1, 1000, 754]) | 5\n",
-      "14624    | torch.Size([1, 1000, 754]) | 5\n",
-      "14625    | torch.Size([1, 1000, 781]) | 5\n",
-      "14626    | torch.Size([1, 1000, 771]) | 5\n",
-      "14627    | torch.Size([1, 1000, 754]) | 5\n",
-      "14628    | torch.Size([1, 1000, 754]) | 5\n",
-      "14629    | torch.Size([1, 1000, 754]) | 5\n",
-      "14630    | torch.Size([1, 1000, 781]) | 5\n",
-      "14631    | torch.Size([1, 1000, 785]) | 5\n",
-      "14632    | torch.Size([1, 1000, 754]) | 5\n",
-      "14633    | torch.Size([1, 1000, 754]) | 5\n",
-      "14634    | torch.Size([1, 1000, 779]) | 5\n",
-      "14635    | torch.Size([1, 1000, 754]) | 5\n",
-      "14636    | torch.Size([1, 1000, 754]) | 5\n",
-      "14637    | torch.Size([1, 1000, 754]) | 5\n",
-      "14638    | torch.Size([1, 1000, 754]) | 5\n",
-      "14639    | torch.Size([1, 1000, 783]) | 5\n",
-      "14640    | torch.Size([1, 1000, 783]) | 5\n",
-      "14641    | torch.Size([1, 1000, 754]) | 5\n",
-      "14642    | torch.Size([1, 1000, 754]) | 5\n",
-      "14643    | torch.Size([1, 1000, 754]) | 5\n",
-      "14644    | torch.Size([1, 1000, 782]) | 5\n",
-      "14645    | torch.Size([1, 1000, 754]) | 5\n",
-      "14646    | torch.Size([1, 1000, 742]) | 5\n",
-      "14647    | torch.Size([1, 1000, 754]) | 5\n",
-      "14648    | torch.Size([1, 1000, 754]) | 5\n",
-      "14649    | torch.Size([1, 1000, 771]) | 5\n",
-      "14650    | torch.Size([1, 1000, 754]) | 5\n",
-      "14651    | torch.Size([1, 1000, 754]) | 5\n",
-      "14652    | torch.Size([1, 1000, 754]) | 5\n",
-      "14653    | torch.Size([1, 1000, 754]) | 5\n",
-      "14654    | torch.Size([1, 1000, 754]) | 5\n",
-      "14655    | torch.Size([1, 1000, 754]) | 5\n",
-      "14656    | torch.Size([1, 1000, 754]) | 5\n",
-      "14657    | torch.Size([1, 1000, 754]) | 5\n",
-      "14658    | torch.Size([1, 1000, 786]) | 5\n",
-      "14659    | torch.Size([1, 1000, 754]) | 5\n",
-      "14660    | torch.Size([1, 1000, 754]) | 5\n",
-      "14661    | torch.Size([1, 1000, 754]) | 5\n",
-      "14662    | torch.Size([1, 1000, 782]) | 5\n",
-      "14663    | torch.Size([1, 1000, 754]) | 5\n",
-      "14664    | torch.Size([1, 1000, 790]) | 5\n",
-      "14665    | torch.Size([1, 1000, 788]) | 5\n",
-      "14666    | torch.Size([1, 1000, 754]) | 5\n",
-      "14667    | torch.Size([1, 1000, 754]) | 5\n",
-      "14668    | torch.Size([1, 1000, 754]) | 5\n",
-      "14669    | torch.Size([1, 1000, 754]) | 5\n",
-      "14670    | torch.Size([1, 1000, 754]) | 5\n",
-      "14671    | torch.Size([1, 1000, 754]) | 5\n",
-      "14672    | torch.Size([1, 1000, 779]) | 5\n",
-      "14673    | torch.Size([1, 1000, 754]) | 5\n",
-      "14674    | torch.Size([1, 1000, 793]) | 5\n",
-      "14675    | torch.Size([1, 1000, 785]) | 5\n",
-      "14676    | torch.Size([1, 1000, 754]) | 5\n",
-      "14677    | torch.Size([1, 1000, 754]) | 5\n",
-      "14678    | torch.Size([1, 1000, 754]) | 5\n",
-      "14679    | torch.Size([1, 1000, 789]) | 5\n",
-      "14680    | torch.Size([1, 1000, 754]) | 5\n",
-      "14681    | torch.Size([1, 1000, 754]) | 5\n",
-      "14682    | torch.Size([1, 1000, 774]) | 5\n",
-      "14683    | torch.Size([1, 1000, 754]) | 5\n",
-      "14684    | torch.Size([1, 1000, 754]) | 5\n",
-      "14685    | torch.Size([1, 1000, 778]) | 5\n",
-      "14686    | torch.Size([1, 1000, 771]) | 5\n",
-      "14687    | torch.Size([1, 1000, 754]) | 5\n",
-      "14688    | torch.Size([1, 1000, 754]) | 5\n",
-      "14689    | torch.Size([1, 1000, 754]) | 5\n",
-      "14690    | torch.Size([1, 1000, 754]) | 5\n",
-      "14691    | torch.Size([1, 1000, 754]) | 5\n",
-      "14692    | torch.Size([1, 1000, 789]) | 5\n",
-      "14693    | torch.Size([1, 1000, 754]) | 5\n",
-      "14694    | torch.Size([1, 1000, 754]) | 5\n",
-      "14695    | torch.Size([1, 1000, 754]) | 5\n",
-      "14696    | torch.Size([1, 1000, 754]) | 5\n",
-      "14697    | torch.Size([1, 1000, 776]) | 5\n",
-      "14698    | torch.Size([1, 1000, 754]) | 5\n",
-      "14699    | torch.Size([1, 1000, 754]) | 5\n",
-      "14700    | torch.Size([1, 1000, 754]) | 5\n",
-      "14701    | torch.Size([1, 1000, 754]) | 5\n",
-      "14702    | torch.Size([1, 1000, 773]) | 5\n",
-      "14703    | torch.Size([1, 1000, 754]) | 5\n",
-      "14704    | torch.Size([1, 1000, 754]) | 5\n",
-      "14705    | torch.Size([1, 1000, 754]) | 5\n",
-      "14706    | torch.Size([1, 1000, 754]) | 5\n",
-      "14707    | torch.Size([1, 1000, 754]) | 5\n",
-      "14708    | torch.Size([1, 1000, 754]) | 5\n",
-      "14709    | torch.Size([1, 1000, 754]) | 5\n",
-      "14710    | torch.Size([1, 1000, 771]) | 5\n",
-      "14711    | torch.Size([1, 1000, 754]) | 5\n",
-      "14712    | torch.Size([1, 1000, 754]) | 5\n",
-      "14713    | torch.Size([1, 1000, 787]) | 5\n",
-      "14714    | torch.Size([1, 1000, 754]) | 5\n",
-      "14715    | torch.Size([1, 1000, 754]) | 5\n",
-      "14716    | torch.Size([1, 1000, 754]) | 5\n",
-      "14717    | torch.Size([1, 1000, 754]) | 5\n",
-      "14718    | torch.Size([1, 1000, 754]) | 5\n",
-      "14719    | torch.Size([1, 1000, 786]) | 5\n",
-      "14720    | torch.Size([1, 1000, 754]) | 5\n",
-      "14721    | torch.Size([1, 1000, 782]) | 5\n",
-      "14722    | torch.Size([1, 1000, 754]) | 5\n",
-      "14723    | torch.Size([1, 1000, 754]) | 5\n",
-      "14724    | torch.Size([1, 1000, 771]) | 5\n",
-      "14725    | torch.Size([1, 1000, 775]) | 5\n",
-      "14726    | torch.Size([1, 1000, 754]) | 5\n",
-      "14727    | torch.Size([1, 1000, 754]) | 5\n",
-      "14728    | torch.Size([1, 1000, 771]) | 5\n",
-      "14729    | torch.Size([1, 1000, 754]) | 5\n",
-      "14730    | torch.Size([1, 1000, 771]) | 5\n",
-      "14731    | torch.Size([1, 1000, 764]) | 5\n",
-      "14732    | torch.Size([1, 1000, 780]) | 5\n",
-      "14733    | torch.Size([1, 1000, 754]) | 5\n",
-      "14734    | torch.Size([1, 1000, 754]) | 5\n",
-      "14735    | torch.Size([1, 1000, 754]) | 5\n",
-      "14736    | torch.Size([1, 1000, 754]) | 5\n",
-      "14737    | torch.Size([1, 1000, 754]) | 5\n",
-      "14738    | torch.Size([1, 1000, 754]) | 5\n",
-      "14739    | torch.Size([1, 1000, 798]) | 5\n",
-      "14740    | torch.Size([1, 1000, 754]) | 5\n",
-      "14741    | torch.Size([1, 1000, 754]) | 5\n",
-      "14742    | torch.Size([1, 1000, 754]) | 5\n",
-      "14743    | torch.Size([1, 1000, 754]) | 5\n",
-      "14744    | torch.Size([1, 1000, 776]) | 5\n",
-      "14745    | torch.Size([1, 1000, 761]) | 5\n",
-      "14746    | torch.Size([1, 1000, 754]) | 5\n",
-      "14747    | torch.Size([1, 1000, 775]) | 5\n",
-      "14748    | torch.Size([1, 1000, 754]) | 5\n",
-      "14749    | torch.Size([1, 1000, 754]) | 5\n",
-      "14750    | torch.Size([1, 1000, 801]) | 5\n",
-      "14751    | torch.Size([1, 1000, 773]) | 5\n",
-      "14752    | torch.Size([1, 1000, 789]) | 5\n",
-      "14753    | torch.Size([1, 1000, 784]) | 5\n",
-      "14754    | torch.Size([1, 1000, 754]) | 5\n",
-      "14755    | torch.Size([1, 1000, 754]) | 5\n",
-      "14756    | torch.Size([1, 1000, 754]) | 5\n",
-      "14757    | torch.Size([1, 1000, 754]) | 5\n",
-      "14758    | torch.Size([1, 1000, 754]) | 5\n",
-      "14759    | torch.Size([1, 1000, 811]) | 5\n",
-      "14760    | torch.Size([1, 1000, 776]) | 5\n",
-      "14761    | torch.Size([1, 1000, 785]) | 5\n",
-      "14762    | torch.Size([1, 1000, 754]) | 5\n",
-      "14763    | torch.Size([1, 1000, 754]) | 5\n",
-      "14764    | torch.Size([1, 1000, 754]) | 5\n",
-      "14765    | torch.Size([1, 1000, 754]) | 5\n",
-      "14766    | torch.Size([1, 1000, 754]) | 5\n",
-      "14767    | torch.Size([1, 1000, 754]) | 5\n",
-      "14768    | torch.Size([1, 1000, 754]) | 5\n",
-      "14769    | torch.Size([1, 1000, 754]) | 5\n",
-      "14770    | torch.Size([1, 1000, 754]) | 5\n",
-      "14771    | torch.Size([1, 1000, 785]) | 5\n",
-      "14772    | torch.Size([1, 1000, 754]) | 5\n",
-      "14773    | torch.Size([1, 1000, 754]) | 5\n",
-      "14774    | torch.Size([1, 1000, 754]) | 5\n",
-      "14775    | torch.Size([1, 1000, 754]) | 5\n",
-      "14776    | torch.Size([1, 1000, 754]) | 5\n",
-      "14777    | torch.Size([1, 1000, 785]) | 5\n",
-      "14778    | torch.Size([1, 1000, 754]) | 5\n",
-      "14779    | torch.Size([1, 1000, 754]) | 5\n",
-      "14780    | torch.Size([1, 1000, 754]) | 5\n",
-      "14781    | torch.Size([1, 1000, 754]) | 5\n",
-      "14782    | torch.Size([1, 1000, 787]) | 5\n",
-      "14783    | torch.Size([1, 1000, 821]) | 5\n",
-      "14784    | torch.Size([1, 1000, 754]) | 5\n",
-      "14785    | torch.Size([1, 1000, 754]) | 5\n",
-      "14786    | torch.Size([1, 1000, 754]) | 5\n",
-      "14787    | torch.Size([1, 1000, 754]) | 5\n",
-      "14788    | torch.Size([1, 1000, 754]) | 5\n",
-      "14789    | torch.Size([1, 1000, 754]) | 5\n",
-      "14790    | torch.Size([1, 1000, 754]) | 5\n",
-      "14791    | torch.Size([1, 1000, 775]) | 5\n",
-      "14792    | torch.Size([1, 1000, 754]) | 5\n",
-      "14793    | torch.Size([1, 1000, 754]) | 5\n",
-      "14794    | torch.Size([1, 1000, 780]) | 5\n",
-      "14795    | torch.Size([1, 1000, 782]) | 5\n",
-      "14796    | torch.Size([1, 1000, 754]) | 5\n",
-      "14797    | torch.Size([1, 1000, 782]) | 5\n",
-      "14798    | torch.Size([1, 1000, 754]) | 5\n",
-      "14799    | torch.Size([1, 1000, 771]) | 5\n",
-      "14800    | torch.Size([1, 1000, 754]) | 5\n",
-      "14801    | torch.Size([1, 1000, 774]) | 5\n",
-      "14802    | torch.Size([1, 1000, 754]) | 5\n",
-      "14803    | torch.Size([1, 1000, 774]) | 5\n",
-      "14804    | torch.Size([1, 1000, 754]) | 5\n",
-      "14805    | torch.Size([1, 1000, 754]) | 5\n",
-      "14806    | torch.Size([1, 1000, 754]) | 5\n",
-      "14807    | torch.Size([1, 1000, 786]) | 5\n",
-      "14808    | torch.Size([1, 1000, 754]) | 5\n",
-      "14809    | torch.Size([1, 1000, 754]) | 5\n",
-      "14810    | torch.Size([1, 1000, 754]) | 5\n",
-      "14811    | torch.Size([1, 1000, 754]) | 5\n",
-      "14812    | torch.Size([1, 1000, 754]) | 5\n",
-      "14813    | torch.Size([1, 1000, 754]) | 5\n",
-      "14814    | torch.Size([1, 1000, 773]) | 5\n",
-      "14815    | torch.Size([1, 1000, 784]) | 5\n",
-      "14816    | torch.Size([1, 1000, 789]) | 5\n",
-      "14817    | torch.Size([1, 1000, 754]) | 5\n",
-      "14818    | torch.Size([1, 1000, 802]) | 5\n",
-      "14819    | torch.Size([1, 1000, 754]) | 5\n",
-      "14820    | torch.Size([1, 1000, 754]) | 5\n",
-      "14821    | torch.Size([1, 1000, 771]) | 5\n",
-      "14822    | torch.Size([1, 1000, 754]) | 5\n",
-      "14823    | torch.Size([1, 1000, 754]) | 5\n",
-      "14824    | torch.Size([1, 1000, 754]) | 5\n",
-      "14825    | torch.Size([1, 1000, 754]) | 5\n",
-      "14826    | torch.Size([1, 1000, 754]) | 5\n",
-      "14827    | torch.Size([1, 1000, 754]) | 5\n",
-      "14828    | torch.Size([1, 1000, 786]) | 5\n",
-      "14829    | torch.Size([1, 1000, 754]) | 5\n",
-      "14830    | torch.Size([1, 1000, 789]) | 5\n",
-      "14831    | torch.Size([1, 1000, 788]) | 5\n",
-      "14832    | torch.Size([1, 1000, 754]) | 5\n",
-      "14833    | torch.Size([1, 1000, 771]) | 5\n",
-      "14834    | torch.Size([1, 1000, 771]) | 5\n",
-      "14835    | torch.Size([1, 1000, 754]) | 5\n",
-      "14836    | torch.Size([1, 1000, 754]) | 5\n",
-      "14837    | torch.Size([1, 1000, 754]) | 5\n",
-      "14838    | torch.Size([1, 1000, 754]) | 5\n",
-      "14839    | torch.Size([1, 1000, 754]) | 5\n",
-      "14840    | torch.Size([1, 1000, 754]) | 5\n",
-      "14841    | torch.Size([1, 1000, 754]) | 5\n",
-      "14842    | torch.Size([1, 1000, 754]) | 5\n",
-      "14843    | torch.Size([1, 1000, 754]) | 5\n",
-      "14844    | torch.Size([1, 1000, 754]) | 5\n",
-      "14845    | torch.Size([1, 1000, 754]) | 5\n",
-      "14846    | torch.Size([1, 1000, 754]) | 5\n",
-      "14847    | torch.Size([1, 1000, 754]) | 5\n",
-      "14848    | torch.Size([1, 1000, 787]) | 5\n",
-      "14849    | torch.Size([1, 1000, 789]) | 5\n",
-      "14850    | torch.Size([1, 1000, 754]) | 5\n",
-      "14851    | torch.Size([1, 1000, 774]) | 5\n",
-      "14852    | torch.Size([1, 1000, 754]) | 5\n",
-      "14853    | torch.Size([1, 1000, 754]) | 5\n",
-      "14854    | torch.Size([1, 1000, 754]) | 5\n",
-      "14855    | torch.Size([1, 1000, 754]) | 5\n",
-      "14856    | torch.Size([1, 1000, 754]) | 5\n",
-      "14857    | torch.Size([1, 1000, 754]) | 5\n",
-      "14858    | torch.Size([1, 1000, 754]) | 5\n",
-      "14859    | torch.Size([1, 1000, 754]) | 5\n",
-      "14860    | torch.Size([1, 1000, 754]) | 5\n",
-      "14861    | torch.Size([1, 1000, 754]) | 5\n",
-      "14862    | torch.Size([1, 1000, 754]) | 5\n",
-      "14863    | torch.Size([1, 1000, 754]) | 5\n",
-      "14864    | torch.Size([1, 1000, 754]) | 5\n",
-      "14865    | torch.Size([1, 1000, 754]) | 5\n",
-      "14866    | torch.Size([1, 1000, 754]) | 5\n",
-      "14867    | torch.Size([1, 1000, 754]) | 5\n",
-      "14868    | torch.Size([1, 1000, 754]) | 5\n",
-      "14869    | torch.Size([1, 1000, 754]) | 5\n",
-      "14870    | torch.Size([1, 1000, 754]) | 5\n",
-      "14871    | torch.Size([1, 1000, 754]) | 5\n",
-      "14872    | torch.Size([1, 1000, 754]) | 5\n",
-      "14873    | torch.Size([1, 1000, 754]) | 5\n",
-      "14874    | torch.Size([1, 1000, 754]) | 5\n",
-      "14875    | torch.Size([1, 1000, 754]) | 5\n",
-      "14876    | torch.Size([1, 1000, 754]) | 5\n",
-      "14877    | torch.Size([1, 1000, 754]) | 5\n",
-      "14878    | torch.Size([1, 1000, 754]) | 5\n",
-      "14879    | torch.Size([1, 1000, 754]) | 5\n",
-      "14880    | torch.Size([1, 1000, 754]) | 5\n",
-      "14881    | torch.Size([1, 1000, 754]) | 5\n",
-      "14882    | torch.Size([1, 1000, 754]) | 5\n",
-      "14883    | torch.Size([1, 1000, 754]) | 5\n",
-      "14884    | torch.Size([1, 1000, 771]) | 5\n",
-      "14885    | torch.Size([1, 1000, 754]) | 5\n",
-      "14886    | torch.Size([1, 1000, 785]) | 5\n",
-      "14887    | torch.Size([1, 1000, 754]) | 5\n",
-      "14888    | torch.Size([1, 1000, 754]) | 5\n",
-      "14889    | torch.Size([1, 1000, 754]) | 5\n",
-      "14890    | torch.Size([1, 1000, 754]) | 5\n",
-      "14891    | torch.Size([1, 1000, 807]) | 5\n",
-      "14892    | torch.Size([1, 1000, 776]) | 5\n",
-      "14893    | torch.Size([1, 1000, 754]) | 5\n",
-      "14894    | torch.Size([1, 1000, 754]) | 5\n",
-      "14895    | torch.Size([1, 1000, 771]) | 5\n",
-      "14896    | torch.Size([1, 1000, 789]) | 5\n",
-      "14897    | torch.Size([1, 1000, 754]) | 5\n",
-      "14898    | torch.Size([1, 1000, 754]) | 5\n",
-      "14899    | torch.Size([1, 1000, 754]) | 5\n",
-      "14900    | torch.Size([1, 1000, 754]) | 5\n",
-      "14901    | torch.Size([1, 1000, 754]) | 5\n",
-      "14902    | torch.Size([1, 1000, 754]) | 5\n",
-      "14903    | torch.Size([1, 1000, 754]) | 5\n",
-      "14904    | torch.Size([1, 1000, 754]) | 5\n",
-      "14905    | torch.Size([1, 1000, 754]) | 5\n",
-      "14906    | torch.Size([1, 1000, 754]) | 5\n",
-      "14907    | torch.Size([1, 1000, 754]) | 5\n",
-      "14908    | torch.Size([1, 1000, 754]) | 5\n",
-      "14909    | torch.Size([1, 1000, 754]) | 5\n",
-      "14910    | torch.Size([1, 1000, 754]) | 5\n",
-      "14911    | torch.Size([1, 1000, 754]) | 5\n",
-      "14912    | torch.Size([1, 1000, 754]) | 5\n",
-      "14913    | torch.Size([1, 1000, 754]) | 5\n",
-      "14914    | torch.Size([1, 1000, 754]) | 5\n",
-      "14915    | torch.Size([1, 1000, 754]) | 5\n",
-      "14916    | torch.Size([1, 1000, 771]) | 5\n",
-      "14917    | torch.Size([1, 1000, 795]) | 5\n",
-      "14918    | torch.Size([1, 1000, 761]) | 5\n",
-      "14919    | torch.Size([1, 1000, 754]) | 5\n",
-      "14920    | torch.Size([1, 1000, 754]) | 5\n",
-      "14921    | torch.Size([1, 1000, 754]) | 5\n",
-      "14922    | torch.Size([1, 1000, 754]) | 5\n",
-      "14923    | torch.Size([1, 1000, 780]) | 5\n",
-      "14924    | torch.Size([1, 1000, 790]) | 5\n",
-      "14925    | torch.Size([1, 1000, 794]) | 5\n",
-      "14926    | torch.Size([1, 1000, 754]) | 5\n",
-      "14927    | torch.Size([1, 1000, 754]) | 5\n",
-      "14928    | torch.Size([1, 1000, 754]) | 5\n",
-      "14929    | torch.Size([1, 1000, 789]) | 5\n",
-      "14930    | torch.Size([1, 1000, 754]) | 5\n",
-      "14931    | torch.Size([1, 1000, 754]) | 5\n",
-      "14932    | torch.Size([1, 1000, 754]) | 5\n",
-      "14933    | torch.Size([1, 1000, 786]) | 5\n",
-      "14934    | torch.Size([1, 1000, 754]) | 5\n",
-      "14935    | torch.Size([1, 1000, 754]) | 5\n",
-      "14936    | torch.Size([1, 1000, 754]) | 5\n",
-      "14937    | torch.Size([1, 1000, 754]) | 5\n",
-      "14938    | torch.Size([1, 1000, 754]) | 5\n",
-      "14939    | torch.Size([1, 1000, 754]) | 5\n",
-      "14940    | torch.Size([1, 1000, 793]) | 5\n",
-      "14941    | torch.Size([1, 1000, 784]) | 5\n",
-      "14942    | torch.Size([1, 1000, 756]) | 5\n",
-      "14943    | torch.Size([1, 1000, 754]) | 5\n",
-      "14944    | torch.Size([1, 1000, 789]) | 5\n",
-      "14945    | torch.Size([1, 1000, 754]) | 5\n",
-      "14946    | torch.Size([1, 1000, 754]) | 5\n",
-      "14947    | torch.Size([1, 1000, 754]) | 5\n",
-      "14948    | torch.Size([1, 1000, 754]) | 5\n",
-      "14949    | torch.Size([1, 1000, 754]) | 5\n",
-      "14950    | torch.Size([1, 1000, 754]) | 5\n",
-      "14951    | torch.Size([1, 1000, 754]) | 5\n",
-      "14952    | torch.Size([1, 1000, 754]) | 5\n",
-      "14953    | torch.Size([1, 1000, 754]) | 5\n",
-      "14954    | torch.Size([1, 1000, 754]) | 5\n",
-      "14955    | torch.Size([1, 1000, 754]) | 5\n",
-      "14956    | torch.Size([1, 1000, 754]) | 5\n",
-      "14957    | torch.Size([1, 1000, 754]) | 5\n",
-      "14958    | torch.Size([1, 1000, 754]) | 5\n",
-      "14959    | torch.Size([1, 1000, 754]) | 5\n",
-      "14960    | torch.Size([1, 1000, 754]) | 5\n",
-      "14961    | torch.Size([1, 1000, 777]) | 5\n",
-      "14962    | torch.Size([1, 1000, 754]) | 5\n",
-      "14963    | torch.Size([1, 1000, 786]) | 5\n",
-      "14964    | torch.Size([1, 1000, 754]) | 5\n",
-      "14965    | torch.Size([1, 1000, 754]) | 5\n",
-      "14966    | torch.Size([1, 1000, 754]) | 5\n",
-      "14967    | torch.Size([1, 1000, 754]) | 5\n",
-      "14968    | torch.Size([1, 1000, 754]) | 5\n",
-      "14969    | torch.Size([1, 1000, 754]) | 5\n",
-      "14970    | torch.Size([1, 1000, 789]) | 5\n",
-      "14971    | torch.Size([1, 1000, 754]) | 5\n",
-      "14972    | torch.Size([1, 1000, 754]) | 5\n",
-      "14973    | torch.Size([1, 1000, 754]) | 5\n",
-      "14974    | torch.Size([1, 1000, 754]) | 5\n",
-      "14975    | torch.Size([1, 1000, 754]) | 5\n",
-      "14976    | torch.Size([1, 1000, 754]) | 5\n",
-      "14977    | torch.Size([1, 1000, 778]) | 5\n",
-      "14978    | torch.Size([1, 1000, 754]) | 5\n",
-      "14979    | torch.Size([1, 1000, 754]) | 5\n",
-      "14980    | torch.Size([1, 1000, 754]) | 5\n",
-      "14981    | torch.Size([1, 1000, 785]) | 5\n",
-      "14982    | torch.Size([1, 1000, 754]) | 5\n",
-      "14983    | torch.Size([1, 1000, 754]) | 5\n",
-      "14984    | torch.Size([1, 1000, 754]) | 5\n",
-      "14985    | torch.Size([1, 1000, 754]) | 5\n",
-      "14986    | torch.Size([1, 1000, 754]) | 5\n",
-      "14987    | torch.Size([1, 1000, 788]) | 5\n",
-      "14988    | torch.Size([1, 1000, 754]) | 5\n",
-      "14989    | torch.Size([1, 1000, 775]) | 5\n",
-      "14990    | torch.Size([1, 1000, 754]) | 5\n",
-      "14991    | torch.Size([1, 1000, 754]) | 5\n",
-      "14992    | torch.Size([1, 1000, 754]) | 5\n",
-      "14993    | torch.Size([1, 1000, 754]) | 5\n",
-      "14994    | torch.Size([1, 1000, 754]) | 5\n",
-      "14995    | torch.Size([1, 1000, 754]) | 5\n",
-      "14996    | torch.Size([1, 1000, 754]) | 5\n",
-      "14997    | torch.Size([1, 1000, 793]) | 5\n",
-      "14998    | torch.Size([1, 1000, 771]) | 5\n",
-      "14999    | torch.Size([1, 1000, 754]) | 5\n",
-      "15000    | torch.Size([1, 1000, 754]) | 5\n",
-      "15001    | torch.Size([1, 1000, 754]) | 5\n",
-      "15002    | torch.Size([1, 1000, 788]) | 5\n",
-      "15003    | torch.Size([1, 1000, 754]) | 5\n",
-      "15004    | torch.Size([1, 1000, 754]) | 5\n",
-      "15005    | torch.Size([1, 1000, 771]) | 5\n",
-      "15006    | torch.Size([1, 1000, 754]) | 5\n",
-      "15007    | torch.Size([1, 1000, 754]) | 5\n",
-      "15008    | torch.Size([1, 1000, 754]) | 5\n",
-      "15009    | torch.Size([1, 1000, 754]) | 5\n",
-      "15010    | torch.Size([1, 1000, 754]) | 5\n",
-      "15011    | torch.Size([1, 1000, 779]) | 5\n",
-      "15012    | torch.Size([1, 1000, 784]) | 5\n",
-      "15013    | torch.Size([1, 1000, 754]) | 5\n",
-      "15014    | torch.Size([1, 1000, 754]) | 5\n",
-      "15015    | torch.Size([1, 1000, 754]) | 5\n",
-      "15016    | torch.Size([1, 1000, 754]) | 5\n",
-      "15017    | torch.Size([1, 1000, 782]) | 5\n",
-      "15018    | torch.Size([1, 1000, 754]) | 5\n",
-      "15019    | torch.Size([1, 1000, 754]) | 5\n",
-      "15020    | torch.Size([1, 1000, 754]) | 5\n",
-      "15021    | torch.Size([1, 1000, 754]) | 5\n",
-      "15022    | torch.Size([1, 1000, 754]) | 5\n",
-      "15023    | torch.Size([1, 1000, 754]) | 5\n",
-      "15024    | torch.Size([1, 1000, 766]) | 5\n",
-      "15025    | torch.Size([1, 1000, 754]) | 5\n",
-      "15026    | torch.Size([1, 1000, 754]) | 5\n",
-      "15027    | torch.Size([1, 1000, 794]) | 5\n",
-      "15028    | torch.Size([1, 1000, 754]) | 5\n",
-      "15029    | torch.Size([1, 1000, 738]) | 5\n",
-      "15030    | torch.Size([1, 1000, 754]) | 5\n",
-      "15031    | torch.Size([1, 1000, 754]) | 5\n",
-      "15032    | torch.Size([1, 1000, 784]) | 5\n",
-      "15033    | torch.Size([1, 1000, 754]) | 5\n",
-      "15034    | torch.Size([1, 1000, 771]) | 5\n",
-      "15035    | torch.Size([1, 1000, 754]) | 5\n",
-      "15036    | torch.Size([1, 1000, 754]) | 5\n",
-      "15037    | torch.Size([1, 1000, 754]) | 5\n",
-      "15038    | torch.Size([1, 1000, 784]) | 5\n",
-      "15039    | torch.Size([1, 1000, 754]) | 5\n",
-      "15040    | torch.Size([1, 1000, 773]) | 5\n",
-      "15041    | torch.Size([1, 1000, 754]) | 5\n",
-      "15042    | torch.Size([1, 1000, 754]) | 5\n",
-      "15043    | torch.Size([1, 1000, 754]) | 5\n",
-      "15044    | torch.Size([1, 1000, 754]) | 5\n",
-      "15045    | torch.Size([1, 1000, 789]) | 5\n",
-      "15046    | torch.Size([1, 1000, 775]) | 5\n",
-      "15047    | torch.Size([1, 1000, 754]) | 5\n",
-      "15048    | torch.Size([1, 1000, 754]) | 5\n",
-      "15049    | torch.Size([1, 1000, 754]) | 5\n",
-      "15050    | torch.Size([1, 1000, 754]) | 5\n",
-      "15051    | torch.Size([1, 1000, 754]) | 5\n",
-      "15052    | torch.Size([1, 1000, 754]) | 5\n",
-      "15053    | torch.Size([1, 1000, 754]) | 5\n",
-      "15054    | torch.Size([1, 1000, 754]) | 5\n",
-      "15055    | torch.Size([1, 1000, 754]) | 5\n",
-      "15056    | torch.Size([1, 1000, 754]) | 5\n",
-      "15057    | torch.Size([1, 1000, 754]) | 5\n",
-      "15058    | torch.Size([1, 1000, 778]) | 5\n",
-      "15059    | torch.Size([1, 1000, 754]) | 5\n",
-      "15060    | torch.Size([1, 1000, 783]) | 5\n",
-      "15061    | torch.Size([1, 1000, 754]) | 5\n",
-      "15062    | torch.Size([1, 1000, 754]) | 5\n",
-      "15063    | torch.Size([1, 1000, 774]) | 5\n",
-      "15064    | torch.Size([1, 1000, 754]) | 5\n",
-      "15065    | torch.Size([1, 1000, 754]) | 5\n",
-      "15066    | torch.Size([1, 1000, 754]) | 5\n",
-      "15067    | torch.Size([1, 1000, 754]) | 5\n",
-      "15068    | torch.Size([1, 1000, 754]) | 5\n",
-      "15069    | torch.Size([1, 1000, 754]) | 5\n",
-      "15070    | torch.Size([1, 1000, 780]) | 5\n",
-      "15071    | torch.Size([1, 1000, 754]) | 5\n",
-      "15072    | torch.Size([1, 1000, 754]) | 5\n",
-      "15073    | torch.Size([1, 1000, 754]) | 5\n",
-      "15074    | torch.Size([1, 1000, 801]) | 5\n",
-      "15075    | torch.Size([1, 1000, 754]) | 5\n",
-      "15076    | torch.Size([1, 1000, 754]) | 5\n",
-      "15077    | torch.Size([1, 1000, 783]) | 5\n",
-      "15078    | torch.Size([1, 1000, 754]) | 5\n",
-      "15079    | torch.Size([1, 1000, 754]) | 5\n",
-      "15080    | torch.Size([1, 1000, 754]) | 5\n",
-      "15081    | torch.Size([1, 1000, 754]) | 5\n",
-      "15082    | torch.Size([1, 1000, 771]) | 5\n",
-      "15083    | torch.Size([1, 1000, 754]) | 5\n",
-      "15084    | torch.Size([1, 1000, 771]) | 5\n",
-      "15085    | torch.Size([1, 1000, 754]) | 5\n",
-      "15086    | torch.Size([1, 1000, 754]) | 5\n",
-      "15087    | torch.Size([1, 1000, 771]) | 5\n",
-      "15088    | torch.Size([1, 1000, 754]) | 5\n",
-      "15089    | torch.Size([1, 1000, 754]) | 5\n",
-      "15090    | torch.Size([1, 1000, 781]) | 5\n",
-      "15091    | torch.Size([1, 1000, 782]) | 5\n",
-      "15092    | torch.Size([1, 1000, 754]) | 5\n",
-      "15093    | torch.Size([1, 1000, 754]) | 5\n",
-      "15094    | torch.Size([1, 1000, 754]) | 5\n",
-      "15095    | torch.Size([1, 1000, 771]) | 5\n",
-      "15096    | torch.Size([1, 1000, 754]) | 5\n",
-      "15097    | torch.Size([1, 1000, 754]) | 5\n",
-      "15098    | torch.Size([1, 1000, 788]) | 5\n",
-      "15099    | torch.Size([1, 1000, 771]) | 5\n",
-      "15100    | torch.Size([1, 1000, 754]) | 5\n",
-      "15101    | torch.Size([1, 1000, 776]) | 6\n",
-      "15102    | torch.Size([1, 1000, 808]) | 6\n",
-      "15103    | torch.Size([1, 1000, 777]) | 6\n",
-      "15104    | torch.Size([1, 1000, 769]) | 6\n",
-      "15105    | torch.Size([1, 1000, 754]) | 6\n",
-      "15106    | torch.Size([1, 1000, 777]) | 6\n",
-      "15107    | torch.Size([1, 1000, 754]) | 6\n",
-      "15108    | torch.Size([1, 1000, 754]) | 6\n",
-      "15109    | torch.Size([1, 1000, 754]) | 6\n",
-      "15110    | torch.Size([1, 1000, 754]) | 6\n",
-      "15111    | torch.Size([1, 1000, 754]) | 6\n",
-      "15112    | torch.Size([1, 1000, 754]) | 6\n",
-      "15113    | torch.Size([1, 1000, 754]) | 6\n",
-      "15114    | torch.Size([1, 1000, 809]) | 6\n",
-      "15115    | torch.Size([1, 1000, 791]) | 6\n",
-      "15116    | torch.Size([1, 1000, 754]) | 6\n",
-      "15117    | torch.Size([1, 1000, 773]) | 6\n",
-      "15118    | torch.Size([1, 1000, 770]) | 6\n",
-      "15119    | torch.Size([1, 1000, 754]) | 6\n",
-      "15120    | torch.Size([1, 1000, 763]) | 6\n",
-      "15121    | torch.Size([1, 1000, 776]) | 6\n",
-      "15122    | torch.Size([1, 1000, 762]) | 6\n",
-      "15123    | torch.Size([1, 1000, 754]) | 6\n",
-      "15124    | torch.Size([1, 1000, 754]) | 6\n",
-      "15125    | torch.Size([1, 1000, 776]) | 6\n",
-      "15126    | torch.Size([1, 1000, 777]) | 6\n",
-      "15127    | torch.Size([1, 1000, 754]) | 6\n",
-      "15128    | torch.Size([1, 1000, 754]) | 6\n",
-      "15129    | torch.Size([1, 1000, 762]) | 6\n",
-      "15130    | torch.Size([1, 1000, 754]) | 6\n",
-      "15131    | torch.Size([1, 1000, 754]) | 6\n",
-      "15132    | torch.Size([1, 1000, 754]) | 6\n",
-      "15133    | torch.Size([1, 1000, 754]) | 6\n",
-      "15134    | torch.Size([1, 1000, 754]) | 6\n",
-      "15135    | torch.Size([1, 1000, 754]) | 6\n",
-      "15136    | torch.Size([1, 1000, 754]) | 6\n",
-      "15137    | torch.Size([1, 1000, 754]) | 6\n",
-      "15138    | torch.Size([1, 1000, 754]) | 6\n",
-      "15139    | torch.Size([1, 1000, 777]) | 6\n",
-      "15140    | torch.Size([1, 1000, 771]) | 6\n",
-      "15141    | torch.Size([1, 1000, 783]) | 6\n",
-      "15142    | torch.Size([1, 1000, 768]) | 6\n",
-      "15143    | torch.Size([1, 1000, 754]) | 6\n",
-      "15144    | torch.Size([1, 1000, 754]) | 6\n",
-      "15145    | torch.Size([1, 1000, 776]) | 6\n",
-      "15146    | torch.Size([1, 1000, 754]) | 6\n",
-      "15147    | torch.Size([1, 1000, 770]) | 6\n",
-      "15148    | torch.Size([1, 1000, 754]) | 6\n",
-      "15149    | torch.Size([1, 1000, 754]) | 6\n",
-      "15150    | torch.Size([1, 1000, 778]) | 6\n",
-      "15151    | torch.Size([1, 1000, 754]) | 6\n",
-      "15152    | torch.Size([1, 1000, 771]) | 6\n",
-      "15153    | torch.Size([1, 1000, 754]) | 6\n",
-      "15154    | torch.Size([1, 1000, 754]) | 6\n",
-      "15155    | torch.Size([1, 1000, 762]) | 6\n",
-      "15156    | torch.Size([1, 1000, 754]) | 6\n",
-      "15157    | torch.Size([1, 1000, 775]) | 6\n",
-      "15158    | torch.Size([1, 1000, 777]) | 6\n",
-      "15159    | torch.Size([1, 1000, 754]) | 6\n",
-      "15160    | torch.Size([1, 1000, 754]) | 6\n",
-      "15161    | torch.Size([1, 1000, 754]) | 6\n",
-      "15162    | torch.Size([1, 1000, 754]) | 6\n",
-      "15163    | torch.Size([1, 1000, 754]) | 6\n",
-      "15164    | torch.Size([1, 1000, 754]) | 6\n",
-      "15165    | torch.Size([1, 1000, 752]) | 6\n",
-      "15166    | torch.Size([1, 1000, 762]) | 6\n",
-      "15167    | torch.Size([1, 1000, 787]) | 6\n",
-      "15168    | torch.Size([1, 1000, 784]) | 6\n",
-      "15169    | torch.Size([1, 1000, 762]) | 6\n",
-      "15170    | torch.Size([1, 1000, 754]) | 6\n",
-      "15171    | torch.Size([1, 1000, 777]) | 6\n",
-      "15172    | torch.Size([1, 1000, 754]) | 6\n",
-      "15173    | torch.Size([1, 1000, 762]) | 6\n",
-      "15174    | torch.Size([1, 1000, 754]) | 6\n",
-      "15175    | torch.Size([1, 1000, 764]) | 6\n",
-      "15176    | torch.Size([1, 1000, 754]) | 6\n",
-      "15177    | torch.Size([1, 1000, 724]) | 6\n",
-      "15178    | torch.Size([1, 1000, 754]) | 6\n",
-      "15179    | torch.Size([1, 1000, 754]) | 6\n",
-      "15180    | torch.Size([1, 1000, 777]) | 6\n",
-      "15181    | torch.Size([1, 1000, 754]) | 6\n",
-      "15182    | torch.Size([1, 1000, 782]) | 6\n",
-      "15183    | torch.Size([1, 1000, 777]) | 6\n",
-      "15184    | torch.Size([1, 1000, 754]) | 6\n",
-      "15185    | torch.Size([1, 1000, 777]) | 6\n",
-      "15186    | torch.Size([1, 1000, 780]) | 6\n",
-      "15187    | torch.Size([1, 1000, 754]) | 6\n",
-      "15188    | torch.Size([1, 1000, 762]) | 6\n",
-      "15189    | torch.Size([1, 1000, 754]) | 6\n",
-      "15190    | torch.Size([1, 1000, 752]) | 6\n",
-      "15191    | torch.Size([1, 1000, 754]) | 6\n",
-      "15192    | torch.Size([1, 1000, 771]) | 6\n",
-      "15193    | torch.Size([1, 1000, 777]) | 6\n",
-      "15194    | torch.Size([1, 1000, 777]) | 6\n",
-      "15195    | torch.Size([1, 1000, 777]) | 6\n",
-      "15196    | torch.Size([1, 1000, 754]) | 6\n",
-      "15197    | torch.Size([1, 1000, 754]) | 6\n",
-      "15198    | torch.Size([1, 1000, 762]) | 6\n",
-      "15199    | torch.Size([1, 1000, 754]) | 6\n",
-      "15200    | torch.Size([1, 1000, 762]) | 6\n",
-      "15201    | torch.Size([1, 1000, 792]) | 6\n",
-      "15202    | torch.Size([1, 1000, 754]) | 6\n",
-      "15203    | torch.Size([1, 1000, 777]) | 6\n",
-      "15204    | torch.Size([1, 1000, 779]) | 6\n",
-      "15205    | torch.Size([1, 1000, 777]) | 6\n",
-      "15206    | torch.Size([1, 1000, 777]) | 6\n",
-      "15207    | torch.Size([1, 1000, 777]) | 6\n",
-      "15208    | torch.Size([1, 1000, 784]) | 6\n",
-      "15209    | torch.Size([1, 1000, 754]) | 6\n",
-      "15210    | torch.Size([1, 1000, 754]) | 6\n",
-      "15211    | torch.Size([1, 1000, 754]) | 6\n",
-      "15212    | torch.Size([1, 1000, 754]) | 6\n",
-      "15213    | torch.Size([1, 1000, 754]) | 6\n",
-      "15214    | torch.Size([1, 1000, 788]) | 6\n",
-      "15215    | torch.Size([1, 1000, 777]) | 6\n",
-      "15216    | torch.Size([1, 1000, 754]) | 6\n",
-      "15217    | torch.Size([1, 1000, 777]) | 6\n",
-      "15218    | torch.Size([1, 1000, 754]) | 6\n",
-      "15219    | torch.Size([1, 1000, 754]) | 6\n",
-      "15220    | torch.Size([1, 1000, 762]) | 6\n",
-      "15221    | torch.Size([1, 1000, 787]) | 6\n",
-      "15222    | torch.Size([1, 1000, 762]) | 6\n",
-      "15223    | torch.Size([1, 1000, 754]) | 6\n",
-      "15224    | torch.Size([1, 1000, 759]) | 6\n",
-      "15225    | torch.Size([1, 1000, 774]) | 6\n",
-      "15226    | torch.Size([1, 1000, 754]) | 6\n",
-      "15227    | torch.Size([1, 1000, 787]) | 6\n",
-      "15228    | torch.Size([1, 1000, 754]) | 6\n",
-      "15229    | torch.Size([1, 1000, 777]) | 6\n",
-      "15230    | torch.Size([1, 1000, 777]) | 6\n",
-      "15231    | torch.Size([1, 1000, 775]) | 6\n",
-      "15232    | torch.Size([1, 1000, 781]) | 6\n",
-      "15233    | torch.Size([1, 1000, 762]) | 6\n",
-      "15234    | torch.Size([1, 1000, 768]) | 6\n",
-      "15235    | torch.Size([1, 1000, 754]) | 6\n",
-      "15236    | torch.Size([1, 1000, 754]) | 6\n",
-      "15237    | torch.Size([1, 1000, 762]) | 6\n",
-      "15238    | torch.Size([1, 1000, 777]) | 6\n",
-      "15239    | torch.Size([1, 1000, 754]) | 6\n",
-      "15240    | torch.Size([1, 1000, 754]) | 6\n",
-      "15241    | torch.Size([1, 1000, 762]) | 6\n",
-      "15242    | torch.Size([1, 1000, 777]) | 6\n",
-      "15243    | torch.Size([1, 1000, 754]) | 6\n",
-      "15244    | torch.Size([1, 1000, 754]) | 6\n",
-      "15245    | torch.Size([1, 1000, 777]) | 6\n",
-      "15246    | torch.Size([1, 1000, 754]) | 6\n",
-      "15247    | torch.Size([1, 1000, 777]) | 6\n",
-      "15248    | torch.Size([1, 1000, 762]) | 6\n",
-      "15249    | torch.Size([1, 1000, 782]) | 6\n",
-      "15250    | torch.Size([1, 1000, 754]) | 6\n",
-      "15251    | torch.Size([1, 1000, 754]) | 6\n",
-      "15252    | torch.Size([1, 1000, 777]) | 6\n",
-      "15253    | torch.Size([1, 1000, 754]) | 6\n",
-      "15254    | torch.Size([1, 1000, 777]) | 6\n",
-      "15255    | torch.Size([1, 1000, 754]) | 6\n",
-      "15256    | torch.Size([1, 1000, 754]) | 6\n",
-      "15257    | torch.Size([1, 1000, 754]) | 6\n",
-      "15258    | torch.Size([1, 1000, 773]) | 6\n",
-      "15259    | torch.Size([1, 1000, 754]) | 6\n",
-      "15260    | torch.Size([1, 1000, 777]) | 6\n",
-      "15261    | torch.Size([1, 1000, 754]) | 6\n",
-      "15262    | torch.Size([1, 1000, 778]) | 6\n",
-      "15263    | torch.Size([1, 1000, 773]) | 6\n",
-      "15264    | torch.Size([1, 1000, 780]) | 6\n",
-      "15265    | torch.Size([1, 1000, 777]) | 6\n",
-      "15266    | torch.Size([1, 1000, 754]) | 6\n",
-      "15267    | torch.Size([1, 1000, 777]) | 6\n",
-      "15268    | torch.Size([1, 1000, 754]) | 6\n",
-      "15269    | torch.Size([1, 1000, 766]) | 6\n",
-      "15270    | torch.Size([1, 1000, 754]) | 6\n",
-      "15271    | torch.Size([1, 1000, 788]) | 6\n",
-      "15272    | torch.Size([1, 1000, 777]) | 6\n",
-      "15273    | torch.Size([1, 1000, 754]) | 6\n",
-      "15274    | torch.Size([1, 1000, 785]) | 6\n",
-      "15275    | torch.Size([1, 1000, 777]) | 6\n",
-      "15276    | torch.Size([1, 1000, 762]) | 6\n",
-      "15277    | torch.Size([1, 1000, 777]) | 6\n",
-      "15278    | torch.Size([1, 1000, 775]) | 6\n",
-      "15279    | torch.Size([1, 1000, 787]) | 6\n",
-      "15280    | torch.Size([1, 1000, 762]) | 6\n",
-      "15281    | torch.Size([1, 1000, 754]) | 6\n",
-      "15282    | torch.Size([1, 1000, 754]) | 6\n",
-      "15283    | torch.Size([1, 1000, 762]) | 6\n",
-      "15284    | torch.Size([1, 1000, 771]) | 6\n",
-      "15285    | torch.Size([1, 1000, 775]) | 6\n",
-      "15286    | torch.Size([1, 1000, 754]) | 6\n",
-      "15287    | torch.Size([1, 1000, 771]) | 6\n",
-      "15288    | torch.Size([1, 1000, 754]) | 6\n",
-      "15289    | torch.Size([1, 1000, 754]) | 6\n",
-      "15290    | torch.Size([1, 1000, 777]) | 6\n",
-      "15291    | torch.Size([1, 1000, 754]) | 6\n",
-      "15292    | torch.Size([1, 1000, 765]) | 6\n",
-      "15293    | torch.Size([1, 1000, 754]) | 6\n",
-      "15294    | torch.Size([1, 1000, 762]) | 6\n",
-      "15295    | torch.Size([1, 1000, 806]) | 6\n",
-      "15296    | torch.Size([1, 1000, 754]) | 6\n",
-      "15297    | torch.Size([1, 1000, 754]) | 6\n",
-      "15298    | torch.Size([1, 1000, 754]) | 6\n",
-      "15299    | torch.Size([1, 1000, 788]) | 6\n",
-      "15300    | torch.Size([1, 1000, 762]) | 6\n",
-      "15301    | torch.Size([1, 1000, 754]) | 6\n",
-      "15302    | torch.Size([1, 1000, 754]) | 6\n",
-      "15303    | torch.Size([1, 1000, 784]) | 6\n",
-      "15304    | torch.Size([1, 1000, 769]) | 6\n",
-      "15305    | torch.Size([1, 1000, 763]) | 6\n",
-      "15306    | torch.Size([1, 1000, 777]) | 6\n",
-      "15307    | torch.Size([1, 1000, 754]) | 6\n",
-      "15308    | torch.Size([1, 1000, 762]) | 6\n",
-      "15309    | torch.Size([1, 1000, 762]) | 6\n",
-      "15310    | torch.Size([1, 1000, 777]) | 6\n",
-      "15311    | torch.Size([1, 1000, 771]) | 6\n",
-      "15312    | torch.Size([1, 1000, 752]) | 6\n",
-      "15313    | torch.Size([1, 1000, 754]) | 6\n",
-      "15314    | torch.Size([1, 1000, 777]) | 6\n",
-      "15315    | torch.Size([1, 1000, 777]) | 6\n",
-      "15316    | torch.Size([1, 1000, 777]) | 6\n",
-      "15317    | torch.Size([1, 1000, 754]) | 6\n",
-      "15318    | torch.Size([1, 1000, 754]) | 6\n",
-      "15319    | torch.Size([1, 1000, 782]) | 6\n",
-      "15320    | torch.Size([1, 1000, 754]) | 6\n",
-      "15321    | torch.Size([1, 1000, 754]) | 6\n",
-      "15322    | torch.Size([1, 1000, 762]) | 6\n",
-      "15323    | torch.Size([1, 1000, 762]) | 6\n",
-      "15324    | torch.Size([1, 1000, 777]) | 6\n",
-      "15325    | torch.Size([1, 1000, 754]) | 6\n",
-      "15326    | torch.Size([1, 1000, 777]) | 6\n",
-      "15327    | torch.Size([1, 1000, 777]) | 6\n",
-      "15328    | torch.Size([1, 1000, 754]) | 6\n",
-      "15329    | torch.Size([1, 1000, 762]) | 6\n",
-      "15330    | torch.Size([1, 1000, 762]) | 6\n",
-      "15331    | torch.Size([1, 1000, 754]) | 6\n",
-      "15332    | torch.Size([1, 1000, 762]) | 6\n",
-      "15333    | torch.Size([1, 1000, 773]) | 6\n",
-      "15334    | torch.Size([1, 1000, 754]) | 6\n",
-      "15335    | torch.Size([1, 1000, 754]) | 6\n",
-      "15336    | torch.Size([1, 1000, 777]) | 6\n",
-      "15337    | torch.Size([1, 1000, 762]) | 6\n",
-      "15338    | torch.Size([1, 1000, 754]) | 6\n",
-      "15339    | torch.Size([1, 1000, 754]) | 6\n",
-      "15340    | torch.Size([1, 1000, 772]) | 6\n",
-      "15341    | torch.Size([1, 1000, 754]) | 6\n",
-      "15342    | torch.Size([1, 1000, 754]) | 6\n",
-      "15343    | torch.Size([1, 1000, 777]) | 6\n",
-      "15344    | torch.Size([1, 1000, 770]) | 6\n",
-      "15345    | torch.Size([1, 1000, 754]) | 6\n",
-      "15346    | torch.Size([1, 1000, 754]) | 6\n",
-      "15347    | torch.Size([1, 1000, 784]) | 6\n",
-      "15348    | torch.Size([1, 1000, 754]) | 6\n",
-      "15349    | torch.Size([1, 1000, 754]) | 6\n",
-      "15350    | torch.Size([1, 1000, 754]) | 6\n",
-      "15351    | torch.Size([1, 1000, 771]) | 6\n",
-      "15352    | torch.Size([1, 1000, 777]) | 6\n",
-      "15353    | torch.Size([1, 1000, 754]) | 6\n",
-      "15354    | torch.Size([1, 1000, 776]) | 6\n",
-      "15355    | torch.Size([1, 1000, 754]) | 6\n",
-      "15356    | torch.Size([1, 1000, 772]) | 6\n",
-      "15357    | torch.Size([1, 1000, 777]) | 6\n",
-      "15358    | torch.Size([1, 1000, 754]) | 6\n",
-      "15359    | torch.Size([1, 1000, 754]) | 6\n",
-      "15360    | torch.Size([1, 1000, 777]) | 6\n",
-      "15361    | torch.Size([1, 1000, 754]) | 6\n",
-      "15362    | torch.Size([1, 1000, 754]) | 6\n",
-      "15363    | torch.Size([1, 1000, 777]) | 6\n",
-      "15364    | torch.Size([1, 1000, 754]) | 6\n",
-      "15365    | torch.Size([1, 1000, 778]) | 6\n",
-      "15366    | torch.Size([1, 1000, 754]) | 6\n",
-      "15367    | torch.Size([1, 1000, 777]) | 6\n",
-      "15368    | torch.Size([1, 1000, 754]) | 6\n",
-      "15369    | torch.Size([1, 1000, 762]) | 6\n",
-      "15370    | torch.Size([1, 1000, 762]) | 6\n",
-      "15371    | torch.Size([1, 1000, 762]) | 6\n",
-      "15372    | torch.Size([1, 1000, 762]) | 6\n",
-      "15373    | torch.Size([1, 1000, 768]) | 6\n",
-      "15374    | torch.Size([1, 1000, 754]) | 6\n",
-      "15375    | torch.Size([1, 1000, 754]) | 6\n",
-      "15376    | torch.Size([1, 1000, 772]) | 6\n",
-      "15377    | torch.Size([1, 1000, 777]) | 6\n",
-      "15378    | torch.Size([1, 1000, 777]) | 6\n",
-      "15379    | torch.Size([1, 1000, 789]) | 6\n",
-      "15380    | torch.Size([1, 1000, 762]) | 6\n",
-      "15381    | torch.Size([1, 1000, 777]) | 6\n",
-      "15382    | torch.Size([1, 1000, 754]) | 6\n",
-      "15383    | torch.Size([1, 1000, 754]) | 6\n",
-      "15384    | torch.Size([1, 1000, 770]) | 6\n",
-      "15385    | torch.Size([1, 1000, 754]) | 6\n",
-      "15386    | torch.Size([1, 1000, 762]) | 6\n",
-      "15387    | torch.Size([1, 1000, 776]) | 6\n",
-      "15388    | torch.Size([1, 1000, 754]) | 6\n",
-      "15389    | torch.Size([1, 1000, 754]) | 6\n",
-      "15390    | torch.Size([1, 1000, 802]) | 6\n",
-      "15391    | torch.Size([1, 1000, 754]) | 6\n",
-      "15392    | torch.Size([1, 1000, 802]) | 6\n",
-      "15393    | torch.Size([1, 1000, 772]) | 6\n",
-      "15394    | torch.Size([1, 1000, 754]) | 6\n",
-      "15395    | torch.Size([1, 1000, 754]) | 6\n",
-      "15396    | torch.Size([1, 1000, 754]) | 6\n",
-      "15397    | torch.Size([1, 1000, 777]) | 6\n",
-      "15398    | torch.Size([1, 1000, 754]) | 6\n",
-      "15399    | torch.Size([1, 1000, 754]) | 6\n",
-      "15400    | torch.Size([1, 1000, 754]) | 6\n",
-      "15401    | torch.Size([1, 1000, 754]) | 6\n",
-      "15402    | torch.Size([1, 1000, 754]) | 6\n",
-      "15403    | torch.Size([1, 1000, 754]) | 6\n",
-      "15404    | torch.Size([1, 1000, 754]) | 6\n",
-      "15405    | torch.Size([1, 1000, 762]) | 6\n",
-      "15406    | torch.Size([1, 1000, 777]) | 6\n",
-      "15407    | torch.Size([1, 1000, 776]) | 6\n",
-      "15408    | torch.Size([1, 1000, 777]) | 6\n",
-      "15409    | torch.Size([1, 1000, 754]) | 6\n",
-      "15410    | torch.Size([1, 1000, 780]) | 6\n",
-      "15411    | torch.Size([1, 1000, 754]) | 6\n",
-      "15412    | torch.Size([1, 1000, 762]) | 6\n",
-      "15413    | torch.Size([1, 1000, 776]) | 6\n",
-      "15414    | torch.Size([1, 1000, 754]) | 6\n",
-      "15415    | torch.Size([1, 1000, 782]) | 6\n",
-      "15416    | torch.Size([1, 1000, 777]) | 6\n",
-      "15417    | torch.Size([1, 1000, 781]) | 6\n",
-      "15418    | torch.Size([1, 1000, 806]) | 6\n",
-      "15419    | torch.Size([1, 1000, 754]) | 6\n",
-      "15420    | torch.Size([1, 1000, 777]) | 6\n",
-      "15421    | torch.Size([1, 1000, 772]) | 6\n",
-      "15422    | torch.Size([1, 1000, 754]) | 6\n",
-      "15423    | torch.Size([1, 1000, 777]) | 6\n",
-      "15424    | torch.Size([1, 1000, 754]) | 6\n",
-      "15425    | torch.Size([1, 1000, 768]) | 6\n",
-      "15426    | torch.Size([1, 1000, 754]) | 6\n",
-      "15427    | torch.Size([1, 1000, 754]) | 6\n",
-      "15428    | torch.Size([1, 1000, 762]) | 6\n",
-      "15429    | torch.Size([1, 1000, 762]) | 6\n",
-      "15430    | torch.Size([1, 1000, 777]) | 6\n",
-      "15431    | torch.Size([1, 1000, 762]) | 6\n",
-      "15432    | torch.Size([1, 1000, 777]) | 6\n",
-      "15433    | torch.Size([1, 1000, 754]) | 6\n",
-      "15434    | torch.Size([1, 1000, 762]) | 6\n",
-      "15435    | torch.Size([1, 1000, 754]) | 6\n",
-      "15436    | torch.Size([1, 1000, 754]) | 6\n",
-      "15437    | torch.Size([1, 1000, 773]) | 6\n",
-      "15438    | torch.Size([1, 1000, 777]) | 6\n",
-      "15439    | torch.Size([1, 1000, 754]) | 6\n",
-      "15440    | torch.Size([1, 1000, 754]) | 6\n",
-      "15441    | torch.Size([1, 1000, 754]) | 6\n",
-      "15442    | torch.Size([1, 1000, 754]) | 6\n",
-      "15443    | torch.Size([1, 1000, 775]) | 6\n",
-      "15444    | torch.Size([1, 1000, 754]) | 6\n",
-      "15445    | torch.Size([1, 1000, 754]) | 6\n",
-      "15446    | torch.Size([1, 1000, 754]) | 6\n",
-      "15447    | torch.Size([1, 1000, 769]) | 6\n",
-      "15448    | torch.Size([1, 1000, 771]) | 6\n",
-      "15449    | torch.Size([1, 1000, 762]) | 6\n",
-      "15450    | torch.Size([1, 1000, 754]) | 6\n",
-      "15451    | torch.Size([1, 1000, 777]) | 6\n",
-      "15452    | torch.Size([1, 1000, 754]) | 6\n",
-      "15453    | torch.Size([1, 1000, 774]) | 6\n",
-      "15454    | torch.Size([1, 1000, 777]) | 6\n",
-      "15455    | torch.Size([1, 1000, 777]) | 6\n",
-      "15456    | torch.Size([1, 1000, 802]) | 6\n",
-      "15457    | torch.Size([1, 1000, 762]) | 6\n",
-      "15458    | torch.Size([1, 1000, 771]) | 6\n",
-      "15459    | torch.Size([1, 1000, 802]) | 6\n",
-      "15460    | torch.Size([1, 1000, 796]) | 6\n",
-      "15461    | torch.Size([1, 1000, 771]) | 6\n",
-      "15462    | torch.Size([1, 1000, 762]) | 6\n",
-      "15463    | torch.Size([1, 1000, 754]) | 6\n",
-      "15464    | torch.Size([1, 1000, 777]) | 6\n",
-      "15465    | torch.Size([1, 1000, 762]) | 6\n",
-      "15466    | torch.Size([1, 1000, 754]) | 6\n",
-      "15467    | torch.Size([1, 1000, 766]) | 6\n",
-      "15468    | torch.Size([1, 1000, 775]) | 6\n",
-      "15469    | torch.Size([1, 1000, 754]) | 6\n",
-      "15470    | torch.Size([1, 1000, 754]) | 6\n",
-      "15471    | torch.Size([1, 1000, 754]) | 6\n",
-      "15472    | torch.Size([1, 1000, 754]) | 6\n",
-      "15473    | torch.Size([1, 1000, 762]) | 6\n",
-      "15474    | torch.Size([1, 1000, 762]) | 6\n",
-      "15475    | torch.Size([1, 1000, 777]) | 6\n",
-      "15476    | torch.Size([1, 1000, 754]) | 6\n",
-      "15477    | torch.Size([1, 1000, 777]) | 6\n",
-      "15478    | torch.Size([1, 1000, 754]) | 6\n",
-      "15479    | torch.Size([1, 1000, 768]) | 6\n",
-      "15480    | torch.Size([1, 1000, 802]) | 6\n",
-      "15481    | torch.Size([1, 1000, 754]) | 6\n",
-      "15482    | torch.Size([1, 1000, 771]) | 6\n",
-      "15483    | torch.Size([1, 1000, 754]) | 6\n",
-      "15484    | torch.Size([1, 1000, 754]) | 6\n",
-      "15485    | torch.Size([1, 1000, 754]) | 6\n",
-      "15486    | torch.Size([1, 1000, 754]) | 6\n",
-      "15487    | torch.Size([1, 1000, 762]) | 6\n",
-      "15488    | torch.Size([1, 1000, 722]) | 6\n",
-      "15489    | torch.Size([1, 1000, 754]) | 6\n",
-      "15490    | torch.Size([1, 1000, 778]) | 6\n",
-      "15491    | torch.Size([1, 1000, 754]) | 6\n",
-      "15492    | torch.Size([1, 1000, 777]) | 6\n",
-      "15493    | torch.Size([1, 1000, 754]) | 6\n",
-      "15494    | torch.Size([1, 1000, 786]) | 6\n",
-      "15495    | torch.Size([1, 1000, 754]) | 6\n",
-      "15496    | torch.Size([1, 1000, 777]) | 6\n",
-      "15497    | torch.Size([1, 1000, 754]) | 6\n",
-      "15498    | torch.Size([1, 1000, 777]) | 6\n",
-      "15499    | torch.Size([1, 1000, 771]) | 6\n",
-      "15500    | torch.Size([1, 1000, 777]) | 6\n",
-      "15501    | torch.Size([1, 1000, 767]) | 6\n",
-      "15502    | torch.Size([1, 1000, 777]) | 6\n",
-      "15503    | torch.Size([1, 1000, 778]) | 6\n",
-      "15504    | torch.Size([1, 1000, 754]) | 6\n",
-      "15505    | torch.Size([1, 1000, 754]) | 6\n",
-      "15506    | torch.Size([1, 1000, 752]) | 6\n",
-      "15507    | torch.Size([1, 1000, 775]) | 6\n",
-      "15508    | torch.Size([1, 1000, 754]) | 6\n",
-      "15509    | torch.Size([1, 1000, 773]) | 6\n",
-      "15510    | torch.Size([1, 1000, 762]) | 6\n",
-      "15511    | torch.Size([1, 1000, 777]) | 6\n",
-      "15512    | torch.Size([1, 1000, 780]) | 6\n",
-      "15513    | torch.Size([1, 1000, 771]) | 6\n",
-      "15514    | torch.Size([1, 1000, 754]) | 6\n",
-      "15515    | torch.Size([1, 1000, 774]) | 6\n",
-      "15516    | torch.Size([1, 1000, 754]) | 6\n",
-      "15517    | torch.Size([1, 1000, 754]) | 6\n",
-      "15518    | torch.Size([1, 1000, 777]) | 6\n",
-      "15519    | torch.Size([1, 1000, 786]) | 6\n",
-      "15520    | torch.Size([1, 1000, 777]) | 6\n",
-      "15521    | torch.Size([1, 1000, 754]) | 6\n",
-      "15522    | torch.Size([1, 1000, 754]) | 6\n",
-      "15523    | torch.Size([1, 1000, 777]) | 6\n",
-      "15524    | torch.Size([1, 1000, 754]) | 6\n",
-      "15525    | torch.Size([1, 1000, 762]) | 6\n",
-      "15526    | torch.Size([1, 1000, 762]) | 6\n",
-      "15527    | torch.Size([1, 1000, 754]) | 6\n",
-      "15528    | torch.Size([1, 1000, 754]) | 6\n",
-      "15529    | torch.Size([1, 1000, 788]) | 6\n",
-      "15530    | torch.Size([1, 1000, 777]) | 6\n",
-      "15531    | torch.Size([1, 1000, 762]) | 6\n",
-      "15532    | torch.Size([1, 1000, 754]) | 6\n",
-      "15533    | torch.Size([1, 1000, 772]) | 6\n",
-      "15534    | torch.Size([1, 1000, 754]) | 6\n",
-      "15535    | torch.Size([1, 1000, 754]) | 6\n",
-      "15536    | torch.Size([1, 1000, 727]) | 6\n",
-      "15537    | torch.Size([1, 1000, 754]) | 6\n",
-      "15538    | torch.Size([1, 1000, 777]) | 6\n",
-      "15539    | torch.Size([1, 1000, 777]) | 6\n",
-      "15540    | torch.Size([1, 1000, 754]) | 6\n",
-      "15541    | torch.Size([1, 1000, 767]) | 6\n",
-      "15542    | torch.Size([1, 1000, 754]) | 6\n",
-      "15543    | torch.Size([1, 1000, 791]) | 6\n",
-      "15544    | torch.Size([1, 1000, 786]) | 6\n",
-      "15545    | torch.Size([1, 1000, 777]) | 6\n",
-      "15546    | torch.Size([1, 1000, 754]) | 6\n",
-      "15547    | torch.Size([1, 1000, 771]) | 6\n",
-      "15548    | torch.Size([1, 1000, 762]) | 6\n",
-      "15549    | torch.Size([1, 1000, 762]) | 6\n",
-      "15550    | torch.Size([1, 1000, 788]) | 6\n",
-      "15551    | torch.Size([1, 1000, 754]) | 6\n",
-      "15552    | torch.Size([1, 1000, 754]) | 6\n",
-      "15553    | torch.Size([1, 1000, 754]) | 6\n",
-      "15554    | torch.Size([1, 1000, 754]) | 6\n",
-      "15555    | torch.Size([1, 1000, 754]) | 6\n",
-      "15556    | torch.Size([1, 1000, 846]) | 6\n",
-      "15557    | torch.Size([1, 1000, 754]) | 6\n",
-      "15558    | torch.Size([1, 1000, 777]) | 6\n",
-      "15559    | torch.Size([1, 1000, 754]) | 6\n",
-      "15560    | torch.Size([1, 1000, 802]) | 6\n",
-      "15561    | torch.Size([1, 1000, 754]) | 6\n",
-      "15562    | torch.Size([1, 1000, 777]) | 6\n",
-      "15563    | torch.Size([1, 1000, 754]) | 6\n",
-      "15564    | torch.Size([1, 1000, 788]) | 6\n",
-      "15565    | torch.Size([1, 1000, 777]) | 6\n",
-      "15566    | torch.Size([1, 1000, 754]) | 6\n",
-      "15567    | torch.Size([1, 1000, 773]) | 6\n",
-      "15568    | torch.Size([1, 1000, 762]) | 6\n",
-      "15569    | torch.Size([1, 1000, 754]) | 6\n",
-      "15570    | torch.Size([1, 1000, 754]) | 6\n",
-      "15571    | torch.Size([1, 1000, 777]) | 6\n",
-      "15572    | torch.Size([1, 1000, 754]) | 6\n",
-      "15573    | torch.Size([1, 1000, 772]) | 6\n",
-      "15574    | torch.Size([1, 1000, 847]) | 6\n",
-      "15575    | torch.Size([1, 1000, 754]) | 6\n",
-      "15576    | torch.Size([1, 1000, 773]) | 6\n",
-      "15577    | torch.Size([1, 1000, 766]) | 6\n",
-      "15578    | torch.Size([1, 1000, 775]) | 6\n",
-      "15579    | torch.Size([1, 1000, 754]) | 6\n",
-      "15580    | torch.Size([1, 1000, 762]) | 6\n",
-      "15581    | torch.Size([1, 1000, 754]) | 6\n",
-      "15582    | torch.Size([1, 1000, 754]) | 6\n",
-      "15583    | torch.Size([1, 1000, 754]) | 6\n",
-      "15584    | torch.Size([1, 1000, 754]) | 6\n",
-      "15585    | torch.Size([1, 1000, 786]) | 6\n",
-      "15586    | torch.Size([1, 1000, 762]) | 6\n",
-      "15587    | torch.Size([1, 1000, 754]) | 6\n",
-      "15588    | torch.Size([1, 1000, 754]) | 6\n",
-      "15589    | torch.Size([1, 1000, 754]) | 6\n",
-      "15590    | torch.Size([1, 1000, 770]) | 6\n",
-      "15591    | torch.Size([1, 1000, 762]) | 6\n",
-      "15592    | torch.Size([1, 1000, 781]) | 6\n",
-      "15593    | torch.Size([1, 1000, 728]) | 6\n",
-      "15594    | torch.Size([1, 1000, 771]) | 6\n",
-      "15595    | torch.Size([1, 1000, 777]) | 6\n",
-      "15596    | torch.Size([1, 1000, 779]) | 6\n",
-      "15597    | torch.Size([1, 1000, 762]) | 6\n",
-      "15598    | torch.Size([1, 1000, 754]) | 6\n",
-      "15599    | torch.Size([1, 1000, 754]) | 6\n",
-      "15600    | torch.Size([1, 1000, 754]) | 6\n",
-      "15601    | torch.Size([1, 1000, 754]) | 6\n",
-      "15602    | torch.Size([1, 1000, 754]) | 6\n",
-      "15603    | torch.Size([1, 1000, 784]) | 6\n",
-      "15604    | torch.Size([1, 1000, 754]) | 6\n",
-      "15605    | torch.Size([1, 1000, 777]) | 6\n",
-      "15606    | torch.Size([1, 1000, 754]) | 6\n",
-      "15607    | torch.Size([1, 1000, 844]) | 6\n",
-      "15608    | torch.Size([1, 1000, 754]) | 6\n",
-      "15609    | torch.Size([1, 1000, 779]) | 6\n",
-      "15610    | torch.Size([1, 1000, 785]) | 6\n",
-      "15611    | torch.Size([1, 1000, 754]) | 6\n",
-      "15612    | torch.Size([1, 1000, 777]) | 6\n",
-      "15613    | torch.Size([1, 1000, 762]) | 6\n",
-      "15614    | torch.Size([1, 1000, 777]) | 6\n",
-      "15615    | torch.Size([1, 1000, 780]) | 6\n",
-      "15616    | torch.Size([1, 1000, 762]) | 6\n",
-      "15617    | torch.Size([1, 1000, 754]) | 6\n",
-      "15618    | torch.Size([1, 1000, 777]) | 6\n",
-      "15619    | torch.Size([1, 1000, 792]) | 6\n",
-      "15620    | torch.Size([1, 1000, 762]) | 6\n",
-      "15621    | torch.Size([1, 1000, 754]) | 6\n",
-      "15622    | torch.Size([1, 1000, 776]) | 6\n",
-      "15623    | torch.Size([1, 1000, 777]) | 6\n",
-      "15624    | torch.Size([1, 1000, 762]) | 6\n",
-      "15625    | torch.Size([1, 1000, 754]) | 6\n",
-      "15626    | torch.Size([1, 1000, 782]) | 6\n",
-      "15627    | torch.Size([1, 1000, 776]) | 6\n",
-      "15628    | torch.Size([1, 1000, 779]) | 6\n",
-      "15629    | torch.Size([1, 1000, 754]) | 6\n",
-      "15630    | torch.Size([1, 1000, 777]) | 6\n",
-      "15631    | torch.Size([1, 1000, 754]) | 6\n",
-      "15632    | torch.Size([1, 1000, 762]) | 6\n",
-      "15633    | torch.Size([1, 1000, 777]) | 6\n",
-      "15634    | torch.Size([1, 1000, 754]) | 6\n",
-      "15635    | torch.Size([1, 1000, 765]) | 6\n",
-      "15636    | torch.Size([1, 1000, 762]) | 6\n",
-      "15637    | torch.Size([1, 1000, 802]) | 6\n",
-      "15638    | torch.Size([1, 1000, 785]) | 6\n",
-      "15639    | torch.Size([1, 1000, 777]) | 6\n",
-      "15640    | torch.Size([1, 1000, 754]) | 6\n",
-      "15641    | torch.Size([1, 1000, 791]) | 6\n",
-      "15642    | torch.Size([1, 1000, 802]) | 6\n",
-      "15643    | torch.Size([1, 1000, 777]) | 6\n",
-      "15644    | torch.Size([1, 1000, 754]) | 6\n",
-      "15645    | torch.Size([1, 1000, 777]) | 6\n",
-      "15646    | torch.Size([1, 1000, 754]) | 6\n",
-      "15647    | torch.Size([1, 1000, 777]) | 6\n",
-      "15648    | torch.Size([1, 1000, 786]) | 6\n",
-      "15649    | torch.Size([1, 1000, 736]) | 6\n",
-      "15650    | torch.Size([1, 1000, 792]) | 6\n",
-      "15651    | torch.Size([1, 1000, 773]) | 6\n",
-      "15652    | torch.Size([1, 1000, 762]) | 6\n",
-      "15653    | torch.Size([1, 1000, 754]) | 6\n",
-      "15654    | torch.Size([1, 1000, 754]) | 6\n",
-      "15655    | torch.Size([1, 1000, 754]) | 6\n",
-      "15656    | torch.Size([1, 1000, 780]) | 6\n",
-      "15657    | torch.Size([1, 1000, 786]) | 6\n",
-      "15658    | torch.Size([1, 1000, 764]) | 6\n",
-      "15659    | torch.Size([1, 1000, 777]) | 6\n",
-      "15660    | torch.Size([1, 1000, 762]) | 6\n",
-      "15661    | torch.Size([1, 1000, 754]) | 6\n",
-      "15662    | torch.Size([1, 1000, 754]) | 6\n",
-      "15663    | torch.Size([1, 1000, 754]) | 6\n",
-      "15664    | torch.Size([1, 1000, 762]) | 6\n",
-      "15665    | torch.Size([1, 1000, 769]) | 6\n",
-      "15666    | torch.Size([1, 1000, 768]) | 6\n",
-      "15667    | torch.Size([1, 1000, 754]) | 6\n",
-      "15668    | torch.Size([1, 1000, 781]) | 6\n",
-      "15669    | torch.Size([1, 1000, 777]) | 6\n",
-      "15670    | torch.Size([1, 1000, 777]) | 6\n",
-      "15671    | torch.Size([1, 1000, 785]) | 6\n",
-      "15672    | torch.Size([1, 1000, 754]) | 6\n",
-      "15673    | torch.Size([1, 1000, 754]) | 6\n",
-      "15674    | torch.Size([1, 1000, 754]) | 6\n",
-      "15675    | torch.Size([1, 1000, 786]) | 6\n",
-      "15676    | torch.Size([1, 1000, 802]) | 6\n",
-      "15677    | torch.Size([1, 1000, 772]) | 6\n",
-      "15678    | torch.Size([1, 1000, 754]) | 6\n",
-      "15679    | torch.Size([1, 1000, 772]) | 6\n",
-      "15680    | torch.Size([1, 1000, 777]) | 6\n",
-      "15681    | torch.Size([1, 1000, 772]) | 6\n",
-      "15682    | torch.Size([1, 1000, 754]) | 6\n",
-      "15683    | torch.Size([1, 1000, 754]) | 6\n",
-      "15684    | torch.Size([1, 1000, 776]) | 6\n",
-      "15685    | torch.Size([1, 1000, 762]) | 6\n",
-      "15686    | torch.Size([1, 1000, 754]) | 6\n",
-      "15687    | torch.Size([1, 1000, 763]) | 6\n",
-      "15688    | torch.Size([1, 1000, 777]) | 6\n",
-      "15689    | torch.Size([1, 1000, 754]) | 6\n",
-      "15690    | torch.Size([1, 1000, 768]) | 6\n",
-      "15691    | torch.Size([1, 1000, 754]) | 6\n",
-      "15692    | torch.Size([1, 1000, 768]) | 6\n",
-      "15693    | torch.Size([1, 1000, 762]) | 6\n",
-      "15694    | torch.Size([1, 1000, 754]) | 6\n",
-      "15695    | torch.Size([1, 1000, 754]) | 6\n",
-      "15696    | torch.Size([1, 1000, 754]) | 6\n",
-      "15697    | torch.Size([1, 1000, 777]) | 6\n",
-      "15698    | torch.Size([1, 1000, 766]) | 6\n",
-      "15699    | torch.Size([1, 1000, 762]) | 6\n",
-      "15700    | torch.Size([1, 1000, 777]) | 6\n",
-      "15701    | torch.Size([1, 1000, 723]) | 6\n",
-      "15702    | torch.Size([1, 1000, 772]) | 6\n",
-      "15703    | torch.Size([1, 1000, 762]) | 6\n",
-      "15704    | torch.Size([1, 1000, 770]) | 6\n",
-      "15705    | torch.Size([1, 1000, 815]) | 6\n",
-      "15706    | torch.Size([1, 1000, 778]) | 6\n",
-      "15707    | torch.Size([1, 1000, 777]) | 6\n",
-      "15708    | torch.Size([1, 1000, 808]) | 6\n",
-      "15709    | torch.Size([1, 1000, 754]) | 6\n",
-      "15710    | torch.Size([1, 1000, 777]) | 6\n",
-      "15711    | torch.Size([1, 1000, 762]) | 6\n",
-      "15712    | torch.Size([1, 1000, 778]) | 6\n",
-      "15713    | torch.Size([1, 1000, 769]) | 6\n",
-      "15714    | torch.Size([1, 1000, 778]) | 6\n",
-      "15715    | torch.Size([1, 1000, 802]) | 6\n",
-      "15716    | torch.Size([1, 1000, 777]) | 6\n",
-      "15717    | torch.Size([1, 1000, 754]) | 6\n",
-      "15718    | torch.Size([1, 1000, 754]) | 6\n",
-      "15719    | torch.Size([1, 1000, 778]) | 6\n",
-      "15720    | torch.Size([1, 1000, 754]) | 6\n",
-      "15721    | torch.Size([1, 1000, 754]) | 6\n",
-      "15722    | torch.Size([1, 1000, 754]) | 6\n",
-      "15723    | torch.Size([1, 1000, 777]) | 6\n",
-      "15724    | torch.Size([1, 1000, 762]) | 6\n",
-      "15725    | torch.Size([1, 1000, 777]) | 6\n",
-      "15726    | torch.Size([1, 1000, 754]) | 6\n",
-      "15727    | torch.Size([1, 1000, 754]) | 6\n",
-      "15728    | torch.Size([1, 1000, 777]) | 6\n",
-      "15729    | torch.Size([1, 1000, 777]) | 6\n",
-      "15730    | torch.Size([1, 1000, 772]) | 6\n",
-      "15731    | torch.Size([1, 1000, 777]) | 6\n",
-      "15732    | torch.Size([1, 1000, 754]) | 6\n",
-      "15733    | torch.Size([1, 1000, 762]) | 6\n",
-      "15734    | torch.Size([1, 1000, 763]) | 6\n",
-      "15735    | torch.Size([1, 1000, 754]) | 6\n",
-      "15736    | torch.Size([1, 1000, 762]) | 6\n",
-      "15737    | torch.Size([1, 1000, 754]) | 6\n",
-      "15738    | torch.Size([1, 1000, 777]) | 6\n",
-      "15739    | torch.Size([1, 1000, 754]) | 6\n",
-      "15740    | torch.Size([1, 1000, 754]) | 6\n",
-      "15741    | torch.Size([1, 1000, 777]) | 6\n",
-      "15742    | torch.Size([1, 1000, 766]) | 6\n",
-      "15743    | torch.Size([1, 1000, 775]) | 6\n",
-      "15744    | torch.Size([1, 1000, 754]) | 6\n",
-      "15745    | torch.Size([1, 1000, 785]) | 6\n",
-      "15746    | torch.Size([1, 1000, 782]) | 6\n",
-      "15747    | torch.Size([1, 1000, 770]) | 6\n",
-      "15748    | torch.Size([1, 1000, 754]) | 6\n",
-      "15749    | torch.Size([1, 1000, 754]) | 6\n",
-      "15750    | torch.Size([1, 1000, 754]) | 6\n",
-      "15751    | torch.Size([1, 1000, 754]) | 6\n",
-      "15752    | torch.Size([1, 1000, 777]) | 6\n",
-      "15753    | torch.Size([1, 1000, 777]) | 6\n",
-      "15754    | torch.Size([1, 1000, 754]) | 6\n",
-      "15755    | torch.Size([1, 1000, 777]) | 6\n",
-      "15756    | torch.Size([1, 1000, 771]) | 6\n",
-      "15757    | torch.Size([1, 1000, 762]) | 6\n",
-      "15758    | torch.Size([1, 1000, 754]) | 6\n",
-      "15759    | torch.Size([1, 1000, 786]) | 6\n",
-      "15760    | torch.Size([1, 1000, 786]) | 6\n",
-      "15761    | torch.Size([1, 1000, 777]) | 6\n",
-      "15762    | torch.Size([1, 1000, 776]) | 6\n",
-      "15763    | torch.Size([1, 1000, 754]) | 6\n",
-      "15764    | torch.Size([1, 1000, 754]) | 6\n",
-      "15765    | torch.Size([1, 1000, 778]) | 6\n",
-      "15766    | torch.Size([1, 1000, 785]) | 6\n",
-      "15767    | torch.Size([1, 1000, 776]) | 6\n",
-      "15768    | torch.Size([1, 1000, 777]) | 6\n",
-      "15769    | torch.Size([1, 1000, 770]) | 6\n",
-      "15770    | torch.Size([1, 1000, 754]) | 6\n",
-      "15771    | torch.Size([1, 1000, 773]) | 6\n",
-      "15772    | torch.Size([1, 1000, 776]) | 6\n",
-      "15773    | torch.Size([1, 1000, 767]) | 6\n",
-      "15774    | torch.Size([1, 1000, 752]) | 6\n",
-      "15775    | torch.Size([1, 1000, 754]) | 6\n",
-      "15776    | torch.Size([1, 1000, 777]) | 6\n",
-      "15777    | torch.Size([1, 1000, 762]) | 6\n",
-      "15778    | torch.Size([1, 1000, 754]) | 6\n",
-      "15779    | torch.Size([1, 1000, 773]) | 6\n",
-      "15780    | torch.Size([1, 1000, 771]) | 6\n",
-      "15781    | torch.Size([1, 1000, 754]) | 6\n",
-      "15782    | torch.Size([1, 1000, 768]) | 6\n",
-      "15783    | torch.Size([1, 1000, 802]) | 6\n",
-      "15784    | torch.Size([1, 1000, 754]) | 6\n",
-      "15785    | torch.Size([1, 1000, 762]) | 6\n",
-      "15786    | torch.Size([1, 1000, 777]) | 6\n",
-      "15787    | torch.Size([1, 1000, 777]) | 6\n",
-      "15788    | torch.Size([1, 1000, 777]) | 6\n",
-      "15789    | torch.Size([1, 1000, 754]) | 6\n",
-      "15790    | torch.Size([1, 1000, 754]) | 6\n",
-      "15791    | torch.Size([1, 1000, 777]) | 6\n",
-      "15792    | torch.Size([1, 1000, 754]) | 6\n",
-      "15793    | torch.Size([1, 1000, 777]) | 6\n",
-      "15794    | torch.Size([1, 1000, 781]) | 6\n",
-      "15795    | torch.Size([1, 1000, 754]) | 6\n",
-      "15796    | torch.Size([1, 1000, 777]) | 6\n",
-      "15797    | torch.Size([1, 1000, 754]) | 6\n",
-      "15798    | torch.Size([1, 1000, 777]) | 6\n",
-      "15799    | torch.Size([1, 1000, 781]) | 6\n",
-      "15800    | torch.Size([1, 1000, 754]) | 6\n",
-      "15801    | torch.Size([1, 1000, 754]) | 6\n",
-      "15802    | torch.Size([1, 1000, 770]) | 6\n",
-      "15803    | torch.Size([1, 1000, 754]) | 6\n",
-      "15804    | torch.Size([1, 1000, 762]) | 6\n",
-      "15805    | torch.Size([1, 1000, 754]) | 6\n",
-      "15806    | torch.Size([1, 1000, 752]) | 6\n",
-      "15807    | torch.Size([1, 1000, 802]) | 6\n",
-      "15808    | torch.Size([1, 1000, 777]) | 6\n",
-      "15809    | torch.Size([1, 1000, 762]) | 6\n",
-      "15810    | torch.Size([1, 1000, 754]) | 6\n",
-      "15811    | torch.Size([1, 1000, 754]) | 6\n",
-      "15812    | torch.Size([1, 1000, 754]) | 6\n",
-      "15813    | torch.Size([1, 1000, 771]) | 6\n",
-      "15814    | torch.Size([1, 1000, 781]) | 6\n",
-      "15815    | torch.Size([1, 1000, 769]) | 6\n",
-      "15816    | torch.Size([1, 1000, 770]) | 6\n",
-      "15817    | torch.Size([1, 1000, 754]) | 6\n",
-      "15818    | torch.Size([1, 1000, 777]) | 6\n",
-      "15819    | torch.Size([1, 1000, 752]) | 6\n",
-      "15820    | torch.Size([1, 1000, 754]) | 6\n",
-      "15821    | torch.Size([1, 1000, 802]) | 6\n",
-      "15822    | torch.Size([1, 1000, 752]) | 6\n",
-      "15823    | torch.Size([1, 1000, 762]) | 6\n",
-      "15824    | torch.Size([1, 1000, 777]) | 6\n",
-      "15825    | torch.Size([1, 1000, 769]) | 6\n",
-      "15826    | torch.Size([1, 1000, 790]) | 6\n",
-      "15827    | torch.Size([1, 1000, 769]) | 6\n",
-      "15828    | torch.Size([1, 1000, 754]) | 6\n",
-      "15829    | torch.Size([1, 1000, 782]) | 6\n",
-      "15830    | torch.Size([1, 1000, 777]) | 6\n",
-      "15831    | torch.Size([1, 1000, 754]) | 6\n",
-      "15832    | torch.Size([1, 1000, 754]) | 6\n",
-      "15833    | torch.Size([1, 1000, 754]) | 6\n",
-      "15834    | torch.Size([1, 1000, 762]) | 6\n",
-      "15835    | torch.Size([1, 1000, 754]) | 6\n",
-      "15836    | torch.Size([1, 1000, 772]) | 6\n",
-      "15837    | torch.Size([1, 1000, 784]) | 6\n",
-      "15838    | torch.Size([1, 1000, 754]) | 6\n",
-      "15839    | torch.Size([1, 1000, 754]) | 6\n",
-      "15840    | torch.Size([1, 1000, 738]) | 6\n",
-      "15841    | torch.Size([1, 1000, 776]) | 6\n",
-      "15842    | torch.Size([1, 1000, 777]) | 6\n",
-      "15843    | torch.Size([1, 1000, 754]) | 6\n",
-      "15844    | torch.Size([1, 1000, 754]) | 6\n",
-      "15845    | torch.Size([1, 1000, 784]) | 6\n",
-      "15846    | torch.Size([1, 1000, 754]) | 6\n",
-      "15847    | torch.Size([1, 1000, 779]) | 6\n",
-      "15848    | torch.Size([1, 1000, 806]) | 6\n",
-      "15849    | torch.Size([1, 1000, 754]) | 6\n",
-      "15850    | torch.Size([1, 1000, 777]) | 6\n",
-      "15851    | torch.Size([1, 1000, 774]) | 6\n",
-      "15852    | torch.Size([1, 1000, 771]) | 6\n",
-      "15853    | torch.Size([1, 1000, 772]) | 6\n",
-      "15854    | torch.Size([1, 1000, 808]) | 6\n",
-      "15855    | torch.Size([1, 1000, 777]) | 6\n",
-      "15856    | torch.Size([1, 1000, 762]) | 6\n",
-      "15857    | torch.Size([1, 1000, 754]) | 6\n",
-      "15858    | torch.Size([1, 1000, 754]) | 6\n",
-      "15859    | torch.Size([1, 1000, 777]) | 6\n",
-      "15860    | torch.Size([1, 1000, 786]) | 6\n",
-      "15861    | torch.Size([1, 1000, 762]) | 6\n",
-      "15862    | torch.Size([1, 1000, 754]) | 6\n",
-      "15863    | torch.Size([1, 1000, 772]) | 6\n",
-      "15864    | torch.Size([1, 1000, 762]) | 6\n",
-      "15865    | torch.Size([1, 1000, 754]) | 6\n",
-      "15866    | torch.Size([1, 1000, 754]) | 6\n",
-      "15867    | torch.Size([1, 1000, 754]) | 6\n",
-      "15868    | torch.Size([1, 1000, 777]) | 6\n",
-      "15869    | torch.Size([1, 1000, 777]) | 6\n",
-      "15870    | torch.Size([1, 1000, 769]) | 6\n",
-      "15871    | torch.Size([1, 1000, 752]) | 6\n",
-      "15872    | torch.Size([1, 1000, 777]) | 6\n",
-      "15873    | torch.Size([1, 1000, 754]) | 6\n",
-      "15874    | torch.Size([1, 1000, 777]) | 6\n",
-      "15875    | torch.Size([1, 1000, 754]) | 6\n",
-      "15876    | torch.Size([1, 1000, 773]) | 6\n",
-      "15877    | torch.Size([1, 1000, 754]) | 6\n",
-      "15878    | torch.Size([1, 1000, 779]) | 6\n",
-      "15879    | torch.Size([1, 1000, 754]) | 6\n",
-      "15880    | torch.Size([1, 1000, 775]) | 6\n",
-      "15881    | torch.Size([1, 1000, 754]) | 6\n",
-      "15882    | torch.Size([1, 1000, 777]) | 6\n",
-      "15883    | torch.Size([1, 1000, 793]) | 6\n",
-      "15884    | torch.Size([1, 1000, 777]) | 6\n",
-      "15885    | torch.Size([1, 1000, 777]) | 6\n",
-      "15886    | torch.Size([1, 1000, 752]) | 6\n",
-      "15887    | torch.Size([1, 1000, 754]) | 6\n",
-      "15888    | torch.Size([1, 1000, 771]) | 6\n",
-      "15889    | torch.Size([1, 1000, 754]) | 6\n",
-      "15890    | torch.Size([1, 1000, 754]) | 6\n",
-      "15891    | torch.Size([1, 1000, 772]) | 6\n",
-      "15892    | torch.Size([1, 1000, 777]) | 6\n",
-      "15893    | torch.Size([1, 1000, 754]) | 6\n",
-      "15894    | torch.Size([1, 1000, 754]) | 6\n",
-      "15895    | torch.Size([1, 1000, 777]) | 6\n",
-      "15896    | torch.Size([1, 1000, 754]) | 6\n",
-      "15897    | torch.Size([1, 1000, 821]) | 6\n",
-      "15898    | torch.Size([1, 1000, 777]) | 6\n",
-      "15899    | torch.Size([1, 1000, 784]) | 6\n",
-      "15900    | torch.Size([1, 1000, 759]) | 6\n",
-      "15901    | torch.Size([1, 1000, 754]) | 6\n",
-      "15902    | torch.Size([1, 1000, 754]) | 6\n",
-      "15903    | torch.Size([1, 1000, 767]) | 6\n",
-      "15904    | torch.Size([1, 1000, 762]) | 6\n",
-      "15905    | torch.Size([1, 1000, 777]) | 6\n",
-      "15906    | torch.Size([1, 1000, 780]) | 6\n",
-      "15907    | torch.Size([1, 1000, 768]) | 6\n",
-      "15908    | torch.Size([1, 1000, 771]) | 6\n",
-      "15909    | torch.Size([1, 1000, 762]) | 6\n",
-      "15910    | torch.Size([1, 1000, 754]) | 6\n",
-      "15911    | torch.Size([1, 1000, 762]) | 6\n",
-      "15912    | torch.Size([1, 1000, 777]) | 6\n",
-      "15913    | torch.Size([1, 1000, 754]) | 6\n",
-      "15914    | torch.Size([1, 1000, 762]) | 6\n",
-      "15915    | torch.Size([1, 1000, 777]) | 6\n",
-      "15916    | torch.Size([1, 1000, 818]) | 6\n",
-      "15917    | torch.Size([1, 1000, 772]) | 6\n",
-      "15918    | torch.Size([1, 1000, 787]) | 6\n",
-      "15919    | torch.Size([1, 1000, 754]) | 6\n",
-      "15920    | torch.Size([1, 1000, 763]) | 6\n",
-      "15921    | torch.Size([1, 1000, 771]) | 6\n",
-      "15922    | torch.Size([1, 1000, 754]) | 6\n",
-      "15923    | torch.Size([1, 1000, 762]) | 6\n",
-      "15924    | torch.Size([1, 1000, 777]) | 6\n",
-      "15925    | torch.Size([1, 1000, 783]) | 6\n",
-      "15926    | torch.Size([1, 1000, 762]) | 6\n",
-      "15927    | torch.Size([1, 1000, 754]) | 6\n",
-      "15928    | torch.Size([1, 1000, 754]) | 6\n",
-      "15929    | torch.Size([1, 1000, 754]) | 6\n",
-      "15930    | torch.Size([1, 1000, 766]) | 6\n",
-      "15931    | torch.Size([1, 1000, 754]) | 6\n",
-      "15932    | torch.Size([1, 1000, 762]) | 6\n",
-      "15933    | torch.Size([1, 1000, 754]) | 6\n",
-      "15934    | torch.Size([1, 1000, 792]) | 6\n",
-      "15935    | torch.Size([1, 1000, 754]) | 6\n",
-      "15936    | torch.Size([1, 1000, 777]) | 6\n",
-      "15937    | torch.Size([1, 1000, 773]) | 6\n",
-      "15938    | torch.Size([1, 1000, 797]) | 6\n",
-      "15939    | torch.Size([1, 1000, 754]) | 6\n",
-      "15940    | torch.Size([1, 1000, 754]) | 6\n",
-      "15941    | torch.Size([1, 1000, 803]) | 6\n",
-      "15942    | torch.Size([1, 1000, 754]) | 6\n",
-      "15943    | torch.Size([1, 1000, 754]) | 6\n",
-      "15944    | torch.Size([1, 1000, 762]) | 6\n",
-      "15945    | torch.Size([1, 1000, 762]) | 6\n",
-      "15946    | torch.Size([1, 1000, 754]) | 6\n",
-      "15947    | torch.Size([1, 1000, 754]) | 6\n",
-      "15948    | torch.Size([1, 1000, 729]) | 6\n",
-      "15949    | torch.Size([1, 1000, 754]) | 6\n",
-      "15950    | torch.Size([1, 1000, 754]) | 6\n",
-      "15951    | torch.Size([1, 1000, 754]) | 6\n",
-      "15952    | torch.Size([1, 1000, 765]) | 6\n",
-      "15953    | torch.Size([1, 1000, 773]) | 6\n",
-      "15954    | torch.Size([1, 1000, 733]) | 6\n",
-      "15955    | torch.Size([1, 1000, 763]) | 6\n",
-      "15956    | torch.Size([1, 1000, 754]) | 6\n",
-      "15957    | torch.Size([1, 1000, 804]) | 6\n",
-      "15958    | torch.Size([1, 1000, 756]) | 6\n",
-      "15959    | torch.Size([1, 1000, 777]) | 6\n",
-      "15960    | torch.Size([1, 1000, 777]) | 6\n",
-      "15961    | torch.Size([1, 1000, 777]) | 6\n",
-      "15962    | torch.Size([1, 1000, 777]) | 6\n",
-      "15963    | torch.Size([1, 1000, 762]) | 6\n",
-      "15964    | torch.Size([1, 1000, 754]) | 6\n",
-      "15965    | torch.Size([1, 1000, 754]) | 6\n",
-      "15966    | torch.Size([1, 1000, 777]) | 6\n",
-      "15967    | torch.Size([1, 1000, 777]) | 6\n",
-      "15968    | torch.Size([1, 1000, 776]) | 6\n",
-      "15969    | torch.Size([1, 1000, 754]) | 6\n",
-      "15970    | torch.Size([1, 1000, 754]) | 6\n",
-      "15971    | torch.Size([1, 1000, 762]) | 6\n",
-      "15972    | torch.Size([1, 1000, 783]) | 6\n",
-      "15973    | torch.Size([1, 1000, 754]) | 6\n",
-      "15974    | torch.Size([1, 1000, 762]) | 6\n",
-      "15975    | torch.Size([1, 1000, 754]) | 6\n",
-      "15976    | torch.Size([1, 1000, 754]) | 6\n",
-      "15977    | torch.Size([1, 1000, 762]) | 6\n",
-      "15978    | torch.Size([1, 1000, 785]) | 6\n",
-      "15979    | torch.Size([1, 1000, 774]) | 6\n",
-      "15980    | torch.Size([1, 1000, 769]) | 6\n",
-      "15981    | torch.Size([1, 1000, 808]) | 6\n",
-      "15982    | torch.Size([1, 1000, 777]) | 6\n",
-      "15983    | torch.Size([1, 1000, 762]) | 6\n",
-      "15984    | torch.Size([1, 1000, 754]) | 6\n",
-      "15985    | torch.Size([1, 1000, 754]) | 6\n",
-      "15986    | torch.Size([1, 1000, 802]) | 6\n",
-      "15987    | torch.Size([1, 1000, 729]) | 6\n",
-      "15988    | torch.Size([1, 1000, 777]) | 6\n",
-      "15989    | torch.Size([1, 1000, 754]) | 6\n",
-      "15990    | torch.Size([1, 1000, 807]) | 6\n",
-      "15991    | torch.Size([1, 1000, 754]) | 6\n",
-      "15992    | torch.Size([1, 1000, 775]) | 6\n",
-      "15993    | torch.Size([1, 1000, 777]) | 6\n",
-      "15994    | torch.Size([1, 1000, 781]) | 6\n",
-      "15995    | torch.Size([1, 1000, 754]) | 6\n",
-      "15996    | torch.Size([1, 1000, 754]) | 6\n",
-      "15997    | torch.Size([1, 1000, 777]) | 6\n",
-      "15998    | torch.Size([1, 1000, 771]) | 6\n",
-      "15999    | torch.Size([1, 1000, 777]) | 6\n",
-      "16000    | torch.Size([1, 1000, 773]) | 6\n",
-      "16001    | torch.Size([1, 1000, 754]) | 6\n",
-      "16002    | torch.Size([1, 1000, 762]) | 6\n",
-      "16003    | torch.Size([1, 1000, 779]) | 6\n",
-      "16004    | torch.Size([1, 1000, 754]) | 6\n",
-      "16005    | torch.Size([1, 1000, 754]) | 6\n",
-      "16006    | torch.Size([1, 1000, 767]) | 6\n",
-      "16007    | torch.Size([1, 1000, 754]) | 6\n",
-      "16008    | torch.Size([1, 1000, 762]) | 6\n",
-      "16009    | torch.Size([1, 1000, 771]) | 6\n",
-      "16010    | torch.Size([1, 1000, 754]) | 6\n",
-      "16011    | torch.Size([1, 1000, 754]) | 6\n",
-      "16012    | torch.Size([1, 1000, 777]) | 6\n",
-      "16013    | torch.Size([1, 1000, 770]) | 6\n",
-      "16014    | torch.Size([1, 1000, 754]) | 6\n",
-      "16015    | torch.Size([1, 1000, 762]) | 6\n",
-      "16016    | torch.Size([1, 1000, 762]) | 6\n",
-      "16017    | torch.Size([1, 1000, 769]) | 6\n",
-      "16018    | torch.Size([1, 1000, 806]) | 6\n",
-      "16019    | torch.Size([1, 1000, 802]) | 6\n",
-      "16020    | torch.Size([1, 1000, 754]) | 6\n",
-      "16021    | torch.Size([1, 1000, 754]) | 6\n",
-      "16022    | torch.Size([1, 1000, 754]) | 6\n",
-      "16023    | torch.Size([1, 1000, 754]) | 6\n",
-      "16024    | torch.Size([1, 1000, 768]) | 6\n",
-      "16025    | torch.Size([1, 1000, 754]) | 6\n",
-      "16026    | torch.Size([1, 1000, 754]) | 6\n",
-      "16027    | torch.Size([1, 1000, 785]) | 6\n",
-      "16028    | torch.Size([1, 1000, 802]) | 6\n",
-      "16029    | torch.Size([1, 1000, 762]) | 6\n",
-      "16030    | torch.Size([1, 1000, 754]) | 6\n",
-      "16031    | torch.Size([1, 1000, 765]) | 6\n",
-      "16032    | torch.Size([1, 1000, 762]) | 6\n",
-      "16033    | torch.Size([1, 1000, 754]) | 6\n",
-      "16034    | torch.Size([1, 1000, 790]) | 6\n",
-      "16035    | torch.Size([1, 1000, 777]) | 6\n",
-      "16036    | torch.Size([1, 1000, 762]) | 6\n",
-      "16037    | torch.Size([1, 1000, 777]) | 6\n",
-      "16038    | torch.Size([1, 1000, 773]) | 6\n",
-      "16039    | torch.Size([1, 1000, 754]) | 6\n",
-      "16040    | torch.Size([1, 1000, 754]) | 6\n",
-      "16041    | torch.Size([1, 1000, 762]) | 6\n",
-      "16042    | torch.Size([1, 1000, 762]) | 6\n",
-      "16043    | torch.Size([1, 1000, 754]) | 6\n",
-      "16044    | torch.Size([1, 1000, 754]) | 6\n",
-      "16045    | torch.Size([1, 1000, 754]) | 6\n",
-      "16046    | torch.Size([1, 1000, 754]) | 6\n",
-      "16047    | torch.Size([1, 1000, 754]) | 6\n",
-      "16048    | torch.Size([1, 1000, 754]) | 6\n",
-      "16049    | torch.Size([1, 1000, 762]) | 6\n",
-      "16050    | torch.Size([1, 1000, 754]) | 6\n",
-      "16051    | torch.Size([1, 1000, 777]) | 6\n",
-      "16052    | torch.Size([1, 1000, 771]) | 6\n",
-      "16053    | torch.Size([1, 1000, 777]) | 6\n",
-      "16054    | torch.Size([1, 1000, 805]) | 6\n",
-      "16055    | torch.Size([1, 1000, 754]) | 6\n",
-      "16056    | torch.Size([1, 1000, 777]) | 6\n",
-      "16057    | torch.Size([1, 1000, 785]) | 6\n",
-      "16058    | torch.Size([1, 1000, 777]) | 6\n",
-      "16059    | torch.Size([1, 1000, 754]) | 6\n",
-      "16060    | torch.Size([1, 1000, 762]) | 6\n",
-      "16061    | torch.Size([1, 1000, 777]) | 6\n",
-      "16062    | torch.Size([1, 1000, 762]) | 6\n",
-      "16063    | torch.Size([1, 1000, 775]) | 6\n",
-      "16064    | torch.Size([1, 1000, 777]) | 6\n",
-      "16065    | torch.Size([1, 1000, 754]) | 6\n",
-      "16066    | torch.Size([1, 1000, 787]) | 6\n",
-      "16067    | torch.Size([1, 1000, 792]) | 6\n",
-      "16068    | torch.Size([1, 1000, 752]) | 6\n",
-      "16069    | torch.Size([1, 1000, 754]) | 6\n",
-      "16070    | torch.Size([1, 1000, 786]) | 6\n",
-      "16071    | torch.Size([1, 1000, 776]) | 6\n",
-      "16072    | torch.Size([1, 1000, 754]) | 6\n",
-      "16073    | torch.Size([1, 1000, 780]) | 6\n",
-      "16074    | torch.Size([1, 1000, 754]) | 6\n",
-      "16075    | torch.Size([1, 1000, 780]) | 6\n",
-      "16076    | torch.Size([1, 1000, 775]) | 6\n",
-      "16077    | torch.Size([1, 1000, 777]) | 6\n",
-      "16078    | torch.Size([1, 1000, 777]) | 6\n",
-      "16079    | torch.Size([1, 1000, 778]) | 6\n",
-      "16080    | torch.Size([1, 1000, 734]) | 6\n",
-      "16081    | torch.Size([1, 1000, 762]) | 6\n",
-      "16082    | torch.Size([1, 1000, 777]) | 6\n",
-      "16083    | torch.Size([1, 1000, 762]) | 6\n",
-      "16084    | torch.Size([1, 1000, 738]) | 6\n",
-      "16085    | torch.Size([1, 1000, 762]) | 6\n",
-      "16086    | torch.Size([1, 1000, 772]) | 6\n",
-      "16087    | torch.Size([1, 1000, 777]) | 6\n",
-      "16088    | torch.Size([1, 1000, 777]) | 6\n",
-      "16089    | torch.Size([1, 1000, 754]) | 6\n",
-      "16090    | torch.Size([1, 1000, 772]) | 6\n",
-      "16091    | torch.Size([1, 1000, 754]) | 6\n",
-      "16092    | torch.Size([1, 1000, 754]) | 6\n",
-      "16093    | torch.Size([1, 1000, 777]) | 6\n",
-      "16094    | torch.Size([1, 1000, 754]) | 6\n",
-      "16095    | torch.Size([1, 1000, 754]) | 6\n",
-      "16096    | torch.Size([1, 1000, 771]) | 6\n",
-      "16097    | torch.Size([1, 1000, 754]) | 6\n",
-      "16098    | torch.Size([1, 1000, 754]) | 6\n",
-      "16099    | torch.Size([1, 1000, 754]) | 6\n",
-      "16100    | torch.Size([1, 1000, 762]) | 6\n",
-      "16101    | torch.Size([1, 1000, 760]) | 6\n",
-      "16102    | torch.Size([1, 1000, 754]) | 6\n",
-      "16103    | torch.Size([1, 1000, 777]) | 6\n",
-      "16104    | torch.Size([1, 1000, 754]) | 6\n",
-      "16105    | torch.Size([1, 1000, 762]) | 6\n",
-      "16106    | torch.Size([1, 1000, 762]) | 6\n",
-      "16107    | torch.Size([1, 1000, 766]) | 6\n",
-      "16108    | torch.Size([1, 1000, 777]) | 6\n",
-      "16109    | torch.Size([1, 1000, 754]) | 6\n",
-      "16110    | torch.Size([1, 1000, 754]) | 6\n",
-      "16111    | torch.Size([1, 1000, 777]) | 6\n",
-      "16112    | torch.Size([1, 1000, 777]) | 6\n",
-      "16113    | torch.Size([1, 1000, 754]) | 6\n",
-      "16114    | torch.Size([1, 1000, 754]) | 6\n",
-      "16115    | torch.Size([1, 1000, 778]) | 6\n",
-      "16116    | torch.Size([1, 1000, 754]) | 6\n",
-      "16117    | torch.Size([1, 1000, 754]) | 6\n",
-      "16118    | torch.Size([1, 1000, 754]) | 6\n",
-      "16119    | torch.Size([1, 1000, 762]) | 6\n",
-      "16120    | torch.Size([1, 1000, 762]) | 6\n",
-      "16121    | torch.Size([1, 1000, 754]) | 6\n",
-      "16122    | torch.Size([1, 1000, 802]) | 6\n",
-      "16123    | torch.Size([1, 1000, 754]) | 6\n",
-      "16124    | torch.Size([1, 1000, 762]) | 6\n",
-      "16125    | torch.Size([1, 1000, 777]) | 6\n",
-      "16126    | torch.Size([1, 1000, 788]) | 6\n",
-      "16127    | torch.Size([1, 1000, 772]) | 6\n",
-      "16128    | torch.Size([1, 1000, 747]) | 6\n",
-      "16129    | torch.Size([1, 1000, 754]) | 6\n",
-      "16130    | torch.Size([1, 1000, 771]) | 6\n",
-      "16131    | torch.Size([1, 1000, 763]) | 6\n",
-      "16132    | torch.Size([1, 1000, 775]) | 6\n",
-      "16133    | torch.Size([1, 1000, 762]) | 6\n",
-      "16134    | torch.Size([1, 1000, 779]) | 6\n",
-      "16135    | torch.Size([1, 1000, 762]) | 6\n",
-      "16136    | torch.Size([1, 1000, 777]) | 6\n",
-      "16137    | torch.Size([1, 1000, 777]) | 6\n",
-      "16138    | torch.Size([1, 1000, 784]) | 6\n",
-      "16139    | torch.Size([1, 1000, 802]) | 6\n",
-      "16140    | torch.Size([1, 1000, 802]) | 6\n",
-      "16141    | torch.Size([1, 1000, 762]) | 6\n",
-      "16142    | torch.Size([1, 1000, 770]) | 6\n",
-      "16143    | torch.Size([1, 1000, 777]) | 6\n",
-      "16144    | torch.Size([1, 1000, 777]) | 6\n",
-      "16145    | torch.Size([1, 1000, 827]) | 6\n",
-      "16146    | torch.Size([1, 1000, 769]) | 6\n",
-      "16147    | torch.Size([1, 1000, 754]) | 6\n",
-      "16148    | torch.Size([1, 1000, 754]) | 6\n",
-      "16149    | torch.Size([1, 1000, 754]) | 6\n",
-      "16150    | torch.Size([1, 1000, 777]) | 6\n",
-      "16151    | torch.Size([1, 1000, 754]) | 6\n",
-      "16152    | torch.Size([1, 1000, 754]) | 6\n",
-      "16153    | torch.Size([1, 1000, 786]) | 6\n",
-      "16154    | torch.Size([1, 1000, 802]) | 6\n",
-      "16155    | torch.Size([1, 1000, 777]) | 6\n",
-      "16156    | torch.Size([1, 1000, 762]) | 6\n",
-      "16157    | torch.Size([1, 1000, 786]) | 6\n",
-      "16158    | torch.Size([1, 1000, 777]) | 6\n",
-      "16159    | torch.Size([1, 1000, 777]) | 6\n",
-      "16160    | torch.Size([1, 1000, 786]) | 6\n",
-      "16161    | torch.Size([1, 1000, 762]) | 6\n",
-      "16162    | torch.Size([1, 1000, 777]) | 6\n",
-      "16163    | torch.Size([1, 1000, 754]) | 6\n",
-      "16164    | torch.Size([1, 1000, 781]) | 6\n",
-      "16165    | torch.Size([1, 1000, 780]) | 6\n",
-      "16166    | torch.Size([1, 1000, 777]) | 6\n",
-      "16167    | torch.Size([1, 1000, 782]) | 6\n",
-      "16168    | torch.Size([1, 1000, 785]) | 6\n",
-      "16169    | torch.Size([1, 1000, 770]) | 6\n",
-      "16170    | torch.Size([1, 1000, 754]) | 6\n",
-      "16171    | torch.Size([1, 1000, 752]) | 6\n",
-      "16172    | torch.Size([1, 1000, 779]) | 6\n",
-      "16173    | torch.Size([1, 1000, 754]) | 6\n",
-      "16174    | torch.Size([1, 1000, 777]) | 6\n",
-      "16175    | torch.Size([1, 1000, 777]) | 6\n",
-      "16176    | torch.Size([1, 1000, 754]) | 6\n",
-      "16177    | torch.Size([1, 1000, 754]) | 6\n",
-      "16178    | torch.Size([1, 1000, 754]) | 6\n",
-      "16179    | torch.Size([1, 1000, 777]) | 6\n",
-      "16180    | torch.Size([1, 1000, 788]) | 6\n",
-      "16181    | torch.Size([1, 1000, 807]) | 6\n",
-      "16182    | torch.Size([1, 1000, 754]) | 6\n",
-      "16183    | torch.Size([1, 1000, 754]) | 6\n",
-      "16184    | torch.Size([1, 1000, 777]) | 6\n",
-      "16185    | torch.Size([1, 1000, 754]) | 6\n",
-      "16186    | torch.Size([1, 1000, 754]) | 6\n",
-      "16187    | torch.Size([1, 1000, 793]) | 6\n",
-      "16188    | torch.Size([1, 1000, 776]) | 6\n",
-      "16189    | torch.Size([1, 1000, 777]) | 6\n",
-      "16190    | torch.Size([1, 1000, 754]) | 6\n",
-      "16191    | torch.Size([1, 1000, 730]) | 6\n",
-      "16192    | torch.Size([1, 1000, 777]) | 6\n",
-      "16193    | torch.Size([1, 1000, 752]) | 6\n",
-      "16194    | torch.Size([1, 1000, 754]) | 6\n",
-      "16195    | torch.Size([1, 1000, 754]) | 6\n",
-      "16196    | torch.Size([1, 1000, 786]) | 6\n",
-      "16197    | torch.Size([1, 1000, 770]) | 6\n",
-      "16198    | torch.Size([1, 1000, 777]) | 6\n",
-      "16199    | torch.Size([1, 1000, 754]) | 6\n",
-      "16200    | torch.Size([1, 1000, 777]) | 6\n",
-      "16201    | torch.Size([1, 1000, 777]) | 6\n",
-      "16202    | torch.Size([1, 1000, 777]) | 6\n",
-      "16203    | torch.Size([1, 1000, 754]) | 6\n",
-      "16204    | torch.Size([1, 1000, 777]) | 6\n",
-      "16205    | torch.Size([1, 1000, 773]) | 6\n",
-      "16206    | torch.Size([1, 1000, 787]) | 6\n",
-      "16207    | torch.Size([1, 1000, 754]) | 6\n",
-      "16208    | torch.Size([1, 1000, 754]) | 6\n",
-      "16209    | torch.Size([1, 1000, 754]) | 6\n",
-      "16210    | torch.Size([1, 1000, 777]) | 6\n",
-      "16211    | torch.Size([1, 1000, 777]) | 6\n",
-      "16212    | torch.Size([1, 1000, 769]) | 6\n",
-      "16213    | torch.Size([1, 1000, 762]) | 6\n",
-      "16214    | torch.Size([1, 1000, 754]) | 6\n",
-      "16215    | torch.Size([1, 1000, 754]) | 6\n",
-      "16216    | torch.Size([1, 1000, 777]) | 6\n",
-      "16217    | torch.Size([1, 1000, 802]) | 6\n",
-      "16218    | torch.Size([1, 1000, 784]) | 6\n",
-      "16219    | torch.Size([1, 1000, 754]) | 6\n",
-      "16220    | torch.Size([1, 1000, 777]) | 6\n",
-      "16221    | torch.Size([1, 1000, 754]) | 6\n",
-      "16222    | torch.Size([1, 1000, 777]) | 6\n",
-      "16223    | torch.Size([1, 1000, 754]) | 6\n",
-      "16224    | torch.Size([1, 1000, 792]) | 6\n",
-      "16225    | torch.Size([1, 1000, 754]) | 6\n",
-      "16226    | torch.Size([1, 1000, 754]) | 6\n",
-      "16227    | torch.Size([1, 1000, 754]) | 6\n",
-      "16228    | torch.Size([1, 1000, 762]) | 6\n",
-      "16229    | torch.Size([1, 1000, 754]) | 6\n",
-      "16230    | torch.Size([1, 1000, 762]) | 6\n",
-      "16231    | torch.Size([1, 1000, 754]) | 6\n",
-      "16232    | torch.Size([1, 1000, 762]) | 6\n",
-      "16233    | torch.Size([1, 1000, 770]) | 6\n",
-      "16234    | torch.Size([1, 1000, 773]) | 6\n",
-      "16235    | torch.Size([1, 1000, 777]) | 6\n",
-      "16236    | torch.Size([1, 1000, 777]) | 6\n",
-      "16237    | torch.Size([1, 1000, 777]) | 6\n",
-      "16238    | torch.Size([1, 1000, 769]) | 6\n",
-      "16239    | torch.Size([1, 1000, 754]) | 6\n",
-      "16240    | torch.Size([1, 1000, 754]) | 6\n",
-      "16241    | torch.Size([1, 1000, 762]) | 6\n",
-      "16242    | torch.Size([1, 1000, 777]) | 6\n",
-      "16243    | torch.Size([1, 1000, 754]) | 6\n",
-      "16244    | torch.Size([1, 1000, 754]) | 6\n",
-      "16245    | torch.Size([1, 1000, 771]) | 6\n",
-      "16246    | torch.Size([1, 1000, 752]) | 6\n",
-      "16247    | torch.Size([1, 1000, 776]) | 6\n",
-      "16248    | torch.Size([1, 1000, 747]) | 6\n",
-      "16249    | torch.Size([1, 1000, 754]) | 6\n",
-      "16250    | torch.Size([1, 1000, 754]) | 6\n",
-      "16251    | torch.Size([1, 1000, 754]) | 6\n",
-      "16252    | torch.Size([1, 1000, 762]) | 6\n",
-      "16253    | torch.Size([1, 1000, 777]) | 6\n",
-      "16254    | torch.Size([1, 1000, 754]) | 6\n",
-      "16255    | torch.Size([1, 1000, 786]) | 6\n",
-      "16256    | torch.Size([1, 1000, 754]) | 6\n",
-      "16257    | torch.Size([1, 1000, 754]) | 6\n",
-      "16258    | torch.Size([1, 1000, 754]) | 6\n",
-      "16259    | torch.Size([1, 1000, 771]) | 6\n",
-      "16260    | torch.Size([1, 1000, 770]) | 6\n",
-      "16261    | torch.Size([1, 1000, 762]) | 6\n",
-      "16262    | torch.Size([1, 1000, 762]) | 6\n",
-      "16263    | torch.Size([1, 1000, 772]) | 6\n",
-      "16264    | torch.Size([1, 1000, 777]) | 6\n",
-      "16265    | torch.Size([1, 1000, 830]) | 6\n",
-      "16266    | torch.Size([1, 1000, 777]) | 6\n",
-      "16267    | torch.Size([1, 1000, 754]) | 6\n",
-      "16268    | torch.Size([1, 1000, 762]) | 6\n",
-      "16269    | torch.Size([1, 1000, 795]) | 6\n",
-      "16270    | torch.Size([1, 1000, 762]) | 6\n",
-      "16271    | torch.Size([1, 1000, 754]) | 6\n",
-      "16272    | torch.Size([1, 1000, 791]) | 6\n",
-      "16273    | torch.Size([1, 1000, 754]) | 6\n",
-      "16274    | torch.Size([1, 1000, 754]) | 6\n",
-      "16275    | torch.Size([1, 1000, 754]) | 6\n",
-      "16276    | torch.Size([1, 1000, 777]) | 6\n",
-      "16277    | torch.Size([1, 1000, 754]) | 6\n",
-      "16278    | torch.Size([1, 1000, 762]) | 6\n",
-      "16279    | torch.Size([1, 1000, 771]) | 6\n",
-      "16280    | torch.Size([1, 1000, 769]) | 6\n",
-      "16281    | torch.Size([1, 1000, 777]) | 6\n",
-      "16282    | torch.Size([1, 1000, 754]) | 6\n",
-      "16283    | torch.Size([1, 1000, 762]) | 6\n",
-      "16284    | torch.Size([1, 1000, 767]) | 6\n",
-      "16285    | torch.Size([1, 1000, 773]) | 6\n",
-      "16286    | torch.Size([1, 1000, 777]) | 6\n",
-      "16287    | torch.Size([1, 1000, 776]) | 6\n",
-      "16288    | torch.Size([1, 1000, 777]) | 6\n",
-      "16289    | torch.Size([1, 1000, 777]) | 6\n",
-      "16290    | torch.Size([1, 1000, 773]) | 6\n",
-      "16291    | torch.Size([1, 1000, 762]) | 6\n",
-      "16292    | torch.Size([1, 1000, 754]) | 6\n",
-      "16293    | torch.Size([1, 1000, 777]) | 6\n",
-      "16294    | torch.Size([1, 1000, 754]) | 6\n",
-      "16295    | torch.Size([1, 1000, 781]) | 6\n",
-      "16296    | torch.Size([1, 1000, 754]) | 6\n",
-      "16297    | torch.Size([1, 1000, 793]) | 6\n",
-      "16298    | torch.Size([1, 1000, 754]) | 6\n",
-      "16299    | torch.Size([1, 1000, 776]) | 6\n",
-      "16300    | torch.Size([1, 1000, 754]) | 6\n",
-      "16301    | torch.Size([1, 1000, 754]) | 6\n",
-      "16302    | torch.Size([1, 1000, 777]) | 6\n",
-      "16303    | torch.Size([1, 1000, 754]) | 6\n",
-      "16304    | torch.Size([1, 1000, 754]) | 6\n",
-      "16305    | torch.Size([1, 1000, 772]) | 6\n",
-      "16306    | torch.Size([1, 1000, 752]) | 6\n",
-      "16307    | torch.Size([1, 1000, 777]) | 6\n",
-      "16308    | torch.Size([1, 1000, 777]) | 6\n",
-      "16309    | torch.Size([1, 1000, 730]) | 6\n",
-      "16310    | torch.Size([1, 1000, 777]) | 6\n",
-      "16311    | torch.Size([1, 1000, 754]) | 6\n",
-      "16312    | torch.Size([1, 1000, 754]) | 6\n",
-      "16313    | torch.Size([1, 1000, 754]) | 6\n",
-      "16314    | torch.Size([1, 1000, 767]) | 6\n",
-      "16315    | torch.Size([1, 1000, 754]) | 6\n",
-      "16316    | torch.Size([1, 1000, 797]) | 6\n",
-      "16317    | torch.Size([1, 1000, 762]) | 6\n",
-      "16318    | torch.Size([1, 1000, 777]) | 6\n",
-      "16319    | torch.Size([1, 1000, 785]) | 6\n",
-      "16320    | torch.Size([1, 1000, 763]) | 6\n",
-      "16321    | torch.Size([1, 1000, 754]) | 6\n",
-      "16322    | torch.Size([1, 1000, 754]) | 6\n",
-      "16323    | torch.Size([1, 1000, 754]) | 6\n",
-      "16324    | torch.Size([1, 1000, 762]) | 6\n",
-      "16325    | torch.Size([1, 1000, 754]) | 6\n",
-      "16326    | torch.Size([1, 1000, 777]) | 6\n",
-      "16327    | torch.Size([1, 1000, 801]) | 6\n",
-      "16328    | torch.Size([1, 1000, 762]) | 6\n",
-      "16329    | torch.Size([1, 1000, 777]) | 6\n",
-      "16330    | torch.Size([1, 1000, 772]) | 6\n",
-      "16331    | torch.Size([1, 1000, 754]) | 6\n",
-      "16332    | torch.Size([1, 1000, 776]) | 6\n",
-      "16333    | torch.Size([1, 1000, 754]) | 6\n",
-      "16334    | torch.Size([1, 1000, 762]) | 6\n",
-      "16335    | torch.Size([1, 1000, 799]) | 6\n",
-      "16336    | torch.Size([1, 1000, 762]) | 6\n",
-      "16337    | torch.Size([1, 1000, 789]) | 6\n",
-      "16338    | torch.Size([1, 1000, 762]) | 6\n",
-      "16339    | torch.Size([1, 1000, 762]) | 6\n",
-      "16340    | torch.Size([1, 1000, 762]) | 6\n",
-      "16341    | torch.Size([1, 1000, 777]) | 6\n",
-      "16342    | torch.Size([1, 1000, 754]) | 6\n",
-      "16343    | torch.Size([1, 1000, 754]) | 6\n",
-      "16344    | torch.Size([1, 1000, 754]) | 6\n",
-      "16345    | torch.Size([1, 1000, 754]) | 6\n",
-      "16346    | torch.Size([1, 1000, 775]) | 6\n",
-      "16347    | torch.Size([1, 1000, 773]) | 6\n",
-      "16348    | torch.Size([1, 1000, 772]) | 6\n",
-      "16349    | torch.Size([1, 1000, 762]) | 6\n",
-      "16350    | torch.Size([1, 1000, 771]) | 6\n",
-      "16351    | torch.Size([1, 1000, 754]) | 6\n",
-      "16352    | torch.Size([1, 1000, 754]) | 6\n",
-      "16353    | torch.Size([1, 1000, 754]) | 6\n",
-      "16354    | torch.Size([1, 1000, 780]) | 6\n",
-      "16355    | torch.Size([1, 1000, 754]) | 6\n",
-      "16356    | torch.Size([1, 1000, 796]) | 6\n",
-      "16357    | torch.Size([1, 1000, 754]) | 6\n",
-      "16358    | torch.Size([1, 1000, 777]) | 6\n",
-      "16359    | torch.Size([1, 1000, 762]) | 6\n",
-      "16360    | torch.Size([1, 1000, 754]) | 6\n",
-      "16361    | torch.Size([1, 1000, 754]) | 6\n",
-      "16362    | torch.Size([1, 1000, 754]) | 6\n",
-      "16363    | torch.Size([1, 1000, 754]) | 6\n",
-      "16364    | torch.Size([1, 1000, 762]) | 6\n",
-      "16365    | torch.Size([1, 1000, 754]) | 6\n",
-      "16366    | torch.Size([1, 1000, 611]) | 6\n",
-      "16367    | torch.Size([1, 1000, 768]) | 6\n",
-      "16368    | torch.Size([1, 1000, 767]) | 6\n",
-      "16369    | torch.Size([1, 1000, 754]) | 6\n",
-      "16370    | torch.Size([1, 1000, 754]) | 6\n",
-      "16371    | torch.Size([1, 1000, 766]) | 6\n",
-      "16372    | torch.Size([1, 1000, 770]) | 6\n",
-      "16373    | torch.Size([1, 1000, 762]) | 6\n",
-      "16374    | torch.Size([1, 1000, 754]) | 6\n",
-      "16375    | torch.Size([1, 1000, 762]) | 6\n",
-      "16376    | torch.Size([1, 1000, 754]) | 6\n",
-      "16377    | torch.Size([1, 1000, 777]) | 6\n",
-      "16378    | torch.Size([1, 1000, 766]) | 6\n",
-      "16379    | torch.Size([1, 1000, 781]) | 6\n",
-      "16380    | torch.Size([1, 1000, 802]) | 6\n",
-      "16381    | torch.Size([1, 1000, 762]) | 6\n",
-      "16382    | torch.Size([1, 1000, 771]) | 6\n",
-      "16383    | torch.Size([1, 1000, 762]) | 6\n",
-      "16384    | torch.Size([1, 1000, 771]) | 6\n",
-      "16385    | torch.Size([1, 1000, 777]) | 6\n",
-      "16386    | torch.Size([1, 1000, 790]) | 6\n",
-      "16387    | torch.Size([1, 1000, 802]) | 6\n",
-      "16388    | torch.Size([1, 1000, 777]) | 6\n",
-      "16389    | torch.Size([1, 1000, 754]) | 6\n",
-      "16390    | torch.Size([1, 1000, 754]) | 6\n",
-      "16391    | torch.Size([1, 1000, 754]) | 6\n",
-      "16392    | torch.Size([1, 1000, 754]) | 6\n",
-      "16393    | torch.Size([1, 1000, 754]) | 6\n",
-      "16394    | torch.Size([1, 1000, 843]) | 6\n",
-      "16395    | torch.Size([1, 1000, 777]) | 6\n",
-      "16396    | torch.Size([1, 1000, 754]) | 6\n",
-      "16397    | torch.Size([1, 1000, 754]) | 6\n",
-      "16398    | torch.Size([1, 1000, 754]) | 6\n",
-      "16399    | torch.Size([1, 1000, 754]) | 6\n",
-      "16400    | torch.Size([1, 1000, 754]) | 6\n",
-      "16401    | torch.Size([1, 1000, 777]) | 6\n",
-      "16402    | torch.Size([1, 1000, 762]) | 6\n",
-      "16403    | torch.Size([1, 1000, 771]) | 6\n",
-      "16404    | torch.Size([1, 1000, 777]) | 6\n",
-      "16405    | torch.Size([1, 1000, 754]) | 6\n",
-      "16406    | torch.Size([1, 1000, 754]) | 6\n",
-      "16407    | torch.Size([1, 1000, 794]) | 6\n",
-      "16408    | torch.Size([1, 1000, 776]) | 6\n",
-      "16409    | torch.Size([1, 1000, 754]) | 6\n",
-      "16410    | torch.Size([1, 1000, 768]) | 6\n",
-      "16411    | torch.Size([1, 1000, 777]) | 6\n",
-      "16412    | torch.Size([1, 1000, 754]) | 6\n",
-      "16413    | torch.Size([1, 1000, 777]) | 6\n",
-      "16414    | torch.Size([1, 1000, 754]) | 6\n",
-      "16415    | torch.Size([1, 1000, 754]) | 6\n",
-      "16416    | torch.Size([1, 1000, 754]) | 6\n",
-      "16417    | torch.Size([1, 1000, 754]) | 6\n",
-      "16418    | torch.Size([1, 1000, 762]) | 6\n",
-      "16419    | torch.Size([1, 1000, 754]) | 6\n",
-      "16420    | torch.Size([1, 1000, 754]) | 6\n",
-      "16421    | torch.Size([1, 1000, 762]) | 6\n",
-      "16422    | torch.Size([1, 1000, 765]) | 6\n",
-      "16423    | torch.Size([1, 1000, 754]) | 6\n",
-      "16424    | torch.Size([1, 1000, 754]) | 6\n",
-      "16425    | torch.Size([1, 1000, 762]) | 6\n",
-      "16426    | torch.Size([1, 1000, 754]) | 6\n",
-      "16427    | torch.Size([1, 1000, 777]) | 6\n",
-      "16428    | torch.Size([1, 1000, 794]) | 6\n",
-      "16429    | torch.Size([1, 1000, 754]) | 6\n",
-      "16430    | torch.Size([1, 1000, 786]) | 6\n",
-      "16431    | torch.Size([1, 1000, 754]) | 6\n",
-      "16432    | torch.Size([1, 1000, 754]) | 6\n",
-      "16433    | torch.Size([1, 1000, 754]) | 6\n",
-      "16434    | torch.Size([1, 1000, 754]) | 6\n",
-      "16435    | torch.Size([1, 1000, 754]) | 6\n",
-      "16436    | torch.Size([1, 1000, 754]) | 6\n",
-      "16437    | torch.Size([1, 1000, 754]) | 6\n",
-      "16438    | torch.Size([1, 1000, 777]) | 6\n",
-      "16439    | torch.Size([1, 1000, 779]) | 6\n",
-      "16440    | torch.Size([1, 1000, 777]) | 6\n",
-      "16441    | torch.Size([1, 1000, 754]) | 6\n",
-      "16442    | torch.Size([1, 1000, 754]) | 6\n",
-      "16443    | torch.Size([1, 1000, 777]) | 6\n",
-      "16444    | torch.Size([1, 1000, 754]) | 6\n",
-      "16445    | torch.Size([1, 1000, 754]) | 6\n",
-      "16446    | torch.Size([1, 1000, 767]) | 6\n",
-      "16447    | torch.Size([1, 1000, 754]) | 6\n",
-      "16448    | torch.Size([1, 1000, 754]) | 6\n",
-      "16449    | torch.Size([1, 1000, 785]) | 6\n",
-      "16450    | torch.Size([1, 1000, 754]) | 6\n",
-      "16451    | torch.Size([1, 1000, 754]) | 6\n",
-      "16452    | torch.Size([1, 1000, 795]) | 6\n",
-      "16453    | torch.Size([1, 1000, 777]) | 6\n",
-      "16454    | torch.Size([1, 1000, 754]) | 6\n",
-      "16455    | torch.Size([1, 1000, 771]) | 6\n",
-      "16456    | torch.Size([1, 1000, 754]) | 6\n",
-      "16457    | torch.Size([1, 1000, 754]) | 6\n",
-      "16458    | torch.Size([1, 1000, 763]) | 6\n",
-      "16459    | torch.Size([1, 1000, 772]) | 6\n",
-      "16460    | torch.Size([1, 1000, 754]) | 6\n",
-      "16461    | torch.Size([1, 1000, 777]) | 6\n",
-      "16462    | torch.Size([1, 1000, 771]) | 6\n",
-      "16463    | torch.Size([1, 1000, 771]) | 6\n",
-      "16464    | torch.Size([1, 1000, 762]) | 6\n",
-      "16465    | torch.Size([1, 1000, 777]) | 6\n",
-      "16466    | torch.Size([1, 1000, 777]) | 6\n",
-      "16467    | torch.Size([1, 1000, 777]) | 6\n",
-      "16468    | torch.Size([1, 1000, 770]) | 6\n",
-      "16469    | torch.Size([1, 1000, 783]) | 6\n",
-      "16470    | torch.Size([1, 1000, 754]) | 6\n",
-      "16471    | torch.Size([1, 1000, 754]) | 6\n",
-      "16472    | torch.Size([1, 1000, 754]) | 6\n",
-      "16473    | torch.Size([1, 1000, 762]) | 6\n",
-      "16474    | torch.Size([1, 1000, 762]) | 6\n",
-      "16475    | torch.Size([1, 1000, 754]) | 6\n",
-      "16476    | torch.Size([1, 1000, 802]) | 6\n",
-      "16477    | torch.Size([1, 1000, 754]) | 6\n",
-      "16478    | torch.Size([1, 1000, 770]) | 6\n",
-      "16479    | torch.Size([1, 1000, 752]) | 6\n",
-      "16480    | torch.Size([1, 1000, 754]) | 6\n",
-      "16481    | torch.Size([1, 1000, 754]) | 6\n",
-      "16482    | torch.Size([1, 1000, 773]) | 6\n",
-      "16483    | torch.Size([1, 1000, 767]) | 6\n",
-      "16484    | torch.Size([1, 1000, 754]) | 6\n",
-      "16485    | torch.Size([1, 1000, 754]) | 6\n",
-      "16486    | torch.Size([1, 1000, 754]) | 6\n",
-      "16487    | torch.Size([1, 1000, 754]) | 6\n",
-      "16488    | torch.Size([1, 1000, 777]) | 6\n",
-      "16489    | torch.Size([1, 1000, 777]) | 6\n",
-      "16490    | torch.Size([1, 1000, 754]) | 6\n",
-      "16491    | torch.Size([1, 1000, 762]) | 6\n",
-      "16492    | torch.Size([1, 1000, 762]) | 6\n",
-      "16493    | torch.Size([1, 1000, 731]) | 6\n",
-      "16494    | torch.Size([1, 1000, 778]) | 6\n",
-      "16495    | torch.Size([1, 1000, 777]) | 6\n",
-      "16496    | torch.Size([1, 1000, 754]) | 6\n",
-      "16497    | torch.Size([1, 1000, 769]) | 6\n",
-      "16498    | torch.Size([1, 1000, 777]) | 6\n",
-      "16499    | torch.Size([1, 1000, 754]) | 6\n",
-      "16500    | torch.Size([1, 1000, 754]) | 6\n",
-      "16501    | torch.Size([1, 1000, 754]) | 6\n",
-      "16502    | torch.Size([1, 1000, 777]) | 6\n",
-      "16503    | torch.Size([1, 1000, 777]) | 6\n",
-      "16504    | torch.Size([1, 1000, 777]) | 6\n",
-      "16505    | torch.Size([1, 1000, 762]) | 6\n",
-      "16506    | torch.Size([1, 1000, 777]) | 6\n",
-      "16507    | torch.Size([1, 1000, 754]) | 6\n",
-      "16508    | torch.Size([1, 1000, 802]) | 6\n",
-      "16509    | torch.Size([1, 1000, 777]) | 6\n",
-      "16510    | torch.Size([1, 1000, 754]) | 6\n",
-      "16511    | torch.Size([1, 1000, 789]) | 6\n",
-      "16512    | torch.Size([1, 1000, 754]) | 6\n",
-      "16513    | torch.Size([1, 1000, 754]) | 6\n",
-      "16514    | torch.Size([1, 1000, 754]) | 6\n",
-      "16515    | torch.Size([1, 1000, 777]) | 6\n",
-      "16516    | torch.Size([1, 1000, 804]) | 6\n",
-      "16517    | torch.Size([1, 1000, 777]) | 6\n",
-      "16518    | torch.Size([1, 1000, 754]) | 6\n",
-      "16519    | torch.Size([1, 1000, 779]) | 6\n",
-      "16520    | torch.Size([1, 1000, 776]) | 6\n",
-      "16521    | torch.Size([1, 1000, 777]) | 6\n",
-      "16522    | torch.Size([1, 1000, 754]) | 6\n",
-      "16523    | torch.Size([1, 1000, 784]) | 6\n",
-      "16524    | torch.Size([1, 1000, 802]) | 6\n",
-      "16525    | torch.Size([1, 1000, 777]) | 6\n",
-      "16526    | torch.Size([1, 1000, 777]) | 6\n",
-      "16527    | torch.Size([1, 1000, 772]) | 6\n",
-      "16528    | torch.Size([1, 1000, 754]) | 6\n",
-      "16529    | torch.Size([1, 1000, 791]) | 6\n",
-      "16530    | torch.Size([1, 1000, 754]) | 6\n",
-      "16531    | torch.Size([1, 1000, 754]) | 6\n",
-      "16532    | torch.Size([1, 1000, 754]) | 6\n",
-      "16533    | torch.Size([1, 1000, 754]) | 6\n",
-      "16534    | torch.Size([1, 1000, 777]) | 6\n",
-      "16535    | torch.Size([1, 1000, 754]) | 6\n",
-      "16536    | torch.Size([1, 1000, 809]) | 6\n",
-      "16537    | torch.Size([1, 1000, 754]) | 6\n",
-      "16538    | torch.Size([1, 1000, 771]) | 6\n",
-      "16539    | torch.Size([1, 1000, 754]) | 6\n",
-      "16540    | torch.Size([1, 1000, 754]) | 6\n",
-      "16541    | torch.Size([1, 1000, 777]) | 6\n",
-      "16542    | torch.Size([1, 1000, 779]) | 6\n",
-      "16543    | torch.Size([1, 1000, 762]) | 6\n",
-      "16544    | torch.Size([1, 1000, 759]) | 6\n",
-      "16545    | torch.Size([1, 1000, 777]) | 6\n",
-      "16546    | torch.Size([1, 1000, 754]) | 6\n",
-      "16547    | torch.Size([1, 1000, 780]) | 6\n",
-      "16548    | torch.Size([1, 1000, 778]) | 6\n",
-      "16549    | torch.Size([1, 1000, 772]) | 6\n",
-      "16550    | torch.Size([1, 1000, 754]) | 6\n",
-      "16551    | torch.Size([1, 1000, 777]) | 6\n",
-      "16552    | torch.Size([1, 1000, 777]) | 6\n",
-      "16553    | torch.Size([1, 1000, 777]) | 6\n",
-      "16554    | torch.Size([1, 1000, 754]) | 6\n",
-      "16555    | torch.Size([1, 1000, 771]) | 6\n",
-      "16556    | torch.Size([1, 1000, 754]) | 6\n",
-      "16557    | torch.Size([1, 1000, 754]) | 6\n",
-      "16558    | torch.Size([1, 1000, 788]) | 6\n",
-      "16559    | torch.Size([1, 1000, 754]) | 6\n",
-      "16560    | torch.Size([1, 1000, 764]) | 6\n",
-      "16561    | torch.Size([1, 1000, 754]) | 6\n",
-      "16562    | torch.Size([1, 1000, 754]) | 6\n",
-      "16563    | torch.Size([1, 1000, 754]) | 6\n",
-      "16564    | torch.Size([1, 1000, 766]) | 6\n",
-      "16565    | torch.Size([1, 1000, 771]) | 6\n",
-      "16566    | torch.Size([1, 1000, 754]) | 6\n",
-      "16567    | torch.Size([1, 1000, 777]) | 6\n",
-      "16568    | torch.Size([1, 1000, 777]) | 6\n",
-      "16569    | torch.Size([1, 1000, 788]) | 6\n",
-      "16570    | torch.Size([1, 1000, 777]) | 6\n",
-      "16571    | torch.Size([1, 1000, 777]) | 6\n",
-      "16572    | torch.Size([1, 1000, 777]) | 6\n",
-      "16573    | torch.Size([1, 1000, 770]) | 6\n",
-      "16574    | torch.Size([1, 1000, 754]) | 6\n",
-      "16575    | torch.Size([1, 1000, 762]) | 6\n",
-      "16576    | torch.Size([1, 1000, 754]) | 6\n",
-      "16577    | torch.Size([1, 1000, 754]) | 6\n",
-      "16578    | torch.Size([1, 1000, 754]) | 6\n",
-      "16579    | torch.Size([1, 1000, 777]) | 6\n",
-      "16580    | torch.Size([1, 1000, 777]) | 6\n",
-      "16581    | torch.Size([1, 1000, 754]) | 6\n",
-      "16582    | torch.Size([1, 1000, 734]) | 6\n",
-      "16583    | torch.Size([1, 1000, 776]) | 6\n",
-      "16584    | torch.Size([1, 1000, 777]) | 6\n",
-      "16585    | torch.Size([1, 1000, 762]) | 6\n",
-      "16586    | torch.Size([1, 1000, 754]) | 6\n",
-      "16587    | torch.Size([1, 1000, 780]) | 6\n",
-      "16588    | torch.Size([1, 1000, 776]) | 6\n",
-      "16589    | torch.Size([1, 1000, 754]) | 6\n",
-      "16590    | torch.Size([1, 1000, 754]) | 6\n",
-      "16591    | torch.Size([1, 1000, 754]) | 6\n",
-      "16592    | torch.Size([1, 1000, 777]) | 6\n",
-      "16593    | torch.Size([1, 1000, 762]) | 6\n",
-      "16594    | torch.Size([1, 1000, 787]) | 6\n",
-      "16595    | torch.Size([1, 1000, 754]) | 6\n",
-      "16596    | torch.Size([1, 1000, 788]) | 6\n",
-      "16597    | torch.Size([1, 1000, 763]) | 6\n",
-      "16598    | torch.Size([1, 1000, 762]) | 6\n",
-      "16599    | torch.Size([1, 1000, 800]) | 6\n",
-      "16600    | torch.Size([1, 1000, 773]) | 6\n",
-      "16601    | torch.Size([1, 1000, 754]) | 6\n",
-      "16602    | torch.Size([1, 1000, 777]) | 6\n",
-      "16603    | torch.Size([1, 1000, 754]) | 6\n",
-      "16604    | torch.Size([1, 1000, 754]) | 6\n",
-      "16605    | torch.Size([1, 1000, 779]) | 6\n",
-      "16606    | torch.Size([1, 1000, 771]) | 6\n",
-      "16607    | torch.Size([1, 1000, 766]) | 6\n",
-      "16608    | torch.Size([1, 1000, 731]) | 6\n",
-      "16609    | torch.Size([1, 1000, 754]) | 6\n",
-      "16610    | torch.Size([1, 1000, 764]) | 6\n",
-      "16611    | torch.Size([1, 1000, 754]) | 6\n",
-      "16612    | torch.Size([1, 1000, 791]) | 6\n",
-      "16613    | torch.Size([1, 1000, 777]) | 6\n",
-      "16614    | torch.Size([1, 1000, 754]) | 6\n",
-      "16615    | torch.Size([1, 1000, 754]) | 6\n",
-      "16616    | torch.Size([1, 1000, 777]) | 6\n",
-      "16617    | torch.Size([1, 1000, 790]) | 6\n",
-      "16618    | torch.Size([1, 1000, 778]) | 6\n",
-      "16619    | torch.Size([1, 1000, 783]) | 6\n",
-      "16620    | torch.Size([1, 1000, 777]) | 6\n",
-      "16621    | torch.Size([1, 1000, 777]) | 6\n",
-      "16622    | torch.Size([1, 1000, 754]) | 6\n",
-      "16623    | torch.Size([1, 1000, 770]) | 6\n",
-      "16624    | torch.Size([1, 1000, 754]) | 6\n",
-      "16625    | torch.Size([1, 1000, 777]) | 6\n",
-      "16626    | torch.Size([1, 1000, 778]) | 6\n",
-      "16627    | torch.Size([1, 1000, 754]) | 6\n",
-      "16628    | torch.Size([1, 1000, 777]) | 6\n",
-      "16629    | torch.Size([1, 1000, 762]) | 6\n",
-      "16630    | torch.Size([1, 1000, 777]) | 6\n",
-      "16631    | torch.Size([1, 1000, 765]) | 6\n",
-      "16632    | torch.Size([1, 1000, 754]) | 6\n",
-      "16633    | torch.Size([1, 1000, 754]) | 6\n",
-      "16634    | torch.Size([1, 1000, 751]) | 6\n",
-      "16635    | torch.Size([1, 1000, 763]) | 6\n",
-      "16636    | torch.Size([1, 1000, 762]) | 6\n",
-      "16637    | torch.Size([1, 1000, 754]) | 6\n",
-      "16638    | torch.Size([1, 1000, 787]) | 6\n",
-      "16639    | torch.Size([1, 1000, 777]) | 6\n",
-      "16640    | torch.Size([1, 1000, 762]) | 6\n",
-      "16641    | torch.Size([1, 1000, 771]) | 6\n",
-      "16642    | torch.Size([1, 1000, 754]) | 6\n",
-      "16643    | torch.Size([1, 1000, 754]) | 6\n",
-      "16644    | torch.Size([1, 1000, 767]) | 6\n",
-      "16645    | torch.Size([1, 1000, 767]) | 6\n",
-      "16646    | torch.Size([1, 1000, 777]) | 6\n",
-      "16647    | torch.Size([1, 1000, 773]) | 6\n",
-      "16648    | torch.Size([1, 1000, 771]) | 6\n",
-      "16649    | torch.Size([1, 1000, 754]) | 6\n",
-      "16650    | torch.Size([1, 1000, 754]) | 6\n",
-      "16651    | torch.Size([1, 1000, 754]) | 6\n",
-      "16652    | torch.Size([1, 1000, 754]) | 6\n",
-      "16653    | torch.Size([1, 1000, 777]) | 6\n",
-      "16654    | torch.Size([1, 1000, 754]) | 6\n",
-      "16655    | torch.Size([1, 1000, 754]) | 6\n",
-      "16656    | torch.Size([1, 1000, 777]) | 6\n",
-      "16657    | torch.Size([1, 1000, 754]) | 6\n",
-      "16658    | torch.Size([1, 1000, 731]) | 6\n",
-      "16659    | torch.Size([1, 1000, 762]) | 6\n",
-      "16660    | torch.Size([1, 1000, 754]) | 6\n",
-      "16661    | torch.Size([1, 1000, 754]) | 6\n",
-      "16662    | torch.Size([1, 1000, 762]) | 6\n",
-      "16663    | torch.Size([1, 1000, 762]) | 6\n",
-      "16664    | torch.Size([1, 1000, 754]) | 6\n",
-      "16665    | torch.Size([1, 1000, 777]) | 6\n",
-      "16666    | torch.Size([1, 1000, 762]) | 6\n",
-      "16667    | torch.Size([1, 1000, 777]) | 6\n",
-      "16668    | torch.Size([1, 1000, 754]) | 6\n",
-      "16669    | torch.Size([1, 1000, 754]) | 6\n",
-      "16670    | torch.Size([1, 1000, 787]) | 6\n",
-      "16671    | torch.Size([1, 1000, 777]) | 6\n",
-      "16672    | torch.Size([1, 1000, 754]) | 6\n",
-      "16673    | torch.Size([1, 1000, 762]) | 6\n",
-      "16674    | torch.Size([1, 1000, 777]) | 6\n",
-      "16675    | torch.Size([1, 1000, 754]) | 6\n",
-      "16676    | torch.Size([1, 1000, 796]) | 6\n",
-      "16677    | torch.Size([1, 1000, 754]) | 6\n",
-      "16678    | torch.Size([1, 1000, 754]) | 6\n",
-      "16679    | torch.Size([1, 1000, 777]) | 6\n",
-      "16680    | torch.Size([1, 1000, 777]) | 6\n",
-      "16681    | torch.Size([1, 1000, 754]) | 6\n",
-      "16682    | torch.Size([1, 1000, 771]) | 6\n",
-      "16683    | torch.Size([1, 1000, 777]) | 6\n",
-      "16684    | torch.Size([1, 1000, 769]) | 6\n",
-      "16685    | torch.Size([1, 1000, 752]) | 6\n",
-      "16686    | torch.Size([1, 1000, 778]) | 6\n",
-      "16687    | torch.Size([1, 1000, 777]) | 6\n",
-      "16688    | torch.Size([1, 1000, 754]) | 6\n",
-      "16689    | torch.Size([1, 1000, 788]) | 6\n",
-      "16690    | torch.Size([1, 1000, 771]) | 6\n",
-      "16691    | torch.Size([1, 1000, 754]) | 6\n",
-      "16692    | torch.Size([1, 1000, 777]) | 6\n",
-      "16693    | torch.Size([1, 1000, 777]) | 6\n",
-      "16694    | torch.Size([1, 1000, 777]) | 6\n",
-      "16695    | torch.Size([1, 1000, 777]) | 6\n",
-      "16696    | torch.Size([1, 1000, 762]) | 6\n",
-      "16697    | torch.Size([1, 1000, 754]) | 6\n",
-      "16698    | torch.Size([1, 1000, 754]) | 6\n",
-      "16699    | torch.Size([1, 1000, 789]) | 6\n",
-      "16700    | torch.Size([1, 1000, 754]) | 6\n",
-      "16701    | torch.Size([1, 1000, 775]) | 6\n",
-      "16702    | torch.Size([1, 1000, 754]) | 6\n",
-      "16703    | torch.Size([1, 1000, 777]) | 6\n",
-      "16704    | torch.Size([1, 1000, 754]) | 6\n",
-      "16705    | torch.Size([1, 1000, 754]) | 6\n",
-      "16706    | torch.Size([1, 1000, 754]) | 6\n",
-      "16707    | torch.Size([1, 1000, 762]) | 6\n",
-      "16708    | torch.Size([1, 1000, 752]) | 6\n",
-      "16709    | torch.Size([1, 1000, 754]) | 6\n",
-      "16710    | torch.Size([1, 1000, 754]) | 6\n",
-      "16711    | torch.Size([1, 1000, 762]) | 6\n",
-      "16712    | torch.Size([1, 1000, 777]) | 6\n",
-      "16713    | torch.Size([1, 1000, 754]) | 6\n",
-      "16714    | torch.Size([1, 1000, 754]) | 6\n",
-      "16715    | torch.Size([1, 1000, 754]) | 6\n",
-      "16716    | torch.Size([1, 1000, 772]) | 6\n",
-      "16717    | torch.Size([1, 1000, 754]) | 6\n",
-      "16718    | torch.Size([1, 1000, 771]) | 6\n",
-      "16719    | torch.Size([1, 1000, 754]) | 6\n",
-      "16720    | torch.Size([1, 1000, 752]) | 6\n",
-      "16721    | torch.Size([1, 1000, 762]) | 6\n",
-      "16722    | torch.Size([1, 1000, 754]) | 6\n",
-      "16723    | torch.Size([1, 1000, 754]) | 6\n",
-      "16724    | torch.Size([1, 1000, 754]) | 6\n",
-      "16725    | torch.Size([1, 1000, 754]) | 6\n",
-      "16726    | torch.Size([1, 1000, 754]) | 6\n",
-      "16727    | torch.Size([1, 1000, 762]) | 6\n",
-      "16728    | torch.Size([1, 1000, 777]) | 6\n",
-      "16729    | torch.Size([1, 1000, 754]) | 6\n",
-      "16730    | torch.Size([1, 1000, 777]) | 6\n",
-      "16731    | torch.Size([1, 1000, 754]) | 6\n",
-      "16732    | torch.Size([1, 1000, 778]) | 6\n",
-      "16733    | torch.Size([1, 1000, 754]) | 6\n",
-      "16734    | torch.Size([1, 1000, 754]) | 6\n",
-      "16735    | torch.Size([1, 1000, 754]) | 6\n",
-      "16736    | torch.Size([1, 1000, 754]) | 6\n",
-      "16737    | torch.Size([1, 1000, 754]) | 6\n",
-      "16738    | torch.Size([1, 1000, 754]) | 6\n",
-      "16739    | torch.Size([1, 1000, 754]) | 6\n",
-      "16740    | torch.Size([1, 1000, 754]) | 6\n",
-      "16741    | torch.Size([1, 1000, 754]) | 6\n",
-      "16742    | torch.Size([1, 1000, 754]) | 6\n",
-      "16743    | torch.Size([1, 1000, 777]) | 6\n",
-      "16744    | torch.Size([1, 1000, 786]) | 6\n",
-      "16745    | torch.Size([1, 1000, 754]) | 6\n",
-      "16746    | torch.Size([1, 1000, 777]) | 6\n",
-      "16747    | torch.Size([1, 1000, 805]) | 6\n",
-      "16748    | torch.Size([1, 1000, 754]) | 6\n",
-      "16749    | torch.Size([1, 1000, 762]) | 6\n",
-      "16750    | torch.Size([1, 1000, 777]) | 6\n",
-      "16751    | torch.Size([1, 1000, 762]) | 6\n",
-      "16752    | torch.Size([1, 1000, 776]) | 6\n",
-      "16753    | torch.Size([1, 1000, 754]) | 6\n",
-      "16754    | torch.Size([1, 1000, 780]) | 6\n",
-      "16755    | torch.Size([1, 1000, 777]) | 6\n",
-      "16756    | torch.Size([1, 1000, 754]) | 6\n",
-      "16757    | torch.Size([1, 1000, 774]) | 6\n",
-      "16758    | torch.Size([1, 1000, 754]) | 6\n",
-      "16759    | torch.Size([1, 1000, 801]) | 6\n",
-      "16760    | torch.Size([1, 1000, 754]) | 6\n",
-      "16761    | torch.Size([1, 1000, 754]) | 6\n",
-      "16762    | torch.Size([1, 1000, 777]) | 6\n",
-      "16763    | torch.Size([1, 1000, 777]) | 6\n",
-      "16764    | torch.Size([1, 1000, 766]) | 6\n",
-      "16765    | torch.Size([1, 1000, 754]) | 6\n",
-      "16766    | torch.Size([1, 1000, 772]) | 6\n",
-      "16767    | torch.Size([1, 1000, 784]) | 6\n",
-      "16768    | torch.Size([1, 1000, 754]) | 6\n",
-      "16769    | torch.Size([1, 1000, 754]) | 6\n",
-      "16770    | torch.Size([1, 1000, 798]) | 6\n",
-      "16771    | torch.Size([1, 1000, 802]) | 6\n",
-      "16772    | torch.Size([1, 1000, 754]) | 6\n",
-      "16773    | torch.Size([1, 1000, 802]) | 6\n",
-      "16774    | torch.Size([1, 1000, 754]) | 6\n",
-      "16775    | torch.Size([1, 1000, 769]) | 6\n",
-      "16776    | torch.Size([1, 1000, 774]) | 6\n",
-      "16777    | torch.Size([1, 1000, 779]) | 6\n",
-      "16778    | torch.Size([1, 1000, 754]) | 6\n",
-      "16779    | torch.Size([1, 1000, 781]) | 6\n",
-      "16780    | torch.Size([1, 1000, 777]) | 6\n",
-      "16781    | torch.Size([1, 1000, 754]) | 6\n",
-      "16782    | torch.Size([1, 1000, 777]) | 6\n",
-      "16783    | torch.Size([1, 1000, 754]) | 6\n",
-      "16784    | torch.Size([1, 1000, 779]) | 6\n",
-      "16785    | torch.Size([1, 1000, 762]) | 6\n",
-      "16786    | torch.Size([1, 1000, 762]) | 6\n",
-      "16787    | torch.Size([1, 1000, 754]) | 6\n",
-      "16788    | torch.Size([1, 1000, 754]) | 6\n",
-      "16789    | torch.Size([1, 1000, 762]) | 6\n",
-      "16790    | torch.Size([1, 1000, 754]) | 6\n",
-      "16791    | torch.Size([1, 1000, 781]) | 6\n",
-      "16792    | torch.Size([1, 1000, 754]) | 6\n",
-      "16793    | torch.Size([1, 1000, 754]) | 6\n",
-      "16794    | torch.Size([1, 1000, 754]) | 6\n",
-      "16795    | torch.Size([1, 1000, 754]) | 6\n",
-      "16796    | torch.Size([1, 1000, 772]) | 6\n",
-      "16797    | torch.Size([1, 1000, 780]) | 6\n",
-      "16798    | torch.Size([1, 1000, 754]) | 6\n",
-      "16799    | torch.Size([1, 1000, 754]) | 6\n",
-      "16800    | torch.Size([1, 1000, 754]) | 6\n",
-      "16801    | torch.Size([1, 1000, 771]) | 6\n",
-      "16802    | torch.Size([1, 1000, 777]) | 6\n",
-      "16803    | torch.Size([1, 1000, 777]) | 6\n",
-      "16804    | torch.Size([1, 1000, 762]) | 6\n",
-      "16805    | torch.Size([1, 1000, 754]) | 6\n",
-      "16806    | torch.Size([1, 1000, 782]) | 6\n",
-      "16807    | torch.Size([1, 1000, 754]) | 6\n",
-      "16808    | torch.Size([1, 1000, 754]) | 6\n",
-      "16809    | torch.Size([1, 1000, 754]) | 6\n",
-      "16810    | torch.Size([1, 1000, 762]) | 6\n",
-      "16811    | torch.Size([1, 1000, 787]) | 6\n",
-      "16812    | torch.Size([1, 1000, 777]) | 6\n",
-      "16813    | torch.Size([1, 1000, 785]) | 6\n",
-      "16814    | torch.Size([1, 1000, 762]) | 6\n",
-      "16815    | torch.Size([1, 1000, 754]) | 6\n",
-      "16816    | torch.Size([1, 1000, 802]) | 6\n",
-      "16817    | torch.Size([1, 1000, 772]) | 6\n",
-      "16818    | torch.Size([1, 1000, 762]) | 6\n",
-      "16819    | torch.Size([1, 1000, 767]) | 6\n",
-      "16820    | torch.Size([1, 1000, 781]) | 6\n",
-      "16821    | torch.Size([1, 1000, 787]) | 6\n",
-      "16822    | torch.Size([1, 1000, 762]) | 6\n",
-      "16823    | torch.Size([1, 1000, 777]) | 6\n",
-      "16824    | torch.Size([1, 1000, 788]) | 6\n",
-      "16825    | torch.Size([1, 1000, 768]) | 6\n",
-      "16826    | torch.Size([1, 1000, 754]) | 6\n",
-      "16827    | torch.Size([1, 1000, 754]) | 6\n",
-      "16828    | torch.Size([1, 1000, 762]) | 6\n",
-      "16829    | torch.Size([1, 1000, 754]) | 6\n",
-      "16830    | torch.Size([1, 1000, 796]) | 6\n",
-      "16831    | torch.Size([1, 1000, 752]) | 6\n",
-      "16832    | torch.Size([1, 1000, 777]) | 6\n",
-      "16833    | torch.Size([1, 1000, 764]) | 6\n",
-      "16834    | torch.Size([1, 1000, 754]) | 6\n",
-      "16835    | torch.Size([1, 1000, 754]) | 6\n",
-      "16836    | torch.Size([1, 1000, 777]) | 6\n",
-      "16837    | torch.Size([1, 1000, 754]) | 6\n",
-      "16838    | torch.Size([1, 1000, 769]) | 6\n",
-      "16839    | torch.Size([1, 1000, 762]) | 6\n",
-      "16840    | torch.Size([1, 1000, 754]) | 6\n",
-      "16841    | torch.Size([1, 1000, 754]) | 6\n",
-      "16842    | torch.Size([1, 1000, 762]) | 6\n",
-      "16843    | torch.Size([1, 1000, 802]) | 6\n",
-      "16844    | torch.Size([1, 1000, 802]) | 6\n",
-      "16845    | torch.Size([1, 1000, 754]) | 6\n",
-      "16846    | torch.Size([1, 1000, 777]) | 6\n",
-      "16847    | torch.Size([1, 1000, 762]) | 6\n",
-      "16848    | torch.Size([1, 1000, 776]) | 6\n",
-      "16849    | torch.Size([1, 1000, 754]) | 6\n",
-      "16850    | torch.Size([1, 1000, 809]) | 6\n",
-      "16851    | torch.Size([1, 1000, 754]) | 6\n",
-      "16852    | torch.Size([1, 1000, 771]) | 6\n",
-      "16853    | torch.Size([1, 1000, 754]) | 6\n",
-      "16854    | torch.Size([1, 1000, 769]) | 6\n",
-      "16855    | torch.Size([1, 1000, 785]) | 6\n",
-      "16856    | torch.Size([1, 1000, 754]) | 6\n",
-      "16857    | torch.Size([1, 1000, 752]) | 6\n",
-      "16858    | torch.Size([1, 1000, 762]) | 6\n",
-      "16859    | torch.Size([1, 1000, 777]) | 6\n",
-      "16860    | torch.Size([1, 1000, 776]) | 6\n",
-      "16861    | torch.Size([1, 1000, 754]) | 6\n",
-      "16862    | torch.Size([1, 1000, 754]) | 6\n",
-      "16863    | torch.Size([1, 1000, 778]) | 6\n",
-      "16864    | torch.Size([1, 1000, 770]) | 6\n",
-      "16865    | torch.Size([1, 1000, 754]) | 6\n",
-      "16866    | torch.Size([1, 1000, 774]) | 6\n",
-      "16867    | torch.Size([1, 1000, 806]) | 6\n",
-      "16868    | torch.Size([1, 1000, 777]) | 6\n",
-      "16869    | torch.Size([1, 1000, 778]) | 6\n",
-      "16870    | torch.Size([1, 1000, 754]) | 6\n",
-      "16871    | torch.Size([1, 1000, 773]) | 6\n",
-      "16872    | torch.Size([1, 1000, 754]) | 6\n",
-      "16873    | torch.Size([1, 1000, 777]) | 6\n",
-      "16874    | torch.Size([1, 1000, 754]) | 6\n",
-      "16875    | torch.Size([1, 1000, 777]) | 6\n",
-      "16876    | torch.Size([1, 1000, 754]) | 6\n",
-      "16877    | torch.Size([1, 1000, 754]) | 6\n",
-      "16878    | torch.Size([1, 1000, 754]) | 6\n",
-      "16879    | torch.Size([1, 1000, 754]) | 6\n",
-      "16880    | torch.Size([1, 1000, 783]) | 6\n",
-      "16881    | torch.Size([1, 1000, 762]) | 6\n",
-      "16882    | torch.Size([1, 1000, 777]) | 6\n",
-      "16883    | torch.Size([1, 1000, 762]) | 6\n",
-      "16884    | torch.Size([1, 1000, 777]) | 6\n",
-      "16885    | torch.Size([1, 1000, 802]) | 6\n",
-      "16886    | torch.Size([1, 1000, 754]) | 6\n",
-      "16887    | torch.Size([1, 1000, 754]) | 6\n",
-      "16888    | torch.Size([1, 1000, 768]) | 6\n",
-      "16889    | torch.Size([1, 1000, 769]) | 6\n",
-      "16890    | torch.Size([1, 1000, 754]) | 6\n",
-      "16891    | torch.Size([1, 1000, 773]) | 6\n",
-      "16892    | torch.Size([1, 1000, 754]) | 6\n",
-      "16893    | torch.Size([1, 1000, 771]) | 6\n",
-      "16894    | torch.Size([1, 1000, 754]) | 6\n",
-      "16895    | torch.Size([1, 1000, 754]) | 6\n",
-      "16896    | torch.Size([1, 1000, 810]) | 6\n",
-      "16897    | torch.Size([1, 1000, 777]) | 6\n",
-      "16898    | torch.Size([1, 1000, 754]) | 6\n",
-      "16899    | torch.Size([1, 1000, 754]) | 6\n",
-      "16900    | torch.Size([1, 1000, 806]) | 6\n",
-      "16901    | torch.Size([1, 1000, 788]) | 6\n",
-      "16902    | torch.Size([1, 1000, 777]) | 6\n",
-      "16903    | torch.Size([1, 1000, 770]) | 6\n",
-      "16904    | torch.Size([1, 1000, 785]) | 6\n",
-      "16905    | torch.Size([1, 1000, 777]) | 6\n",
-      "16906    | torch.Size([1, 1000, 777]) | 6\n",
-      "16907    | torch.Size([1, 1000, 754]) | 6\n",
-      "16908    | torch.Size([1, 1000, 762]) | 6\n",
-      "16909    | torch.Size([1, 1000, 767]) | 6\n",
-      "16910    | torch.Size([1, 1000, 782]) | 6\n",
-      "16911    | torch.Size([1, 1000, 777]) | 6\n",
-      "16912    | torch.Size([1, 1000, 754]) | 6\n",
-      "16913    | torch.Size([1, 1000, 754]) | 6\n",
-      "16914    | torch.Size([1, 1000, 754]) | 6\n",
-      "16915    | torch.Size([1, 1000, 792]) | 6\n",
-      "16916    | torch.Size([1, 1000, 754]) | 6\n",
-      "16917    | torch.Size([1, 1000, 754]) | 6\n",
-      "16918    | torch.Size([1, 1000, 767]) | 6\n",
-      "16919    | torch.Size([1, 1000, 754]) | 6\n",
-      "16920    | torch.Size([1, 1000, 754]) | 6\n",
-      "16921    | torch.Size([1, 1000, 777]) | 6\n",
-      "16922    | torch.Size([1, 1000, 754]) | 6\n",
-      "16923    | torch.Size([1, 1000, 754]) | 6\n",
-      "16924    | torch.Size([1, 1000, 754]) | 6\n",
-      "16925    | torch.Size([1, 1000, 777]) | 6\n",
-      "16926    | torch.Size([1, 1000, 787]) | 6\n",
-      "16927    | torch.Size([1, 1000, 778]) | 6\n",
-      "16928    | torch.Size([1, 1000, 777]) | 6\n",
-      "16929    | torch.Size([1, 1000, 754]) | 6\n",
-      "16930    | torch.Size([1, 1000, 754]) | 6\n",
-      "16931    | torch.Size([1, 1000, 780]) | 6\n",
-      "16932    | torch.Size([1, 1000, 777]) | 6\n",
-      "16933    | torch.Size([1, 1000, 786]) | 6\n",
-      "16934    | torch.Size([1, 1000, 777]) | 6\n",
-      "16935    | torch.Size([1, 1000, 754]) | 6\n",
-      "16936    | torch.Size([1, 1000, 778]) | 6\n",
-      "16937    | torch.Size([1, 1000, 784]) | 6\n",
-      "16938    | torch.Size([1, 1000, 777]) | 6\n",
-      "16939    | torch.Size([1, 1000, 777]) | 6\n",
-      "16940    | torch.Size([1, 1000, 790]) | 6\n",
-      "16941    | torch.Size([1, 1000, 777]) | 6\n",
-      "16942    | torch.Size([1, 1000, 777]) | 6\n",
-      "16943    | torch.Size([1, 1000, 766]) | 6\n",
-      "16944    | torch.Size([1, 1000, 754]) | 6\n",
-      "16945    | torch.Size([1, 1000, 772]) | 6\n",
-      "16946    | torch.Size([1, 1000, 754]) | 6\n",
-      "16947    | torch.Size([1, 1000, 784]) | 6\n",
-      "16948    | torch.Size([1, 1000, 770]) | 6\n",
-      "16949    | torch.Size([1, 1000, 754]) | 6\n",
-      "16950    | torch.Size([1, 1000, 762]) | 6\n",
-      "16951    | torch.Size([1, 1000, 754]) | 6\n",
-      "16952    | torch.Size([1, 1000, 754]) | 6\n",
-      "16953    | torch.Size([1, 1000, 777]) | 6\n",
-      "16954    | torch.Size([1, 1000, 754]) | 6\n",
-      "16955    | torch.Size([1, 1000, 799]) | 6\n",
-      "16956    | torch.Size([1, 1000, 794]) | 6\n",
-      "16957    | torch.Size([1, 1000, 774]) | 6\n",
-      "16958    | torch.Size([1, 1000, 754]) | 6\n",
-      "16959    | torch.Size([1, 1000, 765]) | 6\n",
-      "16960    | torch.Size([1, 1000, 754]) | 6\n",
-      "16961    | torch.Size([1, 1000, 754]) | 6\n",
-      "16962    | torch.Size([1, 1000, 762]) | 6\n",
-      "16963    | torch.Size([1, 1000, 754]) | 6\n",
-      "16964    | torch.Size([1, 1000, 777]) | 6\n",
-      "16965    | torch.Size([1, 1000, 754]) | 6\n",
-      "16966    | torch.Size([1, 1000, 776]) | 6\n",
-      "16967    | torch.Size([1, 1000, 762]) | 6\n",
-      "16968    | torch.Size([1, 1000, 646]) | 6\n",
-      "16969    | torch.Size([1, 1000, 754]) | 6\n",
-      "16970    | torch.Size([1, 1000, 793]) | 6\n",
-      "16971    | torch.Size([1, 1000, 777]) | 6\n",
-      "16972    | torch.Size([1, 1000, 771]) | 6\n",
-      "16973    | torch.Size([1, 1000, 777]) | 6\n",
-      "16974    | torch.Size([1, 1000, 762]) | 6\n",
-      "16975    | torch.Size([1, 1000, 754]) | 6\n",
-      "16976    | torch.Size([1, 1000, 783]) | 6\n",
-      "16977    | torch.Size([1, 1000, 802]) | 6\n",
-      "16978    | torch.Size([1, 1000, 762]) | 6\n",
-      "16979    | torch.Size([1, 1000, 771]) | 6\n",
-      "16980    | torch.Size([1, 1000, 786]) | 6\n",
-      "16981    | torch.Size([1, 1000, 754]) | 6\n",
-      "16982    | torch.Size([1, 1000, 754]) | 6\n",
-      "16983    | torch.Size([1, 1000, 762]) | 6\n",
-      "16984    | torch.Size([1, 1000, 786]) | 6\n",
-      "16985    | torch.Size([1, 1000, 762]) | 6\n",
-      "16986    | torch.Size([1, 1000, 754]) | 6\n",
-      "16987    | torch.Size([1, 1000, 769]) | 6\n",
-      "16988    | torch.Size([1, 1000, 762]) | 6\n",
-      "16989    | torch.Size([1, 1000, 762]) | 6\n",
-      "16990    | torch.Size([1, 1000, 777]) | 6\n",
-      "16991    | torch.Size([1, 1000, 762]) | 6\n",
-      "16992    | torch.Size([1, 1000, 777]) | 6\n",
-      "16993    | torch.Size([1, 1000, 754]) | 6\n",
-      "16994    | torch.Size([1, 1000, 772]) | 6\n",
-      "16995    | torch.Size([1, 1000, 754]) | 6\n",
-      "16996    | torch.Size([1, 1000, 762]) | 6\n",
-      "16997    | torch.Size([1, 1000, 802]) | 6\n",
-      "16998    | torch.Size([1, 1000, 754]) | 6\n",
-      "16999    | torch.Size([1, 1000, 754]) | 6\n",
-      "17000    | torch.Size([1, 1000, 754]) | 6\n",
-      "17001    | torch.Size([1, 1000, 778]) | 6\n",
-      "17002    | torch.Size([1, 1000, 754]) | 6\n",
-      "17003    | torch.Size([1, 1000, 777]) | 6\n",
-      "17004    | torch.Size([1, 1000, 754]) | 6\n",
-      "17005    | torch.Size([1, 1000, 788]) | 6\n",
-      "17006    | torch.Size([1, 1000, 754]) | 6\n",
-      "17007    | torch.Size([1, 1000, 780]) | 6\n",
-      "17008    | torch.Size([1, 1000, 754]) | 6\n",
-      "17009    | torch.Size([1, 1000, 762]) | 6\n",
-      "17010    | torch.Size([1, 1000, 777]) | 6\n",
-      "17011    | torch.Size([1, 1000, 780]) | 6\n",
-      "17012    | torch.Size([1, 1000, 776]) | 6\n",
-      "17013    | torch.Size([1, 1000, 772]) | 6\n",
-      "17014    | torch.Size([1, 1000, 754]) | 6\n",
-      "17015    | torch.Size([1, 1000, 762]) | 6\n",
-      "17016    | torch.Size([1, 1000, 790]) | 6\n",
-      "17017    | torch.Size([1, 1000, 772]) | 6\n",
-      "17018    | torch.Size([1, 1000, 754]) | 6\n",
-      "17019    | torch.Size([1, 1000, 765]) | 6\n",
-      "17020    | torch.Size([1, 1000, 770]) | 6\n",
-      "17021    | torch.Size([1, 1000, 754]) | 6\n",
-      "17022    | torch.Size([1, 1000, 777]) | 6\n",
-      "17023    | torch.Size([1, 1000, 762]) | 6\n",
-      "17024    | torch.Size([1, 1000, 762]) | 6\n",
-      "17025    | torch.Size([1, 1000, 754]) | 6\n",
-      "17026    | torch.Size([1, 1000, 752]) | 6\n",
-      "17027    | torch.Size([1, 1000, 754]) | 6\n",
-      "17028    | torch.Size([1, 1000, 768]) | 6\n",
-      "17029    | torch.Size([1, 1000, 762]) | 6\n",
-      "17030    | torch.Size([1, 1000, 754]) | 6\n",
-      "17031    | torch.Size([1, 1000, 754]) | 6\n",
-      "17032    | torch.Size([1, 1000, 771]) | 6\n",
-      "17033    | torch.Size([1, 1000, 789]) | 6\n",
-      "17034    | torch.Size([1, 1000, 777]) | 6\n",
-      "17035    | torch.Size([1, 1000, 752]) | 6\n",
-      "17036    | torch.Size([1, 1000, 754]) | 6\n",
-      "17037    | torch.Size([1, 1000, 777]) | 6\n",
-      "17038    | torch.Size([1, 1000, 786]) | 6\n",
-      "17039    | torch.Size([1, 1000, 762]) | 6\n",
-      "17040    | torch.Size([1, 1000, 797]) | 6\n",
-      "17041    | torch.Size([1, 1000, 796]) | 6\n",
-      "17042    | torch.Size([1, 1000, 771]) | 6\n",
-      "17043    | torch.Size([1, 1000, 754]) | 6\n",
-      "17044    | torch.Size([1, 1000, 785]) | 6\n",
-      "17045    | torch.Size([1, 1000, 754]) | 6\n",
-      "17046    | torch.Size([1, 1000, 770]) | 6\n",
-      "17047    | torch.Size([1, 1000, 777]) | 6\n",
-      "17048    | torch.Size([1, 1000, 754]) | 6\n",
-      "17049    | torch.Size([1, 1000, 754]) | 6\n",
-      "17050    | torch.Size([1, 1000, 772]) | 6\n",
-      "17051    | torch.Size([1, 1000, 754]) | 6\n",
-      "17052    | torch.Size([1, 1000, 754]) | 6\n",
-      "17053    | torch.Size([1, 1000, 767]) | 6\n",
-      "17054    | torch.Size([1, 1000, 762]) | 6\n",
-      "17055    | torch.Size([1, 1000, 778]) | 6\n",
-      "17056    | torch.Size([1, 1000, 754]) | 6\n",
-      "17057    | torch.Size([1, 1000, 777]) | 6\n",
-      "17058    | torch.Size([1, 1000, 754]) | 6\n",
-      "17059    | torch.Size([1, 1000, 762]) | 6\n",
-      "17060    | torch.Size([1, 1000, 771]) | 6\n",
-      "17061    | torch.Size([1, 1000, 754]) | 6\n",
-      "17062    | torch.Size([1, 1000, 770]) | 6\n",
-      "17063    | torch.Size([1, 1000, 777]) | 6\n",
-      "17064    | torch.Size([1, 1000, 762]) | 6\n",
-      "17065    | torch.Size([1, 1000, 778]) | 6\n",
-      "17066    | torch.Size([1, 1000, 754]) | 6\n",
-      "17067    | torch.Size([1, 1000, 808]) | 6\n",
-      "17068    | torch.Size([1, 1000, 773]) | 6\n",
-      "17069    | torch.Size([1, 1000, 777]) | 6\n",
-      "17070    | torch.Size([1, 1000, 802]) | 6\n",
-      "17071    | torch.Size([1, 1000, 754]) | 6\n",
-      "17072    | torch.Size([1, 1000, 777]) | 6\n",
-      "17073    | torch.Size([1, 1000, 772]) | 6\n",
-      "17074    | torch.Size([1, 1000, 785]) | 6\n",
-      "17075    | torch.Size([1, 1000, 754]) | 6\n",
-      "17076    | torch.Size([1, 1000, 803]) | 6\n",
-      "17077    | torch.Size([1, 1000, 813]) | 6\n",
-      "17078    | torch.Size([1, 1000, 754]) | 6\n",
-      "17079    | torch.Size([1, 1000, 771]) | 6\n",
-      "17080    | torch.Size([1, 1000, 841]) | 6\n",
-      "17081    | torch.Size([1, 1000, 777]) | 6\n",
-      "17082    | torch.Size([1, 1000, 757]) | 6\n",
-      "17083    | torch.Size([1, 1000, 786]) | 6\n",
-      "17084    | torch.Size([1, 1000, 777]) | 6\n",
-      "17085    | torch.Size([1, 1000, 754]) | 6\n",
-      "17086    | torch.Size([1, 1000, 754]) | 6\n",
-      "17087    | torch.Size([1, 1000, 777]) | 6\n",
-      "17088    | torch.Size([1, 1000, 754]) | 6\n",
-      "17089    | torch.Size([1, 1000, 754]) | 6\n",
-      "17090    | torch.Size([1, 1000, 771]) | 6\n",
-      "17091    | torch.Size([1, 1000, 767]) | 6\n",
-      "17092    | torch.Size([1, 1000, 754]) | 6\n",
-      "17093    | torch.Size([1, 1000, 754]) | 6\n",
-      "17094    | torch.Size([1, 1000, 762]) | 6\n",
-      "17095    | torch.Size([1, 1000, 787]) | 6\n",
-      "17096    | torch.Size([1, 1000, 777]) | 6\n",
-      "17097    | torch.Size([1, 1000, 783]) | 6\n",
-      "17098    | torch.Size([1, 1000, 754]) | 6\n",
-      "17099    | torch.Size([1, 1000, 777]) | 6\n",
-      "17100    | torch.Size([1, 1000, 777]) | 6\n",
-      "17101    | torch.Size([1, 1000, 762]) | 6\n",
-      "17102    | torch.Size([1, 1000, 777]) | 6\n",
-      "17103    | torch.Size([1, 1000, 771]) | 6\n",
-      "17104    | torch.Size([1, 1000, 754]) | 6\n",
-      "17105    | torch.Size([1, 1000, 754]) | 6\n",
-      "17106    | torch.Size([1, 1000, 754]) | 6\n",
-      "17107    | torch.Size([1, 1000, 777]) | 6\n",
-      "17108    | torch.Size([1, 1000, 782]) | 6\n",
-      "17109    | torch.Size([1, 1000, 754]) | 6\n",
-      "17110    | torch.Size([1, 1000, 786]) | 6\n",
-      "17111    | torch.Size([1, 1000, 777]) | 6\n",
-      "17112    | torch.Size([1, 1000, 770]) | 6\n",
-      "17113    | torch.Size([1, 1000, 770]) | 6\n",
-      "17114    | torch.Size([1, 1000, 777]) | 6\n",
-      "17115    | torch.Size([1, 1000, 762]) | 6\n",
-      "17116    | torch.Size([1, 1000, 754]) | 6\n",
-      "17117    | torch.Size([1, 1000, 777]) | 6\n",
-      "17118    | torch.Size([1, 1000, 771]) | 6\n",
-      "17119    | torch.Size([1, 1000, 754]) | 6\n",
-      "17120    | torch.Size([1, 1000, 754]) | 6\n",
-      "17121    | torch.Size([1, 1000, 762]) | 6\n",
-      "17122    | torch.Size([1, 1000, 754]) | 6\n",
-      "17123    | torch.Size([1, 1000, 802]) | 6\n",
-      "17124    | torch.Size([1, 1000, 776]) | 6\n",
-      "17125    | torch.Size([1, 1000, 777]) | 6\n",
-      "17126    | torch.Size([1, 1000, 777]) | 6\n",
-      "17127    | torch.Size([1, 1000, 754]) | 6\n",
-      "17128    | torch.Size([1, 1000, 777]) | 6\n",
-      "17129    | torch.Size([1, 1000, 754]) | 6\n",
-      "17130    | torch.Size([1, 1000, 754]) | 6\n",
-      "17131    | torch.Size([1, 1000, 754]) | 6\n",
-      "17132    | torch.Size([1, 1000, 777]) | 6\n",
-      "17133    | torch.Size([1, 1000, 762]) | 6\n",
-      "17134    | torch.Size([1, 1000, 777]) | 6\n",
-      "17135    | torch.Size([1, 1000, 754]) | 6\n",
-      "17136    | torch.Size([1, 1000, 761]) | 6\n",
-      "17137    | torch.Size([1, 1000, 762]) | 6\n",
-      "17138    | torch.Size([1, 1000, 762]) | 6\n",
-      "17139    | torch.Size([1, 1000, 754]) | 6\n",
-      "17140    | torch.Size([1, 1000, 771]) | 6\n",
-      "17141    | torch.Size([1, 1000, 766]) | 6\n",
-      "17142    | torch.Size([1, 1000, 754]) | 6\n",
-      "17143    | torch.Size([1, 1000, 777]) | 6\n",
-      "17144    | torch.Size([1, 1000, 762]) | 6\n",
-      "17145    | torch.Size([1, 1000, 767]) | 6\n",
-      "17146    | torch.Size([1, 1000, 773]) | 6\n",
-      "17147    | torch.Size([1, 1000, 778]) | 6\n",
-      "17148    | torch.Size([1, 1000, 777]) | 6\n",
-      "17149    | torch.Size([1, 1000, 777]) | 6\n",
-      "17150    | torch.Size([1, 1000, 754]) | 6\n",
-      "17151    | torch.Size([1, 1000, 754]) | 6\n",
-      "17152    | torch.Size([1, 1000, 754]) | 6\n",
-      "17153    | torch.Size([1, 1000, 754]) | 6\n",
-      "17154    | torch.Size([1, 1000, 754]) | 6\n",
-      "17155    | torch.Size([1, 1000, 786]) | 6\n",
-      "17156    | torch.Size([1, 1000, 754]) | 6\n",
-      "17157    | torch.Size([1, 1000, 754]) | 6\n",
-      "17158    | torch.Size([1, 1000, 777]) | 6\n",
-      "17159    | torch.Size([1, 1000, 777]) | 6\n",
-      "17160    | torch.Size([1, 1000, 777]) | 6\n",
-      "17161    | torch.Size([1, 1000, 754]) | 6\n",
-      "17162    | torch.Size([1, 1000, 772]) | 6\n",
-      "17163    | torch.Size([1, 1000, 775]) | 6\n",
-      "17164    | torch.Size([1, 1000, 754]) | 6\n",
-      "17165    | torch.Size([1, 1000, 783]) | 6\n",
-      "17166    | torch.Size([1, 1000, 754]) | 6\n",
-      "17167    | torch.Size([1, 1000, 777]) | 6\n",
-      "17168    | torch.Size([1, 1000, 784]) | 6\n",
-      "17169    | torch.Size([1, 1000, 768]) | 6\n",
-      "17170    | torch.Size([1, 1000, 768]) | 6\n",
-      "17171    | torch.Size([1, 1000, 802]) | 6\n",
-      "17172    | torch.Size([1, 1000, 754]) | 6\n",
-      "17173    | torch.Size([1, 1000, 780]) | 6\n",
-      "17174    | torch.Size([1, 1000, 780]) | 6\n",
-      "17175    | torch.Size([1, 1000, 777]) | 6\n",
-      "17176    | torch.Size([1, 1000, 754]) | 6\n",
-      "17177    | torch.Size([1, 1000, 754]) | 6\n",
-      "17178    | torch.Size([1, 1000, 785]) | 6\n",
-      "17179    | torch.Size([1, 1000, 771]) | 6\n",
-      "17180    | torch.Size([1, 1000, 776]) | 6\n",
-      "17181    | torch.Size([1, 1000, 777]) | 6\n",
-      "17182    | torch.Size([1, 1000, 754]) | 6\n",
-      "17183    | torch.Size([1, 1000, 754]) | 6\n",
-      "17184    | torch.Size([1, 1000, 769]) | 6\n",
-      "17185    | torch.Size([1, 1000, 777]) | 6\n",
-      "17186    | torch.Size([1, 1000, 777]) | 6\n",
-      "17187    | torch.Size([1, 1000, 769]) | 6\n",
-      "17188    | torch.Size([1, 1000, 793]) | 6\n",
-      "17189    | torch.Size([1, 1000, 754]) | 6\n",
-      "17190    | torch.Size([1, 1000, 772]) | 6\n",
-      "17191    | torch.Size([1, 1000, 762]) | 6\n",
-      "17192    | torch.Size([1, 1000, 754]) | 6\n",
-      "17193    | torch.Size([1, 1000, 754]) | 6\n",
-      "17194    | torch.Size([1, 1000, 773]) | 6\n",
-      "17195    | torch.Size([1, 1000, 762]) | 6\n",
-      "17196    | torch.Size([1, 1000, 775]) | 6\n",
-      "17197    | torch.Size([1, 1000, 754]) | 6\n",
-      "17198    | torch.Size([1, 1000, 754]) | 6\n",
-      "17199    | torch.Size([1, 1000, 754]) | 6\n",
-      "17200    | torch.Size([1, 1000, 772]) | 6\n",
-      "17201    | torch.Size([1, 1000, 754]) | 6\n",
-      "17202    | torch.Size([1, 1000, 754]) | 6\n",
-      "17203    | torch.Size([1, 1000, 755]) | 6\n",
-      "17204    | torch.Size([1, 1000, 754]) | 6\n",
-      "17205    | torch.Size([1, 1000, 754]) | 6\n",
-      "17206    | torch.Size([1, 1000, 762]) | 6\n",
-      "17207    | torch.Size([1, 1000, 754]) | 6\n",
-      "17208    | torch.Size([1, 1000, 754]) | 6\n",
-      "17209    | torch.Size([1, 1000, 754]) | 6\n",
-      "17210    | torch.Size([1, 1000, 754]) | 6\n",
-      "17211    | torch.Size([1, 1000, 754]) | 6\n",
-      "17212    | torch.Size([1, 1000, 843]) | 6\n",
-      "17213    | torch.Size([1, 1000, 762]) | 6\n",
-      "17214    | torch.Size([1, 1000, 777]) | 6\n",
-      "17215    | torch.Size([1, 1000, 754]) | 6\n",
-      "17216    | torch.Size([1, 1000, 777]) | 6\n",
-      "17217    | torch.Size([1, 1000, 762]) | 6\n",
-      "17218    | torch.Size([1, 1000, 754]) | 6\n",
-      "17219    | torch.Size([1, 1000, 762]) | 6\n",
-      "17220    | torch.Size([1, 1000, 754]) | 6\n",
-      "17221    | torch.Size([1, 1000, 773]) | 6\n",
-      "17222    | torch.Size([1, 1000, 777]) | 6\n",
-      "17223    | torch.Size([1, 1000, 762]) | 6\n",
-      "17224    | torch.Size([1, 1000, 762]) | 6\n",
-      "17225    | torch.Size([1, 1000, 754]) | 6\n",
-      "17226    | torch.Size([1, 1000, 762]) | 6\n",
-      "17227    | torch.Size([1, 1000, 754]) | 6\n",
-      "17228    | torch.Size([1, 1000, 754]) | 6\n",
-      "17229    | torch.Size([1, 1000, 780]) | 6\n",
-      "17230    | torch.Size([1, 1000, 777]) | 6\n",
-      "17231    | torch.Size([1, 1000, 762]) | 6\n",
-      "17232    | torch.Size([1, 1000, 762]) | 6\n",
-      "17233    | torch.Size([1, 1000, 773]) | 6\n",
-      "17234    | torch.Size([1, 1000, 772]) | 6\n",
-      "17235    | torch.Size([1, 1000, 754]) | 6\n",
-      "17236    | torch.Size([1, 1000, 777]) | 6\n",
-      "17237    | torch.Size([1, 1000, 776]) | 6\n",
-      "17238    | torch.Size([1, 1000, 754]) | 6\n",
-      "17239    | torch.Size([1, 1000, 777]) | 6\n",
-      "17240    | torch.Size([1, 1000, 771]) | 6\n",
-      "17241    | torch.Size([1, 1000, 777]) | 6\n",
-      "17242    | torch.Size([1, 1000, 754]) | 6\n",
-      "17243    | torch.Size([1, 1000, 777]) | 6\n",
-      "17244    | torch.Size([1, 1000, 766]) | 6\n",
-      "17245    | torch.Size([1, 1000, 754]) | 6\n",
-      "17246    | torch.Size([1, 1000, 752]) | 6\n",
-      "17247    | torch.Size([1, 1000, 754]) | 6\n",
-      "17248    | torch.Size([1, 1000, 767]) | 6\n",
-      "17249    | torch.Size([1, 1000, 777]) | 6\n",
-      "17250    | torch.Size([1, 1000, 774]) | 6\n",
-      "17251    | torch.Size([1, 1000, 777]) | 6\n",
-      "17252    | torch.Size([1, 1000, 768]) | 6\n",
-      "17253    | torch.Size([1, 1000, 782]) | 6\n",
-      "17254    | torch.Size([1, 1000, 762]) | 6\n",
-      "17255    | torch.Size([1, 1000, 754]) | 6\n",
-      "17256    | torch.Size([1, 1000, 791]) | 6\n",
-      "17257    | torch.Size([1, 1000, 754]) | 6\n",
-      "17258    | torch.Size([1, 1000, 802]) | 6\n",
-      "17259    | torch.Size([1, 1000, 777]) | 6\n",
-      "17260    | torch.Size([1, 1000, 786]) | 6\n",
-      "17261    | torch.Size([1, 1000, 777]) | 6\n",
-      "17262    | torch.Size([1, 1000, 769]) | 6\n",
-      "17263    | torch.Size([1, 1000, 777]) | 6\n",
-      "17264    | torch.Size([1, 1000, 754]) | 6\n",
-      "17265    | torch.Size([1, 1000, 777]) | 6\n",
-      "17266    | torch.Size([1, 1000, 777]) | 6\n",
-      "17267    | torch.Size([1, 1000, 767]) | 6\n",
-      "17268    | torch.Size([1, 1000, 754]) | 6\n",
-      "17269    | torch.Size([1, 1000, 754]) | 6\n",
-      "17270    | torch.Size([1, 1000, 776]) | 6\n",
-      "17271    | torch.Size([1, 1000, 772]) | 6\n",
-      "17272    | torch.Size([1, 1000, 777]) | 6\n",
-      "17273    | torch.Size([1, 1000, 754]) | 6\n",
-      "17274    | torch.Size([1, 1000, 754]) | 6\n",
-      "17275    | torch.Size([1, 1000, 822]) | 6\n",
-      "17276    | torch.Size([1, 1000, 754]) | 6\n",
-      "17277    | torch.Size([1, 1000, 777]) | 6\n",
-      "17278    | torch.Size([1, 1000, 761]) | 6\n",
-      "17279    | torch.Size([1, 1000, 754]) | 6\n",
-      "17280    | torch.Size([1, 1000, 772]) | 6\n",
-      "17281    | torch.Size([1, 1000, 775]) | 6\n",
-      "17282    | torch.Size([1, 1000, 757]) | 6\n",
-      "17283    | torch.Size([1, 1000, 777]) | 6\n",
-      "17284    | torch.Size([1, 1000, 762]) | 6\n",
-      "17285    | torch.Size([1, 1000, 777]) | 6\n",
-      "17286    | torch.Size([1, 1000, 780]) | 6\n",
-      "17287    | torch.Size([1, 1000, 752]) | 6\n",
-      "17288    | torch.Size([1, 1000, 780]) | 6\n",
-      "17289    | torch.Size([1, 1000, 771]) | 6\n",
-      "17290    | torch.Size([1, 1000, 772]) | 6\n",
-      "17291    | torch.Size([1, 1000, 754]) | 6\n",
-      "17292    | torch.Size([1, 1000, 754]) | 6\n",
-      "17293    | torch.Size([1, 1000, 754]) | 6\n",
-      "17294    | torch.Size([1, 1000, 787]) | 6\n",
-      "17295    | torch.Size([1, 1000, 767]) | 6\n",
-      "17296    | torch.Size([1, 1000, 754]) | 6\n",
-      "17297    | torch.Size([1, 1000, 787]) | 6\n",
-      "17298    | torch.Size([1, 1000, 786]) | 6\n",
-      "17299    | torch.Size([1, 1000, 762]) | 6\n",
-      "17300    | torch.Size([1, 1000, 762]) | 6\n",
-      "17301    | torch.Size([1, 1000, 754]) | 6\n",
-      "17302    | torch.Size([1, 1000, 754]) | 6\n",
-      "17303    | torch.Size([1, 1000, 772]) | 6\n",
-      "17304    | torch.Size([1, 1000, 754]) | 6\n",
-      "17305    | torch.Size([1, 1000, 754]) | 6\n",
-      "17306    | torch.Size([1, 1000, 777]) | 6\n",
-      "17307    | torch.Size([1, 1000, 782]) | 6\n",
-      "17308    | torch.Size([1, 1000, 789]) | 6\n",
-      "17309    | torch.Size([1, 1000, 764]) | 6\n",
-      "17310    | torch.Size([1, 1000, 754]) | 6\n",
-      "17311    | torch.Size([1, 1000, 770]) | 6\n",
-      "17312    | torch.Size([1, 1000, 782]) | 6\n",
-      "17313    | torch.Size([1, 1000, 772]) | 6\n",
-      "17314    | torch.Size([1, 1000, 771]) | 6\n",
-      "17315    | torch.Size([1, 1000, 777]) | 6\n",
-      "17316    | torch.Size([1, 1000, 752]) | 6\n",
-      "17317    | torch.Size([1, 1000, 777]) | 6\n",
-      "17318    | torch.Size([1, 1000, 752]) | 6\n",
-      "17319    | torch.Size([1, 1000, 754]) | 6\n",
-      "17320    | torch.Size([1, 1000, 762]) | 6\n",
-      "17321    | torch.Size([1, 1000, 776]) | 6\n",
-      "17322    | torch.Size([1, 1000, 777]) | 6\n",
-      "17323    | torch.Size([1, 1000, 754]) | 6\n",
-      "17324    | torch.Size([1, 1000, 754]) | 6\n",
-      "17325    | torch.Size([1, 1000, 754]) | 6\n",
-      "17326    | torch.Size([1, 1000, 777]) | 6\n",
-      "17327    | torch.Size([1, 1000, 775]) | 6\n",
-      "17328    | torch.Size([1, 1000, 762]) | 6\n",
-      "17329    | torch.Size([1, 1000, 808]) | 6\n",
-      "17330    | torch.Size([1, 1000, 754]) | 6\n",
-      "17331    | torch.Size([1, 1000, 777]) | 6\n",
-      "17332    | torch.Size([1, 1000, 754]) | 6\n",
-      "17333    | torch.Size([1, 1000, 754]) | 6\n",
-      "17334    | torch.Size([1, 1000, 762]) | 6\n",
-      "17335    | torch.Size([1, 1000, 754]) | 6\n",
-      "17336    | torch.Size([1, 1000, 779]) | 6\n",
-      "17337    | torch.Size([1, 1000, 754]) | 6\n",
-      "17338    | torch.Size([1, 1000, 723]) | 6\n",
-      "17339    | torch.Size([1, 1000, 762]) | 6\n",
-      "17340    | torch.Size([1, 1000, 762]) | 6\n",
-      "17341    | torch.Size([1, 1000, 754]) | 6\n",
-      "17342    | torch.Size([1, 1000, 787]) | 6\n",
-      "17343    | torch.Size([1, 1000, 777]) | 6\n",
-      "17344    | torch.Size([1, 1000, 773]) | 6\n",
-      "17345    | torch.Size([1, 1000, 720]) | 6\n",
-      "17346    | torch.Size([1, 1000, 777]) | 6\n",
-      "17347    | torch.Size([1, 1000, 754]) | 6\n",
-      "17348    | torch.Size([1, 1000, 762]) | 6\n",
-      "17349    | torch.Size([1, 1000, 774]) | 6\n",
-      "17350    | torch.Size([1, 1000, 754]) | 6\n",
-      "17351    | torch.Size([1, 1000, 754]) | 6\n",
-      "17352    | torch.Size([1, 1000, 754]) | 6\n",
-      "17353    | torch.Size([1, 1000, 754]) | 6\n",
-      "17354    | torch.Size([1, 1000, 787]) | 6\n",
-      "17355    | torch.Size([1, 1000, 754]) | 6\n",
-      "17356    | torch.Size([1, 1000, 772]) | 6\n",
-      "17357    | torch.Size([1, 1000, 773]) | 6\n",
-      "17358    | torch.Size([1, 1000, 781]) | 6\n",
-      "17359    | torch.Size([1, 1000, 777]) | 6\n",
-      "17360    | torch.Size([1, 1000, 754]) | 6\n",
-      "17361    | torch.Size([1, 1000, 757]) | 6\n",
-      "17362    | torch.Size([1, 1000, 777]) | 6\n",
-      "17363    | torch.Size([1, 1000, 768]) | 6\n",
-      "17364    | torch.Size([1, 1000, 777]) | 6\n",
-      "17365    | torch.Size([1, 1000, 802]) | 6\n",
-      "17366    | torch.Size([1, 1000, 762]) | 6\n",
-      "17367    | torch.Size([1, 1000, 777]) | 6\n",
-      "17368    | torch.Size([1, 1000, 754]) | 6\n",
-      "17369    | torch.Size([1, 1000, 776]) | 6\n",
-      "17370    | torch.Size([1, 1000, 762]) | 6\n",
-      "17371    | torch.Size([1, 1000, 762]) | 6\n",
-      "17372    | torch.Size([1, 1000, 762]) | 6\n",
-      "17373    | torch.Size([1, 1000, 777]) | 6\n",
-      "17374    | torch.Size([1, 1000, 754]) | 6\n",
-      "17375    | torch.Size([1, 1000, 784]) | 6\n",
-      "17376    | torch.Size([1, 1000, 754]) | 6\n",
-      "17377    | torch.Size([1, 1000, 754]) | 6\n",
-      "17378    | torch.Size([1, 1000, 777]) | 6\n",
-      "17379    | torch.Size([1, 1000, 754]) | 6\n",
-      "17380    | torch.Size([1, 1000, 777]) | 6\n",
-      "17381    | torch.Size([1, 1000, 754]) | 6\n",
-      "17382    | torch.Size([1, 1000, 775]) | 6\n",
-      "17383    | torch.Size([1, 1000, 777]) | 6\n",
-      "17384    | torch.Size([1, 1000, 777]) | 6\n",
-      "17385    | torch.Size([1, 1000, 754]) | 6\n",
-      "17386    | torch.Size([1, 1000, 754]) | 6\n",
-      "17387    | torch.Size([1, 1000, 754]) | 6\n",
-      "17388    | torch.Size([1, 1000, 754]) | 6\n",
-      "17389    | torch.Size([1, 1000, 783]) | 6\n",
-      "17390    | torch.Size([1, 1000, 762]) | 6\n",
-      "17391    | torch.Size([1, 1000, 777]) | 6\n",
-      "17392    | torch.Size([1, 1000, 776]) | 6\n",
-      "17393    | torch.Size([1, 1000, 759]) | 6\n",
-      "17394    | torch.Size([1, 1000, 754]) | 6\n",
-      "17395    | torch.Size([1, 1000, 754]) | 6\n",
-      "17396    | torch.Size([1, 1000, 754]) | 6\n",
-      "17397    | torch.Size([1, 1000, 754]) | 6\n",
-      "17398    | torch.Size([1, 1000, 777]) | 6\n",
-      "17399    | torch.Size([1, 1000, 754]) | 6\n",
-      "17400    | torch.Size([1, 1000, 762]) | 6\n",
-      "17401    | torch.Size([1, 1000, 762]) | 6\n",
-      "17402    | torch.Size([1, 1000, 754]) | 6\n",
-      "17403    | torch.Size([1, 1000, 762]) | 6\n",
-      "17404    | torch.Size([1, 1000, 762]) | 6\n",
-      "17405    | torch.Size([1, 1000, 754]) | 6\n",
-      "17406    | torch.Size([1, 1000, 777]) | 6\n",
-      "17407    | torch.Size([1, 1000, 762]) | 6\n",
-      "17408    | torch.Size([1, 1000, 761]) | 6\n",
-      "17409    | torch.Size([1, 1000, 754]) | 6\n",
-      "17410    | torch.Size([1, 1000, 762]) | 6\n",
-      "17411    | torch.Size([1, 1000, 754]) | 6\n",
-      "17412    | torch.Size([1, 1000, 802]) | 6\n",
-      "17413    | torch.Size([1, 1000, 777]) | 6\n",
-      "17414    | torch.Size([1, 1000, 779]) | 6\n",
-      "17415    | torch.Size([1, 1000, 754]) | 6\n",
-      "17416    | torch.Size([1, 1000, 777]) | 6\n",
-      "17417    | torch.Size([1, 1000, 782]) | 6\n",
-      "17418    | torch.Size([1, 1000, 761]) | 6\n",
-      "17419    | torch.Size([1, 1000, 777]) | 6\n",
-      "17420    | torch.Size([1, 1000, 754]) | 6\n",
-      "17421    | torch.Size([1, 1000, 754]) | 6\n",
-      "17422    | torch.Size([1, 1000, 754]) | 6\n",
-      "17423    | torch.Size([1, 1000, 754]) | 6\n",
-      "17424    | torch.Size([1, 1000, 777]) | 6\n",
-      "17425    | torch.Size([1, 1000, 783]) | 6\n",
-      "17426    | torch.Size([1, 1000, 777]) | 6\n",
-      "17427    | torch.Size([1, 1000, 762]) | 6\n",
-      "17428    | torch.Size([1, 1000, 770]) | 6\n",
-      "17429    | torch.Size([1, 1000, 802]) | 6\n",
-      "17430    | torch.Size([1, 1000, 777]) | 6\n",
-      "17431    | torch.Size([1, 1000, 771]) | 6\n",
-      "17432    | torch.Size([1, 1000, 754]) | 6\n",
-      "17433    | torch.Size([1, 1000, 752]) | 6\n",
-      "17434    | torch.Size([1, 1000, 754]) | 6\n",
-      "17435    | torch.Size([1, 1000, 777]) | 6\n",
-      "17436    | torch.Size([1, 1000, 754]) | 6\n",
-      "17437    | torch.Size([1, 1000, 777]) | 6\n",
-      "17438    | torch.Size([1, 1000, 754]) | 6\n",
-      "17439    | torch.Size([1, 1000, 754]) | 6\n",
-      "17440    | torch.Size([1, 1000, 777]) | 6\n",
-      "17441    | torch.Size([1, 1000, 777]) | 6\n",
-      "17442    | torch.Size([1, 1000, 775]) | 6\n",
-      "17443    | torch.Size([1, 1000, 777]) | 6\n",
-      "17444    | torch.Size([1, 1000, 762]) | 6\n",
-      "17445    | torch.Size([1, 1000, 754]) | 6\n",
-      "17446    | torch.Size([1, 1000, 777]) | 6\n",
-      "17447    | torch.Size([1, 1000, 754]) | 6\n",
-      "17448    | torch.Size([1, 1000, 786]) | 6\n",
-      "17449    | torch.Size([1, 1000, 782]) | 6\n",
-      "17450    | torch.Size([1, 1000, 777]) | 6\n",
-      "17451    | torch.Size([1, 1000, 777]) | 6\n",
-      "17452    | torch.Size([1, 1000, 787]) | 6\n",
-      "17453    | torch.Size([1, 1000, 775]) | 6\n",
-      "17454    | torch.Size([1, 1000, 769]) | 6\n",
-      "17455    | torch.Size([1, 1000, 762]) | 6\n",
-      "17456    | torch.Size([1, 1000, 754]) | 6\n",
-      "17457    | torch.Size([1, 1000, 777]) | 6\n",
-      "17458    | torch.Size([1, 1000, 754]) | 6\n",
-      "17459    | torch.Size([1, 1000, 777]) | 6\n",
-      "17460    | torch.Size([1, 1000, 775]) | 6\n",
-      "17461    | torch.Size([1, 1000, 754]) | 6\n",
-      "17462    | torch.Size([1, 1000, 754]) | 6\n",
-      "17463    | torch.Size([1, 1000, 770]) | 6\n",
-      "17464    | torch.Size([1, 1000, 754]) | 6\n",
-      "17465    | torch.Size([1, 1000, 754]) | 6\n",
-      "17466    | torch.Size([1, 1000, 777]) | 6\n",
-      "17467    | torch.Size([1, 1000, 754]) | 6\n",
-      "17468    | torch.Size([1, 1000, 777]) | 6\n",
-      "17469    | torch.Size([1, 1000, 754]) | 6\n",
-      "17470    | torch.Size([1, 1000, 773]) | 6\n",
-      "17471    | torch.Size([1, 1000, 765]) | 6\n",
-      "17472    | torch.Size([1, 1000, 772]) | 6\n",
-      "17473    | torch.Size([1, 1000, 754]) | 6\n",
-      "17474    | torch.Size([1, 1000, 783]) | 6\n",
-      "17475    | torch.Size([1, 1000, 777]) | 6\n",
-      "17476    | torch.Size([1, 1000, 777]) | 6\n",
-      "17477    | torch.Size([1, 1000, 754]) | 6\n",
-      "17478    | torch.Size([1, 1000, 779]) | 6\n",
-      "17479    | torch.Size([1, 1000, 762]) | 6\n",
-      "17480    | torch.Size([1, 1000, 754]) | 6\n",
-      "17481    | torch.Size([1, 1000, 783]) | 6\n",
-      "17482    | torch.Size([1, 1000, 754]) | 6\n",
-      "17483    | torch.Size([1, 1000, 754]) | 6\n",
-      "17484    | torch.Size([1, 1000, 767]) | 6\n",
-      "17485    | torch.Size([1, 1000, 762]) | 6\n",
-      "17486    | torch.Size([1, 1000, 778]) | 6\n",
-      "17487    | torch.Size([1, 1000, 796]) | 6\n",
-      "17488    | torch.Size([1, 1000, 777]) | 6\n",
-      "17489    | torch.Size([1, 1000, 720]) | 6\n",
-      "17490    | torch.Size([1, 1000, 754]) | 6\n",
-      "17491    | torch.Size([1, 1000, 754]) | 6\n",
-      "17492    | torch.Size([1, 1000, 762]) | 6\n",
-      "17493    | torch.Size([1, 1000, 754]) | 6\n",
-      "17494    | torch.Size([1, 1000, 763]) | 6\n",
-      "17495    | torch.Size([1, 1000, 754]) | 6\n",
-      "17496    | torch.Size([1, 1000, 777]) | 6\n",
-      "17497    | torch.Size([1, 1000, 777]) | 6\n",
-      "17498    | torch.Size([1, 1000, 754]) | 6\n",
-      "17499    | torch.Size([1, 1000, 783]) | 6\n",
-      "17500    | torch.Size([1, 1000, 778]) | 6\n",
-      "17501    | torch.Size([1, 1000, 777]) | 6\n",
-      "17502    | torch.Size([1, 1000, 754]) | 6\n",
-      "17503    | torch.Size([1, 1000, 777]) | 6\n",
-      "17504    | torch.Size([1, 1000, 754]) | 6\n",
-      "17505    | torch.Size([1, 1000, 754]) | 6\n",
-      "17506    | torch.Size([1, 1000, 762]) | 6\n",
-      "17507    | torch.Size([1, 1000, 777]) | 6\n",
-      "17508    | torch.Size([1, 1000, 754]) | 6\n",
-      "17509    | torch.Size([1, 1000, 777]) | 6\n",
-      "17510    | torch.Size([1, 1000, 777]) | 6\n",
-      "17511    | torch.Size([1, 1000, 796]) | 6\n",
-      "17512    | torch.Size([1, 1000, 777]) | 6\n",
-      "17513    | torch.Size([1, 1000, 754]) | 6\n",
-      "17514    | torch.Size([1, 1000, 754]) | 6\n",
-      "17515    | torch.Size([1, 1000, 754]) | 6\n",
-      "17516    | torch.Size([1, 1000, 754]) | 6\n",
-      "17517    | torch.Size([1, 1000, 777]) | 6\n",
-      "17518    | torch.Size([1, 1000, 754]) | 6\n",
-      "17519    | torch.Size([1, 1000, 768]) | 6\n",
-      "17520    | torch.Size([1, 1000, 791]) | 6\n",
-      "17521    | torch.Size([1, 1000, 754]) | 6\n",
-      "17522    | torch.Size([1, 1000, 777]) | 6\n",
-      "17523    | torch.Size([1, 1000, 777]) | 6\n",
-      "17524    | torch.Size([1, 1000, 772]) | 6\n",
-      "17525    | torch.Size([1, 1000, 773]) | 6\n",
-      "17526    | torch.Size([1, 1000, 762]) | 6\n",
-      "17527    | torch.Size([1, 1000, 754]) | 6\n",
-      "17528    | torch.Size([1, 1000, 766]) | 6\n",
-      "17529    | torch.Size([1, 1000, 777]) | 6\n",
-      "17530    | torch.Size([1, 1000, 807]) | 6\n",
-      "17531    | torch.Size([1, 1000, 773]) | 6\n",
-      "17532    | torch.Size([1, 1000, 764]) | 6\n",
-      "17533    | torch.Size([1, 1000, 754]) | 6\n",
-      "17534    | torch.Size([1, 1000, 779]) | 6\n",
-      "17535    | torch.Size([1, 1000, 768]) | 6\n",
-      "17536    | torch.Size([1, 1000, 762]) | 6\n",
-      "17537    | torch.Size([1, 1000, 777]) | 6\n",
-      "17538    | torch.Size([1, 1000, 754]) | 6\n",
-      "17539    | torch.Size([1, 1000, 754]) | 6\n",
-      "17540    | torch.Size([1, 1000, 752]) | 6\n",
-      "17541    | torch.Size([1, 1000, 763]) | 6\n",
-      "17542    | torch.Size([1, 1000, 754]) | 6\n",
-      "17543    | torch.Size([1, 1000, 778]) | 6\n",
-      "17544    | torch.Size([1, 1000, 777]) | 6\n",
-      "17545    | torch.Size([1, 1000, 773]) | 6\n",
-      "17546    | torch.Size([1, 1000, 754]) | 6\n",
-      "17547    | torch.Size([1, 1000, 754]) | 6\n",
-      "17548    | torch.Size([1, 1000, 782]) | 6\n",
-      "17549    | torch.Size([1, 1000, 754]) | 6\n",
-      "17550    | torch.Size([1, 1000, 754]) | 6\n",
-      "17551    | torch.Size([1, 1000, 777]) | 6\n",
-      "17552    | torch.Size([1, 1000, 767]) | 6\n",
-      "17553    | torch.Size([1, 1000, 762]) | 6\n",
-      "17554    | torch.Size([1, 1000, 754]) | 6\n",
-      "17555    | torch.Size([1, 1000, 754]) | 6\n",
-      "17556    | torch.Size([1, 1000, 762]) | 6\n",
-      "17557    | torch.Size([1, 1000, 784]) | 6\n",
-      "17558    | torch.Size([1, 1000, 802]) | 6\n",
-      "17559    | torch.Size([1, 1000, 777]) | 6\n",
-      "17560    | torch.Size([1, 1000, 754]) | 6\n",
-      "17561    | torch.Size([1, 1000, 777]) | 6\n",
-      "17562    | torch.Size([1, 1000, 754]) | 6\n",
-      "17563    | torch.Size([1, 1000, 754]) | 6\n",
-      "17564    | torch.Size([1, 1000, 777]) | 6\n",
-      "17565    | torch.Size([1, 1000, 784]) | 6\n",
-      "17566    | torch.Size([1, 1000, 777]) | 6\n",
-      "17567    | torch.Size([1, 1000, 754]) | 6\n",
-      "17568    | torch.Size([1, 1000, 777]) | 6\n",
-      "17569    | torch.Size([1, 1000, 754]) | 6\n",
-      "17570    | torch.Size([1, 1000, 778]) | 6\n",
-      "17571    | torch.Size([1, 1000, 754]) | 6\n",
-      "17572    | torch.Size([1, 1000, 754]) | 6\n",
-      "17573    | torch.Size([1, 1000, 754]) | 6\n",
-      "17574    | torch.Size([1, 1000, 754]) | 6\n",
-      "17575    | torch.Size([1, 1000, 776]) | 6\n",
-      "17576    | torch.Size([1, 1000, 795]) | 6\n",
-      "17577    | torch.Size([1, 1000, 775]) | 6\n",
-      "17578    | torch.Size([1, 1000, 781]) | 7\n",
-      "17579    | torch.Size([1, 1000, 784]) | 7\n",
-      "17580    | torch.Size([1, 1000, 754]) | 7\n",
-      "17581    | torch.Size([1, 1000, 781]) | 7\n",
-      "17582    | torch.Size([1, 1000, 754]) | 7\n",
-      "17583    | torch.Size([1, 1000, 754]) | 7\n",
-      "17584    | torch.Size([1, 1000, 768]) | 7\n",
-      "17585    | torch.Size([1, 1000, 762]) | 7\n",
-      "17586    | torch.Size([1, 1000, 779]) | 7\n",
-      "17587    | torch.Size([1, 1000, 762]) | 7\n",
-      "17588    | torch.Size([1, 1000, 754]) | 7\n",
-      "17589    | torch.Size([1, 1000, 762]) | 7\n",
-      "17590    | torch.Size([1, 1000, 754]) | 7\n",
-      "17591    | torch.Size([1, 1000, 762]) | 7\n",
-      "17592    | torch.Size([1, 1000, 767]) | 7\n",
-      "17593    | torch.Size([1, 1000, 779]) | 7\n",
-      "17594    | torch.Size([1, 1000, 777]) | 7\n",
-      "17595    | torch.Size([1, 1000, 762]) | 7\n",
-      "17596    | torch.Size([1, 1000, 800]) | 7\n",
-      "17597    | torch.Size([1, 1000, 754]) | 7\n",
-      "17598    | torch.Size([1, 1000, 785]) | 7\n",
-      "17599    | torch.Size([1, 1000, 754]) | 7\n",
-      "17600    | torch.Size([1, 1000, 754]) | 7\n",
-      "17601    | torch.Size([1, 1000, 771]) | 7\n",
-      "17602    | torch.Size([1, 1000, 754]) | 7\n",
-      "17603    | torch.Size([1, 1000, 777]) | 7\n",
-      "17604    | torch.Size([1, 1000, 777]) | 7\n",
-      "17605    | torch.Size([1, 1000, 814]) | 7\n",
-      "17606    | torch.Size([1, 1000, 740]) | 7\n",
-      "17607    | torch.Size([1, 1000, 754]) | 7\n",
-      "17608    | torch.Size([1, 1000, 785]) | 7\n",
-      "17609    | torch.Size([1, 1000, 762]) | 7\n",
-      "17610    | torch.Size([1, 1000, 781]) | 7\n",
-      "17611    | torch.Size([1, 1000, 774]) | 7\n",
-      "17612    | torch.Size([1, 1000, 754]) | 7\n",
-      "17613    | torch.Size([1, 1000, 754]) | 7\n",
-      "17614    | torch.Size([1, 1000, 754]) | 7\n",
-      "17615    | torch.Size([1, 1000, 774]) | 7\n",
-      "17616    | torch.Size([1, 1000, 754]) | 7\n",
-      "17617    | torch.Size([1, 1000, 788]) | 7\n",
-      "17618    | torch.Size([1, 1000, 754]) | 7\n",
-      "17619    | torch.Size([1, 1000, 762]) | 7\n",
-      "17620    | torch.Size([1, 1000, 759]) | 7\n",
-      "17621    | torch.Size([1, 1000, 776]) | 7\n",
-      "17622    | torch.Size([1, 1000, 782]) | 7\n",
-      "17623    | torch.Size([1, 1000, 774]) | 7\n",
-      "17624    | torch.Size([1, 1000, 754]) | 7\n",
-      "17625    | torch.Size([1, 1000, 762]) | 7\n",
-      "17626    | torch.Size([1, 1000, 771]) | 7\n",
-      "17627    | torch.Size([1, 1000, 772]) | 7\n",
-      "17628    | torch.Size([1, 1000, 793]) | 7\n",
-      "17629    | torch.Size([1, 1000, 811]) | 7\n",
-      "17630    | torch.Size([1, 1000, 771]) | 7\n",
-      "17631    | torch.Size([1, 1000, 784]) | 7\n",
-      "17632    | torch.Size([1, 1000, 799]) | 7\n",
-      "17633    | torch.Size([1, 1000, 754]) | 7\n",
-      "17634    | torch.Size([1, 1000, 770]) | 7\n",
-      "17635    | torch.Size([1, 1000, 777]) | 7\n",
-      "17636    | torch.Size([1, 1000, 777]) | 7\n",
-      "17637    | torch.Size([1, 1000, 777]) | 7\n",
-      "17638    | torch.Size([1, 1000, 762]) | 7\n",
-      "17639    | torch.Size([1, 1000, 754]) | 7\n",
-      "17640    | torch.Size([1, 1000, 754]) | 7\n",
-      "17641    | torch.Size([1, 1000, 762]) | 7\n",
-      "17642    | torch.Size([1, 1000, 754]) | 7\n",
-      "17643    | torch.Size([1, 1000, 770]) | 7\n",
-      "17644    | torch.Size([1, 1000, 784]) | 7\n",
-      "17645    | torch.Size([1, 1000, 754]) | 7\n",
-      "17646    | torch.Size([1, 1000, 778]) | 7\n",
-      "17647    | torch.Size([1, 1000, 762]) | 7\n",
-      "17648    | torch.Size([1, 1000, 780]) | 7\n",
-      "17649    | torch.Size([1, 1000, 754]) | 7\n",
-      "17650    | torch.Size([1, 1000, 771]) | 7\n",
-      "17651    | torch.Size([1, 1000, 775]) | 7\n",
-      "17652    | torch.Size([1, 1000, 754]) | 7\n",
-      "17653    | torch.Size([1, 1000, 779]) | 7\n",
-      "17654    | torch.Size([1, 1000, 786]) | 7\n",
-      "17655    | torch.Size([1, 1000, 771]) | 7\n",
-      "17656    | torch.Size([1, 1000, 779]) | 7\n",
-      "17657    | torch.Size([1, 1000, 796]) | 7\n",
-      "17658    | torch.Size([1, 1000, 754]) | 7\n",
-      "17659    | torch.Size([1, 1000, 802]) | 7\n",
-      "17660    | torch.Size([1, 1000, 754]) | 7\n",
-      "17661    | torch.Size([1, 1000, 777]) | 7\n",
-      "17662    | torch.Size([1, 1000, 754]) | 7\n",
-      "17663    | torch.Size([1, 1000, 771]) | 7\n",
-      "17664    | torch.Size([1, 1000, 762]) | 7\n",
-      "17665    | torch.Size([1, 1000, 787]) | 7\n",
-      "17666    | torch.Size([1, 1000, 754]) | 7\n",
-      "17667    | torch.Size([1, 1000, 754]) | 7\n",
-      "17668    | torch.Size([1, 1000, 777]) | 7\n",
-      "17669    | torch.Size([1, 1000, 774]) | 7\n",
-      "17670    | torch.Size([1, 1000, 774]) | 7\n",
-      "17671    | torch.Size([1, 1000, 754]) | 7\n",
-      "17672    | torch.Size([1, 1000, 762]) | 7\n",
-      "17673    | torch.Size([1, 1000, 780]) | 7\n",
-      "17674    | torch.Size([1, 1000, 762]) | 7\n",
-      "17675    | torch.Size([1, 1000, 754]) | 7\n",
-      "17676    | torch.Size([1, 1000, 754]) | 7\n",
-      "17677    | torch.Size([1, 1000, 776]) | 7\n",
-      "17678    | torch.Size([1, 1000, 786]) | 7\n",
-      "17679    | torch.Size([1, 1000, 754]) | 7\n",
-      "17680    | torch.Size([1, 1000, 803]) | 7\n",
-      "17681    | torch.Size([1, 1000, 754]) | 7\n",
-      "17682    | torch.Size([1, 1000, 754]) | 7\n",
-      "17683    | torch.Size([1, 1000, 769]) | 7\n",
-      "17684    | torch.Size([1, 1000, 754]) | 7\n",
-      "17685    | torch.Size([1, 1000, 754]) | 7\n",
-      "17686    | torch.Size([1, 1000, 771]) | 7\n",
-      "17687    | torch.Size([1, 1000, 780]) | 7\n",
-      "17688    | torch.Size([1, 1000, 777]) | 7\n",
-      "17689    | torch.Size([1, 1000, 782]) | 7\n",
-      "17690    | torch.Size([1, 1000, 762]) | 7\n",
-      "17691    | torch.Size([1, 1000, 801]) | 7\n",
-      "17692    | torch.Size([1, 1000, 754]) | 7\n",
-      "17693    | torch.Size([1, 1000, 771]) | 7\n",
-      "17694    | torch.Size([1, 1000, 754]) | 7\n",
-      "17695    | torch.Size([1, 1000, 771]) | 7\n",
-      "17696    | torch.Size([1, 1000, 770]) | 7\n",
-      "17697    | torch.Size([1, 1000, 754]) | 7\n",
-      "17698    | torch.Size([1, 1000, 754]) | 7\n",
-      "17699    | torch.Size([1, 1000, 762]) | 7\n",
-      "17700    | torch.Size([1, 1000, 754]) | 7\n",
-      "17701    | torch.Size([1, 1000, 777]) | 7\n",
-      "17702    | torch.Size([1, 1000, 754]) | 7\n",
-      "17703    | torch.Size([1, 1000, 787]) | 7\n",
-      "17704    | torch.Size([1, 1000, 769]) | 7\n",
-      "17705    | torch.Size([1, 1000, 777]) | 7\n",
-      "17706    | torch.Size([1, 1000, 788]) | 7\n",
-      "17707    | torch.Size([1, 1000, 783]) | 7\n",
-      "17708    | torch.Size([1, 1000, 754]) | 7\n",
-      "17709    | torch.Size([1, 1000, 754]) | 7\n",
-      "17710    | torch.Size([1, 1000, 771]) | 7\n",
-      "17711    | torch.Size([1, 1000, 754]) | 7\n",
-      "17712    | torch.Size([1, 1000, 754]) | 7\n",
-      "17713    | torch.Size([1, 1000, 799]) | 7\n",
-      "17714    | torch.Size([1, 1000, 778]) | 7\n",
-      "17715    | torch.Size([1, 1000, 773]) | 7\n",
-      "17716    | torch.Size([1, 1000, 754]) | 7\n",
-      "17717    | torch.Size([1, 1000, 772]) | 7\n",
-      "17718    | torch.Size([1, 1000, 754]) | 7\n",
-      "17719    | torch.Size([1, 1000, 754]) | 7\n",
-      "17720    | torch.Size([1, 1000, 777]) | 7\n",
-      "17721    | torch.Size([1, 1000, 754]) | 7\n",
-      "17722    | torch.Size([1, 1000, 762]) | 7\n",
-      "17723    | torch.Size([1, 1000, 781]) | 7\n",
-      "17724    | torch.Size([1, 1000, 754]) | 7\n",
-      "17725    | torch.Size([1, 1000, 754]) | 7\n",
-      "17726    | torch.Size([1, 1000, 807]) | 7\n",
-      "17727    | torch.Size([1, 1000, 779]) | 7\n",
-      "17728    | torch.Size([1, 1000, 754]) | 7\n",
-      "17729    | torch.Size([1, 1000, 827]) | 7\n",
-      "17730    | torch.Size([1, 1000, 784]) | 7\n",
-      "17731    | torch.Size([1, 1000, 781]) | 7\n",
-      "17732    | torch.Size([1, 1000, 754]) | 7\n",
-      "17733    | torch.Size([1, 1000, 771]) | 7\n",
-      "17734    | torch.Size([1, 1000, 754]) | 7\n",
-      "17735    | torch.Size([1, 1000, 785]) | 7\n",
-      "17736    | torch.Size([1, 1000, 771]) | 7\n",
-      "17737    | torch.Size([1, 1000, 754]) | 7\n",
-      "17738    | torch.Size([1, 1000, 778]) | 7\n",
-      "17739    | torch.Size([1, 1000, 785]) | 7\n",
-      "17740    | torch.Size([1, 1000, 784]) | 7\n",
-      "17741    | torch.Size([1, 1000, 754]) | 7\n",
-      "17742    | torch.Size([1, 1000, 754]) | 7\n",
-      "17743    | torch.Size([1, 1000, 777]) | 7\n",
-      "17744    | torch.Size([1, 1000, 754]) | 7\n",
-      "17745    | torch.Size([1, 1000, 777]) | 7\n",
-      "17746    | torch.Size([1, 1000, 826]) | 7\n",
-      "17747    | torch.Size([1, 1000, 787]) | 7\n",
-      "17748    | torch.Size([1, 1000, 776]) | 7\n",
-      "17749    | torch.Size([1, 1000, 754]) | 7\n",
-      "17750    | torch.Size([1, 1000, 777]) | 7\n",
-      "17751    | torch.Size([1, 1000, 754]) | 7\n",
-      "17752    | torch.Size([1, 1000, 777]) | 7\n",
-      "17753    | torch.Size([1, 1000, 779]) | 7\n",
-      "17754    | torch.Size([1, 1000, 762]) | 7\n",
-      "17755    | torch.Size([1, 1000, 774]) | 7\n",
-      "17756    | torch.Size([1, 1000, 806]) | 7\n",
-      "17757    | torch.Size([1, 1000, 777]) | 7\n",
-      "17758    | torch.Size([1, 1000, 754]) | 7\n",
-      "17759    | torch.Size([1, 1000, 772]) | 7\n",
-      "17760    | torch.Size([1, 1000, 754]) | 7\n",
-      "17761    | torch.Size([1, 1000, 754]) | 7\n",
-      "17762    | torch.Size([1, 1000, 754]) | 7\n",
-      "17763    | torch.Size([1, 1000, 762]) | 7\n",
-      "17764    | torch.Size([1, 1000, 762]) | 7\n",
-      "17765    | torch.Size([1, 1000, 799]) | 7\n",
-      "17766    | torch.Size([1, 1000, 771]) | 7\n",
-      "17767    | torch.Size([1, 1000, 774]) | 7\n",
-      "17768    | torch.Size([1, 1000, 754]) | 7\n",
-      "17769    | torch.Size([1, 1000, 784]) | 7\n",
-      "17770    | torch.Size([1, 1000, 754]) | 7\n",
-      "17771    | torch.Size([1, 1000, 762]) | 7\n",
-      "17772    | torch.Size([1, 1000, 777]) | 7\n",
-      "17773    | torch.Size([1, 1000, 771]) | 7\n",
-      "17774    | torch.Size([1, 1000, 754]) | 7\n",
-      "17775    | torch.Size([1, 1000, 772]) | 7\n",
-      "17776    | torch.Size([1, 1000, 754]) | 7\n",
-      "17777    | torch.Size([1, 1000, 762]) | 7\n",
-      "17778    | torch.Size([1, 1000, 771]) | 7\n",
-      "17779    | torch.Size([1, 1000, 777]) | 7\n",
-      "17780    | torch.Size([1, 1000, 754]) | 7\n",
-      "17781    | torch.Size([1, 1000, 713]) | 7\n",
-      "17782    | torch.Size([1, 1000, 789]) | 7\n",
-      "17783    | torch.Size([1, 1000, 772]) | 7\n",
-      "17784    | torch.Size([1, 1000, 754]) | 7\n",
-      "17785    | torch.Size([1, 1000, 762]) | 7\n",
-      "17786    | torch.Size([1, 1000, 792]) | 7\n",
-      "17787    | torch.Size([1, 1000, 771]) | 7\n",
-      "17788    | torch.Size([1, 1000, 754]) | 7\n",
-      "17789    | torch.Size([1, 1000, 789]) | 7\n",
-      "17790    | torch.Size([1, 1000, 754]) | 7\n",
-      "17791    | torch.Size([1, 1000, 754]) | 7\n",
-      "17792    | torch.Size([1, 1000, 766]) | 7\n",
-      "17793    | torch.Size([1, 1000, 762]) | 7\n",
-      "17794    | torch.Size([1, 1000, 762]) | 7\n",
-      "17795    | torch.Size([1, 1000, 771]) | 7\n",
-      "17796    | torch.Size([1, 1000, 777]) | 7\n",
-      "17797    | torch.Size([1, 1000, 782]) | 7\n",
-      "17798    | torch.Size([1, 1000, 798]) | 7\n",
-      "17799    | torch.Size([1, 1000, 754]) | 7\n",
-      "17800    | torch.Size([1, 1000, 782]) | 7\n",
-      "17801    | torch.Size([1, 1000, 776]) | 7\n",
-      "17802    | torch.Size([1, 1000, 775]) | 7\n",
-      "17803    | torch.Size([1, 1000, 772]) | 7\n",
-      "17804    | torch.Size([1, 1000, 754]) | 7\n",
-      "17805    | torch.Size([1, 1000, 771]) | 7\n",
-      "17806    | torch.Size([1, 1000, 766]) | 7\n",
-      "17807    | torch.Size([1, 1000, 762]) | 7\n",
-      "17808    | torch.Size([1, 1000, 762]) | 7\n",
-      "17809    | torch.Size([1, 1000, 771]) | 7\n",
-      "17810    | torch.Size([1, 1000, 776]) | 7\n",
-      "17811    | torch.Size([1, 1000, 772]) | 7\n",
-      "17812    | torch.Size([1, 1000, 784]) | 7\n",
-      "17813    | torch.Size([1, 1000, 762]) | 7\n",
-      "17814    | torch.Size([1, 1000, 802]) | 7\n",
-      "17815    | torch.Size([1, 1000, 754]) | 7\n",
-      "17816    | torch.Size([1, 1000, 786]) | 7\n",
-      "17817    | torch.Size([1, 1000, 754]) | 7\n",
-      "17818    | torch.Size([1, 1000, 777]) | 7\n",
-      "17819    | torch.Size([1, 1000, 703]) | 7\n",
-      "17820    | torch.Size([1, 1000, 762]) | 7\n",
-      "17821    | torch.Size([1, 1000, 754]) | 7\n",
-      "17822    | torch.Size([1, 1000, 777]) | 7\n",
-      "17823    | torch.Size([1, 1000, 768]) | 7\n",
-      "17824    | torch.Size([1, 1000, 754]) | 7\n",
-      "17825    | torch.Size([1, 1000, 802]) | 7\n",
-      "17826    | torch.Size([1, 1000, 777]) | 7\n",
-      "17827    | torch.Size([1, 1000, 777]) | 7\n",
-      "17828    | torch.Size([1, 1000, 771]) | 7\n",
-      "17829    | torch.Size([1, 1000, 775]) | 7\n",
-      "17830    | torch.Size([1, 1000, 771]) | 7\n",
-      "17831    | torch.Size([1, 1000, 768]) | 7\n",
-      "17832    | torch.Size([1, 1000, 754]) | 7\n",
-      "17833    | torch.Size([1, 1000, 754]) | 7\n",
-      "17834    | torch.Size([1, 1000, 754]) | 7\n",
-      "17835    | torch.Size([1, 1000, 809]) | 7\n",
-      "17836    | torch.Size([1, 1000, 754]) | 7\n",
-      "17837    | torch.Size([1, 1000, 754]) | 7\n",
-      "17838    | torch.Size([1, 1000, 798]) | 7\n",
-      "17839    | torch.Size([1, 1000, 781]) | 7\n",
-      "17840    | torch.Size([1, 1000, 777]) | 7\n",
-      "17841    | torch.Size([1, 1000, 762]) | 7\n",
-      "17842    | torch.Size([1, 1000, 754]) | 7\n",
-      "17843    | torch.Size([1, 1000, 774]) | 7\n",
-      "17844    | torch.Size([1, 1000, 754]) | 7\n",
-      "17845    | torch.Size([1, 1000, 754]) | 7\n",
-      "17846    | torch.Size([1, 1000, 789]) | 7\n",
-      "17847    | torch.Size([1, 1000, 777]) | 7\n",
-      "17848    | torch.Size([1, 1000, 774]) | 7\n",
-      "17849    | torch.Size([1, 1000, 754]) | 7\n",
-      "17850    | torch.Size([1, 1000, 754]) | 7\n",
-      "17851    | torch.Size([1, 1000, 762]) | 7\n",
-      "17852    | torch.Size([1, 1000, 754]) | 7\n",
-      "17853    | torch.Size([1, 1000, 805]) | 7\n",
-      "17854    | torch.Size([1, 1000, 785]) | 7\n",
-      "17855    | torch.Size([1, 1000, 777]) | 7\n",
-      "17856    | torch.Size([1, 1000, 771]) | 7\n",
-      "17857    | torch.Size([1, 1000, 754]) | 7\n",
-      "17858    | torch.Size([1, 1000, 766]) | 7\n",
-      "17859    | torch.Size([1, 1000, 794]) | 7\n",
-      "17860    | torch.Size([1, 1000, 754]) | 7\n",
-      "17861    | torch.Size([1, 1000, 788]) | 7\n",
-      "17862    | torch.Size([1, 1000, 761]) | 7\n",
-      "17863    | torch.Size([1, 1000, 772]) | 7\n",
-      "17864    | torch.Size([1, 1000, 850]) | 7\n",
-      "17865    | torch.Size([1, 1000, 754]) | 7\n",
-      "17866    | torch.Size([1, 1000, 754]) | 7\n",
-      "17867    | torch.Size([1, 1000, 754]) | 7\n",
-      "17868    | torch.Size([1, 1000, 777]) | 7\n",
-      "17869    | torch.Size([1, 1000, 754]) | 7\n",
-      "17870    | torch.Size([1, 1000, 796]) | 7\n",
-      "17871    | torch.Size([1, 1000, 786]) | 7\n",
-      "17872    | torch.Size([1, 1000, 754]) | 7\n",
-      "17873    | torch.Size([1, 1000, 777]) | 7\n",
-      "17874    | torch.Size([1, 1000, 777]) | 7\n",
-      "17875    | torch.Size([1, 1000, 788]) | 7\n",
-      "17876    | torch.Size([1, 1000, 754]) | 7\n",
-      "17877    | torch.Size([1, 1000, 770]) | 7\n",
-      "17878    | torch.Size([1, 1000, 775]) | 7\n",
-      "17879    | torch.Size([1, 1000, 773]) | 7\n",
-      "17880    | torch.Size([1, 1000, 803]) | 7\n",
-      "17881    | torch.Size([1, 1000, 754]) | 7\n",
-      "17882    | torch.Size([1, 1000, 783]) | 7\n",
-      "17883    | torch.Size([1, 1000, 789]) | 7\n",
-      "17884    | torch.Size([1, 1000, 777]) | 7\n",
-      "17885    | torch.Size([1, 1000, 779]) | 7\n",
-      "17886    | torch.Size([1, 1000, 754]) | 7\n",
-      "17887    | torch.Size([1, 1000, 754]) | 7\n",
-      "17888    | torch.Size([1, 1000, 775]) | 7\n",
-      "17889    | torch.Size([1, 1000, 754]) | 7\n",
-      "17890    | torch.Size([1, 1000, 762]) | 7\n",
-      "17891    | torch.Size([1, 1000, 770]) | 7\n",
-      "17892    | torch.Size([1, 1000, 754]) | 7\n",
-      "17893    | torch.Size([1, 1000, 762]) | 7\n",
-      "17894    | torch.Size([1, 1000, 773]) | 7\n",
-      "17895    | torch.Size([1, 1000, 782]) | 7\n",
-      "17896    | torch.Size([1, 1000, 784]) | 7\n",
-      "17897    | torch.Size([1, 1000, 762]) | 7\n",
-      "17898    | torch.Size([1, 1000, 762]) | 7\n",
-      "17899    | torch.Size([1, 1000, 754]) | 7\n",
-      "17900    | torch.Size([1, 1000, 754]) | 7\n",
-      "17901    | torch.Size([1, 1000, 779]) | 7\n",
-      "17902    | torch.Size([1, 1000, 754]) | 7\n",
-      "17903    | torch.Size([1, 1000, 754]) | 7\n",
-      "17904    | torch.Size([1, 1000, 754]) | 7\n",
-      "17905    | torch.Size([1, 1000, 754]) | 7\n",
-      "17906    | torch.Size([1, 1000, 810]) | 7\n",
-      "17907    | torch.Size([1, 1000, 754]) | 7\n",
-      "17908    | torch.Size([1, 1000, 769]) | 7\n",
-      "17909    | torch.Size([1, 1000, 783]) | 7\n",
-      "17910    | torch.Size([1, 1000, 781]) | 7\n",
-      "17911    | torch.Size([1, 1000, 762]) | 7\n",
-      "17912    | torch.Size([1, 1000, 762]) | 7\n",
-      "17913    | torch.Size([1, 1000, 777]) | 7\n",
-      "17914    | torch.Size([1, 1000, 762]) | 7\n",
-      "17915    | torch.Size([1, 1000, 762]) | 7\n",
-      "17916    | torch.Size([1, 1000, 776]) | 7\n",
-      "17917    | torch.Size([1, 1000, 777]) | 7\n",
-      "17918    | torch.Size([1, 1000, 754]) | 7\n",
-      "17919    | torch.Size([1, 1000, 762]) | 7\n",
-      "17920    | torch.Size([1, 1000, 777]) | 7\n",
-      "17921    | torch.Size([1, 1000, 762]) | 7\n",
-      "17922    | torch.Size([1, 1000, 807]) | 7\n",
-      "17923    | torch.Size([1, 1000, 793]) | 7\n",
-      "17924    | torch.Size([1, 1000, 801]) | 7\n",
-      "17925    | torch.Size([1, 1000, 754]) | 7\n",
-      "17926    | torch.Size([1, 1000, 754]) | 7\n",
-      "17927    | torch.Size([1, 1000, 778]) | 7\n",
-      "17928    | torch.Size([1, 1000, 754]) | 7\n",
-      "17929    | torch.Size([1, 1000, 771]) | 7\n",
-      "17930    | torch.Size([1, 1000, 777]) | 7\n",
-      "17931    | torch.Size([1, 1000, 794]) | 7\n",
-      "17932    | torch.Size([1, 1000, 754]) | 7\n",
-      "17933    | torch.Size([1, 1000, 754]) | 7\n",
-      "17934    | torch.Size([1, 1000, 771]) | 7\n",
-      "17935    | torch.Size([1, 1000, 762]) | 7\n",
-      "17936    | torch.Size([1, 1000, 777]) | 7\n",
-      "17937    | torch.Size([1, 1000, 779]) | 7\n",
-      "17938    | torch.Size([1, 1000, 777]) | 7\n",
-      "17939    | torch.Size([1, 1000, 754]) | 7\n",
-      "17940    | torch.Size([1, 1000, 777]) | 7\n",
-      "17941    | torch.Size([1, 1000, 754]) | 7\n",
-      "17942    | torch.Size([1, 1000, 762]) | 7\n",
-      "17943    | torch.Size([1, 1000, 762]) | 7\n",
-      "17944    | torch.Size([1, 1000, 777]) | 7\n",
-      "17945    | torch.Size([1, 1000, 754]) | 7\n",
-      "17946    | torch.Size([1, 1000, 754]) | 7\n",
-      "17947    | torch.Size([1, 1000, 777]) | 7\n",
-      "17948    | torch.Size([1, 1000, 754]) | 7\n",
-      "17949    | torch.Size([1, 1000, 786]) | 7\n",
-      "17950    | torch.Size([1, 1000, 793]) | 7\n",
-      "17951    | torch.Size([1, 1000, 776]) | 7\n",
-      "17952    | torch.Size([1, 1000, 754]) | 7\n",
-      "17953    | torch.Size([1, 1000, 783]) | 7\n",
-      "17954    | torch.Size([1, 1000, 754]) | 7\n",
-      "17955    | torch.Size([1, 1000, 754]) | 7\n",
-      "17956    | torch.Size([1, 1000, 787]) | 7\n",
-      "17957    | torch.Size([1, 1000, 754]) | 7\n",
-      "17958    | torch.Size([1, 1000, 813]) | 7\n",
-      "17959    | torch.Size([1, 1000, 777]) | 7\n",
-      "17960    | torch.Size([1, 1000, 754]) | 7\n",
-      "17961    | torch.Size([1, 1000, 771]) | 7\n",
-      "17962    | torch.Size([1, 1000, 754]) | 7\n",
-      "17963    | torch.Size([1, 1000, 754]) | 7\n",
-      "17964    | torch.Size([1, 1000, 771]) | 7\n",
-      "17965    | torch.Size([1, 1000, 798]) | 7\n",
-      "17966    | torch.Size([1, 1000, 774]) | 7\n",
-      "17967    | torch.Size([1, 1000, 754]) | 7\n",
-      "17968    | torch.Size([1, 1000, 754]) | 7\n",
-      "17969    | torch.Size([1, 1000, 772]) | 7\n",
-      "17970    | torch.Size([1, 1000, 762]) | 7\n",
-      "17971    | torch.Size([1, 1000, 754]) | 7\n",
-      "17972    | torch.Size([1, 1000, 778]) | 7\n",
-      "17973    | torch.Size([1, 1000, 764]) | 7\n",
-      "17974    | torch.Size([1, 1000, 754]) | 7\n",
-      "17975    | torch.Size([1, 1000, 752]) | 7\n",
-      "17976    | torch.Size([1, 1000, 771]) | 7\n",
-      "17977    | torch.Size([1, 1000, 777]) | 7\n",
-      "17978    | torch.Size([1, 1000, 772]) | 7\n",
-      "17979    | torch.Size([1, 1000, 754]) | 7\n",
-      "17980    | torch.Size([1, 1000, 754]) | 7\n",
-      "17981    | torch.Size([1, 1000, 754]) | 7\n",
-      "17982    | torch.Size([1, 1000, 777]) | 7\n",
-      "17983    | torch.Size([1, 1000, 754]) | 7\n",
-      "17984    | torch.Size([1, 1000, 754]) | 7\n",
-      "17985    | torch.Size([1, 1000, 790]) | 7\n",
-      "17986    | torch.Size([1, 1000, 779]) | 7\n",
-      "17987    | torch.Size([1, 1000, 777]) | 7\n",
-      "17988    | torch.Size([1, 1000, 777]) | 7\n",
-      "17989    | torch.Size([1, 1000, 777]) | 7\n",
-      "17990    | torch.Size([1, 1000, 754]) | 7\n",
-      "17991    | torch.Size([1, 1000, 768]) | 7\n",
-      "17992    | torch.Size([1, 1000, 754]) | 7\n",
-      "17993    | torch.Size([1, 1000, 754]) | 7\n",
-      "17994    | torch.Size([1, 1000, 754]) | 7\n",
-      "17995    | torch.Size([1, 1000, 777]) | 7\n",
-      "17996    | torch.Size([1, 1000, 754]) | 7\n",
-      "17997    | torch.Size([1, 1000, 752]) | 7\n",
-      "17998    | torch.Size([1, 1000, 754]) | 7\n",
-      "17999    | torch.Size([1, 1000, 762]) | 7\n",
-      "18000    | torch.Size([1, 1000, 777]) | 7\n",
-      "18001    | torch.Size([1, 1000, 781]) | 7\n",
-      "18002    | torch.Size([1, 1000, 777]) | 7\n",
-      "18003    | torch.Size([1, 1000, 754]) | 7\n",
-      "18004    | torch.Size([1, 1000, 765]) | 7\n",
-      "18005    | torch.Size([1, 1000, 772]) | 7\n",
-      "18006    | torch.Size([1, 1000, 754]) | 7\n",
-      "18007    | torch.Size([1, 1000, 762]) | 7\n",
-      "18008    | torch.Size([1, 1000, 762]) | 7\n",
-      "18009    | torch.Size([1, 1000, 802]) | 7\n",
-      "18010    | torch.Size([1, 1000, 772]) | 7\n",
-      "18011    | torch.Size([1, 1000, 754]) | 7\n",
-      "18012    | torch.Size([1, 1000, 771]) | 7\n",
-      "18013    | torch.Size([1, 1000, 808]) | 7\n",
-      "18014    | torch.Size([1, 1000, 754]) | 7\n",
-      "18015    | torch.Size([1, 1000, 788]) | 7\n",
-      "18016    | torch.Size([1, 1000, 703]) | 7\n",
-      "18017    | torch.Size([1, 1000, 771]) | 7\n",
-      "18018    | torch.Size([1, 1000, 754]) | 7\n",
-      "18019    | torch.Size([1, 1000, 754]) | 7\n",
-      "18020    | torch.Size([1, 1000, 771]) | 7\n",
-      "18021    | torch.Size([1, 1000, 779]) | 7\n",
-      "18022    | torch.Size([1, 1000, 726]) | 7\n",
-      "18023    | torch.Size([1, 1000, 754]) | 7\n",
-      "18024    | torch.Size([1, 1000, 754]) | 7\n",
-      "18025    | torch.Size([1, 1000, 777]) | 7\n",
-      "18026    | torch.Size([1, 1000, 754]) | 7\n",
-      "18027    | torch.Size([1, 1000, 777]) | 7\n",
-      "18028    | torch.Size([1, 1000, 762]) | 7\n",
-      "18029    | torch.Size([1, 1000, 762]) | 7\n",
-      "18030    | torch.Size([1, 1000, 771]) | 7\n",
-      "18031    | torch.Size([1, 1000, 772]) | 7\n",
-      "18032    | torch.Size([1, 1000, 754]) | 7\n",
-      "18033    | torch.Size([1, 1000, 754]) | 7\n",
-      "18034    | torch.Size([1, 1000, 804]) | 7\n",
-      "18035    | torch.Size([1, 1000, 754]) | 7\n",
-      "18036    | torch.Size([1, 1000, 771]) | 7\n",
-      "18037    | torch.Size([1, 1000, 754]) | 7\n",
-      "18038    | torch.Size([1, 1000, 776]) | 7\n",
-      "18039    | torch.Size([1, 1000, 762]) | 7\n",
-      "18040    | torch.Size([1, 1000, 776]) | 7\n",
-      "18041    | torch.Size([1, 1000, 771]) | 7\n",
-      "18042    | torch.Size([1, 1000, 754]) | 7\n",
-      "18043    | torch.Size([1, 1000, 777]) | 7\n",
-      "18044    | torch.Size([1, 1000, 754]) | 7\n",
-      "18045    | torch.Size([1, 1000, 754]) | 7\n",
-      "18046    | torch.Size([1, 1000, 777]) | 7\n",
-      "18047    | torch.Size([1, 1000, 754]) | 7\n",
-      "18048    | torch.Size([1, 1000, 771]) | 7\n",
-      "18049    | torch.Size([1, 1000, 777]) | 7\n",
-      "18050    | torch.Size([1, 1000, 754]) | 7\n",
-      "18051    | torch.Size([1, 1000, 754]) | 7\n",
-      "18052    | torch.Size([1, 1000, 804]) | 7\n",
-      "18053    | torch.Size([1, 1000, 762]) | 7\n",
-      "18054    | torch.Size([1, 1000, 754]) | 7\n",
-      "18055    | torch.Size([1, 1000, 754]) | 7\n",
-      "18056    | torch.Size([1, 1000, 754]) | 7\n",
-      "18057    | torch.Size([1, 1000, 754]) | 7\n",
-      "18058    | torch.Size([1, 1000, 762]) | 7\n",
-      "18059    | torch.Size([1, 1000, 754]) | 7\n",
-      "18060    | torch.Size([1, 1000, 772]) | 7\n",
-      "18061    | torch.Size([1, 1000, 754]) | 7\n",
-      "18062    | torch.Size([1, 1000, 762]) | 7\n",
-      "18063    | torch.Size([1, 1000, 754]) | 7\n",
-      "18064    | torch.Size([1, 1000, 772]) | 7\n",
-      "18065    | torch.Size([1, 1000, 782]) | 7\n",
-      "18066    | torch.Size([1, 1000, 754]) | 7\n",
-      "18067    | torch.Size([1, 1000, 780]) | 7\n",
-      "18068    | torch.Size([1, 1000, 779]) | 7\n",
-      "18069    | torch.Size([1, 1000, 793]) | 7\n",
-      "18070    | torch.Size([1, 1000, 768]) | 7\n",
-      "18071    | torch.Size([1, 1000, 754]) | 7\n",
-      "18072    | torch.Size([1, 1000, 754]) | 7\n",
-      "18073    | torch.Size([1, 1000, 754]) | 7\n",
-      "18074    | torch.Size([1, 1000, 802]) | 7\n",
-      "18075    | torch.Size([1, 1000, 754]) | 7\n",
-      "18076    | torch.Size([1, 1000, 754]) | 7\n",
-      "18077    | torch.Size([1, 1000, 754]) | 7\n",
-      "18078    | torch.Size([1, 1000, 754]) | 7\n",
-      "18079    | torch.Size([1, 1000, 754]) | 7\n",
-      "18080    | torch.Size([1, 1000, 771]) | 7\n",
-      "18081    | torch.Size([1, 1000, 764]) | 7\n",
-      "18082    | torch.Size([1, 1000, 788]) | 7\n",
-      "18083    | torch.Size([1, 1000, 754]) | 7\n",
-      "18084    | torch.Size([1, 1000, 784]) | 7\n",
-      "18085    | torch.Size([1, 1000, 762]) | 7\n",
-      "18086    | torch.Size([1, 1000, 754]) | 7\n",
-      "18087    | torch.Size([1, 1000, 754]) | 7\n",
-      "18088    | torch.Size([1, 1000, 782]) | 7\n",
-      "18089    | torch.Size([1, 1000, 762]) | 7\n",
-      "18090    | torch.Size([1, 1000, 762]) | 7\n",
-      "18091    | torch.Size([1, 1000, 781]) | 7\n",
-      "18092    | torch.Size([1, 1000, 830]) | 7\n",
-      "18093    | torch.Size([1, 1000, 754]) | 7\n",
-      "18094    | torch.Size([1, 1000, 787]) | 7\n",
-      "18095    | torch.Size([1, 1000, 771]) | 7\n",
-      "18096    | torch.Size([1, 1000, 781]) | 7\n",
-      "18097    | torch.Size([1, 1000, 754]) | 7\n",
-      "18098    | torch.Size([1, 1000, 782]) | 7\n",
-      "18099    | torch.Size([1, 1000, 754]) | 7\n",
-      "18100    | torch.Size([1, 1000, 766]) | 7\n",
-      "18101    | torch.Size([1, 1000, 782]) | 7\n",
-      "18102    | torch.Size([1, 1000, 783]) | 7\n",
-      "18103    | torch.Size([1, 1000, 777]) | 7\n",
-      "18104    | torch.Size([1, 1000, 754]) | 7\n",
-      "18105    | torch.Size([1, 1000, 754]) | 7\n",
-      "18106    | torch.Size([1, 1000, 754]) | 7\n",
-      "18107    | torch.Size([1, 1000, 771]) | 7\n",
-      "18108    | torch.Size([1, 1000, 777]) | 7\n",
-      "18109    | torch.Size([1, 1000, 779]) | 7\n",
-      "18110    | torch.Size([1, 1000, 754]) | 7\n",
-      "18111    | torch.Size([1, 1000, 800]) | 7\n",
-      "18112    | torch.Size([1, 1000, 754]) | 7\n",
-      "18113    | torch.Size([1, 1000, 754]) | 7\n",
-      "18114    | torch.Size([1, 1000, 787]) | 7\n",
-      "18115    | torch.Size([1, 1000, 754]) | 7\n",
-      "18116    | torch.Size([1, 1000, 778]) | 7\n",
-      "18117    | torch.Size([1, 1000, 777]) | 7\n",
-      "18118    | torch.Size([1, 1000, 754]) | 7\n",
-      "18119    | torch.Size([1, 1000, 745]) | 7\n",
-      "18120    | torch.Size([1, 1000, 785]) | 7\n",
-      "18121    | torch.Size([1, 1000, 780]) | 7\n",
-      "18122    | torch.Size([1, 1000, 785]) | 7\n",
-      "18123    | torch.Size([1, 1000, 777]) | 7\n",
-      "18124    | torch.Size([1, 1000, 762]) | 7\n",
-      "18125    | torch.Size([1, 1000, 754]) | 7\n",
-      "18126    | torch.Size([1, 1000, 845]) | 7\n",
-      "18127    | torch.Size([1, 1000, 797]) | 7\n",
-      "18128    | torch.Size([1, 1000, 754]) | 7\n",
-      "18129    | torch.Size([1, 1000, 865]) | 7\n",
-      "18130    | torch.Size([1, 1000, 774]) | 7\n",
-      "18131    | torch.Size([1, 1000, 777]) | 7\n",
-      "18132    | torch.Size([1, 1000, 754]) | 7\n",
-      "18133    | torch.Size([1, 1000, 762]) | 7\n",
-      "18134    | torch.Size([1, 1000, 754]) | 7\n",
-      "18135    | torch.Size([1, 1000, 777]) | 7\n",
-      "18136    | torch.Size([1, 1000, 762]) | 7\n",
-      "18137    | torch.Size([1, 1000, 787]) | 7\n",
-      "18138    | torch.Size([1, 1000, 778]) | 7\n",
-      "18139    | torch.Size([1, 1000, 771]) | 7\n",
-      "18140    | torch.Size([1, 1000, 780]) | 7\n",
-      "18141    | torch.Size([1, 1000, 762]) | 7\n",
-      "18142    | torch.Size([1, 1000, 781]) | 7\n",
-      "18143    | torch.Size([1, 1000, 771]) | 7\n",
-      "18144    | torch.Size([1, 1000, 754]) | 7\n",
-      "18145    | torch.Size([1, 1000, 754]) | 7\n",
-      "18146    | torch.Size([1, 1000, 754]) | 7\n",
-      "18147    | torch.Size([1, 1000, 754]) | 7\n",
-      "18148    | torch.Size([1, 1000, 754]) | 7\n",
-      "18149    | torch.Size([1, 1000, 786]) | 7\n",
-      "18150    | torch.Size([1, 1000, 754]) | 7\n",
-      "18151    | torch.Size([1, 1000, 762]) | 7\n",
-      "18152    | torch.Size([1, 1000, 802]) | 7\n",
-      "18153    | torch.Size([1, 1000, 754]) | 7\n",
-      "18154    | torch.Size([1, 1000, 771]) | 7\n",
-      "18155    | torch.Size([1, 1000, 771]) | 7\n",
-      "18156    | torch.Size([1, 1000, 775]) | 7\n",
-      "18157    | torch.Size([1, 1000, 797]) | 7\n",
-      "18158    | torch.Size([1, 1000, 803]) | 7\n",
-      "18159    | torch.Size([1, 1000, 771]) | 7\n",
-      "18160    | torch.Size([1, 1000, 762]) | 7\n",
-      "18161    | torch.Size([1, 1000, 765]) | 7\n",
-      "18162    | torch.Size([1, 1000, 778]) | 7\n",
-      "18163    | torch.Size([1, 1000, 754]) | 7\n",
-      "18164    | torch.Size([1, 1000, 754]) | 7\n",
-      "18165    | torch.Size([1, 1000, 762]) | 7\n",
-      "18166    | torch.Size([1, 1000, 754]) | 7\n",
-      "18167    | torch.Size([1, 1000, 777]) | 7\n",
-      "18168    | torch.Size([1, 1000, 802]) | 7\n",
-      "18169    | torch.Size([1, 1000, 802]) | 7\n",
-      "18170    | torch.Size([1, 1000, 754]) | 7\n",
-      "18171    | torch.Size([1, 1000, 785]) | 7\n",
-      "18172    | torch.Size([1, 1000, 771]) | 7\n",
-      "18173    | torch.Size([1, 1000, 805]) | 7\n",
-      "18174    | torch.Size([1, 1000, 754]) | 7\n",
-      "18175    | torch.Size([1, 1000, 754]) | 7\n",
-      "18176    | torch.Size([1, 1000, 804]) | 7\n",
-      "18177    | torch.Size([1, 1000, 754]) | 7\n",
-      "18178    | torch.Size([1, 1000, 754]) | 7\n",
-      "18179    | torch.Size([1, 1000, 771]) | 7\n",
-      "18180    | torch.Size([1, 1000, 754]) | 7\n",
-      "18181    | torch.Size([1, 1000, 754]) | 7\n",
-      "18182    | torch.Size([1, 1000, 777]) | 7\n",
-      "18183    | torch.Size([1, 1000, 781]) | 7\n",
-      "18184    | torch.Size([1, 1000, 749]) | 7\n",
-      "18185    | torch.Size([1, 1000, 772]) | 7\n",
-      "18186    | torch.Size([1, 1000, 754]) | 7\n",
-      "18187    | torch.Size([1, 1000, 784]) | 7\n",
-      "18188    | torch.Size([1, 1000, 786]) | 7\n",
-      "18189    | torch.Size([1, 1000, 774]) | 7\n",
-      "18190    | torch.Size([1, 1000, 803]) | 7\n",
-      "18191    | torch.Size([1, 1000, 777]) | 7\n",
-      "18192    | torch.Size([1, 1000, 816]) | 7\n",
-      "18193    | torch.Size([1, 1000, 777]) | 7\n",
-      "18194    | torch.Size([1, 1000, 777]) | 7\n",
-      "18195    | torch.Size([1, 1000, 762]) | 7\n",
-      "18196    | torch.Size([1, 1000, 782]) | 7\n",
-      "18197    | torch.Size([1, 1000, 771]) | 7\n",
-      "18198    | torch.Size([1, 1000, 772]) | 7\n",
-      "18199    | torch.Size([1, 1000, 754]) | 7\n",
-      "18200    | torch.Size([1, 1000, 754]) | 7\n",
-      "18201    | torch.Size([1, 1000, 790]) | 7\n",
-      "18202    | torch.Size([1, 1000, 783]) | 7\n",
-      "18203    | torch.Size([1, 1000, 754]) | 7\n",
-      "18204    | torch.Size([1, 1000, 777]) | 7\n",
-      "18205    | torch.Size([1, 1000, 754]) | 7\n",
-      "18206    | torch.Size([1, 1000, 754]) | 7\n",
-      "18207    | torch.Size([1, 1000, 762]) | 7\n",
-      "18208    | torch.Size([1, 1000, 777]) | 7\n",
-      "18209    | torch.Size([1, 1000, 754]) | 7\n",
-      "18210    | torch.Size([1, 1000, 762]) | 7\n",
-      "18211    | torch.Size([1, 1000, 762]) | 7\n",
-      "18212    | torch.Size([1, 1000, 754]) | 7\n",
-      "18213    | torch.Size([1, 1000, 813]) | 7\n",
-      "18214    | torch.Size([1, 1000, 752]) | 7\n",
-      "18215    | torch.Size([1, 1000, 754]) | 7\n",
-      "18216    | torch.Size([1, 1000, 784]) | 7\n",
-      "18217    | torch.Size([1, 1000, 771]) | 7\n",
-      "18218    | torch.Size([1, 1000, 783]) | 7\n",
-      "18219    | torch.Size([1, 1000, 761]) | 7\n",
-      "18220    | torch.Size([1, 1000, 795]) | 7\n",
-      "18221    | torch.Size([1, 1000, 771]) | 7\n",
-      "18222    | torch.Size([1, 1000, 766]) | 7\n",
-      "18223    | torch.Size([1, 1000, 781]) | 7\n",
-      "18224    | torch.Size([1, 1000, 762]) | 7\n",
-      "18225    | torch.Size([1, 1000, 780]) | 7\n",
-      "18226    | torch.Size([1, 1000, 762]) | 7\n",
-      "18227    | torch.Size([1, 1000, 776]) | 7\n",
-      "18228    | torch.Size([1, 1000, 754]) | 7\n",
-      "18229    | torch.Size([1, 1000, 777]) | 7\n",
-      "18230    | torch.Size([1, 1000, 778]) | 7\n",
-      "18231    | torch.Size([1, 1000, 797]) | 7\n",
-      "18232    | torch.Size([1, 1000, 771]) | 7\n",
-      "18233    | torch.Size([1, 1000, 790]) | 7\n",
-      "18234    | torch.Size([1, 1000, 754]) | 7\n",
-      "18235    | torch.Size([1, 1000, 754]) | 7\n",
-      "18236    | torch.Size([1, 1000, 794]) | 7\n",
-      "18237    | torch.Size([1, 1000, 792]) | 7\n",
-      "18238    | torch.Size([1, 1000, 772]) | 7\n",
-      "18239    | torch.Size([1, 1000, 754]) | 7\n",
-      "18240    | torch.Size([1, 1000, 754]) | 7\n",
-      "18241    | torch.Size([1, 1000, 813]) | 7\n",
-      "18242    | torch.Size([1, 1000, 754]) | 7\n",
-      "18243    | torch.Size([1, 1000, 779]) | 7\n",
-      "18244    | torch.Size([1, 1000, 786]) | 7\n",
-      "18245    | torch.Size([1, 1000, 762]) | 7\n",
-      "18246    | torch.Size([1, 1000, 762]) | 7\n",
-      "18247    | torch.Size([1, 1000, 762]) | 7\n",
-      "18248    | torch.Size([1, 1000, 762]) | 7\n",
-      "18249    | torch.Size([1, 1000, 846]) | 7\n",
-      "18250    | torch.Size([1, 1000, 786]) | 7\n",
-      "18251    | torch.Size([1, 1000, 754]) | 7\n",
-      "18252    | torch.Size([1, 1000, 762]) | 7\n",
-      "18253    | torch.Size([1, 1000, 763]) | 7\n",
-      "18254    | torch.Size([1, 1000, 754]) | 7\n",
-      "18255    | torch.Size([1, 1000, 778]) | 7\n",
-      "18256    | torch.Size([1, 1000, 798]) | 7\n",
-      "18257    | torch.Size([1, 1000, 793]) | 7\n",
-      "18258    | torch.Size([1, 1000, 789]) | 7\n",
-      "18259    | torch.Size([1, 1000, 754]) | 7\n",
-      "18260    | torch.Size([1, 1000, 762]) | 7\n",
-      "18261    | torch.Size([1, 1000, 762]) | 7\n",
-      "18262    | torch.Size([1, 1000, 762]) | 7\n",
-      "18263    | torch.Size([1, 1000, 808]) | 7\n",
-      "18264    | torch.Size([1, 1000, 762]) | 7\n",
-      "18265    | torch.Size([1, 1000, 754]) | 7\n",
-      "18266    | torch.Size([1, 1000, 777]) | 7\n",
-      "18267    | torch.Size([1, 1000, 762]) | 7\n",
-      "18268    | torch.Size([1, 1000, 762]) | 7\n",
-      "18269    | torch.Size([1, 1000, 786]) | 7\n",
-      "18270    | torch.Size([1, 1000, 776]) | 7\n",
-      "18271    | torch.Size([1, 1000, 754]) | 7\n",
-      "18272    | torch.Size([1, 1000, 754]) | 7\n",
-      "18273    | torch.Size([1, 1000, 772]) | 7\n",
-      "18274    | torch.Size([1, 1000, 754]) | 7\n",
-      "18275    | torch.Size([1, 1000, 811]) | 7\n",
-      "18276    | torch.Size([1, 1000, 771]) | 7\n",
-      "18277    | torch.Size([1, 1000, 754]) | 7\n",
-      "18278    | torch.Size([1, 1000, 754]) | 7\n",
-      "18279    | torch.Size([1, 1000, 754]) | 7\n",
-      "18280    | torch.Size([1, 1000, 774]) | 7\n",
-      "18281    | torch.Size([1, 1000, 762]) | 7\n",
-      "18282    | torch.Size([1, 1000, 802]) | 7\n",
-      "18283    | torch.Size([1, 1000, 754]) | 7\n",
-      "18284    | torch.Size([1, 1000, 754]) | 7\n",
-      "18285    | torch.Size([1, 1000, 817]) | 7\n",
-      "18286    | torch.Size([1, 1000, 779]) | 7\n",
-      "18287    | torch.Size([1, 1000, 790]) | 7\n",
-      "18288    | torch.Size([1, 1000, 772]) | 7\n",
-      "18289    | torch.Size([1, 1000, 771]) | 7\n",
-      "18290    | torch.Size([1, 1000, 754]) | 7\n",
-      "18291    | torch.Size([1, 1000, 779]) | 7\n",
-      "18292    | torch.Size([1, 1000, 777]) | 7\n",
-      "18293    | torch.Size([1, 1000, 772]) | 7\n",
-      "18294    | torch.Size([1, 1000, 754]) | 7\n",
-      "18295    | torch.Size([1, 1000, 772]) | 7\n",
-      "18296    | torch.Size([1, 1000, 754]) | 7\n",
-      "18297    | torch.Size([1, 1000, 762]) | 7\n",
-      "18298    | torch.Size([1, 1000, 777]) | 7\n",
-      "18299    | torch.Size([1, 1000, 774]) | 7\n",
-      "18300    | torch.Size([1, 1000, 862]) | 7\n",
-      "18301    | torch.Size([1, 1000, 754]) | 7\n",
-      "18302    | torch.Size([1, 1000, 771]) | 7\n",
-      "18303    | torch.Size([1, 1000, 782]) | 7\n",
-      "18304    | torch.Size([1, 1000, 754]) | 7\n",
-      "18305    | torch.Size([1, 1000, 786]) | 7\n",
-      "18306    | torch.Size([1, 1000, 787]) | 7\n",
-      "18307    | torch.Size([1, 1000, 784]) | 7\n",
-      "18308    | torch.Size([1, 1000, 762]) | 7\n",
-      "18309    | torch.Size([1, 1000, 775]) | 7\n",
-      "18310    | torch.Size([1, 1000, 777]) | 7\n",
-      "18311    | torch.Size([1, 1000, 754]) | 7\n",
-      "18312    | torch.Size([1, 1000, 762]) | 7\n",
-      "18313    | torch.Size([1, 1000, 777]) | 7\n",
-      "18314    | torch.Size([1, 1000, 754]) | 7\n",
-      "18315    | torch.Size([1, 1000, 754]) | 7\n",
-      "18316    | torch.Size([1, 1000, 781]) | 7\n",
-      "18317    | torch.Size([1, 1000, 754]) | 7\n",
-      "18318    | torch.Size([1, 1000, 754]) | 7\n",
-      "18319    | torch.Size([1, 1000, 754]) | 7\n",
-      "18320    | torch.Size([1, 1000, 783]) | 7\n",
-      "18321    | torch.Size([1, 1000, 764]) | 7\n",
-      "18322    | torch.Size([1, 1000, 800]) | 7\n",
-      "18323    | torch.Size([1, 1000, 777]) | 7\n",
-      "18324    | torch.Size([1, 1000, 762]) | 7\n",
-      "18325    | torch.Size([1, 1000, 724]) | 7\n",
-      "18326    | torch.Size([1, 1000, 780]) | 7\n",
-      "18327    | torch.Size([1, 1000, 754]) | 7\n",
-      "18328    | torch.Size([1, 1000, 777]) | 7\n",
-      "18329    | torch.Size([1, 1000, 754]) | 7\n",
-      "18330    | torch.Size([1, 1000, 777]) | 7\n",
-      "18331    | torch.Size([1, 1000, 777]) | 7\n",
-      "18332    | torch.Size([1, 1000, 754]) | 7\n",
-      "18333    | torch.Size([1, 1000, 754]) | 7\n",
-      "18334    | torch.Size([1, 1000, 777]) | 7\n",
-      "18335    | torch.Size([1, 1000, 762]) | 7\n",
-      "18336    | torch.Size([1, 1000, 762]) | 7\n",
-      "18337    | torch.Size([1, 1000, 793]) | 7\n",
-      "18338    | torch.Size([1, 1000, 754]) | 7\n",
-      "18339    | torch.Size([1, 1000, 777]) | 7\n",
-      "18340    | torch.Size([1, 1000, 754]) | 7\n",
-      "18341    | torch.Size([1, 1000, 777]) | 7\n",
-      "18342    | torch.Size([1, 1000, 777]) | 7\n",
-      "18343    | torch.Size([1, 1000, 777]) | 7\n",
-      "18344    | torch.Size([1, 1000, 777]) | 7\n",
-      "18345    | torch.Size([1, 1000, 762]) | 7\n",
-      "18346    | torch.Size([1, 1000, 754]) | 7\n",
-      "18347    | torch.Size([1, 1000, 754]) | 7\n",
-      "18348    | torch.Size([1, 1000, 762]) | 7\n",
-      "18349    | torch.Size([1, 1000, 754]) | 7\n",
-      "18350    | torch.Size([1, 1000, 754]) | 7\n",
-      "18351    | torch.Size([1, 1000, 776]) | 7\n",
-      "18352    | torch.Size([1, 1000, 754]) | 7\n",
-      "18353    | torch.Size([1, 1000, 754]) | 7\n",
-      "18354    | torch.Size([1, 1000, 777]) | 7\n",
-      "18355    | torch.Size([1, 1000, 777]) | 7\n",
-      "18356    | torch.Size([1, 1000, 754]) | 7\n",
-      "18357    | torch.Size([1, 1000, 754]) | 7\n",
-      "18358    | torch.Size([1, 1000, 801]) | 7\n",
-      "18359    | torch.Size([1, 1000, 781]) | 7\n",
-      "18360    | torch.Size([1, 1000, 802]) | 7\n",
-      "18361    | torch.Size([1, 1000, 777]) | 7\n",
-      "18362    | torch.Size([1, 1000, 762]) | 7\n",
-      "18363    | torch.Size([1, 1000, 754]) | 7\n",
-      "18364    | torch.Size([1, 1000, 754]) | 7\n",
-      "18365    | torch.Size([1, 1000, 777]) | 7\n",
-      "18366    | torch.Size([1, 1000, 754]) | 7\n",
-      "18367    | torch.Size([1, 1000, 754]) | 7\n",
-      "18368    | torch.Size([1, 1000, 754]) | 7\n",
-      "18369    | torch.Size([1, 1000, 762]) | 7\n",
-      "18370    | torch.Size([1, 1000, 754]) | 7\n",
-      "18371    | torch.Size([1, 1000, 754]) | 7\n",
-      "18372    | torch.Size([1, 1000, 754]) | 7\n",
-      "18373    | torch.Size([1, 1000, 754]) | 7\n",
-      "18374    | torch.Size([1, 1000, 772]) | 7\n",
-      "18375    | torch.Size([1, 1000, 774]) | 7\n",
-      "18376    | torch.Size([1, 1000, 754]) | 7\n",
-      "18377    | torch.Size([1, 1000, 776]) | 7\n",
-      "18378    | torch.Size([1, 1000, 1473]) | 7\n",
-      "18379    | torch.Size([1, 1000, 754]) | 7\n",
-      "18380    | torch.Size([1, 1000, 777]) | 7\n",
-      "18381    | torch.Size([1, 1000, 754]) | 7\n",
-      "18382    | torch.Size([1, 1000, 762]) | 7\n",
-      "18383    | torch.Size([1, 1000, 762]) | 7\n",
-      "18384    | torch.Size([1, 1000, 780]) | 7\n",
-      "18385    | torch.Size([1, 1000, 781]) | 7\n",
-      "18386    | torch.Size([1, 1000, 780]) | 7\n",
-      "18387    | torch.Size([1, 1000, 787]) | 7\n",
-      "18388    | torch.Size([1, 1000, 800]) | 7\n",
-      "18389    | torch.Size([1, 1000, 810]) | 7\n",
-      "18390    | torch.Size([1, 1000, 771]) | 7\n",
-      "18391    | torch.Size([1, 1000, 762]) | 7\n",
-      "18392    | torch.Size([1, 1000, 767]) | 7\n",
-      "18393    | torch.Size([1, 1000, 754]) | 7\n",
-      "18394    | torch.Size([1, 1000, 777]) | 7\n",
-      "18395    | torch.Size([1, 1000, 771]) | 7\n",
-      "18396    | torch.Size([1, 1000, 754]) | 7\n",
-      "18397    | torch.Size([1, 1000, 754]) | 7\n",
-      "18398    | torch.Size([1, 1000, 754]) | 7\n",
-      "18399    | torch.Size([1, 1000, 754]) | 7\n",
-      "18400    | torch.Size([1, 1000, 801]) | 7\n",
-      "18401    | torch.Size([1, 1000, 787]) | 7\n",
-      "18402    | torch.Size([1, 1000, 776]) | 7\n",
-      "18403    | torch.Size([1, 1000, 775]) | 7\n",
-      "18404    | torch.Size([1, 1000, 762]) | 7\n",
-      "18405    | torch.Size([1, 1000, 762]) | 7\n",
-      "18406    | torch.Size([1, 1000, 804]) | 7\n",
-      "18407    | torch.Size([1, 1000, 777]) | 7\n",
-      "18408    | torch.Size([1, 1000, 776]) | 7\n",
-      "18409    | torch.Size([1, 1000, 762]) | 7\n",
-      "18410    | torch.Size([1, 1000, 798]) | 7\n",
-      "18411    | torch.Size([1, 1000, 773]) | 7\n",
-      "18412    | torch.Size([1, 1000, 815]) | 7\n",
-      "18413    | torch.Size([1, 1000, 741]) | 7\n",
-      "18414    | torch.Size([1, 1000, 775]) | 7\n",
-      "18415    | torch.Size([1, 1000, 777]) | 7\n",
-      "18416    | torch.Size([1, 1000, 754]) | 7\n",
-      "18417    | torch.Size([1, 1000, 762]) | 7\n",
-      "18418    | torch.Size([1, 1000, 754]) | 7\n",
-      "18419    | torch.Size([1, 1000, 781]) | 7\n",
-      "18420    | torch.Size([1, 1000, 754]) | 7\n",
-      "18421    | torch.Size([1, 1000, 771]) | 7\n",
-      "18422    | torch.Size([1, 1000, 783]) | 7\n",
-      "18423    | torch.Size([1, 1000, 777]) | 7\n",
-      "18424    | torch.Size([1, 1000, 754]) | 7\n",
-      "18425    | torch.Size([1, 1000, 754]) | 7\n",
-      "18426    | torch.Size([1, 1000, 754]) | 7\n",
-      "18427    | torch.Size([1, 1000, 762]) | 7\n",
-      "18428    | torch.Size([1, 1000, 754]) | 7\n",
-      "18429    | torch.Size([1, 1000, 775]) | 7\n",
-      "18430    | torch.Size([1, 1000, 754]) | 7\n",
-      "18431    | torch.Size([1, 1000, 754]) | 7\n",
-      "18432    | torch.Size([1, 1000, 754]) | 7\n",
-      "18433    | torch.Size([1, 1000, 754]) | 7\n",
-      "18434    | torch.Size([1, 1000, 762]) | 7\n",
-      "18435    | torch.Size([1, 1000, 754]) | 7\n",
-      "18436    | torch.Size([1, 1000, 754]) | 7\n",
-      "18437    | torch.Size([1, 1000, 771]) | 7\n",
-      "18438    | torch.Size([1, 1000, 788]) | 7\n",
-      "18439    | torch.Size([1, 1000, 754]) | 7\n",
-      "18440    | torch.Size([1, 1000, 754]) | 7\n",
-      "18441    | torch.Size([1, 1000, 777]) | 7\n",
-      "18442    | torch.Size([1, 1000, 762]) | 7\n",
-      "18443    | torch.Size([1, 1000, 754]) | 7\n",
-      "18444    | torch.Size([1, 1000, 754]) | 7\n",
-      "18445    | torch.Size([1, 1000, 762]) | 7\n",
-      "18446    | torch.Size([1, 1000, 724]) | 7\n",
-      "18447    | torch.Size([1, 1000, 788]) | 7\n",
-      "18448    | torch.Size([1, 1000, 754]) | 7\n",
-      "18449    | torch.Size([1, 1000, 754]) | 7\n",
-      "18450    | torch.Size([1, 1000, 762]) | 7\n",
-      "18451    | torch.Size([1, 1000, 754]) | 7\n",
-      "18452    | torch.Size([1, 1000, 787]) | 7\n",
-      "18453    | torch.Size([1, 1000, 772]) | 7\n",
-      "18454    | torch.Size([1, 1000, 773]) | 7\n",
-      "18455    | torch.Size([1, 1000, 788]) | 7\n",
-      "18456    | torch.Size([1, 1000, 787]) | 7\n",
-      "18457    | torch.Size([1, 1000, 786]) | 7\n",
-      "18458    | torch.Size([1, 1000, 762]) | 7\n",
-      "18459    | torch.Size([1, 1000, 754]) | 7\n",
-      "18460    | torch.Size([1, 1000, 777]) | 7\n",
-      "18461    | torch.Size([1, 1000, 754]) | 7\n",
-      "18462    | torch.Size([1, 1000, 754]) | 7\n",
-      "18463    | torch.Size([1, 1000, 779]) | 7\n",
-      "18464    | torch.Size([1, 1000, 762]) | 7\n",
-      "18465    | torch.Size([1, 1000, 762]) | 7\n",
-      "18466    | torch.Size([1, 1000, 754]) | 7\n",
-      "18467    | torch.Size([1, 1000, 772]) | 7\n",
-      "18468    | torch.Size([1, 1000, 766]) | 7\n",
-      "18469    | torch.Size([1, 1000, 754]) | 7\n",
-      "18470    | torch.Size([1, 1000, 783]) | 7\n",
-      "18471    | torch.Size([1, 1000, 770]) | 7\n",
-      "18472    | torch.Size([1, 1000, 754]) | 7\n",
-      "18473    | torch.Size([1, 1000, 777]) | 7\n",
-      "18474    | torch.Size([1, 1000, 792]) | 7\n",
-      "18475    | torch.Size([1, 1000, 771]) | 7\n",
-      "18476    | torch.Size([1, 1000, 754]) | 7\n",
-      "18477    | torch.Size([1, 1000, 754]) | 7\n",
-      "18478    | torch.Size([1, 1000, 762]) | 7\n",
-      "18479    | torch.Size([1, 1000, 754]) | 7\n",
-      "18480    | torch.Size([1, 1000, 772]) | 7\n",
-      "18481    | torch.Size([1, 1000, 802]) | 7\n",
-      "18482    | torch.Size([1, 1000, 754]) | 7\n",
-      "18483    | torch.Size([1, 1000, 754]) | 7\n",
-      "18484    | torch.Size([1, 1000, 777]) | 7\n",
-      "18485    | torch.Size([1, 1000, 762]) | 7\n",
-      "18486    | torch.Size([1, 1000, 771]) | 7\n",
-      "18487    | torch.Size([1, 1000, 754]) | 7\n",
-      "18488    | torch.Size([1, 1000, 771]) | 7\n",
-      "18489    | torch.Size([1, 1000, 754]) | 7\n",
-      "18490    | torch.Size([1, 1000, 782]) | 7\n",
-      "18491    | torch.Size([1, 1000, 777]) | 7\n",
-      "18492    | torch.Size([1, 1000, 754]) | 7\n",
-      "18493    | torch.Size([1, 1000, 781]) | 7\n",
-      "18494    | torch.Size([1, 1000, 782]) | 7\n",
-      "18495    | torch.Size([1, 1000, 754]) | 7\n",
-      "18496    | torch.Size([1, 1000, 754]) | 7\n",
-      "18497    | torch.Size([1, 1000, 762]) | 7\n",
-      "18498    | torch.Size([1, 1000, 754]) | 7\n",
-      "18499    | torch.Size([1, 1000, 787]) | 7\n",
-      "18500    | torch.Size([1, 1000, 762]) | 7\n",
-      "18501    | torch.Size([1, 1000, 754]) | 7\n",
-      "18502    | torch.Size([1, 1000, 785]) | 7\n",
-      "18503    | torch.Size([1, 1000, 777]) | 7\n",
-      "18504    | torch.Size([1, 1000, 754]) | 7\n",
-      "18505    | torch.Size([1, 1000, 778]) | 7\n",
-      "18506    | torch.Size([1, 1000, 777]) | 7\n",
-      "18507    | torch.Size([1, 1000, 762]) | 7\n",
-      "18508    | torch.Size([1, 1000, 799]) | 7\n",
-      "18509    | torch.Size([1, 1000, 762]) | 7\n",
-      "18510    | torch.Size([1, 1000, 782]) | 7\n",
-      "18511    | torch.Size([1, 1000, 781]) | 7\n",
-      "18512    | torch.Size([1, 1000, 754]) | 7\n",
-      "18513    | torch.Size([1, 1000, 777]) | 7\n",
-      "18514    | torch.Size([1, 1000, 754]) | 7\n",
-      "18515    | torch.Size([1, 1000, 785]) | 7\n",
-      "18516    | torch.Size([1, 1000, 809]) | 7\n",
-      "18517    | torch.Size([1, 1000, 754]) | 7\n",
-      "18518    | torch.Size([1, 1000, 789]) | 7\n",
-      "18519    | torch.Size([1, 1000, 777]) | 7\n",
-      "18520    | torch.Size([1, 1000, 782]) | 7\n",
-      "18521    | torch.Size([1, 1000, 793]) | 7\n",
-      "18522    | torch.Size([1, 1000, 787]) | 7\n",
-      "18523    | torch.Size([1, 1000, 779]) | 7\n",
-      "18524    | torch.Size([1, 1000, 782]) | 7\n",
-      "18525    | torch.Size([1, 1000, 754]) | 7\n",
-      "18526    | torch.Size([1, 1000, 762]) | 7\n",
-      "18527    | torch.Size([1, 1000, 762]) | 7\n",
-      "18528    | torch.Size([1, 1000, 754]) | 7\n",
-      "18529    | torch.Size([1, 1000, 754]) | 7\n",
-      "18530    | torch.Size([1, 1000, 773]) | 7\n",
-      "18531    | torch.Size([1, 1000, 777]) | 7\n",
-      "18532    | torch.Size([1, 1000, 754]) | 7\n",
-      "18533    | torch.Size([1, 1000, 763]) | 7\n",
-      "18534    | torch.Size([1, 1000, 661]) | 7\n",
-      "18535    | torch.Size([1, 1000, 778]) | 7\n",
-      "18536    | torch.Size([1, 1000, 784]) | 7\n",
-      "18537    | torch.Size([1, 1000, 783]) | 7\n",
-      "18538    | torch.Size([1, 1000, 771]) | 7\n",
-      "18539    | torch.Size([1, 1000, 778]) | 7\n",
-      "18540    | torch.Size([1, 1000, 754]) | 7\n",
-      "18541    | torch.Size([1, 1000, 762]) | 7\n",
-      "18542    | torch.Size([1, 1000, 767]) | 7\n",
-      "18543    | torch.Size([1, 1000, 790]) | 7\n",
-      "18544    | torch.Size([1, 1000, 754]) | 7\n",
-      "18545    | torch.Size([1, 1000, 771]) | 7\n",
-      "18546    | torch.Size([1, 1000, 754]) | 7\n",
-      "18547    | torch.Size([1, 1000, 767]) | 7\n",
-      "18548    | torch.Size([1, 1000, 761]) | 7\n",
-      "18549    | torch.Size([1, 1000, 777]) | 7\n",
-      "18550    | torch.Size([1, 1000, 762]) | 7\n",
-      "18551    | torch.Size([1, 1000, 754]) | 7\n",
-      "18552    | torch.Size([1, 1000, 754]) | 7\n",
-      "18553    | torch.Size([1, 1000, 754]) | 7\n",
-      "18554    | torch.Size([1, 1000, 767]) | 7\n",
-      "18555    | torch.Size([1, 1000, 754]) | 7\n",
-      "18556    | torch.Size([1, 1000, 779]) | 7\n",
-      "18557    | torch.Size([1, 1000, 754]) | 7\n",
-      "18558    | torch.Size([1, 1000, 754]) | 7\n",
-      "18559    | torch.Size([1, 1000, 802]) | 7\n",
-      "18560    | torch.Size([1, 1000, 771]) | 7\n",
-      "18561    | torch.Size([1, 1000, 777]) | 7\n",
-      "18562    | torch.Size([1, 1000, 762]) | 7\n",
-      "18563    | torch.Size([1, 1000, 776]) | 7\n",
-      "18564    | torch.Size([1, 1000, 760]) | 7\n",
-      "18565    | torch.Size([1, 1000, 754]) | 7\n",
-      "18566    | torch.Size([1, 1000, 754]) | 7\n",
-      "18567    | torch.Size([1, 1000, 769]) | 7\n",
-      "18568    | torch.Size([1, 1000, 765]) | 7\n",
-      "18569    | torch.Size([1, 1000, 732]) | 7\n",
-      "18570    | torch.Size([1, 1000, 782]) | 7\n",
-      "18571    | torch.Size([1, 1000, 775]) | 7\n",
-      "18572    | torch.Size([1, 1000, 771]) | 7\n",
-      "18573    | torch.Size([1, 1000, 777]) | 7\n",
-      "18574    | torch.Size([1, 1000, 807]) | 7\n",
-      "18575    | torch.Size([1, 1000, 754]) | 7\n",
-      "18576    | torch.Size([1, 1000, 754]) | 7\n",
-      "18577    | torch.Size([1, 1000, 784]) | 7\n",
-      "18578    | torch.Size([1, 1000, 754]) | 7\n",
-      "18579    | torch.Size([1, 1000, 783]) | 7\n",
-      "18580    | torch.Size([1, 1000, 754]) | 7\n",
-      "18581    | torch.Size([1, 1000, 790]) | 7\n",
-      "18582    | torch.Size([1, 1000, 762]) | 7\n",
-      "18583    | torch.Size([1, 1000, 789]) | 7\n",
-      "18584    | torch.Size([1, 1000, 762]) | 7\n",
-      "18585    | torch.Size([1, 1000, 786]) | 7\n",
-      "18586    | torch.Size([1, 1000, 777]) | 7\n",
-      "18587    | torch.Size([1, 1000, 754]) | 7\n",
-      "18588    | torch.Size([1, 1000, 786]) | 7\n",
-      "18589    | torch.Size([1, 1000, 772]) | 7\n",
-      "18590    | torch.Size([1, 1000, 762]) | 7\n",
-      "18591    | torch.Size([1, 1000, 778]) | 7\n",
-      "18592    | torch.Size([1, 1000, 754]) | 7\n",
-      "18593    | torch.Size([1, 1000, 754]) | 7\n",
-      "18594    | torch.Size([1, 1000, 810]) | 7\n",
-      "18595    | torch.Size([1, 1000, 779]) | 7\n",
-      "18596    | torch.Size([1, 1000, 804]) | 7\n",
-      "18597    | torch.Size([1, 1000, 754]) | 7\n",
-      "18598    | torch.Size([1, 1000, 802]) | 7\n",
-      "18599    | torch.Size([1, 1000, 754]) | 7\n",
-      "18600    | torch.Size([1, 1000, 760]) | 7\n",
-      "18601    | torch.Size([1, 1000, 754]) | 7\n",
-      "18602    | torch.Size([1, 1000, 754]) | 7\n",
-      "18603    | torch.Size([1, 1000, 777]) | 7\n",
-      "18604    | torch.Size([1, 1000, 784]) | 7\n",
-      "18605    | torch.Size([1, 1000, 803]) | 7\n",
-      "18606    | torch.Size([1, 1000, 779]) | 7\n",
-      "18607    | torch.Size([1, 1000, 767]) | 7\n",
-      "18608    | torch.Size([1, 1000, 802]) | 7\n",
-      "18609    | torch.Size([1, 1000, 776]) | 7\n",
-      "18610    | torch.Size([1, 1000, 754]) | 7\n",
-      "18611    | torch.Size([1, 1000, 754]) | 7\n",
-      "18612    | torch.Size([1, 1000, 754]) | 7\n",
-      "18613    | torch.Size([1, 1000, 786]) | 7\n",
-      "18614    | torch.Size([1, 1000, 754]) | 7\n",
-      "18615    | torch.Size([1, 1000, 754]) | 7\n",
-      "18616    | torch.Size([1, 1000, 762]) | 7\n",
-      "18617    | torch.Size([1, 1000, 802]) | 7\n",
-      "18618    | torch.Size([1, 1000, 754]) | 7\n",
-      "18619    | torch.Size([1, 1000, 791]) | 7\n",
-      "18620    | torch.Size([1, 1000, 754]) | 7\n",
-      "18621    | torch.Size([1, 1000, 787]) | 7\n",
-      "18622    | torch.Size([1, 1000, 754]) | 7\n",
-      "18623    | torch.Size([1, 1000, 770]) | 7\n",
-      "18624    | torch.Size([1, 1000, 754]) | 7\n",
-      "18625    | torch.Size([1, 1000, 779]) | 7\n",
-      "18626    | torch.Size([1, 1000, 754]) | 7\n",
-      "18627    | torch.Size([1, 1000, 784]) | 7\n",
-      "18628    | torch.Size([1, 1000, 777]) | 7\n",
-      "18629    | torch.Size([1, 1000, 781]) | 7\n",
-      "18630    | torch.Size([1, 1000, 780]) | 7\n",
-      "18631    | torch.Size([1, 1000, 754]) | 7\n",
-      "18632    | torch.Size([1, 1000, 778]) | 7\n",
-      "18633    | torch.Size([1, 1000, 733]) | 7\n",
-      "18634    | torch.Size([1, 1000, 754]) | 7\n",
-      "18635    | torch.Size([1, 1000, 754]) | 7\n",
-      "18636    | torch.Size([1, 1000, 754]) | 7\n",
-      "18637    | torch.Size([1, 1000, 815]) | 7\n",
-      "18638    | torch.Size([1, 1000, 780]) | 7\n",
-      "18639    | torch.Size([1, 1000, 754]) | 7\n",
-      "18640    | torch.Size([1, 1000, 777]) | 7\n",
-      "18641    | torch.Size([1, 1000, 787]) | 7\n",
-      "18642    | torch.Size([1, 1000, 777]) | 7\n",
-      "18643    | torch.Size([1, 1000, 767]) | 7\n",
-      "18644    | torch.Size([1, 1000, 771]) | 7\n",
-      "18645    | torch.Size([1, 1000, 783]) | 7\n",
-      "18646    | torch.Size([1, 1000, 778]) | 7\n",
-      "18647    | torch.Size([1, 1000, 754]) | 7\n",
-      "18648    | torch.Size([1, 1000, 754]) | 7\n",
-      "18649    | torch.Size([1, 1000, 754]) | 7\n",
-      "18650    | torch.Size([1, 1000, 754]) | 7\n",
-      "18651    | torch.Size([1, 1000, 754]) | 7\n",
-      "18652    | torch.Size([1, 1000, 771]) | 7\n",
-      "18653    | torch.Size([1, 1000, 754]) | 7\n",
-      "18654    | torch.Size([1, 1000, 779]) | 7\n",
-      "18655    | torch.Size([1, 1000, 769]) | 7\n",
-      "18656    | torch.Size([1, 1000, 754]) | 7\n",
-      "18657    | torch.Size([1, 1000, 771]) | 7\n",
-      "18658    | torch.Size([1, 1000, 788]) | 7\n",
-      "18659    | torch.Size([1, 1000, 754]) | 7\n",
-      "18660    | torch.Size([1, 1000, 788]) | 7\n",
-      "18661    | torch.Size([1, 1000, 771]) | 7\n",
-      "18662    | torch.Size([1, 1000, 784]) | 7\n",
-      "18663    | torch.Size([1, 1000, 779]) | 7\n",
-      "18664    | torch.Size([1, 1000, 762]) | 7\n",
-      "18665    | torch.Size([1, 1000, 762]) | 7\n",
-      "18666    | torch.Size([1, 1000, 754]) | 7\n",
-      "18667    | torch.Size([1, 1000, 778]) | 7\n",
-      "18668    | torch.Size([1, 1000, 754]) | 7\n",
-      "18669    | torch.Size([1, 1000, 754]) | 7\n",
-      "18670    | torch.Size([1, 1000, 754]) | 7\n",
-      "18671    | torch.Size([1, 1000, 786]) | 7\n",
-      "18672    | torch.Size([1, 1000, 754]) | 7\n",
-      "18673    | torch.Size([1, 1000, 762]) | 7\n",
-      "18674    | torch.Size([1, 1000, 754]) | 7\n",
-      "18675    | torch.Size([1, 1000, 762]) | 7\n",
-      "18676    | torch.Size([1, 1000, 754]) | 7\n",
-      "18677    | torch.Size([1, 1000, 754]) | 7\n",
-      "18678    | torch.Size([1, 1000, 762]) | 7\n",
-      "18679    | torch.Size([1, 1000, 754]) | 7\n",
-      "18680    | torch.Size([1, 1000, 772]) | 7\n",
-      "18681    | torch.Size([1, 1000, 762]) | 7\n",
-      "18682    | torch.Size([1, 1000, 724]) | 7\n",
-      "18683    | torch.Size([1, 1000, 771]) | 7\n",
-      "18684    | torch.Size([1, 1000, 785]) | 7\n",
-      "18685    | torch.Size([1, 1000, 782]) | 7\n",
-      "18686    | torch.Size([1, 1000, 750]) | 7\n",
-      "18687    | torch.Size([1, 1000, 754]) | 7\n",
-      "18688    | torch.Size([1, 1000, 775]) | 7\n",
-      "18689    | torch.Size([1, 1000, 767]) | 7\n",
-      "18690    | torch.Size([1, 1000, 788]) | 7\n",
-      "18691    | torch.Size([1, 1000, 777]) | 7\n",
-      "18692    | torch.Size([1, 1000, 790]) | 7\n",
-      "18693    | torch.Size([1, 1000, 732]) | 7\n",
-      "18694    | torch.Size([1, 1000, 754]) | 7\n",
-      "18695    | torch.Size([1, 1000, 784]) | 7\n",
-      "18696    | torch.Size([1, 1000, 790]) | 7\n",
-      "18697    | torch.Size([1, 1000, 771]) | 7\n",
-      "18698    | torch.Size([1, 1000, 754]) | 7\n",
-      "18699    | torch.Size([1, 1000, 789]) | 7\n",
-      "18700    | torch.Size([1, 1000, 777]) | 7\n",
-      "18701    | torch.Size([1, 1000, 777]) | 7\n",
-      "18702    | torch.Size([1, 1000, 754]) | 7\n",
-      "18703    | torch.Size([1, 1000, 754]) | 7\n",
-      "18704    | torch.Size([1, 1000, 777]) | 7\n",
-      "18705    | torch.Size([1, 1000, 754]) | 7\n",
-      "18706    | torch.Size([1, 1000, 762]) | 7\n",
-      "18707    | torch.Size([1, 1000, 762]) | 7\n",
-      "18708    | torch.Size([1, 1000, 779]) | 7\n",
-      "18709    | torch.Size([1, 1000, 754]) | 7\n",
-      "18710    | torch.Size([1, 1000, 754]) | 7\n",
-      "18711    | torch.Size([1, 1000, 754]) | 7\n",
-      "18712    | torch.Size([1, 1000, 762]) | 7\n",
-      "18713    | torch.Size([1, 1000, 754]) | 7\n",
-      "18714    | torch.Size([1, 1000, 785]) | 7\n",
-      "18715    | torch.Size([1, 1000, 809]) | 7\n",
-      "18716    | torch.Size([1, 1000, 790]) | 7\n",
-      "18717    | torch.Size([1, 1000, 793]) | 7\n",
-      "18718    | torch.Size([1, 1000, 771]) | 7\n",
-      "18719    | torch.Size([1, 1000, 748]) | 7\n",
-      "18720    | torch.Size([1, 1000, 780]) | 7\n",
-      "18721    | torch.Size([1, 1000, 771]) | 7\n",
-      "18722    | torch.Size([1, 1000, 762]) | 7\n",
-      "18723    | torch.Size([1, 1000, 754]) | 7\n",
-      "18724    | torch.Size([1, 1000, 772]) | 7\n",
-      "18725    | torch.Size([1, 1000, 761]) | 7\n",
-      "18726    | torch.Size([1, 1000, 771]) | 7\n",
-      "18727    | torch.Size([1, 1000, 752]) | 7\n",
-      "18728    | torch.Size([1, 1000, 778]) | 7\n",
-      "18729    | torch.Size([1, 1000, 781]) | 7\n",
-      "18730    | torch.Size([1, 1000, 787]) | 7\n",
-      "18731    | torch.Size([1, 1000, 754]) | 7\n",
-      "18732    | torch.Size([1, 1000, 785]) | 7\n",
-      "18733    | torch.Size([1, 1000, 754]) | 7\n",
-      "18734    | torch.Size([1, 1000, 754]) | 7\n",
-      "18735    | torch.Size([1, 1000, 762]) | 7\n",
-      "18736    | torch.Size([1, 1000, 788]) | 7\n",
-      "18737    | torch.Size([1, 1000, 762]) | 7\n",
-      "18738    | torch.Size([1, 1000, 777]) | 7\n",
-      "18739    | torch.Size([1, 1000, 754]) | 7\n",
-      "18740    | torch.Size([1, 1000, 789]) | 7\n",
-      "18741    | torch.Size([1, 1000, 767]) | 7\n",
-      "18742    | torch.Size([1, 1000, 784]) | 7\n",
-      "18743    | torch.Size([1, 1000, 779]) | 7\n",
-      "18744    | torch.Size([1, 1000, 790]) | 7\n",
-      "18745    | torch.Size([1, 1000, 771]) | 7\n",
-      "18746    | torch.Size([1, 1000, 777]) | 7\n",
-      "18747    | torch.Size([1, 1000, 754]) | 7\n",
-      "18748    | torch.Size([1, 1000, 777]) | 7\n",
-      "18749    | torch.Size([1, 1000, 754]) | 7\n",
-      "18750    | torch.Size([1, 1000, 754]) | 7\n",
-      "18751    | torch.Size([1, 1000, 754]) | 7\n",
-      "18752    | torch.Size([1, 1000, 754]) | 7\n",
-      "18753    | torch.Size([1, 1000, 771]) | 7\n",
-      "18754    | torch.Size([1, 1000, 762]) | 7\n",
-      "18755    | torch.Size([1, 1000, 789]) | 7\n",
-      "18756    | torch.Size([1, 1000, 777]) | 7\n",
-      "18757    | torch.Size([1, 1000, 777]) | 7\n",
-      "18758    | torch.Size([1, 1000, 771]) | 7\n",
-      "18759    | torch.Size([1, 1000, 789]) | 7\n",
-      "18760    | torch.Size([1, 1000, 774]) | 7\n",
-      "18761    | torch.Size([1, 1000, 772]) | 7\n",
-      "18762    | torch.Size([1, 1000, 791]) | 7\n",
-      "18763    | torch.Size([1, 1000, 754]) | 7\n",
-      "18764    | torch.Size([1, 1000, 766]) | 7\n",
-      "18765    | torch.Size([1, 1000, 806]) | 7\n",
-      "18766    | torch.Size([1, 1000, 754]) | 7\n",
-      "18767    | torch.Size([1, 1000, 787]) | 7\n",
-      "18768    | torch.Size([1, 1000, 789]) | 7\n",
-      "18769    | torch.Size([1, 1000, 771]) | 7\n",
-      "18770    | torch.Size([1, 1000, 810]) | 7\n",
-      "18771    | torch.Size([1, 1000, 754]) | 7\n",
-      "18772    | torch.Size([1, 1000, 754]) | 7\n",
-      "18773    | torch.Size([1, 1000, 754]) | 7\n",
-      "18774    | torch.Size([1, 1000, 754]) | 7\n",
-      "18775    | torch.Size([1, 1000, 754]) | 7\n",
-      "18776    | torch.Size([1, 1000, 754]) | 7\n",
-      "18777    | torch.Size([1, 1000, 809]) | 7\n",
-      "18778    | torch.Size([1, 1000, 774]) | 7\n",
-      "18779    | torch.Size([1, 1000, 784]) | 7\n",
-      "18780    | torch.Size([1, 1000, 777]) | 7\n",
-      "18781    | torch.Size([1, 1000, 762]) | 7\n",
-      "18782    | torch.Size([1, 1000, 779]) | 7\n",
-      "18783    | torch.Size([1, 1000, 754]) | 7\n",
-      "18784    | torch.Size([1, 1000, 754]) | 7\n",
-      "18785    | torch.Size([1, 1000, 771]) | 7\n",
-      "18786    | torch.Size([1, 1000, 777]) | 7\n",
-      "18787    | torch.Size([1, 1000, 754]) | 7\n",
-      "18788    | torch.Size([1, 1000, 805]) | 7\n",
-      "18789    | torch.Size([1, 1000, 775]) | 7\n",
-      "18790    | torch.Size([1, 1000, 762]) | 7\n",
-      "18791    | torch.Size([1, 1000, 791]) | 7\n",
-      "18792    | torch.Size([1, 1000, 776]) | 7\n",
-      "18793    | torch.Size([1, 1000, 777]) | 7\n",
-      "18794    | torch.Size([1, 1000, 782]) | 7\n",
-      "18795    | torch.Size([1, 1000, 778]) | 7\n",
-      "18796    | torch.Size([1, 1000, 754]) | 7\n",
-      "18797    | torch.Size([1, 1000, 754]) | 7\n",
-      "18798    | torch.Size([1, 1000, 754]) | 7\n",
-      "18799    | torch.Size([1, 1000, 754]) | 7\n",
-      "18800    | torch.Size([1, 1000, 658]) | 7\n",
-      "18801    | torch.Size([1, 1000, 777]) | 7\n",
-      "18802    | torch.Size([1, 1000, 806]) | 7\n",
-      "18803    | torch.Size([1, 1000, 793]) | 7\n",
-      "18804    | torch.Size([1, 1000, 754]) | 7\n",
-      "18805    | torch.Size([1, 1000, 762]) | 7\n",
-      "18806    | torch.Size([1, 1000, 754]) | 7\n",
-      "18807    | torch.Size([1, 1000, 771]) | 7\n",
-      "18808    | torch.Size([1, 1000, 807]) | 7\n",
-      "18809    | torch.Size([1, 1000, 802]) | 7\n",
-      "18810    | torch.Size([1, 1000, 754]) | 7\n",
-      "18811    | torch.Size([1, 1000, 767]) | 7\n",
-      "18812    | torch.Size([1, 1000, 754]) | 7\n",
-      "18813    | torch.Size([1, 1000, 777]) | 7\n",
-      "18814    | torch.Size([1, 1000, 788]) | 7\n",
-      "18815    | torch.Size([1, 1000, 774]) | 7\n",
-      "18816    | torch.Size([1, 1000, 766]) | 7\n",
-      "18817    | torch.Size([1, 1000, 771]) | 7\n",
-      "18818    | torch.Size([1, 1000, 762]) | 7\n",
-      "18819    | torch.Size([1, 1000, 802]) | 7\n",
-      "18820    | torch.Size([1, 1000, 754]) | 7\n",
-      "18821    | torch.Size([1, 1000, 784]) | 7\n",
-      "18822    | torch.Size([1, 1000, 754]) | 7\n",
-      "18823    | torch.Size([1, 1000, 754]) | 7\n",
-      "18824    | torch.Size([1, 1000, 780]) | 7\n",
-      "18825    | torch.Size([1, 1000, 754]) | 7\n",
-      "18826    | torch.Size([1, 1000, 792]) | 7\n",
-      "18827    | torch.Size([1, 1000, 754]) | 7\n",
-      "18828    | torch.Size([1, 1000, 754]) | 7\n",
-      "18829    | torch.Size([1, 1000, 754]) | 7\n",
-      "18830    | torch.Size([1, 1000, 754]) | 7\n",
-      "18831    | torch.Size([1, 1000, 762]) | 7\n",
-      "18832    | torch.Size([1, 1000, 754]) | 7\n",
-      "18833    | torch.Size([1, 1000, 767]) | 7\n",
-      "18834    | torch.Size([1, 1000, 771]) | 7\n",
-      "18835    | torch.Size([1, 1000, 754]) | 7\n",
-      "18836    | torch.Size([1, 1000, 754]) | 7\n",
-      "18837    | torch.Size([1, 1000, 762]) | 7\n",
-      "18838    | torch.Size([1, 1000, 777]) | 7\n",
-      "18839    | torch.Size([1, 1000, 771]) | 7\n",
-      "18840    | torch.Size([1, 1000, 762]) | 7\n",
-      "18841    | torch.Size([1, 1000, 780]) | 7\n",
-      "18842    | torch.Size([1, 1000, 776]) | 7\n",
-      "18843    | torch.Size([1, 1000, 771]) | 7\n",
-      "18844    | torch.Size([1, 1000, 754]) | 7\n",
-      "18845    | torch.Size([1, 1000, 754]) | 7\n",
-      "18846    | torch.Size([1, 1000, 777]) | 7\n",
-      "18847    | torch.Size([1, 1000, 763]) | 7\n",
-      "18848    | torch.Size([1, 1000, 781]) | 7\n",
-      "18849    | torch.Size([1, 1000, 754]) | 7\n",
-      "18850    | torch.Size([1, 1000, 754]) | 7\n",
-      "18851    | torch.Size([1, 1000, 762]) | 7\n",
-      "18852    | torch.Size([1, 1000, 787]) | 7\n",
-      "18853    | torch.Size([1, 1000, 742]) | 7\n",
-      "18854    | torch.Size([1, 1000, 777]) | 7\n",
-      "18855    | torch.Size([1, 1000, 754]) | 7\n",
-      "18856    | torch.Size([1, 1000, 771]) | 7\n",
-      "18857    | torch.Size([1, 1000, 777]) | 7\n",
-      "18858    | torch.Size([1, 1000, 754]) | 7\n",
-      "18859    | torch.Size([1, 1000, 786]) | 7\n",
-      "18860    | torch.Size([1, 1000, 767]) | 7\n",
-      "18861    | torch.Size([1, 1000, 754]) | 7\n",
-      "18862    | torch.Size([1, 1000, 754]) | 7\n",
-      "18863    | torch.Size([1, 1000, 787]) | 7\n",
-      "18864    | torch.Size([1, 1000, 808]) | 7\n",
-      "18865    | torch.Size([1, 1000, 754]) | 7\n",
-      "18866    | torch.Size([1, 1000, 762]) | 7\n",
-      "18867    | torch.Size([1, 1000, 762]) | 7\n",
-      "18868    | torch.Size([1, 1000, 771]) | 7\n",
-      "18869    | torch.Size([1, 1000, 762]) | 7\n",
-      "18870    | torch.Size([1, 1000, 772]) | 7\n",
-      "18871    | torch.Size([1, 1000, 821]) | 7\n",
-      "18872    | torch.Size([1, 1000, 754]) | 7\n",
-      "18873    | torch.Size([1, 1000, 778]) | 7\n",
-      "18874    | torch.Size([1, 1000, 780]) | 7\n",
-      "18875    | torch.Size([1, 1000, 771]) | 7\n",
-      "18876    | torch.Size([1, 1000, 762]) | 7\n",
-      "18877    | torch.Size([1, 1000, 771]) | 7\n",
-      "18878    | torch.Size([1, 1000, 762]) | 7\n",
-      "18879    | torch.Size([1, 1000, 771]) | 7\n",
-      "18880    | torch.Size([1, 1000, 777]) | 7\n",
-      "18881    | torch.Size([1, 1000, 773]) | 7\n",
-      "18882    | torch.Size([1, 1000, 754]) | 7\n",
-      "18883    | torch.Size([1, 1000, 779]) | 7\n",
-      "18884    | torch.Size([1, 1000, 813]) | 7\n",
-      "18885    | torch.Size([1, 1000, 792]) | 7\n",
-      "18886    | torch.Size([1, 1000, 754]) | 7\n",
-      "18887    | torch.Size([1, 1000, 762]) | 7\n",
-      "18888    | torch.Size([1, 1000, 778]) | 7\n",
-      "18889    | torch.Size([1, 1000, 771]) | 7\n",
-      "18890    | torch.Size([1, 1000, 754]) | 7\n",
-      "18891    | torch.Size([1, 1000, 762]) | 7\n",
-      "18892    | torch.Size([1, 1000, 772]) | 7\n",
-      "18893    | torch.Size([1, 1000, 792]) | 7\n",
-      "18894    | torch.Size([1, 1000, 794]) | 7\n",
-      "18895    | torch.Size([1, 1000, 762]) | 7\n",
-      "18896    | torch.Size([1, 1000, 776]) | 7\n",
-      "18897    | torch.Size([1, 1000, 754]) | 7\n",
-      "18898    | torch.Size([1, 1000, 784]) | 7\n",
-      "18899    | torch.Size([1, 1000, 754]) | 7\n",
-      "18900    | torch.Size([1, 1000, 754]) | 7\n",
-      "18901    | torch.Size([1, 1000, 786]) | 7\n",
-      "18902    | torch.Size([1, 1000, 770]) | 7\n",
-      "18903    | torch.Size([1, 1000, 754]) | 7\n",
-      "18904    | torch.Size([1, 1000, 771]) | 7\n",
-      "18905    | torch.Size([1, 1000, 780]) | 7\n",
-      "18906    | torch.Size([1, 1000, 785]) | 7\n",
-      "18907    | torch.Size([1, 1000, 768]) | 7\n",
-      "18908    | torch.Size([1, 1000, 771]) | 7\n",
-      "18909    | torch.Size([1, 1000, 777]) | 7\n",
-      "18910    | torch.Size([1, 1000, 777]) | 7\n",
-      "18911    | torch.Size([1, 1000, 754]) | 7\n",
-      "18912    | torch.Size([1, 1000, 754]) | 7\n",
-      "18913    | torch.Size([1, 1000, 754]) | 7\n",
-      "18914    | torch.Size([1, 1000, 762]) | 7\n",
-      "18915    | torch.Size([1, 1000, 786]) | 7\n",
-      "18916    | torch.Size([1, 1000, 777]) | 7\n",
-      "18917    | torch.Size([1, 1000, 762]) | 7\n",
-      "18918    | torch.Size([1, 1000, 762]) | 7\n",
-      "18919    | torch.Size([1, 1000, 802]) | 7\n",
-      "18920    | torch.Size([1, 1000, 754]) | 7\n",
-      "18921    | torch.Size([1, 1000, 771]) | 7\n",
-      "18922    | torch.Size([1, 1000, 754]) | 7\n",
-      "18923    | torch.Size([1, 1000, 769]) | 7\n",
-      "18924    | torch.Size([1, 1000, 762]) | 7\n",
-      "18925    | torch.Size([1, 1000, 754]) | 7\n",
-      "18926    | torch.Size([1, 1000, 754]) | 7\n",
-      "18927    | torch.Size([1, 1000, 788]) | 7\n",
-      "18928    | torch.Size([1, 1000, 754]) | 7\n",
-      "18929    | torch.Size([1, 1000, 777]) | 7\n",
-      "18930    | torch.Size([1, 1000, 787]) | 7\n",
-      "18931    | torch.Size([1, 1000, 754]) | 7\n",
-      "18932    | torch.Size([1, 1000, 775]) | 7\n",
-      "18933    | torch.Size([1, 1000, 762]) | 7\n",
-      "18934    | torch.Size([1, 1000, 754]) | 7\n",
-      "18935    | torch.Size([1, 1000, 772]) | 7\n",
-      "18936    | torch.Size([1, 1000, 754]) | 7\n",
-      "18937    | torch.Size([1, 1000, 773]) | 7\n",
-      "18938    | torch.Size([1, 1000, 754]) | 7\n",
-      "18939    | torch.Size([1, 1000, 772]) | 7\n",
-      "18940    | torch.Size([1, 1000, 754]) | 7\n",
-      "18941    | torch.Size([1, 1000, 754]) | 7\n",
-      "18942    | torch.Size([1, 1000, 754]) | 7\n",
-      "18943    | torch.Size([1, 1000, 807]) | 7\n",
-      "18944    | torch.Size([1, 1000, 780]) | 7\n",
-      "18945    | torch.Size([1, 1000, 762]) | 7\n",
-      "18946    | torch.Size([1, 1000, 754]) | 7\n",
-      "18947    | torch.Size([1, 1000, 762]) | 7\n",
-      "18948    | torch.Size([1, 1000, 754]) | 7\n",
-      "18949    | torch.Size([1, 1000, 779]) | 7\n",
-      "18950    | torch.Size([1, 1000, 762]) | 7\n",
-      "18951    | torch.Size([1, 1000, 777]) | 7\n",
-      "18952    | torch.Size([1, 1000, 778]) | 7\n",
-      "18953    | torch.Size([1, 1000, 767]) | 7\n",
-      "18954    | torch.Size([1, 1000, 792]) | 7\n",
-      "18955    | torch.Size([1, 1000, 800]) | 7\n",
-      "18956    | torch.Size([1, 1000, 794]) | 7\n",
-      "18957    | torch.Size([1, 1000, 783]) | 7\n",
-      "18958    | torch.Size([1, 1000, 754]) | 7\n",
-      "18959    | torch.Size([1, 1000, 754]) | 7\n",
-      "18960    | torch.Size([1, 1000, 762]) | 7\n",
-      "18961    | torch.Size([1, 1000, 754]) | 7\n",
-      "18962    | torch.Size([1, 1000, 770]) | 7\n",
-      "18963    | torch.Size([1, 1000, 786]) | 7\n",
-      "18964    | torch.Size([1, 1000, 754]) | 7\n",
-      "18965    | torch.Size([1, 1000, 784]) | 7\n",
-      "18966    | torch.Size([1, 1000, 805]) | 7\n",
-      "18967    | torch.Size([1, 1000, 790]) | 7\n",
-      "18968    | torch.Size([1, 1000, 771]) | 7\n",
-      "18969    | torch.Size([1, 1000, 754]) | 7\n",
-      "18970    | torch.Size([1, 1000, 766]) | 7\n",
-      "18971    | torch.Size([1, 1000, 785]) | 7\n",
-      "18972    | torch.Size([1, 1000, 777]) | 7\n",
-      "18973    | torch.Size([1, 1000, 754]) | 7\n",
-      "18974    | torch.Size([1, 1000, 754]) | 7\n",
-      "18975    | torch.Size([1, 1000, 762]) | 7\n",
-      "18976    | torch.Size([1, 1000, 802]) | 7\n",
-      "18977    | torch.Size([1, 1000, 777]) | 7\n",
-      "18978    | torch.Size([1, 1000, 788]) | 7\n",
-      "18979    | torch.Size([1, 1000, 759]) | 7\n",
-      "18980    | torch.Size([1, 1000, 754]) | 7\n",
-      "18981    | torch.Size([1, 1000, 762]) | 7\n",
-      "18982    | torch.Size([1, 1000, 754]) | 7\n",
-      "18983    | torch.Size([1, 1000, 771]) | 7\n",
-      "18984    | torch.Size([1, 1000, 762]) | 7\n",
-      "18985    | torch.Size([1, 1000, 774]) | 7\n",
-      "18986    | torch.Size([1, 1000, 777]) | 7\n",
-      "18987    | torch.Size([1, 1000, 754]) | 7\n",
-      "18988    | torch.Size([1, 1000, 754]) | 7\n",
-      "18989    | torch.Size([1, 1000, 762]) | 7\n",
-      "18990    | torch.Size([1, 1000, 773]) | 7\n",
-      "18991    | torch.Size([1, 1000, 781]) | 7\n",
-      "18992    | torch.Size([1, 1000, 783]) | 7\n",
-      "18993    | torch.Size([1, 1000, 754]) | 7\n",
-      "18994    | torch.Size([1, 1000, 771]) | 7\n",
-      "18995    | torch.Size([1, 1000, 779]) | 7\n",
-      "18996    | torch.Size([1, 1000, 771]) | 7\n",
-      "18997    | torch.Size([1, 1000, 802]) | 7\n",
-      "18998    | torch.Size([1, 1000, 777]) | 7\n",
-      "18999    | torch.Size([1, 1000, 762]) | 7\n",
-      "19000    | torch.Size([1, 1000, 785]) | 7\n",
-      "19001    | torch.Size([1, 1000, 818]) | 7\n",
-      "19002    | torch.Size([1, 1000, 754]) | 7\n",
-      "19003    | torch.Size([1, 1000, 777]) | 7\n",
-      "19004    | torch.Size([1, 1000, 754]) | 7\n",
-      "19005    | torch.Size([1, 1000, 754]) | 7\n",
-      "19006    | torch.Size([1, 1000, 754]) | 7\n",
-      "19007    | torch.Size([1, 1000, 777]) | 7\n",
-      "19008    | torch.Size([1, 1000, 754]) | 7\n",
-      "19009    | torch.Size([1, 1000, 789]) | 7\n",
-      "19010    | torch.Size([1, 1000, 784]) | 7\n",
-      "19011    | torch.Size([1, 1000, 772]) | 7\n",
-      "19012    | torch.Size([1, 1000, 754]) | 7\n",
-      "19013    | torch.Size([1, 1000, 777]) | 7\n",
-      "19014    | torch.Size([1, 1000, 802]) | 7\n",
-      "19015    | torch.Size([1, 1000, 762]) | 7\n",
-      "19016    | torch.Size([1, 1000, 774]) | 7\n",
-      "19017    | torch.Size([1, 1000, 754]) | 7\n",
-      "19018    | torch.Size([1, 1000, 782]) | 7\n",
-      "19019    | torch.Size([1, 1000, 779]) | 7\n",
-      "19020    | torch.Size([1, 1000, 787]) | 7\n",
-      "19021    | torch.Size([1, 1000, 771]) | 7\n",
-      "19022    | torch.Size([1, 1000, 754]) | 7\n",
-      "19023    | torch.Size([1, 1000, 778]) | 7\n",
-      "19024    | torch.Size([1, 1000, 776]) | 7\n",
-      "19025    | torch.Size([1, 1000, 754]) | 7\n",
-      "19026    | torch.Size([1, 1000, 754]) | 7\n",
-      "19027    | torch.Size([1, 1000, 754]) | 7\n",
-      "19028    | torch.Size([1, 1000, 767]) | 7\n",
-      "19029    | torch.Size([1, 1000, 776]) | 7\n",
-      "19030    | torch.Size([1, 1000, 754]) | 7\n",
-      "19031    | torch.Size([1, 1000, 777]) | 7\n",
-      "19032    | torch.Size([1, 1000, 787]) | 7\n",
-      "19033    | torch.Size([1, 1000, 754]) | 7\n",
-      "19034    | torch.Size([1, 1000, 777]) | 7\n",
-      "19035    | torch.Size([1, 1000, 754]) | 7\n",
-      "19036    | torch.Size([1, 1000, 754]) | 7\n",
-      "19037    | torch.Size([1, 1000, 754]) | 7\n",
-      "19038    | torch.Size([1, 1000, 754]) | 7\n",
-      "19039    | torch.Size([1, 1000, 777]) | 7\n",
-      "19040    | torch.Size([1, 1000, 771]) | 7\n",
-      "19041    | torch.Size([1, 1000, 771]) | 7\n",
-      "19042    | torch.Size([1, 1000, 754]) | 7\n",
-      "19043    | torch.Size([1, 1000, 754]) | 7\n",
-      "19044    | torch.Size([1, 1000, 754]) | 7\n",
-      "19045    | torch.Size([1, 1000, 771]) | 7\n",
-      "19046    | torch.Size([1, 1000, 754]) | 7\n",
-      "19047    | torch.Size([1, 1000, 761]) | 7\n",
-      "19048    | torch.Size([1, 1000, 754]) | 7\n",
-      "19049    | torch.Size([1, 1000, 777]) | 7\n",
-      "19050    | torch.Size([1, 1000, 754]) | 7\n",
-      "19051    | torch.Size([1, 1000, 802]) | 7\n",
-      "19052    | torch.Size([1, 1000, 774]) | 7\n",
-      "19053    | torch.Size([1, 1000, 754]) | 7\n",
-      "19054    | torch.Size([1, 1000, 754]) | 7\n",
-      "19055    | torch.Size([1, 1000, 754]) | 7\n",
-      "19056    | torch.Size([1, 1000, 778]) | 7\n",
-      "19057    | torch.Size([1, 1000, 754]) | 7\n",
-      "19058    | torch.Size([1, 1000, 780]) | 7\n",
-      "19059    | torch.Size([1, 1000, 785]) | 7\n",
-      "19060    | torch.Size([1, 1000, 754]) | 7\n",
-      "19061    | torch.Size([1, 1000, 777]) | 7\n",
-      "19062    | torch.Size([1, 1000, 782]) | 7\n",
-      "19063    | torch.Size([1, 1000, 809]) | 7\n",
-      "19064    | torch.Size([1, 1000, 754]) | 7\n",
-      "19065    | torch.Size([1, 1000, 752]) | 7\n",
-      "19066    | torch.Size([1, 1000, 778]) | 7\n",
-      "19067    | torch.Size([1, 1000, 785]) | 7\n",
-      "19068    | torch.Size([1, 1000, 773]) | 7\n",
-      "19069    | torch.Size([1, 1000, 778]) | 7\n",
-      "19070    | torch.Size([1, 1000, 782]) | 7\n",
-      "19071    | torch.Size([1, 1000, 762]) | 7\n",
-      "19072    | torch.Size([1, 1000, 754]) | 7\n",
-      "19073    | torch.Size([1, 1000, 777]) | 7\n",
-      "19074    | torch.Size([1, 1000, 762]) | 7\n",
-      "19075    | torch.Size([1, 1000, 754]) | 7\n",
-      "19076    | torch.Size([1, 1000, 777]) | 7\n",
-      "19077    | torch.Size([1, 1000, 754]) | 7\n",
-      "19078    | torch.Size([1, 1000, 781]) | 7\n",
-      "19079    | torch.Size([1, 1000, 779]) | 7\n",
-      "19080    | torch.Size([1, 1000, 789]) | 7\n",
-      "19081    | torch.Size([1, 1000, 777]) | 7\n",
-      "19082    | torch.Size([1, 1000, 777]) | 7\n",
-      "19083    | torch.Size([1, 1000, 776]) | 7\n",
-      "19084    | torch.Size([1, 1000, 754]) | 7\n",
-      "19085    | torch.Size([1, 1000, 754]) | 7\n",
-      "19086    | torch.Size([1, 1000, 777]) | 7\n",
-      "19087    | torch.Size([1, 1000, 762]) | 7\n",
-      "19088    | torch.Size([1, 1000, 777]) | 7\n",
-      "19089    | torch.Size([1, 1000, 740]) | 7\n",
-      "19090    | torch.Size([1, 1000, 754]) | 7\n",
-      "19091    | torch.Size([1, 1000, 754]) | 7\n",
-      "19092    | torch.Size([1, 1000, 778]) | 7\n",
-      "19093    | torch.Size([1, 1000, 782]) | 7\n",
-      "19094    | torch.Size([1, 1000, 754]) | 7\n",
-      "19095    | torch.Size([1, 1000, 780]) | 7\n",
-      "19096    | torch.Size([1, 1000, 771]) | 7\n",
-      "19097    | torch.Size([1, 1000, 754]) | 7\n",
-      "19098    | torch.Size([1, 1000, 754]) | 7\n",
-      "19099    | torch.Size([1, 1000, 778]) | 7\n",
-      "19100    | torch.Size([1, 1000, 754]) | 7\n",
-      "19101    | torch.Size([1, 1000, 771]) | 7\n",
-      "19102    | torch.Size([1, 1000, 777]) | 7\n",
-      "19103    | torch.Size([1, 1000, 762]) | 7\n",
-      "19104    | torch.Size([1, 1000, 740]) | 7\n",
-      "19105    | torch.Size([1, 1000, 762]) | 7\n",
-      "19106    | torch.Size([1, 1000, 754]) | 7\n",
-      "19107    | torch.Size([1, 1000, 804]) | 7\n",
-      "19108    | torch.Size([1, 1000, 754]) | 7\n",
-      "19109    | torch.Size([1, 1000, 792]) | 7\n",
-      "19110    | torch.Size([1, 1000, 754]) | 7\n",
-      "19111    | torch.Size([1, 1000, 770]) | 7\n",
-      "19112    | torch.Size([1, 1000, 754]) | 7\n",
-      "19113    | torch.Size([1, 1000, 803]) | 7\n",
-      "19114    | torch.Size([1, 1000, 754]) | 7\n",
-      "19115    | torch.Size([1, 1000, 777]) | 7\n",
-      "19116    | torch.Size([1, 1000, 754]) | 7\n",
-      "19117    | torch.Size([1, 1000, 777]) | 7\n",
-      "19118    | torch.Size([1, 1000, 762]) | 7\n",
-      "19119    | torch.Size([1, 1000, 777]) | 7\n",
-      "19120    | torch.Size([1, 1000, 805]) | 7\n",
-      "19121    | torch.Size([1, 1000, 773]) | 7\n",
-      "19122    | torch.Size([1, 1000, 754]) | 7\n",
-      "19123    | torch.Size([1, 1000, 762]) | 7\n",
-      "19124    | torch.Size([1, 1000, 771]) | 7\n",
-      "19125    | torch.Size([1, 1000, 771]) | 7\n",
-      "19126    | torch.Size([1, 1000, 777]) | 7\n",
-      "19127    | torch.Size([1, 1000, 782]) | 7\n",
-      "19128    | torch.Size([1, 1000, 807]) | 7\n",
-      "19129    | torch.Size([1, 1000, 754]) | 7\n",
-      "19130    | torch.Size([1, 1000, 802]) | 7\n",
-      "19131    | torch.Size([1, 1000, 771]) | 7\n",
-      "19132    | torch.Size([1, 1000, 776]) | 7\n",
-      "19133    | torch.Size([1, 1000, 784]) | 7\n",
-      "19134    | torch.Size([1, 1000, 754]) | 7\n",
-      "19135    | torch.Size([1, 1000, 777]) | 7\n",
-      "19136    | torch.Size([1, 1000, 762]) | 7\n",
-      "19137    | torch.Size([1, 1000, 804]) | 7\n",
-      "19138    | torch.Size([1, 1000, 771]) | 7\n",
-      "19139    | torch.Size([1, 1000, 775]) | 7\n",
-      "19140    | torch.Size([1, 1000, 777]) | 7\n",
-      "19141    | torch.Size([1, 1000, 754]) | 7\n",
-      "19142    | torch.Size([1, 1000, 782]) | 7\n",
-      "19143    | torch.Size([1, 1000, 771]) | 7\n",
-      "19144    | torch.Size([1, 1000, 777]) | 7\n",
-      "19145    | torch.Size([1, 1000, 785]) | 7\n",
-      "19146    | torch.Size([1, 1000, 772]) | 7\n",
-      "19147    | torch.Size([1, 1000, 771]) | 7\n",
-      "19148    | torch.Size([1, 1000, 762]) | 7\n",
-      "19149    | torch.Size([1, 1000, 754]) | 7\n",
-      "19150    | torch.Size([1, 1000, 785]) | 7\n",
-      "19151    | torch.Size([1, 1000, 754]) | 7\n",
-      "19152    | torch.Size([1, 1000, 767]) | 7\n",
-      "19153    | torch.Size([1, 1000, 754]) | 7\n",
-      "19154    | torch.Size([1, 1000, 754]) | 7\n",
-      "19155    | torch.Size([1, 1000, 756]) | 7\n",
-      "19156    | torch.Size([1, 1000, 801]) | 7\n",
-      "19157    | torch.Size([1, 1000, 754]) | 7\n",
-      "19158    | torch.Size([1, 1000, 754]) | 7\n",
-      "19159    | torch.Size([1, 1000, 754]) | 7\n",
-      "19160    | torch.Size([1, 1000, 776]) | 7\n",
-      "19161    | torch.Size([1, 1000, 802]) | 7\n",
-      "19162    | torch.Size([1, 1000, 762]) | 7\n",
-      "19163    | torch.Size([1, 1000, 754]) | 7\n",
-      "19164    | torch.Size([1, 1000, 771]) | 7\n",
-      "19165    | torch.Size([1, 1000, 775]) | 7\n",
-      "19166    | torch.Size([1, 1000, 778]) | 7\n",
-      "19167    | torch.Size([1, 1000, 777]) | 7\n",
-      "19168    | torch.Size([1, 1000, 762]) | 7\n",
-      "19169    | torch.Size([1, 1000, 754]) | 7\n",
-      "19170    | torch.Size([1, 1000, 754]) | 7\n",
-      "19171    | torch.Size([1, 1000, 783]) | 7\n",
-      "19172    | torch.Size([1, 1000, 762]) | 7\n",
-      "19173    | torch.Size([1, 1000, 829]) | 7\n",
-      "19174    | torch.Size([1, 1000, 754]) | 7\n",
-      "19175    | torch.Size([1, 1000, 754]) | 7\n",
-      "19176    | torch.Size([1, 1000, 785]) | 7\n",
-      "19177    | torch.Size([1, 1000, 754]) | 7\n",
-      "19178    | torch.Size([1, 1000, 772]) | 7\n",
-      "19179    | torch.Size([1, 1000, 788]) | 7\n",
-      "19180    | torch.Size([1, 1000, 797]) | 7\n",
-      "19181    | torch.Size([1, 1000, 754]) | 7\n",
-      "19182    | torch.Size([1, 1000, 754]) | 7\n",
-      "19183    | torch.Size([1, 1000, 771]) | 7\n",
-      "19184    | torch.Size([1, 1000, 799]) | 7\n",
-      "19185    | torch.Size([1, 1000, 754]) | 7\n",
-      "19186    | torch.Size([1, 1000, 762]) | 7\n",
-      "19187    | torch.Size([1, 1000, 754]) | 7\n",
-      "19188    | torch.Size([1, 1000, 806]) | 7\n",
-      "19189    | torch.Size([1, 1000, 762]) | 7\n",
-      "19190    | torch.Size([1, 1000, 754]) | 7\n",
-      "19191    | torch.Size([1, 1000, 789]) | 7\n",
-      "19192    | torch.Size([1, 1000, 762]) | 7\n",
-      "19193    | torch.Size([1, 1000, 796]) | 7\n",
-      "19194    | torch.Size([1, 1000, 754]) | 7\n",
-      "19195    | torch.Size([1, 1000, 762]) | 7\n",
-      "19196    | torch.Size([1, 1000, 777]) | 7\n",
-      "19197    | torch.Size([1, 1000, 781]) | 7\n",
-      "19198    | torch.Size([1, 1000, 773]) | 7\n",
-      "19199    | torch.Size([1, 1000, 754]) | 7\n",
-      "19200    | torch.Size([1, 1000, 778]) | 7\n",
-      "19201    | torch.Size([1, 1000, 777]) | 7\n",
-      "19202    | torch.Size([1, 1000, 754]) | 7\n",
-      "19203    | torch.Size([1, 1000, 795]) | 7\n",
-      "19204    | torch.Size([1, 1000, 754]) | 7\n",
-      "19205    | torch.Size([1, 1000, 715]) | 7\n",
-      "19206    | torch.Size([1, 1000, 766]) | 7\n",
-      "19207    | torch.Size([1, 1000, 754]) | 7\n",
-      "19208    | torch.Size([1, 1000, 762]) | 7\n",
-      "19209    | torch.Size([1, 1000, 792]) | 7\n",
-      "19210    | torch.Size([1, 1000, 788]) | 7\n",
-      "19211    | torch.Size([1, 1000, 777]) | 7\n",
-      "19212    | torch.Size([1, 1000, 777]) | 7\n",
-      "19213    | torch.Size([1, 1000, 777]) | 7\n",
-      "19214    | torch.Size([1, 1000, 785]) | 7\n",
-      "19215    | torch.Size([1, 1000, 776]) | 7\n",
-      "19216    | torch.Size([1, 1000, 791]) | 7\n",
-      "19217    | torch.Size([1, 1000, 787]) | 7\n",
-      "19218    | torch.Size([1, 1000, 754]) | 7\n",
-      "19219    | torch.Size([1, 1000, 754]) | 7\n",
-      "19220    | torch.Size([1, 1000, 754]) | 7\n",
-      "19221    | torch.Size([1, 1000, 773]) | 7\n",
-      "19222    | torch.Size([1, 1000, 779]) | 7\n",
-      "19223    | torch.Size([1, 1000, 769]) | 7\n",
-      "19224    | torch.Size([1, 1000, 754]) | 7\n",
-      "19225    | torch.Size([1, 1000, 762]) | 7\n",
-      "19226    | torch.Size([1, 1000, 771]) | 7\n",
-      "19227    | torch.Size([1, 1000, 729]) | 7\n",
-      "19228    | torch.Size([1, 1000, 754]) | 7\n",
-      "19229    | torch.Size([1, 1000, 771]) | 7\n",
-      "19230    | torch.Size([1, 1000, 754]) | 7\n",
-      "19231    | torch.Size([1, 1000, 792]) | 7\n",
-      "19232    | torch.Size([1, 1000, 777]) | 7\n",
-      "19233    | torch.Size([1, 1000, 754]) | 7\n",
-      "19234    | torch.Size([1, 1000, 754]) | 7\n",
-      "19235    | torch.Size([1, 1000, 754]) | 7\n",
-      "19236    | torch.Size([1, 1000, 784]) | 7\n",
-      "19237    | torch.Size([1, 1000, 792]) | 7\n",
-      "19238    | torch.Size([1, 1000, 771]) | 7\n",
-      "19239    | torch.Size([1, 1000, 762]) | 7\n",
-      "19240    | torch.Size([1, 1000, 756]) | 7\n",
-      "19241    | torch.Size([1, 1000, 754]) | 7\n",
-      "19242    | torch.Size([1, 1000, 787]) | 7\n",
-      "19243    | torch.Size([1, 1000, 754]) | 7\n",
-      "19244    | torch.Size([1, 1000, 811]) | 7\n",
-      "19245    | torch.Size([1, 1000, 754]) | 7\n",
-      "19246    | torch.Size([1, 1000, 762]) | 7\n",
-      "19247    | torch.Size([1, 1000, 754]) | 7\n",
-      "19248    | torch.Size([1, 1000, 762]) | 7\n",
-      "19249    | torch.Size([1, 1000, 762]) | 7\n",
-      "19250    | torch.Size([1, 1000, 754]) | 7\n",
-      "19251    | torch.Size([1, 1000, 807]) | 7\n",
-      "19252    | torch.Size([1, 1000, 762]) | 7\n",
-      "19253    | torch.Size([1, 1000, 784]) | 7\n",
-      "19254    | torch.Size([1, 1000, 754]) | 7\n",
-      "19255    | torch.Size([1, 1000, 780]) | 7\n",
-      "19256    | torch.Size([1, 1000, 787]) | 7\n",
-      "19257    | torch.Size([1, 1000, 754]) | 7\n",
-      "19258    | torch.Size([1, 1000, 777]) | 7\n",
-      "19259    | torch.Size([1, 1000, 754]) | 7\n",
-      "19260    | torch.Size([1, 1000, 807]) | 7\n",
-      "19261    | torch.Size([1, 1000, 782]) | 7\n",
-      "19262    | torch.Size([1, 1000, 762]) | 7\n",
-      "19263    | torch.Size([1, 1000, 772]) | 7\n",
-      "19264    | torch.Size([1, 1000, 762]) | 7\n",
-      "19265    | torch.Size([1, 1000, 754]) | 7\n",
-      "19266    | torch.Size([1, 1000, 754]) | 7\n",
-      "19267    | torch.Size([1, 1000, 789]) | 7\n",
-      "19268    | torch.Size([1, 1000, 754]) | 7\n",
-      "19269    | torch.Size([1, 1000, 754]) | 7\n",
-      "19270    | torch.Size([1, 1000, 772]) | 7\n",
-      "19271    | torch.Size([1, 1000, 775]) | 7\n",
-      "19272    | torch.Size([1, 1000, 773]) | 7\n",
-      "19273    | torch.Size([1, 1000, 792]) | 7\n",
-      "19274    | torch.Size([1, 1000, 754]) | 7\n",
-      "19275    | torch.Size([1, 1000, 778]) | 7\n",
-      "19276    | torch.Size([1, 1000, 807]) | 7\n",
-      "19277    | torch.Size([1, 1000, 781]) | 7\n",
-      "19278    | torch.Size([1, 1000, 754]) | 7\n",
-      "19279    | torch.Size([1, 1000, 762]) | 7\n",
-      "19280    | torch.Size([1, 1000, 802]) | 7\n",
-      "19281    | torch.Size([1, 1000, 754]) | 7\n",
-      "19282    | torch.Size([1, 1000, 754]) | 7\n",
-      "19283    | torch.Size([1, 1000, 754]) | 7\n",
-      "19284    | torch.Size([1, 1000, 754]) | 7\n",
-      "19285    | torch.Size([1, 1000, 754]) | 7\n",
-      "19286    | torch.Size([1, 1000, 754]) | 7\n",
-      "19287    | torch.Size([1, 1000, 754]) | 7\n",
-      "19288    | torch.Size([1, 1000, 762]) | 7\n",
-      "19289    | torch.Size([1, 1000, 754]) | 7\n",
-      "19290    | torch.Size([1, 1000, 779]) | 7\n",
-      "19291    | torch.Size([1, 1000, 784]) | 7\n",
-      "19292    | torch.Size([1, 1000, 754]) | 7\n",
-      "19293    | torch.Size([1, 1000, 762]) | 7\n",
-      "19294    | torch.Size([1, 1000, 754]) | 7\n",
-      "19295    | torch.Size([1, 1000, 789]) | 7\n",
-      "19296    | torch.Size([1, 1000, 784]) | 7\n",
-      "19297    | torch.Size([1, 1000, 792]) | 7\n",
-      "19298    | torch.Size([1, 1000, 777]) | 7\n",
-      "19299    | torch.Size([1, 1000, 778]) | 7\n",
-      "19300    | torch.Size([1, 1000, 762]) | 7\n",
-      "19301    | torch.Size([1, 1000, 754]) | 7\n",
-      "19302    | torch.Size([1, 1000, 762]) | 7\n",
-      "19303    | torch.Size([1, 1000, 780]) | 7\n",
-      "19304    | torch.Size([1, 1000, 754]) | 7\n",
-      "19305    | torch.Size([1, 1000, 770]) | 7\n",
-      "19306    | torch.Size([1, 1000, 772]) | 7\n",
-      "19307    | torch.Size([1, 1000, 787]) | 7\n",
-      "19308    | torch.Size([1, 1000, 783]) | 7\n",
-      "19309    | torch.Size([1, 1000, 754]) | 7\n",
-      "19310    | torch.Size([1, 1000, 762]) | 7\n",
-      "19311    | torch.Size([1, 1000, 754]) | 7\n",
-      "19312    | torch.Size([1, 1000, 790]) | 7\n",
-      "19313    | torch.Size([1, 1000, 762]) | 7\n",
-      "19314    | torch.Size([1, 1000, 747]) | 7\n",
-      "19315    | torch.Size([1, 1000, 779]) | 7\n",
-      "19316    | torch.Size([1, 1000, 777]) | 7\n",
-      "19317    | torch.Size([1, 1000, 771]) | 7\n",
-      "19318    | torch.Size([1, 1000, 754]) | 7\n",
-      "19319    | torch.Size([1, 1000, 754]) | 7\n",
-      "19320    | torch.Size([1, 1000, 754]) | 7\n",
-      "19321    | torch.Size([1, 1000, 777]) | 7\n",
-      "19322    | torch.Size([1, 1000, 754]) | 7\n",
-      "19323    | torch.Size([1, 1000, 784]) | 7\n",
-      "19324    | torch.Size([1, 1000, 754]) | 7\n",
-      "19325    | torch.Size([1, 1000, 788]) | 7\n",
-      "19326    | torch.Size([1, 1000, 771]) | 7\n",
-      "19327    | torch.Size([1, 1000, 777]) | 7\n",
-      "19328    | torch.Size([1, 1000, 771]) | 7\n",
-      "19329    | torch.Size([1, 1000, 774]) | 7\n",
-      "19330    | torch.Size([1, 1000, 777]) | 7\n",
-      "19331    | torch.Size([1, 1000, 802]) | 7\n",
-      "19332    | torch.Size([1, 1000, 754]) | 7\n",
-      "19333    | torch.Size([1, 1000, 776]) | 7\n",
-      "19334    | torch.Size([1, 1000, 796]) | 7\n",
-      "19335    | torch.Size([1, 1000, 771]) | 7\n",
-      "19336    | torch.Size([1, 1000, 754]) | 7\n",
-      "19337    | torch.Size([1, 1000, 754]) | 7\n",
-      "19338    | torch.Size([1, 1000, 770]) | 7\n",
-      "19339    | torch.Size([1, 1000, 774]) | 7\n",
-      "19340    | torch.Size([1, 1000, 783]) | 7\n",
-      "19341    | torch.Size([1, 1000, 806]) | 7\n",
-      "19342    | torch.Size([1, 1000, 778]) | 7\n",
-      "19343    | torch.Size([1, 1000, 779]) | 7\n",
-      "19344    | torch.Size([1, 1000, 798]) | 7\n",
-      "19345    | torch.Size([1, 1000, 754]) | 7\n",
-      "19346    | torch.Size([1, 1000, 769]) | 7\n",
-      "19347    | torch.Size([1, 1000, 754]) | 7\n",
-      "19348    | torch.Size([1, 1000, 762]) | 7\n",
-      "19349    | torch.Size([1, 1000, 786]) | 7\n",
-      "19350    | torch.Size([1, 1000, 771]) | 7\n",
-      "19351    | torch.Size([1, 1000, 787]) | 7\n",
-      "19352    | torch.Size([1, 1000, 754]) | 7\n",
-      "19353    | torch.Size([1, 1000, 780]) | 7\n",
-      "19354    | torch.Size([1, 1000, 792]) | 7\n",
-      "19355    | torch.Size([1, 1000, 767]) | 7\n",
-      "19356    | torch.Size([1, 1000, 772]) | 7\n",
-      "19357    | torch.Size([1, 1000, 784]) | 7\n",
-      "19358    | torch.Size([1, 1000, 771]) | 7\n",
-      "19359    | torch.Size([1, 1000, 777]) | 7\n",
-      "19360    | torch.Size([1, 1000, 786]) | 7\n",
-      "19361    | torch.Size([1, 1000, 754]) | 7\n",
-      "19362    | torch.Size([1, 1000, 781]) | 7\n",
-      "19363    | torch.Size([1, 1000, 754]) | 7\n",
-      "19364    | torch.Size([1, 1000, 754]) | 7\n",
-      "19365    | torch.Size([1, 1000, 765]) | 7\n",
-      "19366    | torch.Size([1, 1000, 773]) | 7\n",
-      "19367    | torch.Size([1, 1000, 754]) | 7\n",
-      "19368    | torch.Size([1, 1000, 777]) | 7\n",
-      "19369    | torch.Size([1, 1000, 754]) | 7\n",
-      "19370    | torch.Size([1, 1000, 754]) | 7\n",
-      "19371    | torch.Size([1, 1000, 762]) | 7\n",
-      "19372    | torch.Size([1, 1000, 754]) | 7\n",
-      "19373    | torch.Size([1, 1000, 762]) | 7\n",
-      "19374    | torch.Size([1, 1000, 780]) | 7\n",
-      "19375    | torch.Size([1, 1000, 771]) | 7\n",
-      "19376    | torch.Size([1, 1000, 768]) | 7\n",
-      "19377    | torch.Size([1, 1000, 754]) | 7\n",
-      "19378    | torch.Size([1, 1000, 741]) | 7\n",
-      "19379    | torch.Size([1, 1000, 754]) | 7\n",
-      "19380    | torch.Size([1, 1000, 754]) | 7\n",
-      "19381    | torch.Size([1, 1000, 754]) | 7\n",
-      "19382    | torch.Size([1, 1000, 754]) | 7\n",
-      "19383    | torch.Size([1, 1000, 772]) | 7\n",
-      "19384    | torch.Size([1, 1000, 762]) | 7\n",
-      "19385    | torch.Size([1, 1000, 777]) | 7\n",
-      "19386    | torch.Size([1, 1000, 754]) | 7\n",
-      "19387    | torch.Size([1, 1000, 776]) | 7\n",
-      "19388    | torch.Size([1, 1000, 776]) | 7\n",
-      "19389    | torch.Size([1, 1000, 774]) | 7\n",
-      "19390    | torch.Size([1, 1000, 754]) | 7\n",
-      "19391    | torch.Size([1, 1000, 781]) | 7\n",
-      "19392    | torch.Size([1, 1000, 767]) | 7\n",
-      "19393    | torch.Size([1, 1000, 762]) | 7\n",
-      "19394    | torch.Size([1, 1000, 754]) | 7\n",
-      "19395    | torch.Size([1, 1000, 754]) | 7\n",
-      "19396    | torch.Size([1, 1000, 772]) | 7\n",
-      "19397    | torch.Size([1, 1000, 754]) | 7\n",
-      "19398    | torch.Size([1, 1000, 754]) | 7\n",
-      "19399    | torch.Size([1, 1000, 754]) | 7\n",
-      "19400    | torch.Size([1, 1000, 754]) | 7\n",
-      "19401    | torch.Size([1, 1000, 754]) | 7\n",
-      "19402    | torch.Size([1, 1000, 777]) | 7\n",
-      "19403    | torch.Size([1, 1000, 771]) | 7\n",
-      "19404    | torch.Size([1, 1000, 779]) | 7\n",
-      "19405    | torch.Size([1, 1000, 798]) | 7\n",
-      "19406    | torch.Size([1, 1000, 754]) | 7\n",
-      "19407    | torch.Size([1, 1000, 784]) | 7\n",
-      "19408    | torch.Size([1, 1000, 777]) | 7\n",
-      "19409    | torch.Size([1, 1000, 754]) | 7\n",
-      "19410    | torch.Size([1, 1000, 754]) | 7\n",
-      "19411    | torch.Size([1, 1000, 772]) | 7\n",
-      "19412    | torch.Size([1, 1000, 762]) | 7\n",
-      "19413    | torch.Size([1, 1000, 805]) | 7\n",
-      "19414    | torch.Size([1, 1000, 780]) | 7\n",
-      "19415    | torch.Size([1, 1000, 767]) | 7\n",
-      "19416    | torch.Size([1, 1000, 754]) | 7\n",
-      "19417    | torch.Size([1, 1000, 777]) | 7\n",
-      "19418    | torch.Size([1, 1000, 782]) | 7\n",
-      "19419    | torch.Size([1, 1000, 782]) | 7\n",
-      "19420    | torch.Size([1, 1000, 782]) | 7\n",
-      "19421    | torch.Size([1, 1000, 754]) | 7\n",
-      "19422    | torch.Size([1, 1000, 794]) | 7\n",
-      "19423    | torch.Size([1, 1000, 762]) | 7\n",
-      "19424    | torch.Size([1, 1000, 754]) | 7\n",
-      "19425    | torch.Size([1, 1000, 785]) | 7\n",
-      "19426    | torch.Size([1, 1000, 778]) | 7\n",
-      "19427    | torch.Size([1, 1000, 789]) | 7\n",
-      "19428    | torch.Size([1, 1000, 754]) | 7\n",
-      "19429    | torch.Size([1, 1000, 754]) | 7\n",
-      "19430    | torch.Size([1, 1000, 777]) | 7\n",
-      "19431    | torch.Size([1, 1000, 762]) | 7\n",
-      "19432    | torch.Size([1, 1000, 804]) | 7\n",
-      "19433    | torch.Size([1, 1000, 754]) | 7\n",
-      "19434    | torch.Size([1, 1000, 775]) | 7\n",
-      "19435    | torch.Size([1, 1000, 777]) | 7\n",
-      "19436    | torch.Size([1, 1000, 778]) | 7\n",
-      "19437    | torch.Size([1, 1000, 771]) | 7\n",
-      "19438    | torch.Size([1, 1000, 777]) | 7\n",
-      "19439    | torch.Size([1, 1000, 778]) | 7\n",
-      "19440    | torch.Size([1, 1000, 754]) | 7\n",
-      "19441    | torch.Size([1, 1000, 754]) | 7\n",
-      "19442    | torch.Size([1, 1000, 754]) | 7\n",
-      "19443    | torch.Size([1, 1000, 754]) | 7\n",
-      "19444    | torch.Size([1, 1000, 754]) | 7\n",
-      "19445    | torch.Size([1, 1000, 762]) | 7\n",
-      "19446    | torch.Size([1, 1000, 785]) | 7\n",
-      "19447    | torch.Size([1, 1000, 754]) | 7\n",
-      "19448    | torch.Size([1, 1000, 754]) | 7\n",
-      "19449    | torch.Size([1, 1000, 773]) | 7\n",
-      "19450    | torch.Size([1, 1000, 813]) | 7\n",
-      "19451    | torch.Size([1, 1000, 754]) | 7\n",
-      "19452    | torch.Size([1, 1000, 785]) | 7\n",
-      "19453    | torch.Size([1, 1000, 775]) | 7\n",
-      "19454    | torch.Size([1, 1000, 774]) | 7\n",
-      "19455    | torch.Size([1, 1000, 757]) | 7\n",
-      "19456    | torch.Size([1, 1000, 754]) | 7\n",
-      "19457    | torch.Size([1, 1000, 754]) | 7\n",
-      "19458    | torch.Size([1, 1000, 762]) | 7\n",
-      "19459    | torch.Size([1, 1000, 754]) | 7\n",
-      "19460    | torch.Size([1, 1000, 754]) | 7\n",
-      "19461    | torch.Size([1, 1000, 754]) | 7\n",
-      "19462    | torch.Size([1, 1000, 762]) | 7\n",
-      "19463    | torch.Size([1, 1000, 754]) | 7\n",
-      "19464    | torch.Size([1, 1000, 754]) | 7\n",
-      "19465    | torch.Size([1, 1000, 762]) | 7\n",
-      "19466    | torch.Size([1, 1000, 762]) | 7\n",
-      "19467    | torch.Size([1, 1000, 754]) | 7\n",
-      "19468    | torch.Size([1, 1000, 754]) | 7\n",
-      "19469    | torch.Size([1, 1000, 781]) | 7\n",
-      "19470    | torch.Size([1, 1000, 754]) | 7\n",
-      "19471    | torch.Size([1, 1000, 762]) | 7\n",
-      "19472    | torch.Size([1, 1000, 754]) | 7\n",
-      "19473    | torch.Size([1, 1000, 807]) | 7\n",
-      "19474    | torch.Size([1, 1000, 771]) | 7\n",
-      "19475    | torch.Size([1, 1000, 762]) | 7\n",
-      "19476    | torch.Size([1, 1000, 789]) | 7\n",
-      "19477    | torch.Size([1, 1000, 777]) | 7\n",
-      "19478    | torch.Size([1, 1000, 762]) | 7\n",
-      "19479    | torch.Size([1, 1000, 762]) | 7\n",
-      "19480    | torch.Size([1, 1000, 847]) | 7\n",
-      "19481    | torch.Size([1, 1000, 752]) | 7\n",
-      "19482    | torch.Size([1, 1000, 754]) | 7\n",
-      "19483    | torch.Size([1, 1000, 777]) | 7\n",
-      "19484    | torch.Size([1, 1000, 771]) | 7\n",
-      "19485    | torch.Size([1, 1000, 754]) | 7\n",
-      "19486    | torch.Size([1, 1000, 802]) | 7\n",
-      "19487    | torch.Size([1, 1000, 799]) | 7\n",
-      "19488    | torch.Size([1, 1000, 754]) | 7\n",
-      "19489    | torch.Size([1, 1000, 754]) | 7\n",
-      "19490    | torch.Size([1, 1000, 762]) | 7\n",
-      "19491    | torch.Size([1, 1000, 777]) | 7\n",
-      "19492    | torch.Size([1, 1000, 784]) | 7\n",
-      "19493    | torch.Size([1, 1000, 777]) | 7\n",
-      "19494    | torch.Size([1, 1000, 754]) | 7\n",
-      "19495    | torch.Size([1, 1000, 772]) | 7\n",
-      "19496    | torch.Size([1, 1000, 762]) | 7\n",
-      "19497    | torch.Size([1, 1000, 776]) | 7\n",
-      "19498    | torch.Size([1, 1000, 811]) | 7\n",
-      "19499    | torch.Size([1, 1000, 754]) | 7\n",
-      "19500    | torch.Size([1, 1000, 778]) | 7\n",
-      "19501    | torch.Size([1, 1000, 792]) | 7\n",
-      "19502    | torch.Size([1, 1000, 777]) | 7\n",
-      "19503    | torch.Size([1, 1000, 754]) | 7\n",
-      "19504    | torch.Size([1, 1000, 788]) | 7\n",
-      "19505    | torch.Size([1, 1000, 762]) | 7\n",
-      "19506    | torch.Size([1, 1000, 788]) | 7\n",
-      "19507    | torch.Size([1, 1000, 780]) | 7\n",
-      "19508    | torch.Size([1, 1000, 792]) | 7\n",
-      "19509    | torch.Size([1, 1000, 785]) | 7\n",
-      "19510    | torch.Size([1, 1000, 778]) | 7\n",
-      "19511    | torch.Size([1, 1000, 779]) | 7\n",
-      "19512    | torch.Size([1, 1000, 754]) | 7\n",
-      "19513    | torch.Size([1, 1000, 742]) | 7\n",
-      "19514    | torch.Size([1, 1000, 754]) | 7\n",
-      "19515    | torch.Size([1, 1000, 777]) | 7\n",
-      "19516    | torch.Size([1, 1000, 777]) | 7\n",
-      "19517    | torch.Size([1, 1000, 762]) | 7\n",
-      "19518    | torch.Size([1, 1000, 781]) | 7\n",
-      "19519    | torch.Size([1, 1000, 777]) | 7\n",
-      "19520    | torch.Size([1, 1000, 777]) | 7\n",
-      "19521    | torch.Size([1, 1000, 777]) | 7\n",
-      "19522    | torch.Size([1, 1000, 789]) | 7\n",
-      "19523    | torch.Size([1, 1000, 785]) | 7\n",
-      "19524    | torch.Size([1, 1000, 812]) | 7\n",
-      "19525    | torch.Size([1, 1000, 777]) | 7\n",
-      "19526    | torch.Size([1, 1000, 767]) | 7\n",
-      "19527    | torch.Size([1, 1000, 781]) | 7\n",
-      "19528    | torch.Size([1, 1000, 762]) | 7\n",
-      "19529    | torch.Size([1, 1000, 762]) | 7\n",
-      "19530    | torch.Size([1, 1000, 793]) | 7\n",
-      "19531    | torch.Size([1, 1000, 762]) | 7\n",
-      "19532    | torch.Size([1, 1000, 787]) | 7\n",
-      "19533    | torch.Size([1, 1000, 777]) | 7\n",
-      "19534    | torch.Size([1, 1000, 777]) | 7\n",
-      "19535    | torch.Size([1, 1000, 791]) | 7\n",
-      "19536    | torch.Size([1, 1000, 752]) | 7\n",
-      "19537    | torch.Size([1, 1000, 771]) | 7\n",
-      "19538    | torch.Size([1, 1000, 787]) | 7\n",
-      "19539    | torch.Size([1, 1000, 777]) | 7\n",
-      "19540    | torch.Size([1, 1000, 754]) | 7\n",
-      "19541    | torch.Size([1, 1000, 771]) | 7\n",
-      "19542    | torch.Size([1, 1000, 754]) | 7\n",
-      "19543    | torch.Size([1, 1000, 754]) | 7\n",
-      "19544    | torch.Size([1, 1000, 810]) | 7\n",
-      "19545    | torch.Size([1, 1000, 775]) | 7\n",
-      "19546    | torch.Size([1, 1000, 783]) | 7\n",
-      "19547    | torch.Size([1, 1000, 754]) | 7\n",
-      "19548    | torch.Size([1, 1000, 771]) | 7\n",
-      "19549    | torch.Size([1, 1000, 754]) | 7\n",
-      "19550    | torch.Size([1, 1000, 754]) | 7\n",
-      "19551    | torch.Size([1, 1000, 816]) | 7\n",
-      "19552    | torch.Size([1, 1000, 788]) | 7\n",
-      "19553    | torch.Size([1, 1000, 802]) | 7\n",
-      "19554    | torch.Size([1, 1000, 777]) | 7\n",
-      "19555    | torch.Size([1, 1000, 778]) | 7\n",
-      "19556    | torch.Size([1, 1000, 754]) | 7\n",
-      "19557    | torch.Size([1, 1000, 742]) | 7\n",
-      "19558    | torch.Size([1, 1000, 754]) | 7\n",
-      "19559    | torch.Size([1, 1000, 802]) | 7\n",
-      "19560    | torch.Size([1, 1000, 754]) | 7\n",
-      "19561    | torch.Size([1, 1000, 777]) | 7\n",
-      "19562    | torch.Size([1, 1000, 762]) | 7\n",
-      "19563    | torch.Size([1, 1000, 762]) | 7\n",
-      "19564    | torch.Size([1, 1000, 762]) | 7\n",
-      "19565    | torch.Size([1, 1000, 762]) | 7\n",
-      "19566    | torch.Size([1, 1000, 762]) | 7\n",
-      "19567    | torch.Size([1, 1000, 775]) | 7\n",
-      "19568    | torch.Size([1, 1000, 845]) | 7\n",
-      "19569    | torch.Size([1, 1000, 754]) | 7\n",
-      "19570    | torch.Size([1, 1000, 762]) | 7\n",
-      "19571    | torch.Size([1, 1000, 777]) | 7\n",
-      "19572    | torch.Size([1, 1000, 777]) | 7\n",
-      "19573    | torch.Size([1, 1000, 744]) | 7\n",
-      "19574    | torch.Size([1, 1000, 772]) | 7\n",
-      "19575    | torch.Size([1, 1000, 754]) | 7\n",
-      "19576    | torch.Size([1, 1000, 762]) | 7\n",
-      "19577    | torch.Size([1, 1000, 779]) | 7\n",
-      "19578    | torch.Size([1, 1000, 783]) | 7\n",
-      "19579    | torch.Size([1, 1000, 754]) | 7\n",
-      "19580    | torch.Size([1, 1000, 754]) | 7\n",
-      "19581    | torch.Size([1, 1000, 802]) | 7\n",
-      "19582    | torch.Size([1, 1000, 754]) | 7\n",
-      "19583    | torch.Size([1, 1000, 788]) | 7\n",
-      "19584    | torch.Size([1, 1000, 767]) | 7\n",
-      "19585    | torch.Size([1, 1000, 754]) | 7\n",
-      "19586    | torch.Size([1, 1000, 773]) | 7\n",
-      "19587    | torch.Size([1, 1000, 802]) | 7\n",
-      "19588    | torch.Size([1, 1000, 754]) | 7\n",
-      "19589    | torch.Size([1, 1000, 1001]) | 7\n",
-      "19590    | torch.Size([1, 1000, 786]) | 7\n",
-      "19591    | torch.Size([1, 1000, 778]) | 7\n",
-      "19592    | torch.Size([1, 1000, 771]) | 7\n",
-      "19593    | torch.Size([1, 1000, 754]) | 7\n",
-      "19594    | torch.Size([1, 1000, 771]) | 7\n",
-      "19595    | torch.Size([1, 1000, 754]) | 7\n",
-      "19596    | torch.Size([1, 1000, 777]) | 7\n",
-      "19597    | torch.Size([1, 1000, 777]) | 7\n",
-      "19598    | torch.Size([1, 1000, 777]) | 7\n",
-      "19599    | torch.Size([1, 1000, 754]) | 7\n",
-      "19600    | torch.Size([1, 1000, 754]) | 7\n",
-      "19601    | torch.Size([1, 1000, 844]) | 7\n",
-      "19602    | torch.Size([1, 1000, 762]) | 7\n",
-      "19603    | torch.Size([1, 1000, 800]) | 7\n",
-      "19604    | torch.Size([1, 1000, 754]) | 7\n",
-      "19605    | torch.Size([1, 1000, 772]) | 7\n",
-      "19606    | torch.Size([1, 1000, 754]) | 7\n",
-      "19607    | torch.Size([1, 1000, 754]) | 7\n",
-      "19608    | torch.Size([1, 1000, 777]) | 7\n",
-      "19609    | torch.Size([1, 1000, 777]) | 7\n",
-      "19610    | torch.Size([1, 1000, 754]) | 7\n",
-      "19611    | torch.Size([1, 1000, 782]) | 7\n",
-      "19612    | torch.Size([1, 1000, 754]) | 7\n",
-      "19613    | torch.Size([1, 1000, 754]) | 7\n",
-      "19614    | torch.Size([1, 1000, 762]) | 7\n",
-      "19615    | torch.Size([1, 1000, 754]) | 7\n",
-      "19616    | torch.Size([1, 1000, 782]) | 7\n",
-      "19617    | torch.Size([1, 1000, 771]) | 7\n",
-      "19618    | torch.Size([1, 1000, 762]) | 7\n",
-      "19619    | torch.Size([1, 1000, 762]) | 7\n",
-      "19620    | torch.Size([1, 1000, 754]) | 7\n",
-      "19621    | torch.Size([1, 1000, 754]) | 7\n",
-      "19622    | torch.Size([1, 1000, 754]) | 7\n",
-      "19623    | torch.Size([1, 1000, 754]) | 7\n",
-      "19624    | torch.Size([1, 1000, 762]) | 7\n",
-      "19625    | torch.Size([1, 1000, 754]) | 7\n",
-      "19626    | torch.Size([1, 1000, 782]) | 7\n",
-      "19627    | torch.Size([1, 1000, 762]) | 7\n",
-      "19628    | torch.Size([1, 1000, 772]) | 7\n",
-      "19629    | torch.Size([1, 1000, 775]) | 7\n",
-      "19630    | torch.Size([1, 1000, 754]) | 7\n",
-      "19631    | torch.Size([1, 1000, 754]) | 7\n",
-      "19632    | torch.Size([1, 1000, 762]) | 7\n",
-      "19633    | torch.Size([1, 1000, 762]) | 7\n",
-      "19634    | torch.Size([1, 1000, 771]) | 7\n",
-      "19635    | torch.Size([1, 1000, 754]) | 7\n",
-      "19636    | torch.Size([1, 1000, 762]) | 7\n",
-      "19637    | torch.Size([1, 1000, 754]) | 7\n",
-      "19638    | torch.Size([1, 1000, 754]) | 7\n",
-      "19639    | torch.Size([1, 1000, 780]) | 7\n",
-      "19640    | torch.Size([1, 1000, 777]) | 7\n",
-      "19641    | torch.Size([1, 1000, 754]) | 7\n",
-      "19642    | torch.Size([1, 1000, 762]) | 7\n",
-      "19643    | torch.Size([1, 1000, 754]) | 7\n",
-      "19644    | torch.Size([1, 1000, 787]) | 7\n",
-      "19645    | torch.Size([1, 1000, 713]) | 7\n",
-      "19646    | torch.Size([1, 1000, 768]) | 7\n",
-      "19647    | torch.Size([1, 1000, 787]) | 7\n",
-      "19648    | torch.Size([1, 1000, 754]) | 7\n",
-      "19649    | torch.Size([1, 1000, 802]) | 7\n",
-      "19650    | torch.Size([1, 1000, 801]) | 7\n",
-      "19651    | torch.Size([1, 1000, 754]) | 7\n",
-      "19652    | torch.Size([1, 1000, 772]) | 7\n",
-      "19653    | torch.Size([1, 1000, 767]) | 7\n",
-      "19654    | torch.Size([1, 1000, 771]) | 7\n",
-      "19655    | torch.Size([1, 1000, 785]) | 7\n",
-      "19656    | torch.Size([1, 1000, 754]) | 7\n",
-      "19657    | torch.Size([1, 1000, 778]) | 7\n",
-      "19658    | torch.Size([1, 1000, 754]) | 7\n",
-      "19659    | torch.Size([1, 1000, 768]) | 7\n",
-      "19660    | torch.Size([1, 1000, 786]) | 7\n",
-      "19661    | torch.Size([1, 1000, 777]) | 7\n",
-      "19662    | torch.Size([1, 1000, 762]) | 7\n",
-      "19663    | torch.Size([1, 1000, 778]) | 7\n",
-      "19664    | torch.Size([1, 1000, 754]) | 7\n",
-      "19665    | torch.Size([1, 1000, 762]) | 7\n",
-      "19666    | torch.Size([1, 1000, 792]) | 7\n",
-      "19667    | torch.Size([1, 1000, 754]) | 7\n",
-      "19668    | torch.Size([1, 1000, 776]) | 7\n",
-      "19669    | torch.Size([1, 1000, 774]) | 7\n",
-      "19670    | torch.Size([1, 1000, 773]) | 7\n",
-      "19671    | torch.Size([1, 1000, 762]) | 7\n",
-      "19672    | torch.Size([1, 1000, 802]) | 7\n",
-      "19673    | torch.Size([1, 1000, 754]) | 7\n",
-      "19674    | torch.Size([1, 1000, 769]) | 7\n",
-      "19675    | torch.Size([1, 1000, 780]) | 7\n",
-      "19676    | torch.Size([1, 1000, 784]) | 7\n",
-      "19677    | torch.Size([1, 1000, 762]) | 7\n",
-      "19678    | torch.Size([1, 1000, 777]) | 7\n",
-      "19679    | torch.Size([1, 1000, 778]) | 7\n",
-      "19680    | torch.Size([1, 1000, 800]) | 7\n",
-      "19681    | torch.Size([1, 1000, 790]) | 7\n",
-      "19682    | torch.Size([1, 1000, 792]) | 7\n",
-      "19683    | torch.Size([1, 1000, 754]) | 7\n",
-      "19684    | torch.Size([1, 1000, 754]) | 7\n",
-      "19685    | torch.Size([1, 1000, 762]) | 7\n",
-      "19686    | torch.Size([1, 1000, 754]) | 7\n",
-      "19687    | torch.Size([1, 1000, 773]) | 7\n",
-      "19688    | torch.Size([1, 1000, 771]) | 7\n",
-      "19689    | torch.Size([1, 1000, 762]) | 7\n",
-      "19690    | torch.Size([1, 1000, 754]) | 7\n",
-      "19691    | torch.Size([1, 1000, 777]) | 7\n",
-      "19692    | torch.Size([1, 1000, 754]) | 7\n",
-      "19693    | torch.Size([1, 1000, 754]) | 7\n",
-      "19694    | torch.Size([1, 1000, 754]) | 7\n",
-      "19695    | torch.Size([1, 1000, 844]) | 7\n",
-      "19696    | torch.Size([1, 1000, 754]) | 7\n",
-      "19697    | torch.Size([1, 1000, 754]) | 7\n",
-      "19698    | torch.Size([1, 1000, 779]) | 7\n",
-      "19699    | torch.Size([1, 1000, 754]) | 7\n",
-      "19700    | torch.Size([1, 1000, 754]) | 7\n",
-      "19701    | torch.Size([1, 1000, 762]) | 7\n",
-      "19702    | torch.Size([1, 1000, 754]) | 7\n",
-      "19703    | torch.Size([1, 1000, 754]) | 7\n",
-      "19704    | torch.Size([1, 1000, 754]) | 7\n",
-      "19705    | torch.Size([1, 1000, 777]) | 7\n",
-      "19706    | torch.Size([1, 1000, 754]) | 7\n",
-      "19707    | torch.Size([1, 1000, 781]) | 7\n",
-      "19708    | torch.Size([1, 1000, 768]) | 7\n",
-      "19709    | torch.Size([1, 1000, 762]) | 7\n",
-      "19710    | torch.Size([1, 1000, 779]) | 7\n",
-      "19711    | torch.Size([1, 1000, 754]) | 7\n",
-      "19712    | torch.Size([1, 1000, 754]) | 7\n",
-      "19713    | torch.Size([1, 1000, 779]) | 7\n",
-      "19714    | torch.Size([1, 1000, 787]) | 7\n",
-      "19715    | torch.Size([1, 1000, 798]) | 7\n",
-      "19716    | torch.Size([1, 1000, 754]) | 7\n",
-      "19717    | torch.Size([1, 1000, 788]) | 7\n",
-      "19718    | torch.Size([1, 1000, 802]) | 7\n",
-      "19719    | torch.Size([1, 1000, 781]) | 7\n",
-      "19720    | torch.Size([1, 1000, 777]) | 7\n",
-      "19721    | torch.Size([1, 1000, 762]) | 7\n",
-      "19722    | torch.Size([1, 1000, 754]) | 7\n",
-      "19723    | torch.Size([1, 1000, 754]) | 7\n",
-      "19724    | torch.Size([1, 1000, 762]) | 7\n",
-      "19725    | torch.Size([1, 1000, 754]) | 7\n",
-      "19726    | torch.Size([1, 1000, 772]) | 7\n",
-      "19727    | torch.Size([1, 1000, 754]) | 7\n",
-      "19728    | torch.Size([1, 1000, 777]) | 7\n",
-      "19729    | torch.Size([1, 1000, 777]) | 7\n",
-      "19730    | torch.Size([1, 1000, 742]) | 7\n",
-      "19731    | torch.Size([1, 1000, 782]) | 7\n",
-      "19732    | torch.Size([1, 1000, 775]) | 7\n",
-      "19733    | torch.Size([1, 1000, 776]) | 7\n",
-      "19734    | torch.Size([1, 1000, 778]) | 7\n",
-      "19735    | torch.Size([1, 1000, 779]) | 7\n",
-      "19736    | torch.Size([1, 1000, 754]) | 7\n",
-      "19737    | torch.Size([1, 1000, 784]) | 7\n",
-      "19738    | torch.Size([1, 1000, 772]) | 7\n",
-      "19739    | torch.Size([1, 1000, 754]) | 7\n",
-      "19740    | torch.Size([1, 1000, 777]) | 7\n",
-      "19741    | torch.Size([1, 1000, 779]) | 7\n",
-      "19742    | torch.Size([1, 1000, 771]) | 7\n",
-      "19743    | torch.Size([1, 1000, 852]) | 7\n",
-      "19744    | torch.Size([1, 1000, 793]) | 7\n",
-      "19745    | torch.Size([1, 1000, 803]) | 7\n",
-      "19746    | torch.Size([1, 1000, 754]) | 7\n",
-      "19747    | torch.Size([1, 1000, 762]) | 7\n",
-      "19748    | torch.Size([1, 1000, 780]) | 7\n",
-      "19749    | torch.Size([1, 1000, 762]) | 7\n",
-      "19750    | torch.Size([1, 1000, 778]) | 7\n",
-      "19751    | torch.Size([1, 1000, 762]) | 7\n",
-      "19752    | torch.Size([1, 1000, 780]) | 7\n",
-      "19753    | torch.Size([1, 1000, 777]) | 7\n",
-      "19754    | torch.Size([1, 1000, 754]) | 7\n",
-      "19755    | torch.Size([1, 1000, 762]) | 7\n",
-      "19756    | torch.Size([1, 1000, 777]) | 7\n",
-      "19757    | torch.Size([1, 1000, 754]) | 7\n",
-      "19758    | torch.Size([1, 1000, 777]) | 7\n",
-      "19759    | torch.Size([1, 1000, 754]) | 7\n",
-      "19760    | torch.Size([1, 1000, 780]) | 7\n",
-      "19761    | torch.Size([1, 1000, 777]) | 7\n",
-      "19762    | torch.Size([1, 1000, 775]) | 7\n",
-      "19763    | torch.Size([1, 1000, 767]) | 7\n",
-      "19764    | torch.Size([1, 1000, 762]) | 7\n",
-      "19765    | torch.Size([1, 1000, 762]) | 7\n",
-      "19766    | torch.Size([1, 1000, 788]) | 7\n",
-      "19767    | torch.Size([1, 1000, 762]) | 7\n",
-      "19768    | torch.Size([1, 1000, 779]) | 7\n",
-      "19769    | torch.Size([1, 1000, 777]) | 7\n",
-      "19770    | torch.Size([1, 1000, 766]) | 7\n",
-      "19771    | torch.Size([1, 1000, 762]) | 7\n",
-      "19772    | torch.Size([1, 1000, 762]) | 7\n",
-      "19773    | torch.Size([1, 1000, 754]) | 7\n",
-      "19774    | torch.Size([1, 1000, 775]) | 7\n",
-      "19775    | torch.Size([1, 1000, 762]) | 7\n",
-      "19776    | torch.Size([1, 1000, 777]) | 7\n",
-      "19777    | torch.Size([1, 1000, 754]) | 7\n",
-      "19778    | torch.Size([1, 1000, 771]) | 7\n",
-      "19779    | torch.Size([1, 1000, 754]) | 7\n",
-      "19780    | torch.Size([1, 1000, 762]) | 7\n",
-      "19781    | torch.Size([1, 1000, 754]) | 7\n",
-      "19782    | torch.Size([1, 1000, 777]) | 7\n",
-      "19783    | torch.Size([1, 1000, 794]) | 7\n",
-      "19784    | torch.Size([1, 1000, 785]) | 7\n",
-      "19785    | torch.Size([1, 1000, 754]) | 7\n",
-      "19786    | torch.Size([1, 1000, 762]) | 7\n",
-      "19787    | torch.Size([1, 1000, 772]) | 7\n",
-      "19788    | torch.Size([1, 1000, 776]) | 7\n",
-      "19789    | torch.Size([1, 1000, 771]) | 7\n",
-      "19790    | torch.Size([1, 1000, 786]) | 7\n",
-      "19791    | torch.Size([1, 1000, 784]) | 7\n",
-      "19792    | torch.Size([1, 1000, 805]) | 7\n",
-      "19793    | torch.Size([1, 1000, 754]) | 7\n",
-      "19794    | torch.Size([1, 1000, 778]) | 7\n",
-      "19795    | torch.Size([1, 1000, 777]) | 7\n",
-      "19796    | torch.Size([1, 1000, 754]) | 7\n",
-      "19797    | torch.Size([1, 1000, 786]) | 7\n",
-      "19798    | torch.Size([1, 1000, 762]) | 7\n",
-      "19799    | torch.Size([1, 1000, 775]) | 7\n",
-      "19800    | torch.Size([1, 1000, 762]) | 7\n",
-      "19801    | torch.Size([1, 1000, 754]) | 7\n",
-      "19802    | torch.Size([1, 1000, 754]) | 7\n",
-      "19803    | torch.Size([1, 1000, 754]) | 7\n",
-      "19804    | torch.Size([1, 1000, 770]) | 7\n",
-      "19805    | torch.Size([1, 1000, 754]) | 7\n",
-      "19806    | torch.Size([1, 1000, 754]) | 7\n",
-      "19807    | torch.Size([1, 1000, 776]) | 7\n",
-      "19808    | torch.Size([1, 1000, 754]) | 7\n",
-      "19809    | torch.Size([1, 1000, 754]) | 7\n",
-      "19810    | torch.Size([1, 1000, 783]) | 7\n",
-      "19811    | torch.Size([1, 1000, 754]) | 7\n",
-      "19812    | torch.Size([1, 1000, 778]) | 7\n",
-      "19813    | torch.Size([1, 1000, 754]) | 7\n",
-      "19814    | torch.Size([1, 1000, 754]) | 7\n",
-      "19815    | torch.Size([1, 1000, 754]) | 7\n",
-      "19816    | torch.Size([1, 1000, 754]) | 7\n",
-      "19817    | torch.Size([1, 1000, 754]) | 7\n",
-      "19818    | torch.Size([1, 1000, 832]) | 7\n",
-      "19819    | torch.Size([1, 1000, 771]) | 7\n",
-      "19820    | torch.Size([1, 1000, 780]) | 7\n",
-      "19821    | torch.Size([1, 1000, 783]) | 7\n",
-      "19822    | torch.Size([1, 1000, 791]) | 7\n",
-      "19823    | torch.Size([1, 1000, 777]) | 7\n",
-      "19824    | torch.Size([1, 1000, 762]) | 7\n",
-      "19825    | torch.Size([1, 1000, 754]) | 7\n",
-      "19826    | torch.Size([1, 1000, 754]) | 7\n",
-      "19827    | torch.Size([1, 1000, 774]) | 7\n",
-      "19828    | torch.Size([1, 1000, 754]) | 7\n",
-      "19829    | torch.Size([1, 1000, 754]) | 7\n",
-      "19830    | torch.Size([1, 1000, 762]) | 7\n",
-      "19831    | torch.Size([1, 1000, 754]) | 7\n",
-      "19832    | torch.Size([1, 1000, 754]) | 7\n",
-      "19833    | torch.Size([1, 1000, 754]) | 7\n",
-      "19834    | torch.Size([1, 1000, 779]) | 7\n",
-      "19835    | torch.Size([1, 1000, 788]) | 7\n",
-      "19836    | torch.Size([1, 1000, 767]) | 7\n",
-      "19837    | torch.Size([1, 1000, 754]) | 7\n",
-      "19838    | torch.Size([1, 1000, 771]) | 7\n",
-      "19839    | torch.Size([1, 1000, 775]) | 7\n",
-      "19840    | torch.Size([1, 1000, 754]) | 7\n",
-      "19841    | torch.Size([1, 1000, 754]) | 7\n",
-      "19842    | torch.Size([1, 1000, 754]) | 7\n",
-      "19843    | torch.Size([1, 1000, 765]) | 7\n",
-      "19844    | torch.Size([1, 1000, 802]) | 7\n",
-      "19845    | torch.Size([1, 1000, 754]) | 7\n",
-      "19846    | torch.Size([1, 1000, 784]) | 7\n",
-      "19847    | torch.Size([1, 1000, 797]) | 7\n",
-      "19848    | torch.Size([1, 1000, 754]) | 7\n",
-      "19849    | torch.Size([1, 1000, 777]) | 7\n",
-      "19850    | torch.Size([1, 1000, 775]) | 7\n",
-      "19851    | torch.Size([1, 1000, 754]) | 7\n",
-      "19852    | torch.Size([1, 1000, 762]) | 7\n",
-      "19853    | torch.Size([1, 1000, 762]) | 7\n",
-      "19854    | torch.Size([1, 1000, 795]) | 7\n",
-      "19855    | torch.Size([1, 1000, 754]) | 7\n",
-      "19856    | torch.Size([1, 1000, 771]) | 7\n",
-      "19857    | torch.Size([1, 1000, 754]) | 7\n",
-      "19858    | torch.Size([1, 1000, 754]) | 7\n",
-      "19859    | torch.Size([1, 1000, 762]) | 7\n",
-      "19860    | torch.Size([1, 1000, 754]) | 7\n",
-      "19861    | torch.Size([1, 1000, 781]) | 7\n",
-      "19862    | torch.Size([1, 1000, 773]) | 7\n",
-      "19863    | torch.Size([1, 1000, 782]) | 7\n",
-      "19864    | torch.Size([1, 1000, 762]) | 7\n",
-      "19865    | torch.Size([1, 1000, 756]) | 7\n",
-      "19866    | torch.Size([1, 1000, 771]) | 7\n",
-      "19867    | torch.Size([1, 1000, 762]) | 7\n",
-      "19868    | torch.Size([1, 1000, 788]) | 7\n",
-      "19869    | torch.Size([1, 1000, 754]) | 7\n",
-      "19870    | torch.Size([1, 1000, 788]) | 7\n",
-      "19871    | torch.Size([1, 1000, 754]) | 7\n",
-      "19872    | torch.Size([1, 1000, 754]) | 7\n",
-      "19873    | torch.Size([1, 1000, 762]) | 7\n",
-      "19874    | torch.Size([1, 1000, 733]) | 7\n",
-      "19875    | torch.Size([1, 1000, 775]) | 7\n",
-      "19876    | torch.Size([1, 1000, 784]) | 7\n",
-      "19877    | torch.Size([1, 1000, 761]) | 7\n",
-      "19878    | torch.Size([1, 1000, 770]) | 7\n",
-      "19879    | torch.Size([1, 1000, 787]) | 7\n",
-      "19880    | torch.Size([1, 1000, 762]) | 7\n",
-      "19881    | torch.Size([1, 1000, 754]) | 7\n",
-      "19882    | torch.Size([1, 1000, 784]) | 7\n",
-      "19883    | torch.Size([1, 1000, 779]) | 7\n",
-      "19884    | torch.Size([1, 1000, 754]) | 7\n",
-      "19885    | torch.Size([1, 1000, 754]) | 7\n",
-      "19886    | torch.Size([1, 1000, 762]) | 7\n",
-      "19887    | torch.Size([1, 1000, 754]) | 7\n",
-      "19888    | torch.Size([1, 1000, 754]) | 7\n",
-      "19889    | torch.Size([1, 1000, 789]) | 7\n",
-      "19890    | torch.Size([1, 1000, 754]) | 7\n",
-      "19891    | torch.Size([1, 1000, 754]) | 7\n",
-      "19892    | torch.Size([1, 1000, 792]) | 7\n",
-      "19893    | torch.Size([1, 1000, 771]) | 7\n",
-      "19894    | torch.Size([1, 1000, 754]) | 7\n",
-      "19895    | torch.Size([1, 1000, 817]) | 7\n",
-      "19896    | torch.Size([1, 1000, 754]) | 7\n",
-      "19897    | torch.Size([1, 1000, 787]) | 7\n",
-      "19898    | torch.Size([1, 1000, 776]) | 7\n",
-      "19899    | torch.Size([1, 1000, 772]) | 7\n",
-      "19900    | torch.Size([1, 1000, 772]) | 7\n",
-      "19901    | torch.Size([1, 1000, 777]) | 7\n",
-      "19902    | torch.Size([1, 1000, 754]) | 7\n",
-      "19903    | torch.Size([1, 1000, 762]) | 7\n",
-      "19904    | torch.Size([1, 1000, 762]) | 7\n",
-      "19905    | torch.Size([1, 1000, 771]) | 7\n",
-      "19906    | torch.Size([1, 1000, 784]) | 7\n",
-      "19907    | torch.Size([1, 1000, 777]) | 7\n",
-      "19908    | torch.Size([1, 1000, 777]) | 7\n",
-      "19909    | torch.Size([1, 1000, 788]) | 7\n",
-      "19910    | torch.Size([1, 1000, 768]) | 7\n",
-      "19911    | torch.Size([1, 1000, 762]) | 7\n",
-      "19912    | torch.Size([1, 1000, 754]) | 7\n",
-      "19913    | torch.Size([1, 1000, 762]) | 7\n",
-      "19914    | torch.Size([1, 1000, 777]) | 7\n",
-      "19915    | torch.Size([1, 1000, 782]) | 7\n",
-      "19916    | torch.Size([1, 1000, 777]) | 7\n",
-      "19917    | torch.Size([1, 1000, 754]) | 7\n",
-      "19918    | torch.Size([1, 1000, 777]) | 7\n",
-      "19919    | torch.Size([1, 1000, 762]) | 7\n",
-      "19920    | torch.Size([1, 1000, 778]) | 7\n",
-      "19921    | torch.Size([1, 1000, 754]) | 7\n",
-      "19922    | torch.Size([1, 1000, 754]) | 7\n",
-      "19923    | torch.Size([1, 1000, 780]) | 7\n",
-      "19924    | torch.Size([1, 1000, 795]) | 7\n",
-      "19925    | torch.Size([1, 1000, 777]) | 7\n",
-      "19926    | torch.Size([1, 1000, 771]) | 7\n",
-      "19927    | torch.Size([1, 1000, 785]) | 7\n",
-      "19928    | torch.Size([1, 1000, 776]) | 7\n",
-      "19929    | torch.Size([1, 1000, 754]) | 7\n",
-      "19930    | torch.Size([1, 1000, 767]) | 7\n",
-      "19931    | torch.Size([1, 1000, 754]) | 7\n",
-      "19932    | torch.Size([1, 1000, 767]) | 7\n",
-      "19933    | torch.Size([1, 1000, 726]) | 7\n",
-      "19934    | torch.Size([1, 1000, 769]) | 7\n",
-      "19935    | torch.Size([1, 1000, 774]) | 7\n",
-      "19936    | torch.Size([1, 1000, 762]) | 7\n",
-      "19937    | torch.Size([1, 1000, 770]) | 7\n",
-      "19938    | torch.Size([1, 1000, 769]) | 7\n",
-      "19939    | torch.Size([1, 1000, 790]) | 7\n",
-      "19940    | torch.Size([1, 1000, 754]) | 7\n",
-      "19941    | torch.Size([1, 1000, 754]) | 7\n",
-      "19942    | torch.Size([1, 1000, 754]) | 7\n",
-      "19943    | torch.Size([1, 1000, 754]) | 7\n",
-      "19944    | torch.Size([1, 1000, 777]) | 7\n",
-      "19945    | torch.Size([1, 1000, 785]) | 7\n",
-      "19946    | torch.Size([1, 1000, 754]) | 7\n",
-      "19947    | torch.Size([1, 1000, 754]) | 7\n",
-      "19948    | torch.Size([1, 1000, 781]) | 7\n",
-      "19949    | torch.Size([1, 1000, 762]) | 7\n",
-      "19950    | torch.Size([1, 1000, 754]) | 7\n",
-      "19951    | torch.Size([1, 1000, 782]) | 7\n",
-      "19952    | torch.Size([1, 1000, 782]) | 7\n",
-      "19953    | torch.Size([1, 1000, 777]) | 7\n",
-      "19954    | torch.Size([1, 1000, 778]) | 7\n",
-      "19955    | torch.Size([1, 1000, 777]) | 7\n",
-      "19956    | torch.Size([1, 1000, 786]) | 7\n",
-      "19957    | torch.Size([1, 1000, 754]) | 7\n",
-      "19958    | torch.Size([1, 1000, 777]) | 7\n",
-      "19959    | torch.Size([1, 1000, 754]) | 7\n",
-      "19960    | torch.Size([1, 1000, 782]) | 7\n",
-      "19961    | torch.Size([1, 1000, 771]) | 7\n",
-      "19962    | torch.Size([1, 1000, 777]) | 7\n",
-      "19963    | torch.Size([1, 1000, 754]) | 7\n",
-      "19964    | torch.Size([1, 1000, 786]) | 7\n",
-      "19965    | torch.Size([1, 1000, 754]) | 7\n",
-      "19966    | torch.Size([1, 1000, 776]) | 7\n",
-      "19967    | torch.Size([1, 1000, 754]) | 7\n",
-      "19968    | torch.Size([1, 1000, 754]) | 7\n",
-      "19969    | torch.Size([1, 1000, 754]) | 7\n",
-      "19970    | torch.Size([1, 1000, 777]) | 7\n",
-      "19971    | torch.Size([1, 1000, 762]) | 7\n",
-      "19972    | torch.Size([1, 1000, 762]) | 7\n",
-      "19973    | torch.Size([1, 1000, 788]) | 7\n",
-      "19974    | torch.Size([1, 1000, 782]) | 7\n",
-      "19975    | torch.Size([1, 1000, 754]) | 7\n",
-      "19976    | torch.Size([1, 1000, 770]) | 7\n",
-      "19977    | torch.Size([1, 1000, 777]) | 7\n",
-      "19978    | torch.Size([1, 1000, 754]) | 7\n",
-      "19979    | torch.Size([1, 1000, 773]) | 7\n",
-      "19980    | torch.Size([1, 1000, 754]) | 7\n",
-      "19981    | torch.Size([1, 1000, 775]) | 7\n",
-      "19982    | torch.Size([1, 1000, 795]) | 7\n",
-      "19983    | torch.Size([1, 1000, 754]) | 7\n",
-      "19984    | torch.Size([1, 1000, 762]) | 7\n",
-      "19985    | torch.Size([1, 1000, 754]) | 7\n",
-      "19986    | torch.Size([1, 1000, 754]) | 7\n",
-      "19987    | torch.Size([1, 1000, 754]) | 7\n",
-      "19988    | torch.Size([1, 1000, 775]) | 7\n",
-      "19989    | torch.Size([1, 1000, 747]) | 7\n",
-      "19990    | torch.Size([1, 1000, 754]) | 7\n",
-      "19991    | torch.Size([1, 1000, 754]) | 7\n",
-      "19992    | torch.Size([1, 1000, 754]) | 7\n",
-      "19993    | torch.Size([1, 1000, 762]) | 7\n",
-      "19994    | torch.Size([1, 1000, 762]) | 7\n",
-      "19995    | torch.Size([1, 1000, 777]) | 7\n",
-      "19996    | torch.Size([1, 1000, 775]) | 7\n",
-      "19997    | torch.Size([1, 1000, 754]) | 7\n",
-      "19998    | torch.Size([1, 1000, 777]) | 7\n",
-      "19999    | torch.Size([1, 1000, 777]) | 7\n",
-      "20000    | torch.Size([1, 1000, 779]) | 7\n",
-      "20001    | torch.Size([1, 1000, 763]) | 7\n",
-      "20002    | torch.Size([1, 1000, 771]) | 7\n",
-      "20003    | torch.Size([1, 1000, 754]) | 7\n",
-      "20004    | torch.Size([1, 1000, 777]) | 7\n",
-      "20005    | torch.Size([1, 1000, 754]) | 7\n",
-      "20006    | torch.Size([1, 1000, 776]) | 7\n",
-      "20007    | torch.Size([1, 1000, 777]) | 7\n",
-      "20008    | torch.Size([1, 1000, 754]) | 7\n",
-      "20009    | torch.Size([1, 1000, 774]) | 7\n",
-      "20010    | torch.Size([1, 1000, 754]) | 7\n",
-      "20011    | torch.Size([1, 1000, 780]) | 7\n",
-      "20012    | torch.Size([1, 1000, 780]) | 7\n",
-      "20013    | torch.Size([1, 1000, 772]) | 7\n",
-      "20014    | torch.Size([1, 1000, 771]) | 7\n",
-      "20015    | torch.Size([1, 1000, 754]) | 7\n",
-      "20016    | torch.Size([1, 1000, 772]) | 7\n",
-      "20017    | torch.Size([1, 1000, 777]) | 7\n",
-      "20018    | torch.Size([1, 1000, 771]) | 7\n",
-      "20019    | torch.Size([1, 1000, 754]) | 7\n",
-      "20020    | torch.Size([1, 1000, 777]) | 7\n",
-      "20021    | torch.Size([1, 1000, 777]) | 7\n",
-      "20022    | torch.Size([1, 1000, 754]) | 7\n",
-      "20023    | torch.Size([1, 1000, 777]) | 7\n",
-      "20024    | torch.Size([1, 1000, 783]) | 7\n",
-      "20025    | torch.Size([1, 1000, 777]) | 7\n",
-      "20026    | torch.Size([1, 1000, 777]) | 7\n",
-      "20027    | torch.Size([1, 1000, 777]) | 7\n",
-      "20028    | torch.Size([1, 1000, 764]) | 7\n",
-      "20029    | torch.Size([1, 1000, 740]) | 7\n",
-      "20030    | torch.Size([1, 1000, 802]) | 7\n",
-      "20031    | torch.Size([1, 1000, 787]) | 7\n",
-      "20032    | torch.Size([1, 1000, 754]) | 7\n",
-      "20033    | torch.Size([1, 1000, 754]) | 7\n",
-      "20034    | torch.Size([1, 1000, 783]) | 7\n",
-      "20035    | torch.Size([1, 1000, 762]) | 7\n",
-      "20036    | torch.Size([1, 1000, 754]) | 7\n",
-      "20037    | torch.Size([1, 1000, 754]) | 7\n",
-      "20038    | torch.Size([1, 1000, 762]) | 7\n",
-      "20039    | torch.Size([1, 1000, 754]) | 7\n",
-      "20040    | torch.Size([1, 1000, 754]) | 7\n",
-      "20041    | torch.Size([1, 1000, 779]) | 7\n",
-      "20042    | torch.Size([1, 1000, 786]) | 8\n",
-      "20043    | torch.Size([1, 1000, 754]) | 8\n",
-      "20044    | torch.Size([1, 1000, 762]) | 8\n",
-      "20045    | torch.Size([1, 1000, 778]) | 8\n",
-      "20046    | torch.Size([1, 1000, 777]) | 8\n",
-      "20047    | torch.Size([1, 1000, 754]) | 8\n",
-      "20048    | torch.Size([1, 1000, 782]) | 8\n",
-      "20049    | torch.Size([1, 1000, 771]) | 8\n",
-      "20050    | torch.Size([1, 1000, 769]) | 8\n",
-      "20051    | torch.Size([1, 1000, 771]) | 8\n",
-      "20052    | torch.Size([1, 1000, 762]) | 8\n",
-      "20053    | torch.Size([1, 1000, 762]) | 8\n",
-      "20054    | torch.Size([1, 1000, 781]) | 8\n",
-      "20055    | torch.Size([1, 1000, 788]) | 8\n",
-      "20056    | torch.Size([1, 1000, 754]) | 8\n",
-      "20057    | torch.Size([1, 1000, 762]) | 8\n",
-      "20058    | torch.Size([1, 1000, 762]) | 8\n",
-      "20059    | torch.Size([1, 1000, 754]) | 8\n",
-      "20060    | torch.Size([1, 1000, 791]) | 8\n",
-      "20061    | torch.Size([1, 1000, 754]) | 8\n",
-      "20062    | torch.Size([1, 1000, 754]) | 8\n",
-      "20063    | torch.Size([1, 1000, 754]) | 8\n",
-      "20064    | torch.Size([1, 1000, 778]) | 8\n",
-      "20065    | torch.Size([1, 1000, 770]) | 8\n",
-      "20066    | torch.Size([1, 1000, 762]) | 8\n",
-      "20067    | torch.Size([1, 1000, 762]) | 8\n",
-      "20068    | torch.Size([1, 1000, 771]) | 8\n",
-      "20069    | torch.Size([1, 1000, 762]) | 8\n",
-      "20070    | torch.Size([1, 1000, 789]) | 8\n",
-      "20071    | torch.Size([1, 1000, 762]) | 8\n",
-      "20072    | torch.Size([1, 1000, 762]) | 8\n",
-      "20073    | torch.Size([1, 1000, 797]) | 8\n",
-      "20074    | torch.Size([1, 1000, 754]) | 8\n",
-      "20075    | torch.Size([1, 1000, 840]) | 8\n",
-      "20076    | torch.Size([1, 1000, 754]) | 8\n",
-      "20077    | torch.Size([1, 1000, 762]) | 8\n",
-      "20078    | torch.Size([1, 1000, 754]) | 8\n",
-      "20079    | torch.Size([1, 1000, 762]) | 8\n",
-      "20080    | torch.Size([1, 1000, 789]) | 8\n",
-      "20081    | torch.Size([1, 1000, 764]) | 8\n",
-      "20082    | torch.Size([1, 1000, 762]) | 8\n",
-      "20083    | torch.Size([1, 1000, 754]) | 8\n",
-      "20084    | torch.Size([1, 1000, 762]) | 8\n",
-      "20085    | torch.Size([1, 1000, 754]) | 8\n",
-      "20086    | torch.Size([1, 1000, 762]) | 8\n",
-      "20087    | torch.Size([1, 1000, 754]) | 8\n",
-      "20088    | torch.Size([1, 1000, 771]) | 8\n",
-      "20089    | torch.Size([1, 1000, 762]) | 8\n",
-      "20090    | torch.Size([1, 1000, 762]) | 8\n",
-      "20091    | torch.Size([1, 1000, 762]) | 8\n",
-      "20092    | torch.Size([1, 1000, 762]) | 8\n",
-      "20093    | torch.Size([1, 1000, 777]) | 8\n",
-      "20094    | torch.Size([1, 1000, 785]) | 8\n",
-      "20095    | torch.Size([1, 1000, 777]) | 8\n",
-      "20096    | torch.Size([1, 1000, 762]) | 8\n",
-      "20097    | torch.Size([1, 1000, 754]) | 8\n",
-      "20098    | torch.Size([1, 1000, 762]) | 8\n",
-      "20099    | torch.Size([1, 1000, 754]) | 8\n",
-      "20100    | torch.Size([1, 1000, 777]) | 8\n",
-      "20101    | torch.Size([1, 1000, 754]) | 8\n",
-      "20102    | torch.Size([1, 1000, 762]) | 8\n",
-      "20103    | torch.Size([1, 1000, 762]) | 8\n",
-      "20104    | torch.Size([1, 1000, 770]) | 8\n",
-      "20105    | torch.Size([1, 1000, 762]) | 8\n",
-      "20106    | torch.Size([1, 1000, 762]) | 8\n",
-      "20107    | torch.Size([1, 1000, 788]) | 8\n",
-      "20108    | torch.Size([1, 1000, 754]) | 8\n",
-      "20109    | torch.Size([1, 1000, 754]) | 8\n",
-      "20110    | torch.Size([1, 1000, 762]) | 8\n",
-      "20111    | torch.Size([1, 1000, 762]) | 8\n",
-      "20112    | torch.Size([1, 1000, 754]) | 8\n",
-      "20113    | torch.Size([1, 1000, 781]) | 8\n",
-      "20114    | torch.Size([1, 1000, 762]) | 8\n",
-      "20115    | torch.Size([1, 1000, 762]) | 8\n",
-      "20116    | torch.Size([1, 1000, 797]) | 8\n",
-      "20117    | torch.Size([1, 1000, 762]) | 8\n",
-      "20118    | torch.Size([1, 1000, 777]) | 8\n",
-      "20119    | torch.Size([1, 1000, 781]) | 8\n",
-      "20120    | torch.Size([1, 1000, 762]) | 8\n",
-      "20121    | torch.Size([1, 1000, 754]) | 8\n",
-      "20122    | torch.Size([1, 1000, 762]) | 8\n",
-      "20123    | torch.Size([1, 1000, 754]) | 8\n",
-      "20124    | torch.Size([1, 1000, 779]) | 8\n",
-      "20125    | torch.Size([1, 1000, 777]) | 8\n",
-      "20126    | torch.Size([1, 1000, 754]) | 8\n",
-      "20127    | torch.Size([1, 1000, 762]) | 8\n",
-      "20128    | torch.Size([1, 1000, 762]) | 8\n",
-      "20129    | torch.Size([1, 1000, 754]) | 8\n",
-      "20130    | torch.Size([1, 1000, 778]) | 8\n",
-      "20131    | torch.Size([1, 1000, 754]) | 8\n",
-      "20132    | torch.Size([1, 1000, 768]) | 8\n",
-      "20133    | torch.Size([1, 1000, 754]) | 8\n",
-      "20134    | torch.Size([1, 1000, 762]) | 8\n",
-      "20135    | torch.Size([1, 1000, 802]) | 8\n",
-      "20136    | torch.Size([1, 1000, 776]) | 8\n",
-      "20137    | torch.Size([1, 1000, 762]) | 8\n",
-      "20138    | torch.Size([1, 1000, 754]) | 8\n",
-      "20139    | torch.Size([1, 1000, 780]) | 8\n",
-      "20140    | torch.Size([1, 1000, 782]) | 8\n",
-      "20141    | torch.Size([1, 1000, 754]) | 8\n",
-      "20142    | torch.Size([1, 1000, 762]) | 8\n",
-      "20143    | torch.Size([1, 1000, 777]) | 8\n",
-      "20144    | torch.Size([1, 1000, 797]) | 8\n",
-      "20145    | torch.Size([1, 1000, 762]) | 8\n",
-      "20146    | torch.Size([1, 1000, 754]) | 8\n",
-      "20147    | torch.Size([1, 1000, 775]) | 8\n",
-      "20148    | torch.Size([1, 1000, 810]) | 8\n",
-      "20149    | torch.Size([1, 1000, 754]) | 8\n",
-      "20150    | torch.Size([1, 1000, 754]) | 8\n",
-      "20151    | torch.Size([1, 1000, 805]) | 8\n",
-      "20152    | torch.Size([1, 1000, 754]) | 8\n",
-      "20153    | torch.Size([1, 1000, 798]) | 8\n",
-      "20154    | torch.Size([1, 1000, 754]) | 8\n",
-      "20155    | torch.Size([1, 1000, 754]) | 8\n",
-      "20156    | torch.Size([1, 1000, 762]) | 8\n",
-      "20157    | torch.Size([1, 1000, 784]) | 8\n",
-      "20158    | torch.Size([1, 1000, 777]) | 8\n",
-      "20159    | torch.Size([1, 1000, 777]) | 8\n",
-      "20160    | torch.Size([1, 1000, 762]) | 8\n",
-      "20161    | torch.Size([1, 1000, 762]) | 8\n",
-      "20162    | torch.Size([1, 1000, 777]) | 8\n",
-      "20163    | torch.Size([1, 1000, 762]) | 8\n",
-      "20164    | torch.Size([1, 1000, 772]) | 8\n",
-      "20165    | torch.Size([1, 1000, 762]) | 8\n",
-      "20166    | torch.Size([1, 1000, 779]) | 8\n",
-      "20167    | torch.Size([1, 1000, 762]) | 8\n",
-      "20168    | torch.Size([1, 1000, 754]) | 8\n",
-      "20169    | torch.Size([1, 1000, 802]) | 8\n",
-      "20170    | torch.Size([1, 1000, 762]) | 8\n",
-      "20171    | torch.Size([1, 1000, 802]) | 8\n",
-      "20172    | torch.Size([1, 1000, 754]) | 8\n",
-      "20173    | torch.Size([1, 1000, 787]) | 8\n",
-      "20174    | torch.Size([1, 1000, 766]) | 8\n",
-      "20175    | torch.Size([1, 1000, 777]) | 8\n",
-      "20176    | torch.Size([1, 1000, 762]) | 8\n",
-      "20177    | torch.Size([1, 1000, 762]) | 8\n",
-      "20178    | torch.Size([1, 1000, 762]) | 8\n",
-      "20179    | torch.Size([1, 1000, 762]) | 8\n",
-      "20180    | torch.Size([1, 1000, 771]) | 8\n",
-      "20181    | torch.Size([1, 1000, 762]) | 8\n",
-      "20182    | torch.Size([1, 1000, 754]) | 8\n",
-      "20183    | torch.Size([1, 1000, 754]) | 8\n",
-      "20184    | torch.Size([1, 1000, 762]) | 8\n",
-      "20185    | torch.Size([1, 1000, 762]) | 8\n",
-      "20186    | torch.Size([1, 1000, 782]) | 8\n",
-      "20187    | torch.Size([1, 1000, 762]) | 8\n",
-      "20188    | torch.Size([1, 1000, 777]) | 8\n",
-      "20189    | torch.Size([1, 1000, 762]) | 8\n",
-      "20190    | torch.Size([1, 1000, 754]) | 8\n",
-      "20191    | torch.Size([1, 1000, 800]) | 8\n",
-      "20192    | torch.Size([1, 1000, 847]) | 8\n",
-      "20193    | torch.Size([1, 1000, 762]) | 8\n",
-      "20194    | torch.Size([1, 1000, 762]) | 8\n",
-      "20195    | torch.Size([1, 1000, 754]) | 8\n",
-      "20196    | torch.Size([1, 1000, 774]) | 8\n",
-      "20197    | torch.Size([1, 1000, 777]) | 8\n",
-      "20198    | torch.Size([1, 1000, 762]) | 8\n",
-      "20199    | torch.Size([1, 1000, 762]) | 8\n",
-      "20200    | torch.Size([1, 1000, 754]) | 8\n",
-      "20201    | torch.Size([1, 1000, 778]) | 8\n",
-      "20202    | torch.Size([1, 1000, 762]) | 8\n",
-      "20203    | torch.Size([1, 1000, 791]) | 8\n",
-      "20204    | torch.Size([1, 1000, 762]) | 8\n",
-      "20205    | torch.Size([1, 1000, 841]) | 8\n",
-      "20206    | torch.Size([1, 1000, 754]) | 8\n",
-      "20207    | torch.Size([1, 1000, 777]) | 8\n",
-      "20208    | torch.Size([1, 1000, 754]) | 8\n",
-      "20209    | torch.Size([1, 1000, 822]) | 8\n",
-      "20210    | torch.Size([1, 1000, 762]) | 8\n",
-      "20211    | torch.Size([1, 1000, 777]) | 8\n",
-      "20212    | torch.Size([1, 1000, 762]) | 8\n",
-      "20213    | torch.Size([1, 1000, 770]) | 8\n",
-      "20214    | torch.Size([1, 1000, 762]) | 8\n",
-      "20215    | torch.Size([1, 1000, 781]) | 8\n",
-      "20216    | torch.Size([1, 1000, 779]) | 8\n",
-      "20217    | torch.Size([1, 1000, 778]) | 8\n",
-      "20218    | torch.Size([1, 1000, 800]) | 8\n",
-      "20219    | torch.Size([1, 1000, 754]) | 8\n",
-      "20220    | torch.Size([1, 1000, 762]) | 8\n",
-      "20221    | torch.Size([1, 1000, 754]) | 8\n",
-      "20222    | torch.Size([1, 1000, 762]) | 8\n",
-      "20223    | torch.Size([1, 1000, 762]) | 8\n",
-      "20224    | torch.Size([1, 1000, 762]) | 8\n",
-      "20225    | torch.Size([1, 1000, 810]) | 8\n",
-      "20226    | torch.Size([1, 1000, 754]) | 8\n",
-      "20227    | torch.Size([1, 1000, 777]) | 8\n",
-      "20228    | torch.Size([1, 1000, 777]) | 8\n",
-      "20229    | torch.Size([1, 1000, 779]) | 8\n",
-      "20230    | torch.Size([1, 1000, 762]) | 8\n",
-      "20231    | torch.Size([1, 1000, 770]) | 8\n",
-      "20232    | torch.Size([1, 1000, 777]) | 8\n",
-      "20233    | torch.Size([1, 1000, 776]) | 8\n",
-      "20234    | torch.Size([1, 1000, 778]) | 8\n",
-      "20235    | torch.Size([1, 1000, 754]) | 8\n",
-      "20236    | torch.Size([1, 1000, 762]) | 8\n",
-      "20237    | torch.Size([1, 1000, 777]) | 8\n",
-      "20238    | torch.Size([1, 1000, 779]) | 8\n",
-      "20239    | torch.Size([1, 1000, 754]) | 8\n",
-      "20240    | torch.Size([1, 1000, 754]) | 8\n",
-      "20241    | torch.Size([1, 1000, 762]) | 8\n",
-      "20242    | torch.Size([1, 1000, 754]) | 8\n",
-      "20243    | torch.Size([1, 1000, 808]) | 8\n",
-      "20244    | torch.Size([1, 1000, 778]) | 8\n",
-      "20245    | torch.Size([1, 1000, 770]) | 8\n",
-      "20246    | torch.Size([1, 1000, 762]) | 8\n",
-      "20247    | torch.Size([1, 1000, 754]) | 8\n",
-      "20248    | torch.Size([1, 1000, 762]) | 8\n",
-      "20249    | torch.Size([1, 1000, 754]) | 8\n",
-      "20250    | torch.Size([1, 1000, 762]) | 8\n",
-      "20251    | torch.Size([1, 1000, 754]) | 8\n",
-      "20252    | torch.Size([1, 1000, 754]) | 8\n",
-      "20253    | torch.Size([1, 1000, 780]) | 8\n",
-      "20254    | torch.Size([1, 1000, 781]) | 8\n",
-      "20255    | torch.Size([1, 1000, 802]) | 8\n",
-      "20256    | torch.Size([1, 1000, 777]) | 8\n",
-      "20257    | torch.Size([1, 1000, 762]) | 8\n",
-      "20258    | torch.Size([1, 1000, 848]) | 8\n",
-      "20259    | torch.Size([1, 1000, 762]) | 8\n",
-      "20260    | torch.Size([1, 1000, 786]) | 8\n",
-      "20261    | torch.Size([1, 1000, 754]) | 8\n",
-      "20262    | torch.Size([1, 1000, 777]) | 8\n",
-      "20263    | torch.Size([1, 1000, 809]) | 8\n",
-      "20264    | torch.Size([1, 1000, 754]) | 8\n",
-      "20265    | torch.Size([1, 1000, 779]) | 8\n",
-      "20266    | torch.Size([1, 1000, 781]) | 8\n",
-      "20267    | torch.Size([1, 1000, 778]) | 8\n",
-      "20268    | torch.Size([1, 1000, 777]) | 8\n",
-      "20269    | torch.Size([1, 1000, 777]) | 8\n",
-      "20270    | torch.Size([1, 1000, 754]) | 8\n",
-      "20271    | torch.Size([1, 1000, 762]) | 8\n",
-      "20272    | torch.Size([1, 1000, 777]) | 8\n",
-      "20273    | torch.Size([1, 1000, 754]) | 8\n",
-      "20274    | torch.Size([1, 1000, 843]) | 8\n",
-      "20275    | torch.Size([1, 1000, 754]) | 8\n",
-      "20276    | torch.Size([1, 1000, 777]) | 8\n",
-      "20277    | torch.Size([1, 1000, 777]) | 8\n",
-      "20278    | torch.Size([1, 1000, 762]) | 8\n",
-      "20279    | torch.Size([1, 1000, 762]) | 8\n",
-      "20280    | torch.Size([1, 1000, 762]) | 8\n",
-      "20281    | torch.Size([1, 1000, 784]) | 8\n",
-      "20282    | torch.Size([1, 1000, 774]) | 8\n",
-      "20283    | torch.Size([1, 1000, 784]) | 8\n",
-      "20284    | torch.Size([1, 1000, 754]) | 8\n",
-      "20285    | torch.Size([1, 1000, 771]) | 8\n",
-      "20286    | torch.Size([1, 1000, 777]) | 8\n",
-      "20287    | torch.Size([1, 1000, 754]) | 8\n",
-      "20288    | torch.Size([1, 1000, 754]) | 8\n",
-      "20289    | torch.Size([1, 1000, 762]) | 8\n",
-      "20290    | torch.Size([1, 1000, 796]) | 8\n",
-      "20291    | torch.Size([1, 1000, 762]) | 8\n",
-      "20292    | torch.Size([1, 1000, 754]) | 8\n",
-      "20293    | torch.Size([1, 1000, 754]) | 8\n",
-      "20294    | torch.Size([1, 1000, 754]) | 8\n",
-      "20295    | torch.Size([1, 1000, 786]) | 8\n",
-      "20296    | torch.Size([1, 1000, 787]) | 8\n",
-      "20297    | torch.Size([1, 1000, 764]) | 8\n",
-      "20298    | torch.Size([1, 1000, 777]) | 8\n",
-      "20299    | torch.Size([1, 1000, 777]) | 8\n",
-      "20300    | torch.Size([1, 1000, 802]) | 8\n",
-      "20301    | torch.Size([1, 1000, 793]) | 8\n",
-      "20302    | torch.Size([1, 1000, 777]) | 8\n",
-      "20303    | torch.Size([1, 1000, 762]) | 8\n",
-      "20304    | torch.Size([1, 1000, 778]) | 8\n",
-      "20305    | torch.Size([1, 1000, 770]) | 8\n",
-      "20306    | torch.Size([1, 1000, 777]) | 8\n",
-      "20307    | torch.Size([1, 1000, 777]) | 8\n",
-      "20308    | torch.Size([1, 1000, 777]) | 8\n",
-      "20309    | torch.Size([1, 1000, 762]) | 8\n",
-      "20310    | torch.Size([1, 1000, 802]) | 8\n",
-      "20311    | torch.Size([1, 1000, 762]) | 8\n",
-      "20312    | torch.Size([1, 1000, 777]) | 8\n",
-      "20313    | torch.Size([1, 1000, 754]) | 8\n",
-      "20314    | torch.Size([1, 1000, 762]) | 8\n",
-      "20315    | torch.Size([1, 1000, 762]) | 8\n",
-      "20316    | torch.Size([1, 1000, 762]) | 8\n",
-      "20317    | torch.Size([1, 1000, 754]) | 8\n",
-      "20318    | torch.Size([1, 1000, 754]) | 8\n",
-      "20319    | torch.Size([1, 1000, 754]) | 8\n",
-      "20320    | torch.Size([1, 1000, 762]) | 8\n",
-      "20321    | torch.Size([1, 1000, 762]) | 8\n",
-      "20322    | torch.Size([1, 1000, 762]) | 8\n",
-      "20323    | torch.Size([1, 1000, 762]) | 8\n",
-      "20324    | torch.Size([1, 1000, 762]) | 8\n",
-      "20325    | torch.Size([1, 1000, 778]) | 8\n",
-      "20326    | torch.Size([1, 1000, 776]) | 8\n",
-      "20327    | torch.Size([1, 1000, 777]) | 8\n",
-      "20328    | torch.Size([1, 1000, 754]) | 8\n",
-      "20329    | torch.Size([1, 1000, 754]) | 8\n",
-      "20330    | torch.Size([1, 1000, 769]) | 8\n",
-      "20331    | torch.Size([1, 1000, 762]) | 8\n",
-      "20332    | torch.Size([1, 1000, 762]) | 8\n",
-      "20333    | torch.Size([1, 1000, 770]) | 8\n",
-      "20334    | torch.Size([1, 1000, 771]) | 8\n",
-      "20335    | torch.Size([1, 1000, 779]) | 8\n",
-      "20336    | torch.Size([1, 1000, 762]) | 8\n",
-      "20337    | torch.Size([1, 1000, 770]) | 8\n",
-      "20338    | torch.Size([1, 1000, 754]) | 8\n",
-      "20339    | torch.Size([1, 1000, 754]) | 8\n",
-      "20340    | torch.Size([1, 1000, 777]) | 8\n",
-      "20341    | torch.Size([1, 1000, 777]) | 8\n",
-      "20342    | torch.Size([1, 1000, 754]) | 8\n",
-      "20343    | torch.Size([1, 1000, 777]) | 8\n",
-      "20344    | torch.Size([1, 1000, 777]) | 8\n",
-      "20345    | torch.Size([1, 1000, 762]) | 8\n",
-      "20346    | torch.Size([1, 1000, 777]) | 8\n",
-      "20347    | torch.Size([1, 1000, 772]) | 8\n",
-      "20348    | torch.Size([1, 1000, 777]) | 8\n",
-      "20349    | torch.Size([1, 1000, 762]) | 8\n",
-      "20350    | torch.Size([1, 1000, 775]) | 8\n",
-      "20351    | torch.Size([1, 1000, 777]) | 8\n",
-      "20352    | torch.Size([1, 1000, 754]) | 8\n",
-      "20353    | torch.Size([1, 1000, 777]) | 8\n",
-      "20354    | torch.Size([1, 1000, 777]) | 8\n",
-      "20355    | torch.Size([1, 1000, 787]) | 8\n",
-      "20356    | torch.Size([1, 1000, 793]) | 8\n",
-      "20357    | torch.Size([1, 1000, 777]) | 8\n",
-      "20358    | torch.Size([1, 1000, 754]) | 8\n",
-      "20359    | torch.Size([1, 1000, 762]) | 8\n",
-      "20360    | torch.Size([1, 1000, 762]) | 8\n",
-      "20361    | torch.Size([1, 1000, 762]) | 8\n",
-      "20362    | torch.Size([1, 1000, 762]) | 8\n",
-      "20363    | torch.Size([1, 1000, 773]) | 8\n",
-      "20364    | torch.Size([1, 1000, 762]) | 8\n",
-      "20365    | torch.Size([1, 1000, 784]) | 8\n",
-      "20366    | torch.Size([1, 1000, 754]) | 8\n",
-      "20367    | torch.Size([1, 1000, 777]) | 8\n",
-      "20368    | torch.Size([1, 1000, 765]) | 8\n",
-      "20369    | torch.Size([1, 1000, 780]) | 8\n",
-      "20370    | torch.Size([1, 1000, 762]) | 8\n",
-      "20371    | torch.Size([1, 1000, 762]) | 8\n",
-      "20372    | torch.Size([1, 1000, 854]) | 8\n",
-      "20373    | torch.Size([1, 1000, 776]) | 8\n",
-      "20374    | torch.Size([1, 1000, 762]) | 8\n",
-      "20375    | torch.Size([1, 1000, 777]) | 8\n",
-      "20376    | torch.Size([1, 1000, 754]) | 8\n",
-      "20377    | torch.Size([1, 1000, 754]) | 8\n",
-      "20378    | torch.Size([1, 1000, 754]) | 8\n",
-      "20379    | torch.Size([1, 1000, 777]) | 8\n",
-      "20380    | torch.Size([1, 1000, 771]) | 8\n",
-      "20381    | torch.Size([1, 1000, 796]) | 8\n",
-      "20382    | torch.Size([1, 1000, 782]) | 8\n",
-      "20383    | torch.Size([1, 1000, 779]) | 8\n",
-      "20384    | torch.Size([1, 1000, 762]) | 8\n",
-      "20385    | torch.Size([1, 1000, 754]) | 8\n",
-      "20386    | torch.Size([1, 1000, 754]) | 8\n",
-      "20387    | torch.Size([1, 1000, 754]) | 8\n",
-      "20388    | torch.Size([1, 1000, 762]) | 8\n",
-      "20389    | torch.Size([1, 1000, 847]) | 8\n",
-      "20390    | torch.Size([1, 1000, 754]) | 8\n",
-      "20391    | torch.Size([1, 1000, 777]) | 8\n",
-      "20392    | torch.Size([1, 1000, 772]) | 8\n",
-      "20393    | torch.Size([1, 1000, 772]) | 8\n",
-      "20394    | torch.Size([1, 1000, 762]) | 8\n",
-      "20395    | torch.Size([1, 1000, 754]) | 8\n",
-      "20396    | torch.Size([1, 1000, 762]) | 8\n",
-      "20397    | torch.Size([1, 1000, 762]) | 8\n",
-      "20398    | torch.Size([1, 1000, 772]) | 8\n",
-      "20399    | torch.Size([1, 1000, 777]) | 8\n",
-      "20400    | torch.Size([1, 1000, 777]) | 8\n",
-      "20401    | torch.Size([1, 1000, 754]) | 8\n",
-      "20402    | torch.Size([1, 1000, 762]) | 8\n",
-      "20403    | torch.Size([1, 1000, 762]) | 8\n",
-      "20404    | torch.Size([1, 1000, 762]) | 8\n",
-      "20405    | torch.Size([1, 1000, 842]) | 8\n",
-      "20406    | torch.Size([1, 1000, 776]) | 8\n",
-      "20407    | torch.Size([1, 1000, 754]) | 8\n",
-      "20408    | torch.Size([1, 1000, 773]) | 8\n",
-      "20409    | torch.Size([1, 1000, 781]) | 8\n",
-      "20410    | torch.Size([1, 1000, 777]) | 8\n",
-      "20411    | torch.Size([1, 1000, 754]) | 8\n",
-      "20412    | torch.Size([1, 1000, 762]) | 8\n",
-      "20413    | torch.Size([1, 1000, 762]) | 8\n",
-      "20414    | torch.Size([1, 1000, 777]) | 8\n",
-      "20415    | torch.Size([1, 1000, 754]) | 8\n",
-      "20416    | torch.Size([1, 1000, 762]) | 8\n",
-      "20417    | torch.Size([1, 1000, 810]) | 8\n",
-      "20418    | torch.Size([1, 1000, 762]) | 8\n",
-      "20419    | torch.Size([1, 1000, 777]) | 8\n",
-      "20420    | torch.Size([1, 1000, 762]) | 8\n",
-      "20421    | torch.Size([1, 1000, 777]) | 8\n",
-      "20422    | torch.Size([1, 1000, 779]) | 8\n",
-      "20423    | torch.Size([1, 1000, 754]) | 8\n",
-      "20424    | torch.Size([1, 1000, 783]) | 8\n",
-      "20425    | torch.Size([1, 1000, 773]) | 8\n",
-      "20426    | torch.Size([1, 1000, 754]) | 8\n",
-      "20427    | torch.Size([1, 1000, 754]) | 8\n",
-      "20428    | torch.Size([1, 1000, 762]) | 8\n",
-      "20429    | torch.Size([1, 1000, 754]) | 8\n",
-      "20430    | torch.Size([1, 1000, 754]) | 8\n",
-      "20431    | torch.Size([1, 1000, 754]) | 8\n",
-      "20432    | torch.Size([1, 1000, 833]) | 8\n",
-      "20433    | torch.Size([1, 1000, 784]) | 8\n",
-      "20434    | torch.Size([1, 1000, 762]) | 8\n",
-      "20435    | torch.Size([1, 1000, 754]) | 8\n",
-      "20436    | torch.Size([1, 1000, 762]) | 8\n",
-      "20437    | torch.Size([1, 1000, 777]) | 8\n",
-      "20438    | torch.Size([1, 1000, 762]) | 8\n",
-      "20439    | torch.Size([1, 1000, 787]) | 8\n",
-      "20440    | torch.Size([1, 1000, 772]) | 8\n",
-      "20441    | torch.Size([1, 1000, 783]) | 8\n",
-      "20442    | torch.Size([1, 1000, 777]) | 8\n",
-      "20443    | torch.Size([1, 1000, 777]) | 8\n",
-      "20444    | torch.Size([1, 1000, 777]) | 8\n",
-      "20445    | torch.Size([1, 1000, 754]) | 8\n",
-      "20446    | torch.Size([1, 1000, 774]) | 8\n",
-      "20447    | torch.Size([1, 1000, 754]) | 8\n",
-      "20448    | torch.Size([1, 1000, 780]) | 8\n",
-      "20449    | torch.Size([1, 1000, 754]) | 8\n",
-      "20450    | torch.Size([1, 1000, 762]) | 8\n",
-      "20451    | torch.Size([1, 1000, 797]) | 8\n",
-      "20452    | torch.Size([1, 1000, 777]) | 8\n",
-      "20453    | torch.Size([1, 1000, 754]) | 8\n",
-      "20454    | torch.Size([1, 1000, 762]) | 8\n",
-      "20455    | torch.Size([1, 1000, 785]) | 8\n",
-      "20456    | torch.Size([1, 1000, 802]) | 8\n",
-      "20457    | torch.Size([1, 1000, 777]) | 8\n",
-      "20458    | torch.Size([1, 1000, 777]) | 8\n",
-      "20459    | torch.Size([1, 1000, 762]) | 8\n",
-      "20460    | torch.Size([1, 1000, 777]) | 8\n",
-      "20461    | torch.Size([1, 1000, 777]) | 8\n",
-      "20462    | torch.Size([1, 1000, 762]) | 8\n",
-      "20463    | torch.Size([1, 1000, 754]) | 8\n",
-      "20464    | torch.Size([1, 1000, 771]) | 8\n",
-      "20465    | torch.Size([1, 1000, 846]) | 8\n",
-      "20466    | torch.Size([1, 1000, 754]) | 8\n",
-      "20467    | torch.Size([1, 1000, 754]) | 8\n",
-      "20468    | torch.Size([1, 1000, 731]) | 8\n",
-      "20469    | torch.Size([1, 1000, 779]) | 8\n",
-      "20470    | torch.Size([1, 1000, 777]) | 8\n",
-      "20471    | torch.Size([1, 1000, 763]) | 8\n",
-      "20472    | torch.Size([1, 1000, 754]) | 8\n",
-      "20473    | torch.Size([1, 1000, 777]) | 8\n",
-      "20474    | torch.Size([1, 1000, 762]) | 8\n",
-      "20475    | torch.Size([1, 1000, 762]) | 8\n",
-      "20476    | torch.Size([1, 1000, 777]) | 8\n",
-      "20477    | torch.Size([1, 1000, 762]) | 8\n",
-      "20478    | torch.Size([1, 1000, 777]) | 8\n",
-      "20479    | torch.Size([1, 1000, 762]) | 8\n",
-      "20480    | torch.Size([1, 1000, 766]) | 8\n",
-      "20481    | torch.Size([1, 1000, 762]) | 8\n",
-      "20482    | torch.Size([1, 1000, 762]) | 8\n",
-      "20483    | torch.Size([1, 1000, 794]) | 8\n",
-      "20484    | torch.Size([1, 1000, 762]) | 8\n",
-      "20485    | torch.Size([1, 1000, 762]) | 8\n",
-      "20486    | torch.Size([1, 1000, 777]) | 8\n",
-      "20487    | torch.Size([1, 1000, 754]) | 8\n",
-      "20488    | torch.Size([1, 1000, 809]) | 8\n",
-      "20489    | torch.Size([1, 1000, 777]) | 8\n",
-      "20490    | torch.Size([1, 1000, 761]) | 8\n",
-      "20491    | torch.Size([1, 1000, 732]) | 8\n",
-      "20492    | torch.Size([1, 1000, 754]) | 8\n",
-      "20493    | torch.Size([1, 1000, 754]) | 8\n",
-      "20494    | torch.Size([1, 1000, 762]) | 8\n",
-      "20495    | torch.Size([1, 1000, 773]) | 8\n",
-      "20496    | torch.Size([1, 1000, 754]) | 8\n",
-      "20497    | torch.Size([1, 1000, 802]) | 8\n",
-      "20498    | torch.Size([1, 1000, 778]) | 8\n",
-      "20499    | torch.Size([1, 1000, 762]) | 8\n",
-      "20500    | torch.Size([1, 1000, 754]) | 8\n",
-      "20501    | torch.Size([1, 1000, 795]) | 8\n",
-      "20502    | torch.Size([1, 1000, 762]) | 8\n",
-      "20503    | torch.Size([1, 1000, 777]) | 8\n",
-      "20504    | torch.Size([1, 1000, 844]) | 8\n",
-      "20505    | torch.Size([1, 1000, 762]) | 8\n",
-      "20506    | torch.Size([1, 1000, 775]) | 8\n",
-      "20507    | torch.Size([1, 1000, 762]) | 8\n",
-      "20508    | torch.Size([1, 1000, 841]) | 8\n",
-      "20509    | torch.Size([1, 1000, 777]) | 8\n",
-      "20510    | torch.Size([1, 1000, 762]) | 8\n",
-      "20511    | torch.Size([1, 1000, 777]) | 8\n",
-      "20512    | torch.Size([1, 1000, 777]) | 8\n",
-      "20513    | torch.Size([1, 1000, 754]) | 8\n",
-      "20514    | torch.Size([1, 1000, 777]) | 8\n",
-      "20515    | torch.Size([1, 1000, 762]) | 8\n",
-      "20516    | torch.Size([1, 1000, 782]) | 8\n",
-      "20517    | torch.Size([1, 1000, 777]) | 8\n",
-      "20518    | torch.Size([1, 1000, 831]) | 8\n",
-      "20519    | torch.Size([1, 1000, 802]) | 8\n",
-      "20520    | torch.Size([1, 1000, 762]) | 8\n",
-      "20521    | torch.Size([1, 1000, 762]) | 8\n",
-      "20522    | torch.Size([1, 1000, 808]) | 8\n",
-      "20523    | torch.Size([1, 1000, 762]) | 8\n",
-      "20524    | torch.Size([1, 1000, 762]) | 8\n",
-      "20525    | torch.Size([1, 1000, 762]) | 8\n",
-      "20526    | torch.Size([1, 1000, 762]) | 8\n",
-      "20527    | torch.Size([1, 1000, 762]) | 8\n",
-      "20528    | torch.Size([1, 1000, 762]) | 8\n",
-      "20529    | torch.Size([1, 1000, 762]) | 8\n",
-      "20530    | torch.Size([1, 1000, 762]) | 8\n",
-      "20531    | torch.Size([1, 1000, 777]) | 8\n",
-      "20532    | torch.Size([1, 1000, 754]) | 8\n",
-      "20533    | torch.Size([1, 1000, 754]) | 8\n",
-      "20534    | torch.Size([1, 1000, 768]) | 8\n",
-      "20535    | torch.Size([1, 1000, 773]) | 8\n",
-      "20536    | torch.Size([1, 1000, 762]) | 8\n",
-      "20537    | torch.Size([1, 1000, 754]) | 8\n",
-      "20538    | torch.Size([1, 1000, 754]) | 8\n",
-      "20539    | torch.Size([1, 1000, 762]) | 8\n",
-      "20540    | torch.Size([1, 1000, 754]) | 8\n",
-      "20541    | torch.Size([1, 1000, 778]) | 8\n",
-      "20542    | torch.Size([1, 1000, 762]) | 8\n",
-      "20543    | torch.Size([1, 1000, 797]) | 8\n",
-      "20544    | torch.Size([1, 1000, 754]) | 8\n",
-      "20545    | torch.Size([1, 1000, 762]) | 8\n",
-      "20546    | torch.Size([1, 1000, 754]) | 8\n",
-      "20547    | torch.Size([1, 1000, 762]) | 8\n",
-      "20548    | torch.Size([1, 1000, 762]) | 8\n",
-      "20549    | torch.Size([1, 1000, 762]) | 8\n",
-      "20550    | torch.Size([1, 1000, 762]) | 8\n",
-      "20551    | torch.Size([1, 1000, 762]) | 8\n",
-      "20552    | torch.Size([1, 1000, 762]) | 8\n",
-      "20553    | torch.Size([1, 1000, 762]) | 8\n",
-      "20554    | torch.Size([1, 1000, 754]) | 8\n",
-      "20555    | torch.Size([1, 1000, 762]) | 8\n",
-      "20556    | torch.Size([1, 1000, 762]) | 8\n",
-      "20557    | torch.Size([1, 1000, 762]) | 8\n",
-      "20558    | torch.Size([1, 1000, 777]) | 8\n",
-      "20559    | torch.Size([1, 1000, 754]) | 8\n",
-      "20560    | torch.Size([1, 1000, 777]) | 8\n",
-      "20561    | torch.Size([1, 1000, 768]) | 8\n",
-      "20562    | torch.Size([1, 1000, 754]) | 8\n",
-      "20563    | torch.Size([1, 1000, 798]) | 8\n",
-      "20564    | torch.Size([1, 1000, 762]) | 8\n",
-      "20565    | torch.Size([1, 1000, 754]) | 8\n",
-      "20566    | torch.Size([1, 1000, 754]) | 8\n",
-      "20567    | torch.Size([1, 1000, 762]) | 8\n",
-      "20568    | torch.Size([1, 1000, 762]) | 8\n",
-      "20569    | torch.Size([1, 1000, 754]) | 8\n",
-      "20570    | torch.Size([1, 1000, 777]) | 8\n",
-      "20571    | torch.Size([1, 1000, 762]) | 8\n",
-      "20572    | torch.Size([1, 1000, 762]) | 8\n",
-      "20573    | torch.Size([1, 1000, 778]) | 8\n",
-      "20574    | torch.Size([1, 1000, 777]) | 8\n",
-      "20575    | torch.Size([1, 1000, 777]) | 8\n",
-      "20576    | torch.Size([1, 1000, 762]) | 8\n",
-      "20577    | torch.Size([1, 1000, 768]) | 8\n",
-      "20578    | torch.Size([1, 1000, 802]) | 8\n",
-      "20579    | torch.Size([1, 1000, 777]) | 8\n",
-      "20580    | torch.Size([1, 1000, 762]) | 8\n",
-      "20581    | torch.Size([1, 1000, 754]) | 8\n",
-      "20582    | torch.Size([1, 1000, 772]) | 8\n",
-      "20583    | torch.Size([1, 1000, 777]) | 8\n",
-      "20584    | torch.Size([1, 1000, 777]) | 8\n",
-      "20585    | torch.Size([1, 1000, 754]) | 8\n",
-      "20586    | torch.Size([1, 1000, 777]) | 8\n",
-      "20587    | torch.Size([1, 1000, 754]) | 8\n",
-      "20588    | torch.Size([1, 1000, 754]) | 8\n",
-      "20589    | torch.Size([1, 1000, 754]) | 8\n",
-      "20590    | torch.Size([1, 1000, 762]) | 8\n",
-      "20591    | torch.Size([1, 1000, 754]) | 8\n",
-      "20592    | torch.Size([1, 1000, 762]) | 8\n",
-      "20593    | torch.Size([1, 1000, 754]) | 8\n",
-      "20594    | torch.Size([1, 1000, 762]) | 8\n",
-      "20595    | torch.Size([1, 1000, 802]) | 8\n",
-      "20596    | torch.Size([1, 1000, 762]) | 8\n",
-      "20597    | torch.Size([1, 1000, 754]) | 8\n",
-      "20598    | torch.Size([1, 1000, 754]) | 8\n",
-      "20599    | torch.Size([1, 1000, 754]) | 8\n",
-      "20600    | torch.Size([1, 1000, 754]) | 8\n",
-      "20601    | torch.Size([1, 1000, 762]) | 8\n",
-      "20602    | torch.Size([1, 1000, 777]) | 8\n",
-      "20603    | torch.Size([1, 1000, 788]) | 8\n",
-      "20604    | torch.Size([1, 1000, 754]) | 8\n",
-      "20605    | torch.Size([1, 1000, 754]) | 8\n",
-      "20606    | torch.Size([1, 1000, 771]) | 8\n",
-      "20607    | torch.Size([1, 1000, 754]) | 8\n",
-      "20608    | torch.Size([1, 1000, 777]) | 8\n",
-      "20609    | torch.Size([1, 1000, 754]) | 8\n",
-      "20610    | torch.Size([1, 1000, 777]) | 8\n",
-      "20611    | torch.Size([1, 1000, 774]) | 8\n",
-      "20612    | torch.Size([1, 1000, 762]) | 8\n",
-      "20613    | torch.Size([1, 1000, 779]) | 8\n",
-      "20614    | torch.Size([1, 1000, 754]) | 8\n",
-      "20615    | torch.Size([1, 1000, 754]) | 8\n",
-      "20616    | torch.Size([1, 1000, 754]) | 8\n",
-      "20617    | torch.Size([1, 1000, 777]) | 8\n",
-      "20618    | torch.Size([1, 1000, 803]) | 8\n",
-      "20619    | torch.Size([1, 1000, 775]) | 8\n",
-      "20620    | torch.Size([1, 1000, 777]) | 8\n",
-      "20621    | torch.Size([1, 1000, 762]) | 8\n",
-      "20622    | torch.Size([1, 1000, 774]) | 8\n",
-      "20623    | torch.Size([1, 1000, 777]) | 8\n",
-      "20624    | torch.Size([1, 1000, 811]) | 8\n",
-      "20625    | torch.Size([1, 1000, 754]) | 8\n",
-      "20626    | torch.Size([1, 1000, 777]) | 8\n",
-      "20627    | torch.Size([1, 1000, 754]) | 8\n",
-      "20628    | torch.Size([1, 1000, 776]) | 8\n",
-      "20629    | torch.Size([1, 1000, 810]) | 8\n",
-      "20630    | torch.Size([1, 1000, 762]) | 8\n",
-      "20631    | torch.Size([1, 1000, 775]) | 8\n",
-      "20632    | torch.Size([1, 1000, 777]) | 8\n",
-      "20633    | torch.Size([1, 1000, 769]) | 8\n",
-      "20634    | torch.Size([1, 1000, 777]) | 8\n",
-      "20635    | torch.Size([1, 1000, 762]) | 8\n",
-      "20636    | torch.Size([1, 1000, 762]) | 8\n",
-      "20637    | torch.Size([1, 1000, 777]) | 8\n",
-      "20638    | torch.Size([1, 1000, 754]) | 8\n",
-      "20639    | torch.Size([1, 1000, 777]) | 8\n",
-      "20640    | torch.Size([1, 1000, 777]) | 8\n",
-      "20641    | torch.Size([1, 1000, 772]) | 8\n",
-      "20642    | torch.Size([1, 1000, 754]) | 8\n",
-      "20643    | torch.Size([1, 1000, 762]) | 8\n",
-      "20644    | torch.Size([1, 1000, 762]) | 8\n",
-      "20645    | torch.Size([1, 1000, 762]) | 8\n",
-      "20646    | torch.Size([1, 1000, 777]) | 8\n",
-      "20647    | torch.Size([1, 1000, 754]) | 8\n",
-      "20648    | torch.Size([1, 1000, 754]) | 8\n",
-      "20649    | torch.Size([1, 1000, 762]) | 8\n",
-      "20650    | torch.Size([1, 1000, 754]) | 8\n",
-      "20651    | torch.Size([1, 1000, 762]) | 8\n",
-      "20652    | torch.Size([1, 1000, 788]) | 8\n",
-      "20653    | torch.Size([1, 1000, 762]) | 8\n",
-      "20654    | torch.Size([1, 1000, 762]) | 8\n",
-      "20655    | torch.Size([1, 1000, 754]) | 8\n",
-      "20656    | torch.Size([1, 1000, 762]) | 8\n",
-      "20657    | torch.Size([1, 1000, 754]) | 8\n",
-      "20658    | torch.Size([1, 1000, 762]) | 8\n",
-      "20659    | torch.Size([1, 1000, 754]) | 8\n",
-      "20660    | torch.Size([1, 1000, 762]) | 8\n",
-      "20661    | torch.Size([1, 1000, 754]) | 8\n",
-      "20662    | torch.Size([1, 1000, 780]) | 8\n",
-      "20663    | torch.Size([1, 1000, 795]) | 8\n",
-      "20664    | torch.Size([1, 1000, 777]) | 8\n",
-      "20665    | torch.Size([1, 1000, 762]) | 8\n",
-      "20666    | torch.Size([1, 1000, 762]) | 8\n",
-      "20667    | torch.Size([1, 1000, 784]) | 8\n",
-      "20668    | torch.Size([1, 1000, 827]) | 8\n",
-      "20669    | torch.Size([1, 1000, 762]) | 8\n",
-      "20670    | torch.Size([1, 1000, 754]) | 8\n",
-      "20671    | torch.Size([1, 1000, 762]) | 8\n",
-      "20672    | torch.Size([1, 1000, 786]) | 8\n",
-      "20673    | torch.Size([1, 1000, 752]) | 8\n",
-      "20674    | torch.Size([1, 1000, 754]) | 8\n",
-      "20675    | torch.Size([1, 1000, 777]) | 8\n",
-      "20676    | torch.Size([1, 1000, 754]) | 8\n",
-      "20677    | torch.Size([1, 1000, 754]) | 8\n",
-      "20678    | torch.Size([1, 1000, 659]) | 8\n",
-      "20679    | torch.Size([1, 1000, 754]) | 8\n",
-      "20680    | torch.Size([1, 1000, 773]) | 8\n",
-      "20681    | torch.Size([1, 1000, 754]) | 8\n",
-      "20682    | torch.Size([1, 1000, 754]) | 8\n",
-      "20683    | torch.Size([1, 1000, 754]) | 8\n",
-      "20684    | torch.Size([1, 1000, 762]) | 8\n",
-      "20685    | torch.Size([1, 1000, 762]) | 8\n",
-      "20686    | torch.Size([1, 1000, 754]) | 8\n",
-      "20687    | torch.Size([1, 1000, 762]) | 8\n",
-      "20688    | torch.Size([1, 1000, 777]) | 8\n",
-      "20689    | torch.Size([1, 1000, 762]) | 8\n",
-      "20690    | torch.Size([1, 1000, 762]) | 8\n",
-      "20691    | torch.Size([1, 1000, 762]) | 8\n",
-      "20692    | torch.Size([1, 1000, 777]) | 8\n",
-      "20693    | torch.Size([1, 1000, 754]) | 8\n",
-      "20694    | torch.Size([1, 1000, 754]) | 8\n",
-      "20695    | torch.Size([1, 1000, 762]) | 8\n",
-      "20696    | torch.Size([1, 1000, 762]) | 8\n",
-      "20697    | torch.Size([1, 1000, 754]) | 8\n",
-      "20698    | torch.Size([1, 1000, 762]) | 8\n",
-      "20699    | torch.Size([1, 1000, 762]) | 8\n",
-      "20700    | torch.Size([1, 1000, 774]) | 8\n",
-      "20701    | torch.Size([1, 1000, 754]) | 8\n",
-      "20702    | torch.Size([1, 1000, 754]) | 8\n",
-      "20703    | torch.Size([1, 1000, 777]) | 8\n",
-      "20704    | torch.Size([1, 1000, 802]) | 8\n",
-      "20705    | torch.Size([1, 1000, 754]) | 8\n",
-      "20706    | torch.Size([1, 1000, 808]) | 8\n",
-      "20707    | torch.Size([1, 1000, 762]) | 8\n",
-      "20708    | torch.Size([1, 1000, 777]) | 8\n",
-      "20709    | torch.Size([1, 1000, 762]) | 8\n",
-      "20710    | torch.Size([1, 1000, 762]) | 8\n",
-      "20711    | torch.Size([1, 1000, 762]) | 8\n",
-      "20712    | torch.Size([1, 1000, 762]) | 8\n",
-      "20713    | torch.Size([1, 1000, 782]) | 8\n",
-      "20714    | torch.Size([1, 1000, 754]) | 8\n",
-      "20715    | torch.Size([1, 1000, 762]) | 8\n",
-      "20716    | torch.Size([1, 1000, 777]) | 8\n",
-      "20717    | torch.Size([1, 1000, 773]) | 8\n",
-      "20718    | torch.Size([1, 1000, 777]) | 8\n",
-      "20719    | torch.Size([1, 1000, 762]) | 8\n",
-      "20720    | torch.Size([1, 1000, 754]) | 8\n",
-      "20721    | torch.Size([1, 1000, 754]) | 8\n",
-      "20722    | torch.Size([1, 1000, 777]) | 8\n",
-      "20723    | torch.Size([1, 1000, 762]) | 8\n",
-      "20724    | torch.Size([1, 1000, 777]) | 8\n",
-      "20725    | torch.Size([1, 1000, 777]) | 8\n",
-      "20726    | torch.Size([1, 1000, 762]) | 8\n",
-      "20727    | torch.Size([1, 1000, 762]) | 8\n",
-      "20728    | torch.Size([1, 1000, 754]) | 8\n",
-      "20729    | torch.Size([1, 1000, 762]) | 8\n",
-      "20730    | torch.Size([1, 1000, 780]) | 8\n",
-      "20731    | torch.Size([1, 1000, 754]) | 8\n",
-      "20732    | torch.Size([1, 1000, 763]) | 8\n",
-      "20733    | torch.Size([1, 1000, 754]) | 8\n",
-      "20734    | torch.Size([1, 1000, 777]) | 8\n",
-      "20735    | torch.Size([1, 1000, 754]) | 8\n",
-      "20736    | torch.Size([1, 1000, 754]) | 8\n",
-      "20737    | torch.Size([1, 1000, 762]) | 8\n",
-      "20738    | torch.Size([1, 1000, 762]) | 8\n",
-      "20739    | torch.Size([1, 1000, 762]) | 8\n",
-      "20740    | torch.Size([1, 1000, 777]) | 8\n",
-      "20741    | torch.Size([1, 1000, 802]) | 8\n",
-      "20742    | torch.Size([1, 1000, 772]) | 8\n",
-      "20743    | torch.Size([1, 1000, 762]) | 8\n",
-      "20744    | torch.Size([1, 1000, 777]) | 8\n",
-      "20745    | torch.Size([1, 1000, 754]) | 8\n",
-      "20746    | torch.Size([1, 1000, 777]) | 8\n",
-      "20747    | torch.Size([1, 1000, 762]) | 8\n",
-      "20748    | torch.Size([1, 1000, 780]) | 8\n",
-      "20749    | torch.Size([1, 1000, 762]) | 8\n",
-      "20750    | torch.Size([1, 1000, 777]) | 8\n",
-      "20751    | torch.Size([1, 1000, 777]) | 8\n",
-      "20752    | torch.Size([1, 1000, 762]) | 8\n",
-      "20753    | torch.Size([1, 1000, 762]) | 8\n",
-      "20754    | torch.Size([1, 1000, 777]) | 8\n",
-      "20755    | torch.Size([1, 1000, 762]) | 8\n",
-      "20756    | torch.Size([1, 1000, 756]) | 8\n",
-      "20757    | torch.Size([1, 1000, 754]) | 8\n",
-      "20758    | torch.Size([1, 1000, 774]) | 8\n",
-      "20759    | torch.Size([1, 1000, 754]) | 8\n",
-      "20760    | torch.Size([1, 1000, 762]) | 8\n",
-      "20761    | torch.Size([1, 1000, 754]) | 8\n",
-      "20762    | torch.Size([1, 1000, 777]) | 8\n",
-      "20763    | torch.Size([1, 1000, 777]) | 8\n",
-      "20764    | torch.Size([1, 1000, 771]) | 8\n",
-      "20765    | torch.Size([1, 1000, 777]) | 8\n",
-      "20766    | torch.Size([1, 1000, 777]) | 8\n",
-      "20767    | torch.Size([1, 1000, 754]) | 8\n",
-      "20768    | torch.Size([1, 1000, 754]) | 8\n",
-      "20769    | torch.Size([1, 1000, 771]) | 8\n",
-      "20770    | torch.Size([1, 1000, 777]) | 8\n",
-      "20771    | torch.Size([1, 1000, 777]) | 8\n",
-      "20772    | torch.Size([1, 1000, 754]) | 8\n",
-      "20773    | torch.Size([1, 1000, 788]) | 8\n",
-      "20774    | torch.Size([1, 1000, 777]) | 8\n",
-      "20775    | torch.Size([1, 1000, 762]) | 8\n",
-      "20776    | torch.Size([1, 1000, 777]) | 8\n",
-      "20777    | torch.Size([1, 1000, 754]) | 8\n",
-      "20778    | torch.Size([1, 1000, 771]) | 8\n",
-      "20779    | torch.Size([1, 1000, 754]) | 8\n",
-      "20780    | torch.Size([1, 1000, 794]) | 8\n",
-      "20781    | torch.Size([1, 1000, 771]) | 8\n",
-      "20782    | torch.Size([1, 1000, 777]) | 8\n",
-      "20783    | torch.Size([1, 1000, 772]) | 8\n",
-      "20784    | torch.Size([1, 1000, 762]) | 8\n",
-      "20785    | torch.Size([1, 1000, 777]) | 8\n",
-      "20786    | torch.Size([1, 1000, 762]) | 8\n",
-      "20787    | torch.Size([1, 1000, 754]) | 8\n",
-      "20788    | torch.Size([1, 1000, 754]) | 8\n",
-      "20789    | torch.Size([1, 1000, 777]) | 8\n",
-      "20790    | torch.Size([1, 1000, 762]) | 8\n",
-      "20791    | torch.Size([1, 1000, 762]) | 8\n",
-      "20792    | torch.Size([1, 1000, 754]) | 8\n",
-      "20793    | torch.Size([1, 1000, 777]) | 8\n",
-      "20794    | torch.Size([1, 1000, 781]) | 8\n",
-      "20795    | torch.Size([1, 1000, 777]) | 8\n",
-      "20796    | torch.Size([1, 1000, 777]) | 8\n",
-      "20797    | torch.Size([1, 1000, 777]) | 8\n",
-      "20798    | torch.Size([1, 1000, 754]) | 8\n",
-      "20799    | torch.Size([1, 1000, 777]) | 8\n",
-      "20800    | torch.Size([1, 1000, 762]) | 8\n",
-      "20801    | torch.Size([1, 1000, 777]) | 8\n",
-      "20802    | torch.Size([1, 1000, 754]) | 8\n",
-      "20803    | torch.Size([1, 1000, 782]) | 8\n",
-      "20804    | torch.Size([1, 1000, 762]) | 8\n",
-      "20805    | torch.Size([1, 1000, 762]) | 8\n",
-      "20806    | torch.Size([1, 1000, 754]) | 8\n",
-      "20807    | torch.Size([1, 1000, 762]) | 8\n",
-      "20808    | torch.Size([1, 1000, 754]) | 8\n",
-      "20809    | torch.Size([1, 1000, 767]) | 8\n",
-      "20810    | torch.Size([1, 1000, 771]) | 8\n",
-      "20811    | torch.Size([1, 1000, 773]) | 8\n",
-      "20812    | torch.Size([1, 1000, 779]) | 8\n",
-      "20813    | torch.Size([1, 1000, 762]) | 8\n",
-      "20814    | torch.Size([1, 1000, 786]) | 8\n",
-      "20815    | torch.Size([1, 1000, 762]) | 8\n",
-      "20816    | torch.Size([1, 1000, 767]) | 8\n",
-      "20817    | torch.Size([1, 1000, 754]) | 8\n",
-      "20818    | torch.Size([1, 1000, 777]) | 8\n",
-      "20819    | torch.Size([1, 1000, 772]) | 8\n",
-      "20820    | torch.Size([1, 1000, 775]) | 8\n",
-      "20821    | torch.Size([1, 1000, 762]) | 8\n",
-      "20822    | torch.Size([1, 1000, 777]) | 8\n",
-      "20823    | torch.Size([1, 1000, 777]) | 8\n",
-      "20824    | torch.Size([1, 1000, 762]) | 8\n",
-      "20825    | torch.Size([1, 1000, 762]) | 8\n",
-      "20826    | torch.Size([1, 1000, 762]) | 8\n",
-      "20827    | torch.Size([1, 1000, 762]) | 8\n",
-      "20828    | torch.Size([1, 1000, 793]) | 8\n",
-      "20829    | torch.Size([1, 1000, 795]) | 8\n",
-      "20830    | torch.Size([1, 1000, 762]) | 8\n",
-      "20831    | torch.Size([1, 1000, 762]) | 8\n",
-      "20832    | torch.Size([1, 1000, 777]) | 8\n",
-      "20833    | torch.Size([1, 1000, 777]) | 8\n",
-      "20834    | torch.Size([1, 1000, 762]) | 8\n",
-      "20835    | torch.Size([1, 1000, 777]) | 8\n",
-      "20836    | torch.Size([1, 1000, 713]) | 8\n",
-      "20837    | torch.Size([1, 1000, 802]) | 8\n",
-      "20838    | torch.Size([1, 1000, 779]) | 8\n",
-      "20839    | torch.Size([1, 1000, 777]) | 8\n",
-      "20840    | torch.Size([1, 1000, 754]) | 8\n",
-      "20841    | torch.Size([1, 1000, 754]) | 8\n",
-      "20842    | torch.Size([1, 1000, 792]) | 8\n",
-      "20843    | torch.Size([1, 1000, 785]) | 8\n",
-      "20844    | torch.Size([1, 1000, 800]) | 8\n",
-      "20845    | torch.Size([1, 1000, 762]) | 8\n",
-      "20846    | torch.Size([1, 1000, 762]) | 8\n",
-      "20847    | torch.Size([1, 1000, 762]) | 8\n",
-      "20848    | torch.Size([1, 1000, 762]) | 8\n",
-      "20849    | torch.Size([1, 1000, 863]) | 8\n",
-      "20850    | torch.Size([1, 1000, 762]) | 8\n",
-      "20851    | torch.Size([1, 1000, 777]) | 8\n",
-      "20852    | torch.Size([1, 1000, 754]) | 8\n",
-      "20853    | torch.Size([1, 1000, 754]) | 8\n",
-      "20854    | torch.Size([1, 1000, 762]) | 8\n",
-      "20855    | torch.Size([1, 1000, 772]) | 8\n",
-      "20856    | torch.Size([1, 1000, 754]) | 8\n",
-      "20857    | torch.Size([1, 1000, 762]) | 8\n",
-      "20858    | torch.Size([1, 1000, 754]) | 8\n",
-      "20859    | torch.Size([1, 1000, 781]) | 8\n",
-      "20860    | torch.Size([1, 1000, 754]) | 8\n",
-      "20861    | torch.Size([1, 1000, 661]) | 8\n",
-      "20862    | torch.Size([1, 1000, 786]) | 8\n",
-      "20863    | torch.Size([1, 1000, 778]) | 8\n",
-      "20864    | torch.Size([1, 1000, 777]) | 8\n",
-      "20865    | torch.Size([1, 1000, 762]) | 8\n",
-      "20866    | torch.Size([1, 1000, 802]) | 8\n",
-      "20867    | torch.Size([1, 1000, 843]) | 8\n",
-      "20868    | torch.Size([1, 1000, 754]) | 8\n",
-      "20869    | torch.Size([1, 1000, 762]) | 8\n",
-      "20870    | torch.Size([1, 1000, 754]) | 8\n",
-      "20871    | torch.Size([1, 1000, 772]) | 8\n",
-      "20872    | torch.Size([1, 1000, 762]) | 8\n",
-      "20873    | torch.Size([1, 1000, 762]) | 8\n",
-      "20874    | torch.Size([1, 1000, 797]) | 8\n",
-      "20875    | torch.Size([1, 1000, 754]) | 8\n",
-      "20876    | torch.Size([1, 1000, 762]) | 8\n",
-      "20877    | torch.Size([1, 1000, 762]) | 8\n",
-      "20878    | torch.Size([1, 1000, 762]) | 8\n",
-      "20879    | torch.Size([1, 1000, 796]) | 8\n",
-      "20880    | torch.Size([1, 1000, 794]) | 8\n",
-      "20881    | torch.Size([1, 1000, 762]) | 8\n",
-      "20882    | torch.Size([1, 1000, 754]) | 8\n",
-      "20883    | torch.Size([1, 1000, 777]) | 8\n",
-      "20884    | torch.Size([1, 1000, 777]) | 8\n",
-      "20885    | torch.Size([1, 1000, 762]) | 8\n",
-      "20886    | torch.Size([1, 1000, 786]) | 8\n",
-      "20887    | torch.Size([1, 1000, 772]) | 8\n",
-      "20888    | torch.Size([1, 1000, 767]) | 8\n",
-      "20889    | torch.Size([1, 1000, 762]) | 8\n",
-      "20890    | torch.Size([1, 1000, 777]) | 8\n",
-      "20891    | torch.Size([1, 1000, 782]) | 8\n",
-      "20892    | torch.Size([1, 1000, 762]) | 8\n",
-      "20893    | torch.Size([1, 1000, 754]) | 8\n",
-      "20894    | torch.Size([1, 1000, 754]) | 8\n",
-      "20895    | torch.Size([1, 1000, 762]) | 8\n",
-      "20896    | torch.Size([1, 1000, 754]) | 8\n",
-      "20897    | torch.Size([1, 1000, 762]) | 8\n",
-      "20898    | torch.Size([1, 1000, 762]) | 8\n",
-      "20899    | torch.Size([1, 1000, 754]) | 8\n",
-      "20900    | torch.Size([1, 1000, 762]) | 8\n",
-      "20901    | torch.Size([1, 1000, 770]) | 8\n",
-      "20902    | torch.Size([1, 1000, 777]) | 8\n",
-      "20903    | torch.Size([1, 1000, 768]) | 8\n",
-      "20904    | torch.Size([1, 1000, 777]) | 8\n",
-      "20905    | torch.Size([1, 1000, 762]) | 8\n",
-      "20906    | torch.Size([1, 1000, 762]) | 8\n",
-      "20907    | torch.Size([1, 1000, 802]) | 8\n",
-      "20908    | torch.Size([1, 1000, 754]) | 8\n",
-      "20909    | torch.Size([1, 1000, 772]) | 8\n",
-      "20910    | torch.Size([1, 1000, 777]) | 8\n",
-      "20911    | torch.Size([1, 1000, 754]) | 8\n",
-      "20912    | torch.Size([1, 1000, 762]) | 8\n",
-      "20913    | torch.Size([1, 1000, 762]) | 8\n",
-      "20914    | torch.Size([1, 1000, 762]) | 8\n",
-      "20915    | torch.Size([1, 1000, 762]) | 8\n",
-      "20916    | torch.Size([1, 1000, 762]) | 8\n",
-      "20917    | torch.Size([1, 1000, 777]) | 8\n",
-      "20918    | torch.Size([1, 1000, 762]) | 8\n",
-      "20919    | torch.Size([1, 1000, 772]) | 8\n",
-      "20920    | torch.Size([1, 1000, 777]) | 8\n",
-      "20921    | torch.Size([1, 1000, 754]) | 8\n",
-      "20922    | torch.Size([1, 1000, 770]) | 8\n",
-      "20923    | torch.Size([1, 1000, 762]) | 8\n",
-      "20924    | torch.Size([1, 1000, 802]) | 8\n",
-      "20925    | torch.Size([1, 1000, 754]) | 8\n",
-      "20926    | torch.Size([1, 1000, 754]) | 8\n",
-      "20927    | torch.Size([1, 1000, 754]) | 8\n",
-      "20928    | torch.Size([1, 1000, 785]) | 8\n",
-      "20929    | torch.Size([1, 1000, 754]) | 8\n",
-      "20930    | torch.Size([1, 1000, 762]) | 8\n",
-      "20931    | torch.Size([1, 1000, 754]) | 8\n",
-      "20932    | torch.Size([1, 1000, 737]) | 8\n",
-      "20933    | torch.Size([1, 1000, 764]) | 8\n",
-      "20934    | torch.Size([1, 1000, 754]) | 8\n",
-      "20935    | torch.Size([1, 1000, 787]) | 8\n",
-      "20936    | torch.Size([1, 1000, 777]) | 8\n",
-      "20937    | torch.Size([1, 1000, 762]) | 8\n",
-      "20938    | torch.Size([1, 1000, 779]) | 8\n",
-      "20939    | torch.Size([1, 1000, 801]) | 8\n",
-      "20940    | torch.Size([1, 1000, 754]) | 8\n",
-      "20941    | torch.Size([1, 1000, 865]) | 8\n",
-      "20942    | torch.Size([1, 1000, 754]) | 8\n",
-      "20943    | torch.Size([1, 1000, 754]) | 8\n",
-      "20944    | torch.Size([1, 1000, 754]) | 8\n",
-      "20945    | torch.Size([1, 1000, 754]) | 8\n",
-      "20946    | torch.Size([1, 1000, 802]) | 8\n",
-      "20947    | torch.Size([1, 1000, 774]) | 8\n",
-      "20948    | torch.Size([1, 1000, 762]) | 8\n",
-      "20949    | torch.Size([1, 1000, 771]) | 8\n",
-      "20950    | torch.Size([1, 1000, 754]) | 8\n",
-      "20951    | torch.Size([1, 1000, 754]) | 8\n",
-      "20952    | torch.Size([1, 1000, 849]) | 8\n",
-      "20953    | torch.Size([1, 1000, 850]) | 8\n",
-      "20954    | torch.Size([1, 1000, 754]) | 8\n",
-      "20955    | torch.Size([1, 1000, 762]) | 8\n",
-      "20956    | torch.Size([1, 1000, 762]) | 8\n",
-      "20957    | torch.Size([1, 1000, 762]) | 8\n",
-      "20958    | torch.Size([1, 1000, 762]) | 8\n",
-      "20959    | torch.Size([1, 1000, 762]) | 8\n",
-      "20960    | torch.Size([1, 1000, 776]) | 8\n",
-      "20961    | torch.Size([1, 1000, 754]) | 8\n",
-      "20962    | torch.Size([1, 1000, 754]) | 8\n",
-      "20963    | torch.Size([1, 1000, 762]) | 8\n",
-      "20964    | torch.Size([1, 1000, 777]) | 8\n",
-      "20965    | torch.Size([1, 1000, 762]) | 8\n",
-      "20966    | torch.Size([1, 1000, 760]) | 8\n",
-      "20967    | torch.Size([1, 1000, 762]) | 8\n",
-      "20968    | torch.Size([1, 1000, 762]) | 8\n",
-      "20969    | torch.Size([1, 1000, 796]) | 8\n",
-      "20970    | torch.Size([1, 1000, 762]) | 8\n",
-      "20971    | torch.Size([1, 1000, 777]) | 8\n",
-      "20972    | torch.Size([1, 1000, 781]) | 8\n",
-      "20973    | torch.Size([1, 1000, 762]) | 8\n",
-      "20974    | torch.Size([1, 1000, 754]) | 8\n",
-      "20975    | torch.Size([1, 1000, 762]) | 8\n",
-      "20976    | torch.Size([1, 1000, 762]) | 8\n",
-      "20977    | torch.Size([1, 1000, 785]) | 8\n",
-      "20978    | torch.Size([1, 1000, 754]) | 8\n",
-      "20979    | torch.Size([1, 1000, 807]) | 8\n",
-      "20980    | torch.Size([1, 1000, 762]) | 8\n",
-      "20981    | torch.Size([1, 1000, 762]) | 8\n",
-      "20982    | torch.Size([1, 1000, 762]) | 8\n",
-      "20983    | torch.Size([1, 1000, 774]) | 8\n",
-      "20984    | torch.Size([1, 1000, 762]) | 8\n",
-      "20985    | torch.Size([1, 1000, 754]) | 8\n",
-      "20986    | torch.Size([1, 1000, 754]) | 8\n",
-      "20987    | torch.Size([1, 1000, 778]) | 8\n",
-      "20988    | torch.Size([1, 1000, 756]) | 8\n",
-      "20989    | torch.Size([1, 1000, 762]) | 8\n",
-      "20990    | torch.Size([1, 1000, 754]) | 8\n",
-      "20991    | torch.Size([1, 1000, 810]) | 8\n",
-      "20992    | torch.Size([1, 1000, 754]) | 8\n",
-      "20993    | torch.Size([1, 1000, 777]) | 8\n",
-      "20994    | torch.Size([1, 1000, 799]) | 8\n",
-      "20995    | torch.Size([1, 1000, 762]) | 8\n",
-      "20996    | torch.Size([1, 1000, 799]) | 8\n",
-      "20997    | torch.Size([1, 1000, 762]) | 8\n",
-      "20998    | torch.Size([1, 1000, 762]) | 8\n",
-      "20999    | torch.Size([1, 1000, 754]) | 8\n",
-      "21000    | torch.Size([1, 1000, 762]) | 8\n",
-      "21001    | torch.Size([1, 1000, 762]) | 8\n",
-      "21002    | torch.Size([1, 1000, 762]) | 8\n",
-      "21003    | torch.Size([1, 1000, 762]) | 8\n",
-      "21004    | torch.Size([1, 1000, 768]) | 8\n",
-      "21005    | torch.Size([1, 1000, 762]) | 8\n",
-      "21006    | torch.Size([1, 1000, 786]) | 8\n",
-      "21007    | torch.Size([1, 1000, 762]) | 8\n",
-      "21008    | torch.Size([1, 1000, 762]) | 8\n",
-      "21009    | torch.Size([1, 1000, 762]) | 8\n",
-      "21010    | torch.Size([1, 1000, 792]) | 8\n",
-      "21011    | torch.Size([1, 1000, 777]) | 8\n",
-      "21012    | torch.Size([1, 1000, 762]) | 8\n",
-      "21013    | torch.Size([1, 1000, 781]) | 8\n",
-      "21014    | torch.Size([1, 1000, 769]) | 8\n",
-      "21015    | torch.Size([1, 1000, 762]) | 8\n",
-      "21016    | torch.Size([1, 1000, 765]) | 8\n",
-      "21017    | torch.Size([1, 1000, 760]) | 8\n",
-      "21018    | torch.Size([1, 1000, 868]) | 8\n",
-      "21019    | torch.Size([1, 1000, 775]) | 8\n",
-      "21020    | torch.Size([1, 1000, 762]) | 8\n",
-      "21021    | torch.Size([1, 1000, 785]) | 8\n",
-      "21022    | torch.Size([1, 1000, 771]) | 8\n",
-      "21023    | torch.Size([1, 1000, 754]) | 8\n",
-      "21024    | torch.Size([1, 1000, 762]) | 8\n",
-      "21025    | torch.Size([1, 1000, 762]) | 8\n",
-      "21026    | torch.Size([1, 1000, 762]) | 8\n",
-      "21027    | torch.Size([1, 1000, 802]) | 8\n",
-      "21028    | torch.Size([1, 1000, 777]) | 8\n",
-      "21029    | torch.Size([1, 1000, 784]) | 8\n",
-      "21030    | torch.Size([1, 1000, 786]) | 8\n",
-      "21031    | torch.Size([1, 1000, 768]) | 8\n",
-      "21032    | torch.Size([1, 1000, 870]) | 8\n",
-      "21033    | torch.Size([1, 1000, 777]) | 8\n",
-      "21034    | torch.Size([1, 1000, 764]) | 8\n",
-      "21035    | torch.Size([1, 1000, 754]) | 8\n",
-      "21036    | torch.Size([1, 1000, 762]) | 8\n",
-      "21037    | torch.Size([1, 1000, 784]) | 8\n",
-      "21038    | torch.Size([1, 1000, 767]) | 8\n",
-      "21039    | torch.Size([1, 1000, 777]) | 8\n",
-      "21040    | torch.Size([1, 1000, 786]) | 8\n",
-      "21041    | torch.Size([1, 1000, 762]) | 8\n",
-      "21042    | torch.Size([1, 1000, 762]) | 8\n",
-      "21043    | torch.Size([1, 1000, 762]) | 8\n",
-      "21044    | torch.Size([1, 1000, 785]) | 8\n",
-      "21045    | torch.Size([1, 1000, 777]) | 8\n",
-      "21046    | torch.Size([1, 1000, 754]) | 8\n",
-      "21047    | torch.Size([1, 1000, 762]) | 8\n",
-      "21048    | torch.Size([1, 1000, 805]) | 8\n",
-      "21049    | torch.Size([1, 1000, 762]) | 8\n",
-      "21050    | torch.Size([1, 1000, 785]) | 8\n",
-      "21051    | torch.Size([1, 1000, 762]) | 8\n",
-      "21052    | torch.Size([1, 1000, 762]) | 8\n",
-      "21053    | torch.Size([1, 1000, 762]) | 8\n",
-      "21054    | torch.Size([1, 1000, 767]) | 8\n",
-      "21055    | torch.Size([1, 1000, 754]) | 8\n",
-      "21056    | torch.Size([1, 1000, 788]) | 8\n",
-      "21057    | torch.Size([1, 1000, 762]) | 8\n",
-      "21058    | torch.Size([1, 1000, 778]) | 8\n",
-      "21059    | torch.Size([1, 1000, 777]) | 8\n",
-      "21060    | torch.Size([1, 1000, 754]) | 8\n",
-      "21061    | torch.Size([1, 1000, 762]) | 8\n",
-      "21062    | torch.Size([1, 1000, 777]) | 8\n",
-      "21063    | torch.Size([1, 1000, 754]) | 8\n",
-      "21064    | torch.Size([1, 1000, 777]) | 8\n",
-      "21065    | torch.Size([1, 1000, 762]) | 8\n",
-      "21066    | torch.Size([1, 1000, 754]) | 8\n",
-      "21067    | torch.Size([1, 1000, 754]) | 8\n",
-      "21068    | torch.Size([1, 1000, 762]) | 8\n",
-      "21069    | torch.Size([1, 1000, 777]) | 8\n",
-      "21070    | torch.Size([1, 1000, 754]) | 8\n",
-      "21071    | torch.Size([1, 1000, 762]) | 8\n",
-      "21072    | torch.Size([1, 1000, 762]) | 8\n",
-      "21073    | torch.Size([1, 1000, 777]) | 8\n",
-      "21074    | torch.Size([1, 1000, 784]) | 8\n",
-      "21075    | torch.Size([1, 1000, 782]) | 8\n",
-      "21076    | torch.Size([1, 1000, 754]) | 8\n",
-      "21077    | torch.Size([1, 1000, 754]) | 8\n",
-      "21078    | torch.Size([1, 1000, 777]) | 8\n",
-      "21079    | torch.Size([1, 1000, 777]) | 8\n",
-      "21080    | torch.Size([1, 1000, 777]) | 8\n",
-      "21081    | torch.Size([1, 1000, 754]) | 8\n",
-      "21082    | torch.Size([1, 1000, 762]) | 8\n",
-      "21083    | torch.Size([1, 1000, 777]) | 8\n",
-      "21084    | torch.Size([1, 1000, 754]) | 8\n",
-      "21085    | torch.Size([1, 1000, 727]) | 8\n",
-      "21086    | torch.Size([1, 1000, 754]) | 8\n",
-      "21087    | torch.Size([1, 1000, 858]) | 8\n",
-      "21088    | torch.Size([1, 1000, 769]) | 8\n",
-      "21089    | torch.Size([1, 1000, 762]) | 8\n",
-      "21090    | torch.Size([1, 1000, 784]) | 8\n",
-      "21091    | torch.Size([1, 1000, 777]) | 8\n",
-      "21092    | torch.Size([1, 1000, 754]) | 8\n",
-      "21093    | torch.Size([1, 1000, 754]) | 8\n",
-      "21094    | torch.Size([1, 1000, 777]) | 8\n",
-      "21095    | torch.Size([1, 1000, 777]) | 8\n",
-      "21096    | torch.Size([1, 1000, 762]) | 8\n",
-      "21097    | torch.Size([1, 1000, 796]) | 8\n",
-      "21098    | torch.Size([1, 1000, 777]) | 8\n",
-      "21099    | torch.Size([1, 1000, 762]) | 8\n",
-      "21100    | torch.Size([1, 1000, 762]) | 8\n",
-      "21101    | torch.Size([1, 1000, 754]) | 8\n",
-      "21102    | torch.Size([1, 1000, 777]) | 8\n",
-      "21103    | torch.Size([1, 1000, 754]) | 8\n",
-      "21104    | torch.Size([1, 1000, 762]) | 8\n",
-      "21105    | torch.Size([1, 1000, 771]) | 8\n",
-      "21106    | torch.Size([1, 1000, 762]) | 8\n",
-      "21107    | torch.Size([1, 1000, 802]) | 8\n",
-      "21108    | torch.Size([1, 1000, 777]) | 8\n",
-      "21109    | torch.Size([1, 1000, 754]) | 8\n",
-      "21110    | torch.Size([1, 1000, 754]) | 8\n",
-      "21111    | torch.Size([1, 1000, 762]) | 8\n",
-      "21112    | torch.Size([1, 1000, 762]) | 8\n",
-      "21113    | torch.Size([1, 1000, 777]) | 8\n",
-      "21114    | torch.Size([1, 1000, 779]) | 8\n",
-      "21115    | torch.Size([1, 1000, 754]) | 8\n",
-      "21116    | torch.Size([1, 1000, 791]) | 8\n",
-      "21117    | torch.Size([1, 1000, 777]) | 8\n",
-      "21118    | torch.Size([1, 1000, 754]) | 8\n",
-      "21119    | torch.Size([1, 1000, 754]) | 8\n",
-      "21120    | torch.Size([1, 1000, 754]) | 8\n",
-      "21121    | torch.Size([1, 1000, 762]) | 8\n",
-      "21122    | torch.Size([1, 1000, 754]) | 8\n",
-      "21123    | torch.Size([1, 1000, 754]) | 8\n",
-      "21124    | torch.Size([1, 1000, 768]) | 8\n",
-      "21125    | torch.Size([1, 1000, 777]) | 8\n",
-      "21126    | torch.Size([1, 1000, 754]) | 8\n",
-      "21127    | torch.Size([1, 1000, 763]) | 8\n",
-      "21128    | torch.Size([1, 1000, 762]) | 8\n",
-      "21129    | torch.Size([1, 1000, 813]) | 8\n",
-      "21130    | torch.Size([1, 1000, 762]) | 8\n",
-      "21131    | torch.Size([1, 1000, 762]) | 8\n",
-      "21132    | torch.Size([1, 1000, 774]) | 8\n",
-      "21133    | torch.Size([1, 1000, 762]) | 8\n",
-      "21134    | torch.Size([1, 1000, 762]) | 8\n",
-      "21135    | torch.Size([1, 1000, 762]) | 8\n",
-      "21136    | torch.Size([1, 1000, 762]) | 8\n",
-      "21137    | torch.Size([1, 1000, 772]) | 8\n",
-      "21138    | torch.Size([1, 1000, 754]) | 8\n",
-      "21139    | torch.Size([1, 1000, 762]) | 8\n",
-      "21140    | torch.Size([1, 1000, 762]) | 8\n",
-      "21141    | torch.Size([1, 1000, 762]) | 8\n",
-      "21142    | torch.Size([1, 1000, 788]) | 8\n",
-      "21143    | torch.Size([1, 1000, 777]) | 8\n",
-      "21144    | torch.Size([1, 1000, 762]) | 8\n",
-      "21145    | torch.Size([1, 1000, 762]) | 8\n",
-      "21146    | torch.Size([1, 1000, 772]) | 8\n",
-      "21147    | torch.Size([1, 1000, 777]) | 8\n",
-      "21148    | torch.Size([1, 1000, 762]) | 8\n",
-      "21149    | torch.Size([1, 1000, 861]) | 8\n",
-      "21150    | torch.Size([1, 1000, 777]) | 8\n",
-      "21151    | torch.Size([1, 1000, 754]) | 8\n",
-      "21152    | torch.Size([1, 1000, 777]) | 8\n",
-      "21153    | torch.Size([1, 1000, 763]) | 8\n",
-      "21154    | torch.Size([1, 1000, 777]) | 8\n",
-      "21155    | torch.Size([1, 1000, 754]) | 8\n",
-      "21156    | torch.Size([1, 1000, 762]) | 8\n",
-      "21157    | torch.Size([1, 1000, 754]) | 8\n",
-      "21158    | torch.Size([1, 1000, 777]) | 8\n",
-      "21159    | torch.Size([1, 1000, 771]) | 8\n",
-      "21160    | torch.Size([1, 1000, 771]) | 8\n",
-      "21161    | torch.Size([1, 1000, 754]) | 8\n",
-      "21162    | torch.Size([1, 1000, 862]) | 8\n",
-      "21163    | torch.Size([1, 1000, 778]) | 8\n",
-      "21164    | torch.Size([1, 1000, 772]) | 8\n",
-      "21165    | torch.Size([1, 1000, 847]) | 8\n",
-      "21166    | torch.Size([1, 1000, 754]) | 8\n",
-      "21167    | torch.Size([1, 1000, 754]) | 8\n",
-      "21168    | torch.Size([1, 1000, 775]) | 8\n",
-      "21169    | torch.Size([1, 1000, 795]) | 8\n",
-      "21170    | torch.Size([1, 1000, 754]) | 8\n",
-      "21171    | torch.Size([1, 1000, 786]) | 8\n",
-      "21172    | torch.Size([1, 1000, 762]) | 8\n",
-      "21173    | torch.Size([1, 1000, 771]) | 8\n",
-      "21174    | torch.Size([1, 1000, 762]) | 8\n",
-      "21175    | torch.Size([1, 1000, 762]) | 8\n",
-      "21176    | torch.Size([1, 1000, 789]) | 8\n",
-      "21177    | torch.Size([1, 1000, 777]) | 8\n",
-      "21178    | torch.Size([1, 1000, 762]) | 8\n",
-      "21179    | torch.Size([1, 1000, 774]) | 8\n",
-      "21180    | torch.Size([1, 1000, 754]) | 8\n",
-      "21181    | torch.Size([1, 1000, 762]) | 8\n",
-      "21182    | torch.Size([1, 1000, 786]) | 8\n",
-      "21183    | torch.Size([1, 1000, 777]) | 8\n",
-      "21184    | torch.Size([1, 1000, 783]) | 8\n",
-      "21185    | torch.Size([1, 1000, 662]) | 8\n",
-      "21186    | torch.Size([1, 1000, 762]) | 8\n",
-      "21187    | torch.Size([1, 1000, 777]) | 8\n",
-      "21188    | torch.Size([1, 1000, 802]) | 8\n",
-      "21189    | torch.Size([1, 1000, 772]) | 8\n",
-      "21190    | torch.Size([1, 1000, 783]) | 8\n",
-      "21191    | torch.Size([1, 1000, 762]) | 8\n",
-      "21192    | torch.Size([1, 1000, 754]) | 8\n",
-      "21193    | torch.Size([1, 1000, 762]) | 8\n",
-      "21194    | torch.Size([1, 1000, 777]) | 8\n",
-      "21195    | torch.Size([1, 1000, 762]) | 8\n",
-      "21196    | torch.Size([1, 1000, 754]) | 8\n",
-      "21197    | torch.Size([1, 1000, 754]) | 8\n",
-      "21198    | torch.Size([1, 1000, 777]) | 8\n",
-      "21199    | torch.Size([1, 1000, 762]) | 8\n",
-      "21200    | torch.Size([1, 1000, 762]) | 8\n",
-      "21201    | torch.Size([1, 1000, 766]) | 8\n",
-      "21202    | torch.Size([1, 1000, 762]) | 8\n",
-      "21203    | torch.Size([1, 1000, 762]) | 8\n",
-      "21204    | torch.Size([1, 1000, 827]) | 8\n",
-      "21205    | torch.Size([1, 1000, 762]) | 8\n",
-      "21206    | torch.Size([1, 1000, 762]) | 8\n",
-      "21207    | torch.Size([1, 1000, 772]) | 8\n",
-      "21208    | torch.Size([1, 1000, 762]) | 8\n",
-      "21209    | torch.Size([1, 1000, 777]) | 8\n",
-      "21210    | torch.Size([1, 1000, 762]) | 8\n",
-      "21211    | torch.Size([1, 1000, 777]) | 8\n",
-      "21212    | torch.Size([1, 1000, 762]) | 8\n",
-      "21213    | torch.Size([1, 1000, 787]) | 8\n",
-      "21214    | torch.Size([1, 1000, 785]) | 8\n",
-      "21215    | torch.Size([1, 1000, 777]) | 8\n",
-      "21216    | torch.Size([1, 1000, 754]) | 8\n",
-      "21217    | torch.Size([1, 1000, 777]) | 8\n",
-      "21218    | torch.Size([1, 1000, 762]) | 8\n",
-      "21219    | torch.Size([1, 1000, 762]) | 8\n",
-      "21220    | torch.Size([1, 1000, 777]) | 8\n",
-      "21221    | torch.Size([1, 1000, 762]) | 8\n",
-      "21222    | torch.Size([1, 1000, 754]) | 8\n",
-      "21223    | torch.Size([1, 1000, 777]) | 8\n",
-      "21224    | torch.Size([1, 1000, 762]) | 8\n",
-      "21225    | torch.Size([1, 1000, 762]) | 8\n",
-      "21226    | torch.Size([1, 1000, 762]) | 8\n",
-      "21227    | torch.Size([1, 1000, 754]) | 8\n",
-      "21228    | torch.Size([1, 1000, 762]) | 8\n",
-      "21229    | torch.Size([1, 1000, 762]) | 8\n",
-      "21230    | torch.Size([1, 1000, 778]) | 8\n",
-      "21231    | torch.Size([1, 1000, 781]) | 8\n",
-      "21232    | torch.Size([1, 1000, 786]) | 8\n",
-      "21233    | torch.Size([1, 1000, 772]) | 8\n",
-      "21234    | torch.Size([1, 1000, 754]) | 8\n",
-      "21235    | torch.Size([1, 1000, 762]) | 8\n",
-      "21236    | torch.Size([1, 1000, 762]) | 8\n",
-      "21237    | torch.Size([1, 1000, 754]) | 8\n",
-      "21238    | torch.Size([1, 1000, 854]) | 8\n",
-      "21239    | torch.Size([1, 1000, 771]) | 8\n",
-      "21240    | torch.Size([1, 1000, 779]) | 8\n",
-      "21241    | torch.Size([1, 1000, 762]) | 8\n",
-      "21242    | torch.Size([1, 1000, 777]) | 8\n",
-      "21243    | torch.Size([1, 1000, 754]) | 8\n",
-      "21244    | torch.Size([1, 1000, 775]) | 8\n",
-      "21245    | torch.Size([1, 1000, 754]) | 8\n",
-      "21246    | torch.Size([1, 1000, 771]) | 8\n",
-      "21247    | torch.Size([1, 1000, 762]) | 8\n",
-      "21248    | torch.Size([1, 1000, 777]) | 8\n",
-      "21249    | torch.Size([1, 1000, 777]) | 8\n",
-      "21250    | torch.Size([1, 1000, 752]) | 8\n",
-      "21251    | torch.Size([1, 1000, 754]) | 8\n",
-      "21252    | torch.Size([1, 1000, 782]) | 8\n",
-      "21253    | torch.Size([1, 1000, 777]) | 8\n",
-      "21254    | torch.Size([1, 1000, 754]) | 8\n",
-      "21255    | torch.Size([1, 1000, 785]) | 8\n",
-      "21256    | torch.Size([1, 1000, 762]) | 8\n",
-      "21257    | torch.Size([1, 1000, 762]) | 8\n",
-      "21258    | torch.Size([1, 1000, 784]) | 8\n",
-      "21259    | torch.Size([1, 1000, 827]) | 8\n",
-      "21260    | torch.Size([1, 1000, 843]) | 8\n",
-      "21261    | torch.Size([1, 1000, 762]) | 8\n",
-      "21262    | torch.Size([1, 1000, 762]) | 8\n",
-      "21263    | torch.Size([1, 1000, 754]) | 8\n",
-      "21264    | torch.Size([1, 1000, 754]) | 8\n",
-      "21265    | torch.Size([1, 1000, 762]) | 8\n",
-      "21266    | torch.Size([1, 1000, 754]) | 8\n",
-      "21267    | torch.Size([1, 1000, 762]) | 8\n",
-      "21268    | torch.Size([1, 1000, 754]) | 8\n",
-      "21269    | torch.Size([1, 1000, 772]) | 8\n",
-      "21270    | torch.Size([1, 1000, 762]) | 8\n",
-      "21271    | torch.Size([1, 1000, 799]) | 8\n",
-      "21272    | torch.Size([1, 1000, 762]) | 8\n",
-      "21273    | torch.Size([1, 1000, 754]) | 8\n",
-      "21274    | torch.Size([1, 1000, 783]) | 8\n",
-      "21275    | torch.Size([1, 1000, 777]) | 8\n",
-      "21276    | torch.Size([1, 1000, 777]) | 8\n",
-      "21277    | torch.Size([1, 1000, 777]) | 8\n",
-      "21278    | torch.Size([1, 1000, 783]) | 8\n",
-      "21279    | torch.Size([1, 1000, 754]) | 8\n",
-      "21280    | torch.Size([1, 1000, 774]) | 8\n",
-      "21281    | torch.Size([1, 1000, 775]) | 8\n",
-      "21282    | torch.Size([1, 1000, 777]) | 8\n",
-      "21283    | torch.Size([1, 1000, 777]) | 8\n",
-      "21284    | torch.Size([1, 1000, 754]) | 8\n",
-      "21285    | torch.Size([1, 1000, 754]) | 8\n",
-      "21286    | torch.Size([1, 1000, 762]) | 8\n",
-      "21287    | torch.Size([1, 1000, 806]) | 8\n",
-      "21288    | torch.Size([1, 1000, 777]) | 8\n",
-      "21289    | torch.Size([1, 1000, 808]) | 8\n",
-      "21290    | torch.Size([1, 1000, 762]) | 8\n",
-      "21291    | torch.Size([1, 1000, 754]) | 8\n",
-      "21292    | torch.Size([1, 1000, 762]) | 8\n",
-      "21293    | torch.Size([1, 1000, 777]) | 8\n",
-      "21294    | torch.Size([1, 1000, 762]) | 8\n",
-      "21295    | torch.Size([1, 1000, 754]) | 8\n",
-      "21296    | torch.Size([1, 1000, 769]) | 8\n",
-      "21297    | torch.Size([1, 1000, 780]) | 8\n",
-      "21298    | torch.Size([1, 1000, 777]) | 8\n",
-      "21299    | torch.Size([1, 1000, 754]) | 8\n",
-      "21300    | torch.Size([1, 1000, 762]) | 8\n",
-      "21301    | torch.Size([1, 1000, 811]) | 8\n",
-      "21302    | torch.Size([1, 1000, 777]) | 8\n",
-      "21303    | torch.Size([1, 1000, 802]) | 8\n",
-      "21304    | torch.Size([1, 1000, 762]) | 8\n",
-      "21305    | torch.Size([1, 1000, 762]) | 8\n",
-      "21306    | torch.Size([1, 1000, 774]) | 8\n",
-      "21307    | torch.Size([1, 1000, 793]) | 8\n",
-      "21308    | torch.Size([1, 1000, 798]) | 8\n",
-      "21309    | torch.Size([1, 1000, 762]) | 8\n",
-      "21310    | torch.Size([1, 1000, 782]) | 8\n",
-      "21311    | torch.Size([1, 1000, 754]) | 8\n",
-      "21312    | torch.Size([1, 1000, 771]) | 8\n",
-      "21313    | torch.Size([1, 1000, 762]) | 8\n",
-      "21314    | torch.Size([1, 1000, 754]) | 8\n",
-      "21315    | torch.Size([1, 1000, 777]) | 8\n",
-      "21316    | torch.Size([1, 1000, 777]) | 8\n",
-      "21317    | torch.Size([1, 1000, 762]) | 8\n",
-      "21318    | torch.Size([1, 1000, 754]) | 8\n",
-      "21319    | torch.Size([1, 1000, 763]) | 8\n",
-      "21320    | torch.Size([1, 1000, 762]) | 8\n",
-      "21321    | torch.Size([1, 1000, 766]) | 8\n",
-      "21322    | torch.Size([1, 1000, 754]) | 8\n",
-      "21323    | torch.Size([1, 1000, 754]) | 8\n",
-      "21324    | torch.Size([1, 1000, 754]) | 8\n",
-      "21325    | torch.Size([1, 1000, 762]) | 8\n",
-      "21326    | torch.Size([1, 1000, 754]) | 8\n",
-      "21327    | torch.Size([1, 1000, 849]) | 8\n",
-      "21328    | torch.Size([1, 1000, 809]) | 8\n",
-      "21329    | torch.Size([1, 1000, 773]) | 8\n",
-      "21330    | torch.Size([1, 1000, 762]) | 8\n",
-      "21331    | torch.Size([1, 1000, 782]) | 8\n",
-      "21332    | torch.Size([1, 1000, 768]) | 8\n",
-      "21333    | torch.Size([1, 1000, 762]) | 8\n",
-      "21334    | torch.Size([1, 1000, 762]) | 8\n",
-      "21335    | torch.Size([1, 1000, 777]) | 8\n",
-      "21336    | torch.Size([1, 1000, 762]) | 8\n",
-      "21337    | torch.Size([1, 1000, 762]) | 8\n",
-      "21338    | torch.Size([1, 1000, 754]) | 8\n",
-      "21339    | torch.Size([1, 1000, 777]) | 8\n",
-      "21340    | torch.Size([1, 1000, 762]) | 8\n",
-      "21341    | torch.Size([1, 1000, 777]) | 8\n",
-      "21342    | torch.Size([1, 1000, 777]) | 8\n",
-      "21343    | torch.Size([1, 1000, 762]) | 8\n",
-      "21344    | torch.Size([1, 1000, 772]) | 8\n",
-      "21345    | torch.Size([1, 1000, 754]) | 8\n",
-      "21346    | torch.Size([1, 1000, 777]) | 8\n",
-      "21347    | torch.Size([1, 1000, 762]) | 8\n",
-      "21348    | torch.Size([1, 1000, 777]) | 8\n",
-      "21349    | torch.Size([1, 1000, 762]) | 8\n",
-      "21350    | torch.Size([1, 1000, 777]) | 8\n",
-      "21351    | torch.Size([1, 1000, 700]) | 8\n",
-      "21352    | torch.Size([1, 1000, 790]) | 8\n",
-      "21353    | torch.Size([1, 1000, 777]) | 8\n",
-      "21354    | torch.Size([1, 1000, 782]) | 8\n",
-      "21355    | torch.Size([1, 1000, 777]) | 8\n",
-      "21356    | torch.Size([1, 1000, 762]) | 8\n",
-      "21357    | torch.Size([1, 1000, 754]) | 8\n",
-      "21358    | torch.Size([1, 1000, 777]) | 8\n",
-      "21359    | torch.Size([1, 1000, 754]) | 8\n",
-      "21360    | torch.Size([1, 1000, 762]) | 8\n",
-      "21361    | torch.Size([1, 1000, 754]) | 8\n",
-      "21362    | torch.Size([1, 1000, 754]) | 8\n",
-      "21363    | torch.Size([1, 1000, 754]) | 8\n",
-      "21364    | torch.Size([1, 1000, 762]) | 8\n",
-      "21365    | torch.Size([1, 1000, 777]) | 8\n",
-      "21366    | torch.Size([1, 1000, 771]) | 8\n",
-      "21367    | torch.Size([1, 1000, 762]) | 8\n",
-      "21368    | torch.Size([1, 1000, 762]) | 8\n",
-      "21369    | torch.Size([1, 1000, 762]) | 8\n",
-      "21370    | torch.Size([1, 1000, 762]) | 8\n",
-      "21371    | torch.Size([1, 1000, 754]) | 8\n",
-      "21372    | torch.Size([1, 1000, 762]) | 8\n",
-      "21373    | torch.Size([1, 1000, 754]) | 8\n",
-      "21374    | torch.Size([1, 1000, 798]) | 8\n",
-      "21375    | torch.Size([1, 1000, 762]) | 8\n",
-      "21376    | torch.Size([1, 1000, 777]) | 8\n",
-      "21377    | torch.Size([1, 1000, 772]) | 8\n",
-      "21378    | torch.Size([1, 1000, 754]) | 8\n",
-      "21379    | torch.Size([1, 1000, 762]) | 8\n",
-      "21380    | torch.Size([1, 1000, 754]) | 8\n",
-      "21381    | torch.Size([1, 1000, 770]) | 8\n",
-      "21382    | torch.Size([1, 1000, 762]) | 8\n",
-      "21383    | torch.Size([1, 1000, 762]) | 8\n",
-      "21384    | torch.Size([1, 1000, 754]) | 8\n",
-      "21385    | torch.Size([1, 1000, 802]) | 8\n",
-      "21386    | torch.Size([1, 1000, 772]) | 8\n",
-      "21387    | torch.Size([1, 1000, 762]) | 8\n",
-      "21388    | torch.Size([1, 1000, 762]) | 8\n",
-      "21389    | torch.Size([1, 1000, 802]) | 8\n",
-      "21390    | torch.Size([1, 1000, 772]) | 8\n",
-      "21391    | torch.Size([1, 1000, 771]) | 8\n",
-      "21392    | torch.Size([1, 1000, 762]) | 8\n",
-      "21393    | torch.Size([1, 1000, 777]) | 8\n",
-      "21394    | torch.Size([1, 1000, 762]) | 8\n",
-      "21395    | torch.Size([1, 1000, 810]) | 8\n",
-      "21396    | torch.Size([1, 1000, 777]) | 8\n",
-      "21397    | torch.Size([1, 1000, 778]) | 8\n",
-      "21398    | torch.Size([1, 1000, 777]) | 8\n",
-      "21399    | torch.Size([1, 1000, 830]) | 8\n",
-      "21400    | torch.Size([1, 1000, 769]) | 8\n",
-      "21401    | torch.Size([1, 1000, 762]) | 8\n",
-      "21402    | torch.Size([1, 1000, 754]) | 8\n",
-      "21403    | torch.Size([1, 1000, 754]) | 8\n",
-      "21404    | torch.Size([1, 1000, 777]) | 8\n",
-      "21405    | torch.Size([1, 1000, 774]) | 8\n",
-      "21406    | torch.Size([1, 1000, 777]) | 8\n",
-      "21407    | torch.Size([1, 1000, 774]) | 8\n",
-      "21408    | torch.Size([1, 1000, 754]) | 8\n",
-      "21409    | torch.Size([1, 1000, 754]) | 8\n",
-      "21410    | torch.Size([1, 1000, 754]) | 8\n",
-      "21411    | torch.Size([1, 1000, 762]) | 8\n",
-      "21412    | torch.Size([1, 1000, 762]) | 8\n",
-      "21413    | torch.Size([1, 1000, 767]) | 8\n",
-      "21414    | torch.Size([1, 1000, 762]) | 8\n",
-      "21415    | torch.Size([1, 1000, 762]) | 8\n",
-      "21416    | torch.Size([1, 1000, 754]) | 8\n",
-      "21417    | torch.Size([1, 1000, 754]) | 8\n",
-      "21418    | torch.Size([1, 1000, 777]) | 8\n",
-      "21419    | torch.Size([1, 1000, 776]) | 8\n",
-      "21420    | torch.Size([1, 1000, 754]) | 8\n",
-      "21421    | torch.Size([1, 1000, 754]) | 8\n",
-      "21422    | torch.Size([1, 1000, 785]) | 8\n",
-      "21423    | torch.Size([1, 1000, 754]) | 8\n",
-      "21424    | torch.Size([1, 1000, 771]) | 8\n",
-      "21425    | torch.Size([1, 1000, 777]) | 8\n",
-      "21426    | torch.Size([1, 1000, 762]) | 8\n",
-      "21427    | torch.Size([1, 1000, 769]) | 8\n",
-      "21428    | torch.Size([1, 1000, 780]) | 8\n",
-      "21429    | torch.Size([1, 1000, 777]) | 8\n",
-      "21430    | torch.Size([1, 1000, 754]) | 8\n",
-      "21431    | torch.Size([1, 1000, 771]) | 8\n",
-      "21432    | torch.Size([1, 1000, 754]) | 8\n",
-      "21433    | torch.Size([1, 1000, 809]) | 8\n",
-      "21434    | torch.Size([1, 1000, 762]) | 8\n",
-      "21435    | torch.Size([1, 1000, 754]) | 8\n",
-      "21436    | torch.Size([1, 1000, 754]) | 8\n",
-      "21437    | torch.Size([1, 1000, 802]) | 8\n",
-      "21438    | torch.Size([1, 1000, 754]) | 8\n",
-      "21439    | torch.Size([1, 1000, 762]) | 8\n",
-      "21440    | torch.Size([1, 1000, 802]) | 8\n",
-      "21441    | torch.Size([1, 1000, 795]) | 8\n",
-      "21442    | torch.Size([1, 1000, 794]) | 8\n",
-      "21443    | torch.Size([1, 1000, 762]) | 8\n",
-      "21444    | torch.Size([1, 1000, 754]) | 8\n",
-      "21445    | torch.Size([1, 1000, 771]) | 8\n",
-      "21446    | torch.Size([1, 1000, 762]) | 8\n",
-      "21447    | torch.Size([1, 1000, 799]) | 8\n",
-      "21448    | torch.Size([1, 1000, 777]) | 8\n",
-      "21449    | torch.Size([1, 1000, 762]) | 8\n",
-      "21450    | torch.Size([1, 1000, 777]) | 8\n",
-      "21451    | torch.Size([1, 1000, 762]) | 8\n",
-      "21452    | torch.Size([1, 1000, 802]) | 8\n",
-      "21453    | torch.Size([1, 1000, 777]) | 8\n",
-      "21454    | torch.Size([1, 1000, 777]) | 8\n",
-      "21455    | torch.Size([1, 1000, 754]) | 8\n",
-      "21456    | torch.Size([1, 1000, 762]) | 8\n",
-      "21457    | torch.Size([1, 1000, 762]) | 8\n",
-      "21458    | torch.Size([1, 1000, 777]) | 8\n",
-      "21459    | torch.Size([1, 1000, 762]) | 8\n",
-      "21460    | torch.Size([1, 1000, 754]) | 8\n",
-      "21461    | torch.Size([1, 1000, 794]) | 8\n",
-      "21462    | torch.Size([1, 1000, 754]) | 8\n",
-      "21463    | torch.Size([1, 1000, 754]) | 8\n",
-      "21464    | torch.Size([1, 1000, 762]) | 8\n",
-      "21465    | torch.Size([1, 1000, 777]) | 8\n",
-      "21466    | torch.Size([1, 1000, 761]) | 8\n",
-      "21467    | torch.Size([1, 1000, 762]) | 8\n",
-      "21468    | torch.Size([1, 1000, 777]) | 8\n",
-      "21469    | torch.Size([1, 1000, 754]) | 8\n",
-      "21470    | torch.Size([1, 1000, 777]) | 8\n",
-      "21471    | torch.Size([1, 1000, 762]) | 8\n",
-      "21472    | torch.Size([1, 1000, 777]) | 8\n",
-      "21473    | torch.Size([1, 1000, 762]) | 8\n",
-      "21474    | torch.Size([1, 1000, 762]) | 8\n",
-      "21475    | torch.Size([1, 1000, 754]) | 8\n",
-      "21476    | torch.Size([1, 1000, 762]) | 8\n",
-      "21477    | torch.Size([1, 1000, 754]) | 8\n",
-      "21478    | torch.Size([1, 1000, 762]) | 8\n",
-      "21479    | torch.Size([1, 1000, 762]) | 8\n",
-      "21480    | torch.Size([1, 1000, 754]) | 8\n",
-      "21481    | torch.Size([1, 1000, 777]) | 8\n",
-      "21482    | torch.Size([1, 1000, 762]) | 8\n",
-      "21483    | torch.Size([1, 1000, 754]) | 8\n",
-      "21484    | torch.Size([1, 1000, 762]) | 8\n",
-      "21485    | torch.Size([1, 1000, 771]) | 8\n",
-      "21486    | torch.Size([1, 1000, 848]) | 8\n",
-      "21487    | torch.Size([1, 1000, 773]) | 8\n",
-      "21488    | torch.Size([1, 1000, 762]) | 8\n",
-      "21489    | torch.Size([1, 1000, 754]) | 8\n",
-      "21490    | torch.Size([1, 1000, 777]) | 8\n",
-      "21491    | torch.Size([1, 1000, 777]) | 8\n",
-      "21492    | torch.Size([1, 1000, 762]) | 8\n",
-      "21493    | torch.Size([1, 1000, 774]) | 8\n",
-      "21494    | torch.Size([1, 1000, 754]) | 8\n",
-      "21495    | torch.Size([1, 1000, 774]) | 8\n",
-      "21496    | torch.Size([1, 1000, 762]) | 8\n",
-      "21497    | torch.Size([1, 1000, 777]) | 8\n",
-      "21498    | torch.Size([1, 1000, 777]) | 8\n",
-      "21499    | torch.Size([1, 1000, 777]) | 8\n",
-      "21500    | torch.Size([1, 1000, 762]) | 8\n",
-      "21501    | torch.Size([1, 1000, 754]) | 8\n",
-      "21502    | torch.Size([1, 1000, 762]) | 8\n",
-      "21503    | torch.Size([1, 1000, 802]) | 8\n",
-      "21504    | torch.Size([1, 1000, 754]) | 8\n",
-      "21505    | torch.Size([1, 1000, 792]) | 8\n",
-      "21506    | torch.Size([1, 1000, 762]) | 8\n",
-      "21507    | torch.Size([1, 1000, 767]) | 8\n",
-      "21508    | torch.Size([1, 1000, 754]) | 8\n",
-      "21509    | torch.Size([1, 1000, 762]) | 8\n",
-      "21510    | torch.Size([1, 1000, 777]) | 8\n",
-      "21511    | torch.Size([1, 1000, 754]) | 8\n",
-      "21512    | torch.Size([1, 1000, 762]) | 8\n",
-      "21513    | torch.Size([1, 1000, 754]) | 8\n",
-      "21514    | torch.Size([1, 1000, 777]) | 8\n",
-      "21515    | torch.Size([1, 1000, 754]) | 8\n",
-      "21516    | torch.Size([1, 1000, 772]) | 8\n",
-      "21517    | torch.Size([1, 1000, 762]) | 8\n",
-      "21518    | torch.Size([1, 1000, 754]) | 8\n",
-      "21519    | torch.Size([1, 1000, 754]) | 8\n",
-      "21520    | torch.Size([1, 1000, 769]) | 8\n",
-      "21521    | torch.Size([1, 1000, 841]) | 8\n",
-      "21522    | torch.Size([1, 1000, 1096]) | 8\n",
-      "21523    | torch.Size([1, 1000, 762]) | 8\n",
-      "21524    | torch.Size([1, 1000, 764]) | 8\n",
-      "21525    | torch.Size([1, 1000, 754]) | 8\n",
-      "21526    | torch.Size([1, 1000, 754]) | 8\n",
-      "21527    | torch.Size([1, 1000, 762]) | 8\n",
-      "21528    | torch.Size([1, 1000, 761]) | 8\n",
-      "21529    | torch.Size([1, 1000, 774]) | 8\n",
-      "21530    | torch.Size([1, 1000, 762]) | 8\n",
-      "21531    | torch.Size([1, 1000, 762]) | 8\n",
-      "21532    | torch.Size([1, 1000, 772]) | 8\n",
-      "21533    | torch.Size([1, 1000, 762]) | 8\n",
-      "21534    | torch.Size([1, 1000, 802]) | 8\n",
-      "21535    | torch.Size([1, 1000, 762]) | 8\n",
-      "21536    | torch.Size([1, 1000, 754]) | 8\n",
-      "21537    | torch.Size([1, 1000, 774]) | 8\n",
-      "21538    | torch.Size([1, 1000, 777]) | 8\n",
-      "21539    | torch.Size([1, 1000, 777]) | 8\n",
-      "21540    | torch.Size([1, 1000, 762]) | 8\n",
-      "21541    | torch.Size([1, 1000, 762]) | 8\n",
-      "21542    | torch.Size([1, 1000, 754]) | 8\n",
-      "21543    | torch.Size([1, 1000, 762]) | 8\n",
-      "21544    | torch.Size([1, 1000, 762]) | 8\n",
-      "21545    | torch.Size([1, 1000, 777]) | 8\n",
-      "21546    | torch.Size([1, 1000, 762]) | 8\n",
-      "21547    | torch.Size([1, 1000, 762]) | 8\n",
-      "21548    | torch.Size([1, 1000, 771]) | 8\n",
-      "21549    | torch.Size([1, 1000, 777]) | 8\n",
-      "21550    | torch.Size([1, 1000, 777]) | 8\n",
-      "21551    | torch.Size([1, 1000, 762]) | 8\n",
-      "21552    | torch.Size([1, 1000, 754]) | 8\n",
-      "21553    | torch.Size([1, 1000, 762]) | 8\n",
-      "21554    | torch.Size([1, 1000, 754]) | 8\n",
-      "21555    | torch.Size([1, 1000, 762]) | 8\n",
-      "21556    | torch.Size([1, 1000, 762]) | 8\n",
-      "21557    | torch.Size([1, 1000, 778]) | 8\n",
-      "21558    | torch.Size([1, 1000, 771]) | 8\n",
-      "21559    | torch.Size([1, 1000, 781]) | 8\n",
-      "21560    | torch.Size([1, 1000, 762]) | 8\n",
-      "21561    | torch.Size([1, 1000, 777]) | 8\n",
-      "21562    | torch.Size([1, 1000, 762]) | 8\n",
-      "21563    | torch.Size([1, 1000, 777]) | 8\n",
-      "21564    | torch.Size([1, 1000, 799]) | 8\n",
-      "21565    | torch.Size([1, 1000, 802]) | 8\n",
-      "21566    | torch.Size([1, 1000, 762]) | 8\n",
-      "21567    | torch.Size([1, 1000, 779]) | 8\n",
-      "21568    | torch.Size([1, 1000, 762]) | 8\n",
-      "21569    | torch.Size([1, 1000, 777]) | 8\n",
-      "21570    | torch.Size([1, 1000, 796]) | 8\n",
-      "21571    | torch.Size([1, 1000, 762]) | 8\n",
-      "21572    | torch.Size([1, 1000, 754]) | 8\n",
-      "21573    | torch.Size([1, 1000, 762]) | 8\n",
-      "21574    | torch.Size([1, 1000, 762]) | 8\n",
-      "21575    | torch.Size([1, 1000, 754]) | 8\n",
-      "21576    | torch.Size([1, 1000, 863]) | 8\n",
-      "21577    | torch.Size([1, 1000, 754]) | 8\n",
-      "21578    | torch.Size([1, 1000, 762]) | 8\n",
-      "21579    | torch.Size([1, 1000, 779]) | 8\n",
-      "21580    | torch.Size([1, 1000, 762]) | 8\n",
-      "21581    | torch.Size([1, 1000, 776]) | 8\n",
-      "21582    | torch.Size([1, 1000, 772]) | 8\n",
-      "21583    | torch.Size([1, 1000, 786]) | 8\n",
-      "21584    | torch.Size([1, 1000, 762]) | 8\n",
-      "21585    | torch.Size([1, 1000, 762]) | 8\n",
-      "21586    | torch.Size([1, 1000, 796]) | 8\n",
-      "21587    | torch.Size([1, 1000, 771]) | 8\n",
-      "21588    | torch.Size([1, 1000, 754]) | 8\n",
-      "21589    | torch.Size([1, 1000, 753]) | 8\n",
-      "21590    | torch.Size([1, 1000, 754]) | 8\n",
-      "21591    | torch.Size([1, 1000, 779]) | 8\n",
-      "21592    | torch.Size([1, 1000, 762]) | 8\n",
-      "21593    | torch.Size([1, 1000, 777]) | 8\n",
-      "21594    | torch.Size([1, 1000, 754]) | 8\n",
-      "21595    | torch.Size([1, 1000, 754]) | 8\n",
-      "21596    | torch.Size([1, 1000, 777]) | 8\n",
-      "21597    | torch.Size([1, 1000, 762]) | 8\n",
-      "21598    | torch.Size([1, 1000, 777]) | 8\n",
-      "21599    | torch.Size([1, 1000, 773]) | 8\n",
-      "21600    | torch.Size([1, 1000, 781]) | 8\n",
-      "21601    | torch.Size([1, 1000, 649]) | 8\n",
-      "21602    | torch.Size([1, 1000, 754]) | 8\n",
-      "21603    | torch.Size([1, 1000, 777]) | 8\n",
-      "21604    | torch.Size([1, 1000, 777]) | 8\n",
-      "21605    | torch.Size([1, 1000, 762]) | 8\n",
-      "21606    | torch.Size([1, 1000, 754]) | 8\n",
-      "21607    | torch.Size([1, 1000, 777]) | 8\n",
-      "21608    | torch.Size([1, 1000, 777]) | 8\n",
-      "21609    | torch.Size([1, 1000, 777]) | 8\n",
-      "21610    | torch.Size([1, 1000, 777]) | 8\n",
-      "21611    | torch.Size([1, 1000, 754]) | 8\n",
-      "21612    | torch.Size([1, 1000, 856]) | 8\n",
-      "21613    | torch.Size([1, 1000, 771]) | 8\n",
-      "21614    | torch.Size([1, 1000, 777]) | 8\n",
-      "21615    | torch.Size([1, 1000, 762]) | 8\n",
-      "21616    | torch.Size([1, 1000, 843]) | 8\n",
-      "21617    | torch.Size([1, 1000, 808]) | 8\n",
-      "21618    | torch.Size([1, 1000, 776]) | 8\n",
-      "21619    | torch.Size([1, 1000, 762]) | 8\n",
-      "21620    | torch.Size([1, 1000, 802]) | 8\n",
-      "21621    | torch.Size([1, 1000, 754]) | 8\n",
-      "21622    | torch.Size([1, 1000, 777]) | 8\n",
-      "21623    | torch.Size([1, 1000, 762]) | 8\n",
-      "21624    | torch.Size([1, 1000, 754]) | 8\n",
-      "21625    | torch.Size([1, 1000, 754]) | 8\n",
-      "21626    | torch.Size([1, 1000, 784]) | 8\n",
-      "21627    | torch.Size([1, 1000, 762]) | 8\n",
-      "21628    | torch.Size([1, 1000, 772]) | 8\n",
-      "21629    | torch.Size([1, 1000, 754]) | 8\n",
-      "21630    | torch.Size([1, 1000, 754]) | 8\n",
-      "21631    | torch.Size([1, 1000, 754]) | 8\n",
-      "21632    | torch.Size([1, 1000, 762]) | 8\n",
-      "21633    | torch.Size([1, 1000, 754]) | 8\n",
-      "21634    | torch.Size([1, 1000, 754]) | 8\n",
-      "21635    | torch.Size([1, 1000, 786]) | 8\n",
-      "21636    | torch.Size([1, 1000, 771]) | 8\n",
-      "21637    | torch.Size([1, 1000, 742]) | 8\n",
-      "21638    | torch.Size([1, 1000, 762]) | 8\n",
-      "21639    | torch.Size([1, 1000, 754]) | 8\n",
-      "21640    | torch.Size([1, 1000, 752]) | 8\n",
-      "21641    | torch.Size([1, 1000, 754]) | 8\n",
-      "21642    | torch.Size([1, 1000, 754]) | 8\n",
-      "21643    | torch.Size([1, 1000, 777]) | 8\n",
-      "21644    | torch.Size([1, 1000, 754]) | 8\n",
-      "21645    | torch.Size([1, 1000, 763]) | 8\n",
-      "21646    | torch.Size([1, 1000, 777]) | 8\n",
-      "21647    | torch.Size([1, 1000, 781]) | 8\n",
-      "21648    | torch.Size([1, 1000, 762]) | 8\n",
-      "21649    | torch.Size([1, 1000, 762]) | 8\n",
-      "21650    | torch.Size([1, 1000, 779]) | 8\n",
-      "21651    | torch.Size([1, 1000, 777]) | 8\n",
-      "21652    | torch.Size([1, 1000, 776]) | 8\n",
-      "21653    | torch.Size([1, 1000, 762]) | 8\n",
-      "21654    | torch.Size([1, 1000, 777]) | 8\n",
-      "21655    | torch.Size([1, 1000, 776]) | 8\n",
-      "21656    | torch.Size([1, 1000, 754]) | 8\n",
-      "21657    | torch.Size([1, 1000, 762]) | 8\n",
-      "21658    | torch.Size([1, 1000, 762]) | 8\n",
-      "21659    | torch.Size([1, 1000, 762]) | 8\n",
-      "21660    | torch.Size([1, 1000, 767]) | 8\n",
-      "21661    | torch.Size([1, 1000, 754]) | 8\n",
-      "21662    | torch.Size([1, 1000, 861]) | 8\n",
-      "21663    | torch.Size([1, 1000, 754]) | 8\n",
-      "21664    | torch.Size([1, 1000, 762]) | 8\n",
-      "21665    | torch.Size([1, 1000, 777]) | 8\n",
-      "21666    | torch.Size([1, 1000, 754]) | 8\n",
-      "21667    | torch.Size([1, 1000, 754]) | 8\n",
-      "21668    | torch.Size([1, 1000, 754]) | 8\n",
-      "21669    | torch.Size([1, 1000, 754]) | 8\n",
-      "21670    | torch.Size([1, 1000, 754]) | 8\n",
-      "21671    | torch.Size([1, 1000, 762]) | 8\n",
-      "21672    | torch.Size([1, 1000, 762]) | 8\n",
-      "21673    | torch.Size([1, 1000, 752]) | 8\n",
-      "21674    | torch.Size([1, 1000, 762]) | 8\n",
-      "21675    | torch.Size([1, 1000, 754]) | 8\n",
-      "21676    | torch.Size([1, 1000, 802]) | 8\n",
-      "21677    | torch.Size([1, 1000, 754]) | 8\n",
-      "21678    | torch.Size([1, 1000, 762]) | 8\n",
-      "21679    | torch.Size([1, 1000, 762]) | 8\n",
-      "21680    | torch.Size([1, 1000, 775]) | 8\n",
-      "21681    | torch.Size([1, 1000, 777]) | 8\n",
-      "21682    | torch.Size([1, 1000, 788]) | 8\n",
-      "21683    | torch.Size([1, 1000, 827]) | 8\n",
-      "21684    | torch.Size([1, 1000, 754]) | 8\n",
-      "21685    | torch.Size([1, 1000, 754]) | 8\n",
-      "21686    | torch.Size([1, 1000, 754]) | 8\n",
-      "21687    | torch.Size([1, 1000, 762]) | 8\n",
-      "21688    | torch.Size([1, 1000, 777]) | 8\n",
-      "21689    | torch.Size([1, 1000, 762]) | 8\n",
-      "21690    | torch.Size([1, 1000, 754]) | 8\n",
-      "21691    | torch.Size([1, 1000, 762]) | 8\n",
-      "21692    | torch.Size([1, 1000, 754]) | 8\n",
-      "21693    | torch.Size([1, 1000, 777]) | 8\n",
-      "21694    | torch.Size([1, 1000, 762]) | 8\n",
-      "21695    | torch.Size([1, 1000, 754]) | 8\n",
-      "21696    | torch.Size([1, 1000, 762]) | 8\n",
-      "21697    | torch.Size([1, 1000, 754]) | 8\n",
-      "21698    | torch.Size([1, 1000, 762]) | 8\n",
-      "21699    | torch.Size([1, 1000, 754]) | 8\n",
-      "21700    | torch.Size([1, 1000, 762]) | 8\n",
-      "21701    | torch.Size([1, 1000, 772]) | 8\n",
-      "21702    | torch.Size([1, 1000, 754]) | 8\n",
-      "21703    | torch.Size([1, 1000, 762]) | 8\n",
-      "21704    | torch.Size([1, 1000, 754]) | 8\n",
-      "21705    | torch.Size([1, 1000, 754]) | 8\n",
-      "21706    | torch.Size([1, 1000, 762]) | 8\n",
-      "21707    | torch.Size([1, 1000, 754]) | 8\n",
-      "21708    | torch.Size([1, 1000, 777]) | 8\n",
-      "21709    | torch.Size([1, 1000, 777]) | 8\n",
-      "21710    | torch.Size([1, 1000, 762]) | 8\n",
-      "21711    | torch.Size([1, 1000, 779]) | 8\n",
-      "21712    | torch.Size([1, 1000, 776]) | 8\n",
-      "21713    | torch.Size([1, 1000, 777]) | 8\n",
-      "21714    | torch.Size([1, 1000, 789]) | 8\n",
-      "21715    | torch.Size([1, 1000, 754]) | 8\n",
-      "21716    | torch.Size([1, 1000, 754]) | 8\n",
-      "21717    | torch.Size([1, 1000, 762]) | 8\n",
-      "21718    | torch.Size([1, 1000, 783]) | 8\n",
-      "21719    | torch.Size([1, 1000, 762]) | 8\n",
-      "21720    | torch.Size([1, 1000, 754]) | 8\n",
-      "21721    | torch.Size([1, 1000, 774]) | 8\n",
-      "21722    | torch.Size([1, 1000, 754]) | 8\n",
-      "21723    | torch.Size([1, 1000, 762]) | 8\n",
-      "21724    | torch.Size([1, 1000, 784]) | 8\n",
-      "21725    | torch.Size([1, 1000, 754]) | 8\n",
-      "21726    | torch.Size([1, 1000, 777]) | 8\n",
-      "21727    | torch.Size([1, 1000, 754]) | 8\n",
-      "21728    | torch.Size([1, 1000, 772]) | 8\n",
-      "21729    | torch.Size([1, 1000, 762]) | 8\n",
-      "21730    | torch.Size([1, 1000, 762]) | 8\n",
-      "21731    | torch.Size([1, 1000, 777]) | 8\n",
-      "21732    | torch.Size([1, 1000, 732]) | 8\n",
-      "21733    | torch.Size([1, 1000, 777]) | 8\n",
-      "21734    | torch.Size([1, 1000, 754]) | 8\n",
-      "21735    | torch.Size([1, 1000, 762]) | 8\n",
-      "21736    | torch.Size([1, 1000, 762]) | 8\n",
-      "21737    | torch.Size([1, 1000, 762]) | 8\n",
-      "21738    | torch.Size([1, 1000, 777]) | 8\n",
-      "21739    | torch.Size([1, 1000, 777]) | 8\n",
-      "21740    | torch.Size([1, 1000, 777]) | 8\n",
-      "21741    | torch.Size([1, 1000, 798]) | 8\n",
-      "21742    | torch.Size([1, 1000, 754]) | 8\n",
-      "21743    | torch.Size([1, 1000, 754]) | 8\n",
-      "21744    | torch.Size([1, 1000, 784]) | 8\n",
-      "21745    | torch.Size([1, 1000, 773]) | 8\n",
-      "21746    | torch.Size([1, 1000, 754]) | 8\n",
-      "21747    | torch.Size([1, 1000, 782]) | 8\n",
-      "21748    | torch.Size([1, 1000, 777]) | 8\n",
-      "21749    | torch.Size([1, 1000, 762]) | 8\n",
-      "21750    | torch.Size([1, 1000, 772]) | 8\n",
-      "21751    | torch.Size([1, 1000, 762]) | 8\n",
-      "21752    | torch.Size([1, 1000, 771]) | 8\n",
-      "21753    | torch.Size([1, 1000, 778]) | 8\n",
-      "21754    | torch.Size([1, 1000, 767]) | 8\n",
-      "21755    | torch.Size([1, 1000, 762]) | 8\n",
-      "21756    | torch.Size([1, 1000, 754]) | 8\n",
-      "21757    | torch.Size([1, 1000, 776]) | 8\n",
-      "21758    | torch.Size([1, 1000, 754]) | 8\n",
-      "21759    | torch.Size([1, 1000, 777]) | 8\n",
-      "21760    | torch.Size([1, 1000, 777]) | 8\n",
-      "21761    | torch.Size([1, 1000, 777]) | 8\n",
-      "21762    | torch.Size([1, 1000, 754]) | 8\n",
-      "21763    | torch.Size([1, 1000, 777]) | 8\n",
-      "21764    | torch.Size([1, 1000, 762]) | 8\n",
-      "21765    | torch.Size([1, 1000, 845]) | 8\n",
-      "21766    | torch.Size([1, 1000, 777]) | 8\n",
-      "21767    | torch.Size([1, 1000, 787]) | 8\n",
-      "21768    | torch.Size([1, 1000, 762]) | 8\n",
-      "21769    | torch.Size([1, 1000, 762]) | 8\n",
-      "21770    | torch.Size([1, 1000, 762]) | 8\n",
-      "21771    | torch.Size([1, 1000, 762]) | 8\n",
-      "21772    | torch.Size([1, 1000, 754]) | 8\n",
-      "21773    | torch.Size([1, 1000, 754]) | 8\n",
-      "21774    | torch.Size([1, 1000, 781]) | 8\n",
-      "21775    | torch.Size([1, 1000, 777]) | 8\n",
-      "21776    | torch.Size([1, 1000, 754]) | 8\n",
-      "21777    | torch.Size([1, 1000, 844]) | 8\n",
-      "21778    | torch.Size([1, 1000, 754]) | 8\n",
-      "21779    | torch.Size([1, 1000, 754]) | 8\n",
-      "21780    | torch.Size([1, 1000, 762]) | 8\n",
-      "21781    | torch.Size([1, 1000, 762]) | 8\n",
-      "21782    | torch.Size([1, 1000, 777]) | 8\n",
-      "21783    | torch.Size([1, 1000, 769]) | 8\n",
-      "21784    | torch.Size([1, 1000, 762]) | 8\n",
-      "21785    | torch.Size([1, 1000, 782]) | 8\n",
-      "21786    | torch.Size([1, 1000, 777]) | 8\n",
-      "21787    | torch.Size([1, 1000, 777]) | 8\n",
-      "21788    | torch.Size([1, 1000, 762]) | 8\n",
-      "21789    | torch.Size([1, 1000, 762]) | 8\n",
-      "21790    | torch.Size([1, 1000, 777]) | 8\n",
-      "21791    | torch.Size([1, 1000, 754]) | 8\n",
-      "21792    | torch.Size([1, 1000, 799]) | 8\n",
-      "21793    | torch.Size([1, 1000, 762]) | 8\n",
-      "21794    | torch.Size([1, 1000, 767]) | 8\n",
-      "21795    | torch.Size([1, 1000, 762]) | 8\n",
-      "21796    | torch.Size([1, 1000, 777]) | 8\n",
-      "21797    | torch.Size([1, 1000, 796]) | 8\n",
-      "21798    | torch.Size([1, 1000, 762]) | 8\n",
-      "21799    | torch.Size([1, 1000, 754]) | 8\n",
-      "21800    | torch.Size([1, 1000, 775]) | 8\n",
-      "21801    | torch.Size([1, 1000, 754]) | 8\n",
-      "21802    | torch.Size([1, 1000, 762]) | 8\n",
-      "21803    | torch.Size([1, 1000, 777]) | 8\n",
-      "21804    | torch.Size([1, 1000, 754]) | 8\n",
-      "21805    | torch.Size([1, 1000, 762]) | 8\n",
-      "21806    | torch.Size([1, 1000, 774]) | 8\n",
-      "21807    | torch.Size([1, 1000, 785]) | 8\n",
-      "21808    | torch.Size([1, 1000, 765]) | 8\n",
-      "21809    | torch.Size([1, 1000, 762]) | 8\n",
-      "21810    | torch.Size([1, 1000, 777]) | 8\n",
-      "21811    | torch.Size([1, 1000, 762]) | 8\n",
-      "21812    | torch.Size([1, 1000, 762]) | 8\n",
-      "21813    | torch.Size([1, 1000, 762]) | 8\n",
-      "21814    | torch.Size([1, 1000, 762]) | 8\n",
-      "21815    | torch.Size([1, 1000, 790]) | 8\n",
-      "21816    | torch.Size([1, 1000, 754]) | 8\n",
-      "21817    | torch.Size([1, 1000, 779]) | 8\n",
-      "21818    | torch.Size([1, 1000, 772]) | 8\n",
-      "21819    | torch.Size([1, 1000, 815]) | 8\n",
-      "21820    | torch.Size([1, 1000, 783]) | 8\n",
-      "21821    | torch.Size([1, 1000, 864]) | 8\n",
-      "21822    | torch.Size([1, 1000, 777]) | 8\n",
-      "21823    | torch.Size([1, 1000, 754]) | 8\n",
-      "21824    | torch.Size([1, 1000, 762]) | 8\n",
-      "21825    | torch.Size([1, 1000, 771]) | 8\n",
-      "21826    | torch.Size([1, 1000, 754]) | 8\n",
-      "21827    | torch.Size([1, 1000, 824]) | 8\n",
-      "21828    | torch.Size([1, 1000, 762]) | 8\n",
-      "21829    | torch.Size([1, 1000, 761]) | 8\n",
-      "21830    | torch.Size([1, 1000, 762]) | 8\n",
-      "21831    | torch.Size([1, 1000, 762]) | 8\n",
-      "21832    | torch.Size([1, 1000, 795]) | 8\n",
-      "21833    | torch.Size([1, 1000, 762]) | 8\n",
-      "21834    | torch.Size([1, 1000, 809]) | 8\n",
-      "21835    | torch.Size([1, 1000, 762]) | 8\n",
-      "21836    | torch.Size([1, 1000, 777]) | 8\n",
-      "21837    | torch.Size([1, 1000, 777]) | 8\n",
-      "21838    | torch.Size([1, 1000, 754]) | 8\n",
-      "21839    | torch.Size([1, 1000, 777]) | 8\n",
-      "21840    | torch.Size([1, 1000, 754]) | 8\n",
-      "21841    | torch.Size([1, 1000, 762]) | 8\n",
-      "21842    | torch.Size([1, 1000, 802]) | 8\n",
-      "21843    | torch.Size([1, 1000, 777]) | 8\n",
-      "21844    | torch.Size([1, 1000, 762]) | 8\n",
-      "21845    | torch.Size([1, 1000, 754]) | 8\n",
-      "21846    | torch.Size([1, 1000, 782]) | 8\n",
-      "21847    | torch.Size([1, 1000, 754]) | 8\n",
-      "21848    | torch.Size([1, 1000, 754]) | 8\n",
-      "21849    | torch.Size([1, 1000, 762]) | 8\n",
-      "21850    | torch.Size([1, 1000, 777]) | 8\n",
-      "21851    | torch.Size([1, 1000, 869]) | 8\n",
-      "21852    | torch.Size([1, 1000, 777]) | 8\n",
-      "21853    | torch.Size([1, 1000, 754]) | 8\n",
-      "21854    | torch.Size([1, 1000, 754]) | 8\n",
-      "21855    | torch.Size([1, 1000, 762]) | 8\n",
-      "21856    | torch.Size([1, 1000, 762]) | 8\n",
-      "21857    | torch.Size([1, 1000, 786]) | 8\n",
-      "21858    | torch.Size([1, 1000, 762]) | 8\n",
-      "21859    | torch.Size([1, 1000, 754]) | 8\n",
-      "21860    | torch.Size([1, 1000, 754]) | 8\n",
-      "21861    | torch.Size([1, 1000, 762]) | 8\n",
-      "21862    | torch.Size([1, 1000, 754]) | 8\n",
-      "21863    | torch.Size([1, 1000, 762]) | 8\n",
-      "21864    | torch.Size([1, 1000, 777]) | 8\n",
-      "21865    | torch.Size([1, 1000, 754]) | 8\n",
-      "21866    | torch.Size([1, 1000, 779]) | 8\n",
-      "21867    | torch.Size([1, 1000, 769]) | 8\n",
-      "21868    | torch.Size([1, 1000, 754]) | 8\n",
-      "21869    | torch.Size([1, 1000, 754]) | 8\n",
-      "21870    | torch.Size([1, 1000, 762]) | 8\n",
-      "21871    | torch.Size([1, 1000, 845]) | 8\n",
-      "21872    | torch.Size([1, 1000, 777]) | 8\n",
-      "21873    | torch.Size([1, 1000, 762]) | 8\n",
-      "21874    | torch.Size([1, 1000, 772]) | 8\n",
-      "21875    | torch.Size([1, 1000, 777]) | 8\n",
-      "21876    | torch.Size([1, 1000, 797]) | 8\n",
-      "21877    | torch.Size([1, 1000, 779]) | 8\n",
-      "21878    | torch.Size([1, 1000, 770]) | 8\n",
-      "21879    | torch.Size([1, 1000, 762]) | 8\n",
-      "21880    | torch.Size([1, 1000, 754]) | 8\n",
-      "21881    | torch.Size([1, 1000, 777]) | 8\n",
-      "21882    | torch.Size([1, 1000, 762]) | 8\n",
-      "21883    | torch.Size([1, 1000, 777]) | 8\n",
-      "21884    | torch.Size([1, 1000, 798]) | 8\n",
-      "21885    | torch.Size([1, 1000, 762]) | 8\n",
-      "21886    | torch.Size([1, 1000, 762]) | 8\n",
-      "21887    | torch.Size([1, 1000, 771]) | 8\n",
-      "21888    | torch.Size([1, 1000, 731]) | 8\n",
-      "21889    | torch.Size([1, 1000, 784]) | 8\n",
-      "21890    | torch.Size([1, 1000, 762]) | 8\n",
-      "21891    | torch.Size([1, 1000, 762]) | 8\n",
-      "21892    | torch.Size([1, 1000, 768]) | 8\n",
-      "21893    | torch.Size([1, 1000, 762]) | 8\n",
-      "21894    | torch.Size([1, 1000, 762]) | 8\n",
-      "21895    | torch.Size([1, 1000, 790]) | 8\n",
-      "21896    | torch.Size([1, 1000, 847]) | 8\n",
-      "21897    | torch.Size([1, 1000, 774]) | 8\n",
-      "21898    | torch.Size([1, 1000, 785]) | 8\n",
-      "21899    | torch.Size([1, 1000, 762]) | 8\n",
-      "21900    | torch.Size([1, 1000, 771]) | 8\n",
-      "21901    | torch.Size([1, 1000, 767]) | 8\n",
-      "21902    | torch.Size([1, 1000, 771]) | 8\n",
-      "21903    | torch.Size([1, 1000, 777]) | 8\n",
-      "21904    | torch.Size([1, 1000, 762]) | 8\n",
-      "21905    | torch.Size([1, 1000, 767]) | 8\n",
-      "21906    | torch.Size([1, 1000, 754]) | 8\n",
-      "21907    | torch.Size([1, 1000, 754]) | 8\n",
-      "21908    | torch.Size([1, 1000, 847]) | 8\n",
-      "21909    | torch.Size([1, 1000, 754]) | 8\n",
-      "21910    | torch.Size([1, 1000, 762]) | 8\n",
-      "21911    | torch.Size([1, 1000, 772]) | 8\n",
-      "21912    | torch.Size([1, 1000, 762]) | 8\n",
-      "21913    | torch.Size([1, 1000, 777]) | 8\n",
-      "21914    | torch.Size([1, 1000, 754]) | 8\n",
-      "21915    | torch.Size([1, 1000, 762]) | 8\n",
-      "21916    | torch.Size([1, 1000, 762]) | 8\n",
-      "21917    | torch.Size([1, 1000, 762]) | 8\n",
-      "21918    | torch.Size([1, 1000, 754]) | 8\n",
-      "21919    | torch.Size([1, 1000, 762]) | 8\n",
-      "21920    | torch.Size([1, 1000, 754]) | 8\n",
-      "21921    | torch.Size([1, 1000, 762]) | 8\n",
-      "21922    | torch.Size([1, 1000, 754]) | 8\n",
-      "21923    | torch.Size([1, 1000, 762]) | 8\n",
-      "21924    | torch.Size([1, 1000, 762]) | 8\n",
-      "21925    | torch.Size([1, 1000, 754]) | 8\n",
-      "21926    | torch.Size([1, 1000, 762]) | 8\n",
-      "21927    | torch.Size([1, 1000, 767]) | 8\n",
-      "21928    | torch.Size([1, 1000, 762]) | 8\n",
-      "21929    | torch.Size([1, 1000, 777]) | 8\n",
-      "21930    | torch.Size([1, 1000, 788]) | 8\n",
-      "21931    | torch.Size([1, 1000, 754]) | 8\n",
-      "21932    | torch.Size([1, 1000, 764]) | 8\n",
-      "21933    | torch.Size([1, 1000, 754]) | 8\n",
-      "21934    | torch.Size([1, 1000, 767]) | 8\n",
-      "21935    | torch.Size([1, 1000, 777]) | 8\n",
-      "21936    | torch.Size([1, 1000, 806]) | 8\n",
-      "21937    | torch.Size([1, 1000, 777]) | 8\n",
-      "21938    | torch.Size([1, 1000, 777]) | 8\n",
-      "21939    | torch.Size([1, 1000, 762]) | 8\n",
-      "21940    | torch.Size([1, 1000, 771]) | 8\n",
-      "21941    | torch.Size([1, 1000, 805]) | 8\n",
-      "21942    | torch.Size([1, 1000, 762]) | 8\n",
-      "21943    | torch.Size([1, 1000, 829]) | 8\n",
-      "21944    | torch.Size([1, 1000, 777]) | 8\n",
-      "21945    | torch.Size([1, 1000, 754]) | 8\n",
-      "21946    | torch.Size([1, 1000, 754]) | 8\n",
-      "21947    | torch.Size([1, 1000, 777]) | 8\n",
-      "21948    | torch.Size([1, 1000, 762]) | 8\n",
-      "21949    | torch.Size([1, 1000, 754]) | 8\n",
-      "21950    | torch.Size([1, 1000, 762]) | 8\n",
-      "21951    | torch.Size([1, 1000, 754]) | 8\n",
-      "21952    | torch.Size([1, 1000, 777]) | 8\n",
-      "21953    | torch.Size([1, 1000, 762]) | 8\n",
-      "21954    | torch.Size([1, 1000, 777]) | 8\n",
-      "21955    | torch.Size([1, 1000, 762]) | 8\n",
-      "21956    | torch.Size([1, 1000, 762]) | 8\n",
-      "21957    | torch.Size([1, 1000, 762]) | 8\n",
-      "21958    | torch.Size([1, 1000, 754]) | 8\n",
-      "21959    | torch.Size([1, 1000, 754]) | 8\n",
-      "21960    | torch.Size([1, 1000, 769]) | 8\n",
-      "21961    | torch.Size([1, 1000, 754]) | 8\n",
-      "21962    | torch.Size([1, 1000, 761]) | 8\n",
-      "21963    | torch.Size([1, 1000, 762]) | 8\n",
-      "21964    | torch.Size([1, 1000, 802]) | 8\n",
-      "21965    | torch.Size([1, 1000, 754]) | 8\n",
-      "21966    | torch.Size([1, 1000, 768]) | 8\n",
-      "21967    | torch.Size([1, 1000, 754]) | 8\n",
-      "21968    | torch.Size([1, 1000, 762]) | 8\n",
-      "21969    | torch.Size([1, 1000, 777]) | 8\n",
-      "21970    | torch.Size([1, 1000, 762]) | 8\n",
-      "21971    | torch.Size([1, 1000, 762]) | 8\n",
-      "21972    | torch.Size([1, 1000, 762]) | 8\n",
-      "21973    | torch.Size([1, 1000, 773]) | 8\n",
-      "21974    | torch.Size([1, 1000, 786]) | 8\n",
-      "21975    | torch.Size([1, 1000, 777]) | 8\n",
-      "21976    | torch.Size([1, 1000, 777]) | 8\n",
-      "21977    | torch.Size([1, 1000, 762]) | 8\n",
-      "21978    | torch.Size([1, 1000, 762]) | 8\n",
-      "21979    | torch.Size([1, 1000, 754]) | 8\n",
-      "21980    | torch.Size([1, 1000, 754]) | 8\n",
-      "21981    | torch.Size([1, 1000, 754]) | 8\n",
-      "21982    | torch.Size([1, 1000, 785]) | 8\n",
-      "21983    | torch.Size([1, 1000, 788]) | 8\n",
-      "21984    | torch.Size([1, 1000, 827]) | 8\n",
-      "21985    | torch.Size([1, 1000, 782]) | 8\n",
-      "21986    | torch.Size([1, 1000, 754]) | 8\n",
-      "21987    | torch.Size([1, 1000, 781]) | 8\n",
-      "21988    | torch.Size([1, 1000, 777]) | 8\n",
-      "21989    | torch.Size([1, 1000, 762]) | 8\n",
-      "21990    | torch.Size([1, 1000, 775]) | 8\n",
-      "21991    | torch.Size([1, 1000, 754]) | 8\n",
-      "21992    | torch.Size([1, 1000, 762]) | 8\n",
-      "21993    | torch.Size([1, 1000, 762]) | 8\n",
-      "21994    | torch.Size([1, 1000, 762]) | 8\n",
-      "21995    | torch.Size([1, 1000, 777]) | 8\n",
-      "21996    | torch.Size([1, 1000, 777]) | 8\n",
-      "21997    | torch.Size([1, 1000, 762]) | 8\n",
-      "21998    | torch.Size([1, 1000, 844]) | 8\n",
-      "21999    | torch.Size([1, 1000, 773]) | 8\n",
-      "22000    | torch.Size([1, 1000, 762]) | 8\n",
-      "22001    | torch.Size([1, 1000, 754]) | 8\n",
-      "22002    | torch.Size([1, 1000, 754]) | 8\n",
-      "22003    | torch.Size([1, 1000, 762]) | 8\n",
-      "22004    | torch.Size([1, 1000, 781]) | 8\n",
-      "22005    | torch.Size([1, 1000, 777]) | 8\n",
-      "22006    | torch.Size([1, 1000, 754]) | 8\n",
-      "22007    | torch.Size([1, 1000, 754]) | 8\n",
-      "22008    | torch.Size([1, 1000, 762]) | 8\n",
-      "22009    | torch.Size([1, 1000, 777]) | 8\n",
-      "22010    | torch.Size([1, 1000, 762]) | 8\n",
-      "22011    | torch.Size([1, 1000, 781]) | 8\n",
-      "22012    | torch.Size([1, 1000, 782]) | 8\n",
-      "22013    | torch.Size([1, 1000, 781]) | 8\n",
-      "22014    | torch.Size([1, 1000, 762]) | 8\n",
-      "22015    | torch.Size([1, 1000, 762]) | 8\n",
-      "22016    | torch.Size([1, 1000, 754]) | 8\n",
-      "22017    | torch.Size([1, 1000, 762]) | 8\n",
-      "22018    | torch.Size([1, 1000, 767]) | 8\n",
-      "22019    | torch.Size([1, 1000, 754]) | 8\n",
-      "22020    | torch.Size([1, 1000, 765]) | 8\n",
-      "22021    | torch.Size([1, 1000, 762]) | 8\n",
-      "22022    | torch.Size([1, 1000, 772]) | 8\n",
-      "22023    | torch.Size([1, 1000, 805]) | 8\n",
-      "22024    | torch.Size([1, 1000, 762]) | 8\n",
-      "22025    | torch.Size([1, 1000, 754]) | 8\n",
-      "22026    | torch.Size([1, 1000, 762]) | 8\n",
-      "22027    | torch.Size([1, 1000, 777]) | 8\n",
-      "22028    | torch.Size([1, 1000, 662]) | 8\n",
-      "22029    | torch.Size([1, 1000, 762]) | 8\n",
-      "22030    | torch.Size([1, 1000, 779]) | 8\n",
-      "22031    | torch.Size([1, 1000, 777]) | 8\n",
-      "22032    | torch.Size([1, 1000, 777]) | 8\n",
-      "22033    | torch.Size([1, 1000, 762]) | 8\n",
-      "22034    | torch.Size([1, 1000, 771]) | 8\n",
-      "22035    | torch.Size([1, 1000, 754]) | 8\n",
-      "22036    | torch.Size([1, 1000, 842]) | 8\n",
-      "22037    | torch.Size([1, 1000, 762]) | 8\n",
-      "22038    | torch.Size([1, 1000, 754]) | 8\n",
-      "22039    | torch.Size([1, 1000, 762]) | 8\n",
-      "22040    | torch.Size([1, 1000, 762]) | 8\n",
-      "22041    | torch.Size([1, 1000, 754]) | 8\n",
-      "22042    | torch.Size([1, 1000, 802]) | 8\n",
-      "22043    | torch.Size([1, 1000, 782]) | 8\n",
-      "22044    | torch.Size([1, 1000, 777]) | 8\n",
-      "22045    | torch.Size([1, 1000, 777]) | 8\n",
-      "22046    | torch.Size([1, 1000, 754]) | 8\n",
-      "22047    | torch.Size([1, 1000, 754]) | 8\n",
-      "22048    | torch.Size([1, 1000, 777]) | 8\n",
-      "22049    | torch.Size([1, 1000, 754]) | 8\n",
-      "22050    | torch.Size([1, 1000, 754]) | 8\n",
-      "22051    | torch.Size([1, 1000, 777]) | 8\n",
-      "22052    | torch.Size([1, 1000, 754]) | 8\n",
-      "22053    | torch.Size([1, 1000, 762]) | 8\n",
-      "22054    | torch.Size([1, 1000, 771]) | 8\n",
-      "22055    | torch.Size([1, 1000, 796]) | 8\n",
-      "22056    | torch.Size([1, 1000, 776]) | 8\n",
-      "22057    | torch.Size([1, 1000, 777]) | 8\n",
-      "22058    | torch.Size([1, 1000, 777]) | 8\n",
-      "22059    | torch.Size([1, 1000, 754]) | 8\n",
-      "22060    | torch.Size([1, 1000, 792]) | 8\n",
-      "22061    | torch.Size([1, 1000, 762]) | 8\n",
-      "22062    | torch.Size([1, 1000, 754]) | 8\n",
-      "22063    | torch.Size([1, 1000, 778]) | 8\n",
-      "22064    | torch.Size([1, 1000, 754]) | 8\n",
-      "22065    | torch.Size([1, 1000, 788]) | 8\n",
-      "22066    | torch.Size([1, 1000, 783]) | 8\n",
-      "22067    | torch.Size([1, 1000, 772]) | 8\n",
-      "22068    | torch.Size([1, 1000, 775]) | 8\n",
-      "22069    | torch.Size([1, 1000, 762]) | 8\n",
-      "22070    | torch.Size([1, 1000, 754]) | 8\n",
-      "22071    | torch.Size([1, 1000, 762]) | 8\n",
-      "22072    | torch.Size([1, 1000, 754]) | 8\n",
-      "22073    | torch.Size([1, 1000, 777]) | 8\n",
-      "22074    | torch.Size([1, 1000, 844]) | 8\n",
-      "22075    | torch.Size([1, 1000, 754]) | 8\n",
-      "22076    | torch.Size([1, 1000, 773]) | 8\n",
-      "22077    | torch.Size([1, 1000, 777]) | 8\n",
-      "22078    | torch.Size([1, 1000, 754]) | 8\n",
-      "22079    | torch.Size([1, 1000, 777]) | 8\n",
-      "22080    | torch.Size([1, 1000, 771]) | 8\n",
-      "22081    | torch.Size([1, 1000, 777]) | 8\n",
-      "22082    | torch.Size([1, 1000, 762]) | 8\n",
-      "22083    | torch.Size([1, 1000, 777]) | 8\n",
-      "22084    | torch.Size([1, 1000, 754]) | 8\n",
-      "22085    | torch.Size([1, 1000, 762]) | 8\n",
-      "22086    | torch.Size([1, 1000, 777]) | 8\n",
-      "22087    | torch.Size([1, 1000, 754]) | 8\n",
-      "22088    | torch.Size([1, 1000, 762]) | 8\n",
-      "22089    | torch.Size([1, 1000, 762]) | 8\n",
-      "22090    | torch.Size([1, 1000, 762]) | 8\n",
-      "22091    | torch.Size([1, 1000, 762]) | 8\n",
-      "22092    | torch.Size([1, 1000, 754]) | 8\n",
-      "22093    | torch.Size([1, 1000, 776]) | 8\n",
-      "22094    | torch.Size([1, 1000, 787]) | 8\n",
-      "22095    | torch.Size([1, 1000, 771]) | 8\n",
-      "22096    | torch.Size([1, 1000, 762]) | 8\n",
-      "22097    | torch.Size([1, 1000, 754]) | 8\n",
-      "22098    | torch.Size([1, 1000, 777]) | 8\n",
-      "22099    | torch.Size([1, 1000, 762]) | 8\n",
-      "22100    | torch.Size([1, 1000, 762]) | 8\n",
-      "22101    | torch.Size([1, 1000, 754]) | 8\n",
-      "22102    | torch.Size([1, 1000, 767]) | 8\n",
-      "22103    | torch.Size([1, 1000, 777]) | 8\n",
-      "22104    | torch.Size([1, 1000, 754]) | 8\n",
-      "22105    | torch.Size([1, 1000, 802]) | 8\n",
-      "22106    | torch.Size([1, 1000, 771]) | 8\n",
-      "22107    | torch.Size([1, 1000, 777]) | 8\n",
-      "22108    | torch.Size([1, 1000, 754]) | 8\n",
-      "22109    | torch.Size([1, 1000, 798]) | 8\n",
-      "22110    | torch.Size([1, 1000, 762]) | 8\n",
-      "22111    | torch.Size([1, 1000, 777]) | 8\n",
-      "22112    | torch.Size([1, 1000, 777]) | 8\n",
-      "22113    | torch.Size([1, 1000, 777]) | 8\n",
-      "22114    | torch.Size([1, 1000, 754]) | 8\n",
-      "22115    | torch.Size([1, 1000, 808]) | 8\n",
-      "22116    | torch.Size([1, 1000, 777]) | 8\n",
-      "22117    | torch.Size([1, 1000, 754]) | 8\n",
-      "22118    | torch.Size([1, 1000, 777]) | 8\n",
-      "22119    | torch.Size([1, 1000, 762]) | 8\n",
-      "22120    | torch.Size([1, 1000, 777]) | 8\n",
-      "22121    | torch.Size([1, 1000, 754]) | 8\n",
-      "22122    | torch.Size([1, 1000, 754]) | 8\n",
-      "22123    | torch.Size([1, 1000, 776]) | 8\n",
-      "22124    | torch.Size([1, 1000, 754]) | 8\n",
-      "22125    | torch.Size([1, 1000, 762]) | 8\n",
-      "22126    | torch.Size([1, 1000, 802]) | 8\n",
-      "22127    | torch.Size([1, 1000, 777]) | 8\n",
-      "22128    | torch.Size([1, 1000, 762]) | 8\n",
-      "22129    | torch.Size([1, 1000, 809]) | 8\n",
-      "22130    | torch.Size([1, 1000, 762]) | 8\n",
-      "22131    | torch.Size([1, 1000, 762]) | 8\n",
-      "22132    | torch.Size([1, 1000, 754]) | 8\n",
-      "22133    | torch.Size([1, 1000, 777]) | 8\n",
-      "22134    | torch.Size([1, 1000, 762]) | 8\n",
-      "22135    | torch.Size([1, 1000, 754]) | 8\n",
-      "22136    | torch.Size([1, 1000, 779]) | 8\n",
-      "22137    | torch.Size([1, 1000, 762]) | 8\n",
-      "22138    | torch.Size([1, 1000, 754]) | 8\n",
-      "22139    | torch.Size([1, 1000, 770]) | 8\n",
-      "22140    | torch.Size([1, 1000, 780]) | 8\n",
-      "22141    | torch.Size([1, 1000, 754]) | 8\n",
-      "22142    | torch.Size([1, 1000, 867]) | 8\n",
-      "22143    | torch.Size([1, 1000, 762]) | 8\n",
-      "22144    | torch.Size([1, 1000, 868]) | 8\n",
-      "22145    | torch.Size([1, 1000, 762]) | 8\n",
-      "22146    | torch.Size([1, 1000, 781]) | 8\n",
-      "22147    | torch.Size([1, 1000, 754]) | 8\n",
-      "22148    | torch.Size([1, 1000, 754]) | 8\n",
-      "22149    | torch.Size([1, 1000, 762]) | 8\n",
-      "22150    | torch.Size([1, 1000, 762]) | 8\n",
-      "22151    | torch.Size([1, 1000, 793]) | 8\n",
-      "22152    | torch.Size([1, 1000, 762]) | 8\n",
-      "22153    | torch.Size([1, 1000, 776]) | 8\n",
-      "22154    | torch.Size([1, 1000, 762]) | 8\n",
-      "22155    | torch.Size([1, 1000, 777]) | 8\n",
-      "22156    | torch.Size([1, 1000, 769]) | 8\n",
-      "22157    | torch.Size([1, 1000, 773]) | 8\n",
-      "22158    | torch.Size([1, 1000, 777]) | 8\n",
-      "22159    | torch.Size([1, 1000, 777]) | 8\n",
-      "22160    | torch.Size([1, 1000, 762]) | 8\n",
-      "22161    | torch.Size([1, 1000, 762]) | 8\n",
-      "22162    | torch.Size([1, 1000, 774]) | 8\n",
-      "22163    | torch.Size([1, 1000, 754]) | 8\n",
-      "22164    | torch.Size([1, 1000, 754]) | 8\n",
-      "22165    | torch.Size([1, 1000, 777]) | 8\n",
-      "22166    | torch.Size([1, 1000, 777]) | 8\n",
-      "22167    | torch.Size([1, 1000, 762]) | 8\n",
-      "22168    | torch.Size([1, 1000, 762]) | 8\n",
-      "22169    | torch.Size([1, 1000, 777]) | 8\n",
-      "22170    | torch.Size([1, 1000, 773]) | 8\n",
-      "22171    | torch.Size([1, 1000, 843]) | 8\n",
-      "22172    | torch.Size([1, 1000, 787]) | 8\n",
-      "22173    | torch.Size([1, 1000, 783]) | 8\n",
-      "22174    | torch.Size([1, 1000, 762]) | 8\n",
-      "22175    | torch.Size([1, 1000, 779]) | 8\n",
-      "22176    | torch.Size([1, 1000, 754]) | 8\n",
-      "22177    | torch.Size([1, 1000, 754]) | 8\n",
-      "22178    | torch.Size([1, 1000, 762]) | 8\n",
-      "22179    | torch.Size([1, 1000, 754]) | 8\n",
-      "22180    | torch.Size([1, 1000, 762]) | 8\n",
-      "22181    | torch.Size([1, 1000, 762]) | 8\n",
-      "22182    | torch.Size([1, 1000, 777]) | 8\n",
-      "22183    | torch.Size([1, 1000, 762]) | 8\n",
-      "22184    | torch.Size([1, 1000, 819]) | 8\n",
-      "22185    | torch.Size([1, 1000, 802]) | 8\n",
-      "22186    | torch.Size([1, 1000, 777]) | 8\n",
-      "22187    | torch.Size([1, 1000, 788]) | 8\n",
-      "22188    | torch.Size([1, 1000, 754]) | 8\n",
-      "22189    | torch.Size([1, 1000, 781]) | 8\n",
-      "22190    | torch.Size([1, 1000, 762]) | 8\n",
-      "22191    | torch.Size([1, 1000, 777]) | 8\n",
-      "22192    | torch.Size([1, 1000, 772]) | 8\n",
-      "22193    | torch.Size([1, 1000, 762]) | 8\n",
-      "22194    | torch.Size([1, 1000, 845]) | 8\n",
-      "22195    | torch.Size([1, 1000, 762]) | 8\n",
-      "22196    | torch.Size([1, 1000, 754]) | 8\n",
-      "22197    | torch.Size([1, 1000, 808]) | 8\n",
-      "22198    | torch.Size([1, 1000, 761]) | 8\n",
-      "22199    | torch.Size([1, 1000, 754]) | 8\n",
-      "22200    | torch.Size([1, 1000, 772]) | 8\n",
-      "22201    | torch.Size([1, 1000, 754]) | 8\n",
-      "22202    | torch.Size([1, 1000, 754]) | 8\n",
-      "22203    | torch.Size([1, 1000, 754]) | 8\n",
-      "22204    | torch.Size([1, 1000, 754]) | 8\n",
-      "22205    | torch.Size([1, 1000, 762]) | 8\n",
-      "22206    | torch.Size([1, 1000, 754]) | 8\n",
-      "22207    | torch.Size([1, 1000, 762]) | 8\n",
-      "22208    | torch.Size([1, 1000, 754]) | 8\n",
-      "22209    | torch.Size([1, 1000, 780]) | 8\n",
-      "22210    | torch.Size([1, 1000, 771]) | 8\n",
-      "22211    | torch.Size([1, 1000, 762]) | 8\n",
-      "22212    | torch.Size([1, 1000, 762]) | 8\n",
-      "22213    | torch.Size([1, 1000, 773]) | 8\n",
-      "22214    | torch.Size([1, 1000, 762]) | 8\n",
-      "22215    | torch.Size([1, 1000, 842]) | 8\n",
-      "22216    | torch.Size([1, 1000, 778]) | 8\n",
-      "22217    | torch.Size([1, 1000, 777]) | 8\n",
-      "22218    | torch.Size([1, 1000, 777]) | 8\n",
-      "22219    | torch.Size([1, 1000, 762]) | 8\n",
-      "22220    | torch.Size([1, 1000, 762]) | 8\n",
-      "22221    | torch.Size([1, 1000, 777]) | 8\n",
-      "22222    | torch.Size([1, 1000, 861]) | 8\n",
-      "22223    | torch.Size([1, 1000, 754]) | 8\n",
-      "22224    | torch.Size([1, 1000, 777]) | 8\n",
-      "22225    | torch.Size([1, 1000, 838]) | 8\n",
-      "22226    | torch.Size([1, 1000, 754]) | 8\n",
-      "22227    | torch.Size([1, 1000, 754]) | 8\n",
-      "22228    | torch.Size([1, 1000, 754]) | 8\n",
-      "22229    | torch.Size([1, 1000, 777]) | 8\n",
-      "22230    | torch.Size([1, 1000, 754]) | 8\n",
-      "22231    | torch.Size([1, 1000, 781]) | 8\n",
-      "22232    | torch.Size([1, 1000, 777]) | 8\n",
-      "22233    | torch.Size([1, 1000, 771]) | 8\n",
-      "22234    | torch.Size([1, 1000, 771]) | 8\n",
-      "22235    | torch.Size([1, 1000, 762]) | 8\n",
-      "22236    | torch.Size([1, 1000, 762]) | 8\n",
-      "22237    | torch.Size([1, 1000, 762]) | 8\n",
-      "22238    | torch.Size([1, 1000, 762]) | 8\n",
-      "22239    | torch.Size([1, 1000, 776]) | 8\n",
-      "22240    | torch.Size([1, 1000, 754]) | 8\n",
-      "22241    | torch.Size([1, 1000, 762]) | 8\n",
-      "22242    | torch.Size([1, 1000, 754]) | 8\n",
-      "22243    | torch.Size([1, 1000, 762]) | 8\n",
-      "22244    | torch.Size([1, 1000, 762]) | 8\n",
-      "22245    | torch.Size([1, 1000, 762]) | 8\n",
-      "22246    | torch.Size([1, 1000, 762]) | 8\n",
-      "22247    | torch.Size([1, 1000, 777]) | 8\n",
-      "22248    | torch.Size([1, 1000, 775]) | 8\n",
-      "22249    | torch.Size([1, 1000, 776]) | 8\n",
-      "22250    | torch.Size([1, 1000, 784]) | 8\n",
-      "22251    | torch.Size([1, 1000, 762]) | 8\n",
-      "22252    | torch.Size([1, 1000, 771]) | 8\n",
-      "22253    | torch.Size([1, 1000, 777]) | 8\n",
-      "22254    | torch.Size([1, 1000, 781]) | 8\n",
-      "22255    | torch.Size([1, 1000, 752]) | 8\n",
-      "22256    | torch.Size([1, 1000, 754]) | 8\n",
-      "22257    | torch.Size([1, 1000, 765]) | 8\n",
-      "22258    | torch.Size([1, 1000, 754]) | 8\n",
-      "22259    | torch.Size([1, 1000, 777]) | 8\n",
-      "22260    | torch.Size([1, 1000, 799]) | 8\n",
-      "22261    | torch.Size([1, 1000, 762]) | 8\n",
-      "22262    | torch.Size([1, 1000, 807]) | 8\n",
-      "22263    | torch.Size([1, 1000, 762]) | 8\n",
-      "22264    | torch.Size([1, 1000, 762]) | 8\n",
-      "22265    | torch.Size([1, 1000, 762]) | 8\n",
-      "22266    | torch.Size([1, 1000, 754]) | 8\n",
-      "22267    | torch.Size([1, 1000, 754]) | 8\n",
-      "22268    | torch.Size([1, 1000, 754]) | 8\n",
-      "22269    | torch.Size([1, 1000, 754]) | 8\n",
-      "22270    | torch.Size([1, 1000, 798]) | 8\n",
-      "22271    | torch.Size([1, 1000, 777]) | 8\n",
-      "22272    | torch.Size([1, 1000, 777]) | 8\n",
-      "22273    | torch.Size([1, 1000, 762]) | 8\n",
-      "22274    | torch.Size([1, 1000, 776]) | 8\n",
-      "22275    | torch.Size([1, 1000, 762]) | 8\n",
-      "22276    | torch.Size([1, 1000, 794]) | 8\n",
-      "22277    | torch.Size([1, 1000, 754]) | 8\n",
-      "22278    | torch.Size([1, 1000, 762]) | 8\n",
-      "22279    | torch.Size([1, 1000, 754]) | 8\n",
-      "22280    | torch.Size([1, 1000, 778]) | 8\n",
-      "22281    | torch.Size([1, 1000, 762]) | 8\n",
-      "22282    | torch.Size([1, 1000, 768]) | 8\n",
-      "22283    | torch.Size([1, 1000, 754]) | 8\n",
-      "22284    | torch.Size([1, 1000, 762]) | 8\n",
-      "22285    | torch.Size([1, 1000, 762]) | 8\n",
-      "22286    | torch.Size([1, 1000, 762]) | 8\n",
-      "22287    | torch.Size([1, 1000, 762]) | 8\n",
-      "22288    | torch.Size([1, 1000, 775]) | 8\n",
-      "22289    | torch.Size([1, 1000, 777]) | 8\n",
-      "22290    | torch.Size([1, 1000, 762]) | 8\n",
-      "22291    | torch.Size([1, 1000, 754]) | 8\n",
-      "22292    | torch.Size([1, 1000, 754]) | 8\n",
-      "22293    | torch.Size([1, 1000, 845]) | 8\n",
-      "22294    | torch.Size([1, 1000, 754]) | 8\n",
-      "22295    | torch.Size([1, 1000, 777]) | 8\n",
-      "22296    | torch.Size([1, 1000, 762]) | 8\n",
-      "22297    | torch.Size([1, 1000, 762]) | 8\n",
-      "22298    | torch.Size([1, 1000, 762]) | 8\n",
-      "22299    | torch.Size([1, 1000, 754]) | 8\n",
-      "22300    | torch.Size([1, 1000, 791]) | 8\n",
-      "22301    | torch.Size([1, 1000, 762]) | 8\n",
-      "22302    | torch.Size([1, 1000, 762]) | 8\n",
-      "22303    | torch.Size([1, 1000, 762]) | 8\n",
-      "22304    | torch.Size([1, 1000, 762]) | 8\n",
-      "22305    | torch.Size([1, 1000, 768]) | 8\n",
-      "22306    | torch.Size([1, 1000, 754]) | 8\n",
-      "22307    | torch.Size([1, 1000, 777]) | 8\n",
-      "22308    | torch.Size([1, 1000, 840]) | 8\n",
-      "22309    | torch.Size([1, 1000, 762]) | 8\n",
-      "22310    | torch.Size([1, 1000, 762]) | 8\n",
-      "22311    | torch.Size([1, 1000, 754]) | 8\n",
-      "22312    | torch.Size([1, 1000, 777]) | 8\n",
-      "22313    | torch.Size([1, 1000, 762]) | 8\n",
-      "22314    | torch.Size([1, 1000, 762]) | 8\n",
-      "22315    | torch.Size([1, 1000, 794]) | 8\n",
-      "22316    | torch.Size([1, 1000, 762]) | 8\n",
-      "22317    | torch.Size([1, 1000, 762]) | 8\n",
-      "22318    | torch.Size([1, 1000, 754]) | 8\n",
-      "22319    | torch.Size([1, 1000, 762]) | 8\n",
-      "22320    | torch.Size([1, 1000, 762]) | 8\n",
-      "22321    | torch.Size([1, 1000, 754]) | 8\n",
-      "22322    | torch.Size([1, 1000, 762]) | 8\n",
-      "22323    | torch.Size([1, 1000, 777]) | 8\n",
-      "22324    | torch.Size([1, 1000, 777]) | 8\n",
-      "22325    | torch.Size([1, 1000, 777]) | 8\n",
-      "22326    | torch.Size([1, 1000, 848]) | 8\n",
-      "22327    | torch.Size([1, 1000, 754]) | 8\n",
-      "22328    | torch.Size([1, 1000, 777]) | 8\n",
-      "22329    | torch.Size([1, 1000, 771]) | 8\n",
-      "22330    | torch.Size([1, 1000, 837]) | 8\n",
-      "22331    | torch.Size([1, 1000, 754]) | 8\n",
-      "22332    | torch.Size([1, 1000, 792]) | 8\n",
-      "22333    | torch.Size([1, 1000, 754]) | 8\n",
-      "22334    | torch.Size([1, 1000, 754]) | 8\n",
-      "22335    | torch.Size([1, 1000, 754]) | 8\n",
-      "22336    | torch.Size([1, 1000, 776]) | 8\n",
-      "22337    | torch.Size([1, 1000, 762]) | 8\n",
-      "22338    | torch.Size([1, 1000, 742]) | 8\n",
-      "22339    | torch.Size([1, 1000, 754]) | 8\n",
-      "22340    | torch.Size([1, 1000, 754]) | 8\n",
-      "22341    | torch.Size([1, 1000, 762]) | 8\n",
-      "22342    | torch.Size([1, 1000, 754]) | 8\n",
-      "22343    | torch.Size([1, 1000, 781]) | 8\n",
-      "22344    | torch.Size([1, 1000, 777]) | 8\n",
-      "22345    | torch.Size([1, 1000, 754]) | 8\n",
-      "22346    | torch.Size([1, 1000, 754]) | 8\n",
-      "22347    | torch.Size([1, 1000, 762]) | 8\n",
-      "22348    | torch.Size([1, 1000, 754]) | 8\n",
-      "22349    | torch.Size([1, 1000, 762]) | 8\n",
-      "22350    | torch.Size([1, 1000, 786]) | 8\n",
-      "22351    | torch.Size([1, 1000, 753]) | 8\n",
-      "22352    | torch.Size([1, 1000, 766]) | 8\n",
-      "22353    | torch.Size([1, 1000, 754]) | 8\n",
-      "22354    | torch.Size([1, 1000, 802]) | 8\n",
-      "22355    | torch.Size([1, 1000, 777]) | 8\n",
-      "22356    | torch.Size([1, 1000, 771]) | 8\n",
-      "22357    | torch.Size([1, 1000, 777]) | 8\n",
-      "22358    | torch.Size([1, 1000, 754]) | 8\n",
-      "22359    | torch.Size([1, 1000, 782]) | 8\n",
-      "22360    | torch.Size([1, 1000, 754]) | 8\n",
-      "22361    | torch.Size([1, 1000, 777]) | 8\n",
-      "22362    | torch.Size([1, 1000, 762]) | 8\n",
-      "22363    | torch.Size([1, 1000, 762]) | 8\n",
-      "22364    | torch.Size([1, 1000, 762]) | 8\n",
-      "22365    | torch.Size([1, 1000, 762]) | 8\n",
-      "22366    | torch.Size([1, 1000, 762]) | 8\n",
-      "22367    | torch.Size([1, 1000, 754]) | 8\n",
-      "22368    | torch.Size([1, 1000, 773]) | 8\n",
-      "22369    | torch.Size([1, 1000, 762]) | 8\n",
-      "22370    | torch.Size([1, 1000, 762]) | 8\n",
-      "22371    | torch.Size([1, 1000, 762]) | 8\n",
-      "22372    | torch.Size([1, 1000, 778]) | 8\n",
-      "22373    | torch.Size([1, 1000, 754]) | 8\n",
-      "22374    | torch.Size([1, 1000, 762]) | 8\n",
-      "22375    | torch.Size([1, 1000, 802]) | 8\n",
-      "22376    | torch.Size([1, 1000, 754]) | 8\n",
-      "22377    | torch.Size([1, 1000, 768]) | 8\n",
-      "22378    | torch.Size([1, 1000, 762]) | 8\n",
-      "22379    | torch.Size([1, 1000, 754]) | 8\n",
-      "22380    | torch.Size([1, 1000, 762]) | 8\n",
-      "22381    | torch.Size([1, 1000, 762]) | 8\n",
-      "22382    | torch.Size([1, 1000, 762]) | 8\n",
-      "22383    | torch.Size([1, 1000, 762]) | 8\n",
-      "22384    | torch.Size([1, 1000, 782]) | 8\n",
-      "22385    | torch.Size([1, 1000, 774]) | 8\n",
-      "22386    | torch.Size([1, 1000, 777]) | 8\n",
-      "22387    | torch.Size([1, 1000, 771]) | 8\n",
-      "22388    | torch.Size([1, 1000, 762]) | 8\n",
-      "22389    | torch.Size([1, 1000, 762]) | 8\n",
-      "22390    | torch.Size([1, 1000, 762]) | 8\n",
-      "22391    | torch.Size([1, 1000, 762]) | 8\n",
-      "22392    | torch.Size([1, 1000, 754]) | 8\n",
-      "22393    | torch.Size([1, 1000, 754]) | 8\n",
-      "22394    | torch.Size([1, 1000, 772]) | 8\n",
-      "22395    | torch.Size([1, 1000, 802]) | 8\n",
-      "22396    | torch.Size([1, 1000, 777]) | 8\n",
-      "22397    | torch.Size([1, 1000, 754]) | 8\n",
-      "22398    | torch.Size([1, 1000, 777]) | 8\n",
-      "22399    | torch.Size([1, 1000, 802]) | 8\n",
-      "22400    | torch.Size([1, 1000, 754]) | 8\n",
-      "22401    | torch.Size([1, 1000, 754]) | 8\n",
-      "22402    | torch.Size([1, 1000, 762]) | 8\n",
-      "22403    | torch.Size([1, 1000, 762]) | 8\n",
-      "22404    | torch.Size([1, 1000, 777]) | 8\n",
-      "22405    | torch.Size([1, 1000, 724]) | 8\n",
-      "22406    | torch.Size([1, 1000, 798]) | 8\n",
-      "22407    | torch.Size([1, 1000, 775]) | 8\n",
-      "22408    | torch.Size([1, 1000, 754]) | 8\n",
-      "22409    | torch.Size([1, 1000, 781]) | 8\n",
-      "22410    | torch.Size([1, 1000, 777]) | 8\n",
-      "22411    | torch.Size([1, 1000, 754]) | 8\n",
-      "22412    | torch.Size([1, 1000, 777]) | 8\n",
-      "22413    | torch.Size([1, 1000, 754]) | 8\n",
-      "22414    | torch.Size([1, 1000, 754]) | 8\n",
-      "22415    | torch.Size([1, 1000, 754]) | 8\n",
-      "22416    | torch.Size([1, 1000, 849]) | 8\n",
-      "22417    | torch.Size([1, 1000, 768]) | 8\n",
-      "22418    | torch.Size([1, 1000, 762]) | 8\n",
-      "22419    | torch.Size([1, 1000, 754]) | 8\n",
-      "22420    | torch.Size([1, 1000, 762]) | 8\n",
-      "22421    | torch.Size([1, 1000, 777]) | 8\n",
-      "22422    | torch.Size([1, 1000, 777]) | 8\n",
-      "22423    | torch.Size([1, 1000, 777]) | 8\n",
-      "22424    | torch.Size([1, 1000, 777]) | 8\n",
-      "22425    | torch.Size([1, 1000, 754]) | 8\n",
-      "22426    | torch.Size([1, 1000, 762]) | 8\n",
-      "22427    | torch.Size([1, 1000, 754]) | 8\n",
-      "22428    | torch.Size([1, 1000, 754]) | 8\n",
-      "22429    | torch.Size([1, 1000, 868]) | 8\n",
-      "22430    | torch.Size([1, 1000, 777]) | 8\n",
-      "22431    | torch.Size([1, 1000, 771]) | 8\n",
-      "22432    | torch.Size([1, 1000, 762]) | 8\n",
-      "22433    | torch.Size([1, 1000, 762]) | 8\n",
-      "22434    | torch.Size([1, 1000, 778]) | 8\n",
-      "22435    | torch.Size([1, 1000, 773]) | 8\n",
-      "22436    | torch.Size([1, 1000, 762]) | 8\n",
-      "22437    | torch.Size([1, 1000, 785]) | 8\n",
-      "22438    | torch.Size([1, 1000, 762]) | 8\n",
-      "22439    | torch.Size([1, 1000, 762]) | 8\n",
-      "22440    | torch.Size([1, 1000, 808]) | 8\n",
-      "22441    | torch.Size([1, 1000, 812]) | 8\n",
-      "22442    | torch.Size([1, 1000, 762]) | 8\n",
-      "22443    | torch.Size([1, 1000, 777]) | 8\n",
-      "22444    | torch.Size([1, 1000, 754]) | 8\n",
-      "22445    | torch.Size([1, 1000, 777]) | 8\n",
-      "22446    | torch.Size([1, 1000, 754]) | 8\n",
-      "22447    | torch.Size([1, 1000, 776]) | 8\n",
-      "22448    | torch.Size([1, 1000, 762]) | 8\n",
-      "22449    | torch.Size([1, 1000, 762]) | 8\n",
-      "22450    | torch.Size([1, 1000, 777]) | 8\n",
-      "22451    | torch.Size([1, 1000, 777]) | 8\n",
-      "22452    | torch.Size([1, 1000, 754]) | 8\n",
-      "22453    | torch.Size([1, 1000, 762]) | 8\n",
-      "22454    | torch.Size([1, 1000, 786]) | 8\n",
-      "22455    | torch.Size([1, 1000, 762]) | 8\n",
-      "22456    | torch.Size([1, 1000, 777]) | 8\n",
-      "22457    | torch.Size([1, 1000, 777]) | 8\n",
-      "22458    | torch.Size([1, 1000, 772]) | 8\n",
-      "22459    | torch.Size([1, 1000, 762]) | 8\n",
-      "22460    | torch.Size([1, 1000, 772]) | 8\n",
-      "22461    | torch.Size([1, 1000, 762]) | 8\n",
-      "22462    | torch.Size([1, 1000, 777]) | 8\n",
-      "22463    | torch.Size([1, 1000, 762]) | 8\n",
-      "22464    | torch.Size([1, 1000, 762]) | 8\n",
-      "22465    | torch.Size([1, 1000, 754]) | 8\n",
-      "22466    | torch.Size([1, 1000, 762]) | 8\n",
-      "22467    | torch.Size([1, 1000, 754]) | 8\n",
-      "22468    | torch.Size([1, 1000, 754]) | 8\n",
-      "22469    | torch.Size([1, 1000, 775]) | 8\n",
-      "22470    | torch.Size([1, 1000, 777]) | 8\n",
-      "22471    | torch.Size([1, 1000, 762]) | 8\n",
-      "22472    | torch.Size([1, 1000, 762]) | 8\n",
-      "22473    | torch.Size([1, 1000, 769]) | 8\n",
-      "22474    | torch.Size([1, 1000, 762]) | 8\n",
-      "22475    | torch.Size([1, 1000, 762]) | 8\n",
-      "22476    | torch.Size([1, 1000, 789]) | 8\n",
-      "22477    | torch.Size([1, 1000, 777]) | 8\n",
-      "22478    | torch.Size([1, 1000, 754]) | 8\n",
-      "22479    | torch.Size([1, 1000, 780]) | 8\n",
-      "22480    | torch.Size([1, 1000, 772]) | 8\n",
-      "22481    | torch.Size([1, 1000, 777]) | 8\n",
-      "22482    | torch.Size([1, 1000, 754]) | 8\n",
-      "22483    | torch.Size([1, 1000, 754]) | 8\n",
-      "22484    | torch.Size([1, 1000, 754]) | 8\n",
-      "22485    | torch.Size([1, 1000, 792]) | 8\n",
-      "22486    | torch.Size([1, 1000, 762]) | 8\n",
-      "22487    | torch.Size([1, 1000, 777]) | 8\n",
-      "22488    | torch.Size([1, 1000, 754]) | 8\n",
-      "22489    | torch.Size([1, 1000, 802]) | 8\n",
-      "22490    | torch.Size([1, 1000, 802]) | 8\n",
-      "22491    | torch.Size([1, 1000, 793]) | 8\n",
-      "22492    | torch.Size([1, 1000, 810]) | 8\n",
-      "22493    | torch.Size([1, 1000, 762]) | 8\n",
-      "22494    | torch.Size([1, 1000, 762]) | 8\n",
-      "22495    | torch.Size([1, 1000, 777]) | 8\n",
-      "22496    | torch.Size([1, 1000, 754]) | 8\n",
-      "22497    | torch.Size([1, 1000, 777]) | 8\n",
-      "22498    | torch.Size([1, 1000, 754]) | 8\n",
-      "22499    | torch.Size([1, 1000, 754]) | 8\n",
-      "22500    | torch.Size([1, 1000, 754]) | 8\n",
-      "22501    | torch.Size([1, 1000, 777]) | 8\n",
-      "22502    | torch.Size([1, 1000, 754]) | 8\n",
-      "22503    | torch.Size([1, 1000, 754]) | 8\n",
-      "22504    | torch.Size([1, 1000, 786]) | 8\n",
-      "22505    | torch.Size([1, 1000, 754]) | 8\n",
-      "22506    | torch.Size([1, 1000, 791]) | 8\n",
-      "22507    | torch.Size([1, 1000, 793]) | 8\n",
-      "22508    | torch.Size([1, 1000, 754]) | 8\n",
-      "22509    | torch.Size([1, 1000, 762]) | 8\n",
-      "22510    | torch.Size([1, 1000, 754]) | 8\n",
-      "22511    | torch.Size([1, 1000, 779]) | 8\n",
-      "22512    | torch.Size([1, 1000, 772]) | 8\n",
-      "22513    | torch.Size([1, 1000, 762]) | 8\n",
-      "22514    | torch.Size([1, 1000, 762]) | 8\n",
-      "22515    | torch.Size([1, 1000, 762]) | 8\n",
-      "22516    | torch.Size([1, 1000, 772]) | 8\n",
-      "22517    | torch.Size([1, 1000, 773]) | 8\n",
-      "22518    | torch.Size([1, 1000, 754]) | 8\n",
-      "22519    | torch.Size([1, 1000, 754]) | 8\n",
-      "22520    | torch.Size([1, 1000, 786]) | 8\n",
-      "22521    | torch.Size([1, 1000, 777]) | 8\n",
-      "22522    | torch.Size([1, 1000, 762]) | 8\n",
-      "22523    | torch.Size([1, 1000, 776]) | 8\n",
-      "22524    | torch.Size([1, 1000, 777]) | 8\n",
-      "22525    | torch.Size([1, 1000, 796]) | 8\n",
-      "22526    | torch.Size([1, 1000, 762]) | 8\n",
-      "22527    | torch.Size([1, 1000, 754]) | 8\n",
-      "22528    | torch.Size([1, 1000, 754]) | 8\n",
-      "22529    | torch.Size([1, 1000, 754]) | 8\n",
-      "22530    | torch.Size([1, 1000, 762]) | 8\n",
-      "22531    | torch.Size([1, 1000, 762]) | 8\n",
-      "22532    | torch.Size([1, 1000, 801]) | 8\n",
-      "22533    | torch.Size([1, 1000, 802]) | 8\n",
-      "22534    | torch.Size([1, 1000, 754]) | 9\n",
-      "22535    | torch.Size([1, 1000, 796]) | 9\n",
-      "22536    | torch.Size([1, 1000, 754]) | 9\n",
-      "22537    | torch.Size([1, 1000, 754]) | 9\n",
-      "22538    | torch.Size([1, 1000, 777]) | 9\n",
-      "22539    | torch.Size([1, 1000, 780]) | 9\n",
-      "22540    | torch.Size([1, 1000, 777]) | 9\n",
-      "22541    | torch.Size([1, 1000, 778]) | 9\n",
-      "22542    | torch.Size([1, 1000, 802]) | 9\n",
-      "22543    | torch.Size([1, 1000, 848]) | 9\n",
-      "22544    | torch.Size([1, 1000, 754]) | 9\n",
-      "22545    | torch.Size([1, 1000, 754]) | 9\n",
-      "22546    | torch.Size([1, 1000, 754]) | 9\n",
-      "22547    | torch.Size([1, 1000, 754]) | 9\n",
-      "22548    | torch.Size([1, 1000, 754]) | 9\n",
-      "22549    | torch.Size([1, 1000, 793]) | 9\n",
-      "22550    | torch.Size([1, 1000, 754]) | 9\n",
-      "22551    | torch.Size([1, 1000, 758]) | 9\n",
-      "22552    | torch.Size([1, 1000, 726]) | 9\n",
-      "22553    | torch.Size([1, 1000, 754]) | 9\n",
-      "22554    | torch.Size([1, 1000, 754]) | 9\n",
-      "22555    | torch.Size([1, 1000, 777]) | 9\n",
-      "22556    | torch.Size([1, 1000, 777]) | 9\n",
-      "22557    | torch.Size([1, 1000, 754]) | 9\n",
-      "22558    | torch.Size([1, 1000, 786]) | 9\n",
-      "22559    | torch.Size([1, 1000, 754]) | 9\n",
-      "22560    | torch.Size([1, 1000, 762]) | 9\n",
-      "22561    | torch.Size([1, 1000, 774]) | 9\n",
-      "22562    | torch.Size([1, 1000, 777]) | 9\n",
-      "22563    | torch.Size([1, 1000, 788]) | 9\n",
-      "22564    | torch.Size([1, 1000, 774]) | 9\n",
-      "22565    | torch.Size([1, 1000, 802]) | 9\n",
-      "22566    | torch.Size([1, 1000, 754]) | 9\n",
-      "22567    | torch.Size([1, 1000, 777]) | 9\n",
-      "22568    | torch.Size([1, 1000, 754]) | 9\n",
-      "22569    | torch.Size([1, 1000, 777]) | 9\n",
-      "22570    | torch.Size([1, 1000, 754]) | 9\n",
-      "22571    | torch.Size([1, 1000, 754]) | 9\n",
-      "22572    | torch.Size([1, 1000, 785]) | 9\n",
-      "22573    | torch.Size([1, 1000, 775]) | 9\n",
-      "22574    | torch.Size([1, 1000, 754]) | 9\n",
-      "22575    | torch.Size([1, 1000, 777]) | 9\n",
-      "22576    | torch.Size([1, 1000, 784]) | 9\n",
-      "22577    | torch.Size([1, 1000, 777]) | 9\n",
-      "22578    | torch.Size([1, 1000, 775]) | 9\n",
-      "22579    | torch.Size([1, 1000, 829]) | 9\n",
-      "22580    | torch.Size([1, 1000, 754]) | 9\n",
-      "22581    | torch.Size([1, 1000, 614]) | 9\n",
-      "22582    | torch.Size([1, 1000, 754]) | 9\n",
-      "22583    | torch.Size([1, 1000, 787]) | 9\n",
-      "22584    | torch.Size([1, 1000, 754]) | 9\n",
-      "22585    | torch.Size([1, 1000, 735]) | 9\n",
-      "22586    | torch.Size([1, 1000, 780]) | 9\n",
-      "22587    | torch.Size([1, 1000, 773]) | 9\n",
-      "22588    | torch.Size([1, 1000, 754]) | 9\n",
-      "22589    | torch.Size([1, 1000, 777]) | 9\n",
-      "22590    | torch.Size([1, 1000, 777]) | 9\n",
-      "22591    | torch.Size([1, 1000, 775]) | 9\n",
-      "22592    | torch.Size([1, 1000, 754]) | 9\n",
-      "22593    | torch.Size([1, 1000, 777]) | 9\n",
-      "22594    | torch.Size([1, 1000, 781]) | 9\n",
-      "22595    | torch.Size([1, 1000, 777]) | 9\n",
-      "22596    | torch.Size([1, 1000, 754]) | 9\n",
-      "22597    | torch.Size([1, 1000, 789]) | 9\n",
-      "22598    | torch.Size([1, 1000, 754]) | 9\n",
-      "22599    | torch.Size([1, 1000, 754]) | 9\n",
-      "22600    | torch.Size([1, 1000, 754]) | 9\n",
-      "22601    | torch.Size([1, 1000, 777]) | 9\n",
-      "22602    | torch.Size([1, 1000, 777]) | 9\n",
-      "22603    | torch.Size([1, 1000, 754]) | 9\n",
-      "22604    | torch.Size([1, 1000, 754]) | 9\n",
-      "22605    | torch.Size([1, 1000, 754]) | 9\n",
-      "22606    | torch.Size([1, 1000, 781]) | 9\n",
-      "22607    | torch.Size([1, 1000, 772]) | 9\n",
-      "22608    | torch.Size([1, 1000, 777]) | 9\n",
-      "22609    | torch.Size([1, 1000, 768]) | 9\n",
-      "22610    | torch.Size([1, 1000, 785]) | 9\n",
-      "22611    | torch.Size([1, 1000, 754]) | 9\n",
-      "22612    | torch.Size([1, 1000, 768]) | 9\n",
-      "22613    | torch.Size([1, 1000, 754]) | 9\n",
-      "22614    | torch.Size([1, 1000, 762]) | 9\n",
-      "22615    | torch.Size([1, 1000, 777]) | 9\n",
-      "22616    | torch.Size([1, 1000, 785]) | 9\n",
-      "22617    | torch.Size([1, 1000, 719]) | 9\n",
-      "22618    | torch.Size([1, 1000, 754]) | 9\n",
-      "22619    | torch.Size([1, 1000, 777]) | 9\n",
-      "22620    | torch.Size([1, 1000, 784]) | 9\n",
-      "22621    | torch.Size([1, 1000, 777]) | 9\n",
-      "22622    | torch.Size([1, 1000, 777]) | 9\n",
-      "22623    | torch.Size([1, 1000, 777]) | 9\n",
-      "22624    | torch.Size([1, 1000, 754]) | 9\n",
-      "22625    | torch.Size([1, 1000, 771]) | 9\n",
-      "22626    | torch.Size([1, 1000, 802]) | 9\n",
-      "22627    | torch.Size([1, 1000, 784]) | 9\n",
-      "22628    | torch.Size([1, 1000, 774]) | 9\n",
-      "22629    | torch.Size([1, 1000, 754]) | 9\n",
-      "22630    | torch.Size([1, 1000, 754]) | 9\n",
-      "22631    | torch.Size([1, 1000, 754]) | 9\n",
-      "22632    | torch.Size([1, 1000, 754]) | 9\n",
-      "22633    | torch.Size([1, 1000, 754]) | 9\n",
-      "22634    | torch.Size([1, 1000, 791]) | 9\n",
-      "22635    | torch.Size([1, 1000, 777]) | 9\n",
-      "22636    | torch.Size([1, 1000, 754]) | 9\n",
-      "22637    | torch.Size([1, 1000, 754]) | 9\n",
-      "22638    | torch.Size([1, 1000, 754]) | 9\n",
-      "22639    | torch.Size([1, 1000, 754]) | 9\n",
-      "22640    | torch.Size([1, 1000, 754]) | 9\n",
-      "22641    | torch.Size([1, 1000, 754]) | 9\n",
-      "22642    | torch.Size([1, 1000, 777]) | 9\n",
-      "22643    | torch.Size([1, 1000, 762]) | 9\n",
-      "22644    | torch.Size([1, 1000, 777]) | 9\n",
-      "22645    | torch.Size([1, 1000, 840]) | 9\n",
-      "22646    | torch.Size([1, 1000, 754]) | 9\n",
-      "22647    | torch.Size([1, 1000, 754]) | 9\n",
-      "22648    | torch.Size([1, 1000, 754]) | 9\n",
-      "22649    | torch.Size([1, 1000, 754]) | 9\n",
-      "22650    | torch.Size([1, 1000, 762]) | 9\n",
-      "22651    | torch.Size([1, 1000, 783]) | 9\n",
-      "22652    | torch.Size([1, 1000, 782]) | 9\n",
-      "22653    | torch.Size([1, 1000, 754]) | 9\n",
-      "22654    | torch.Size([1, 1000, 754]) | 9\n",
-      "22655    | torch.Size([1, 1000, 777]) | 9\n",
-      "22656    | torch.Size([1, 1000, 786]) | 9\n",
-      "22657    | torch.Size([1, 1000, 777]) | 9\n",
-      "22658    | torch.Size([1, 1000, 777]) | 9\n",
-      "22659    | torch.Size([1, 1000, 773]) | 9\n",
-      "22660    | torch.Size([1, 1000, 788]) | 9\n",
-      "22661    | torch.Size([1, 1000, 754]) | 9\n",
-      "22662    | torch.Size([1, 1000, 783]) | 9\n",
-      "22663    | torch.Size([1, 1000, 766]) | 9\n",
-      "22664    | torch.Size([1, 1000, 868]) | 9\n",
-      "22665    | torch.Size([1, 1000, 754]) | 9\n",
-      "22666    | torch.Size([1, 1000, 781]) | 9\n",
-      "22667    | torch.Size([1, 1000, 778]) | 9\n",
-      "22668    | torch.Size([1, 1000, 777]) | 9\n",
-      "22669    | torch.Size([1, 1000, 796]) | 9\n",
-      "22670    | torch.Size([1, 1000, 754]) | 9\n",
-      "22671    | torch.Size([1, 1000, 762]) | 9\n",
-      "22672    | torch.Size([1, 1000, 754]) | 9\n",
-      "22673    | torch.Size([1, 1000, 783]) | 9\n",
-      "22674    | torch.Size([1, 1000, 754]) | 9\n",
-      "22675    | torch.Size([1, 1000, 754]) | 9\n",
-      "22676    | torch.Size([1, 1000, 754]) | 9\n",
-      "22677    | torch.Size([1, 1000, 777]) | 9\n",
-      "22678    | torch.Size([1, 1000, 781]) | 9\n",
-      "22679    | torch.Size([1, 1000, 777]) | 9\n",
-      "22680    | torch.Size([1, 1000, 703]) | 9\n",
-      "22681    | torch.Size([1, 1000, 754]) | 9\n",
-      "22682    | torch.Size([1, 1000, 754]) | 9\n",
-      "22683    | torch.Size([1, 1000, 754]) | 9\n",
-      "22684    | torch.Size([1, 1000, 750]) | 9\n",
-      "22685    | torch.Size([1, 1000, 754]) | 9\n",
-      "22686    | torch.Size([1, 1000, 772]) | 9\n",
-      "22687    | torch.Size([1, 1000, 777]) | 9\n",
-      "22688    | torch.Size([1, 1000, 754]) | 9\n",
-      "22689    | torch.Size([1, 1000, 758]) | 9\n",
-      "22690    | torch.Size([1, 1000, 777]) | 9\n",
-      "22691    | torch.Size([1, 1000, 754]) | 9\n",
-      "22692    | torch.Size([1, 1000, 754]) | 9\n",
-      "22693    | torch.Size([1, 1000, 754]) | 9\n",
-      "22694    | torch.Size([1, 1000, 754]) | 9\n",
-      "22695    | torch.Size([1, 1000, 777]) | 9\n",
-      "22696    | torch.Size([1, 1000, 703]) | 9\n",
-      "22697    | torch.Size([1, 1000, 777]) | 9\n",
-      "22698    | torch.Size([1, 1000, 777]) | 9\n",
-      "22699    | torch.Size([1, 1000, 785]) | 9\n",
-      "22700    | torch.Size([1, 1000, 781]) | 9\n",
-      "22701    | torch.Size([1, 1000, 777]) | 9\n",
-      "22702    | torch.Size([1, 1000, 754]) | 9\n",
-      "22703    | torch.Size([1, 1000, 781]) | 9\n",
-      "22704    | torch.Size([1, 1000, 781]) | 9\n",
-      "22705    | torch.Size([1, 1000, 777]) | 9\n",
-      "22706    | torch.Size([1, 1000, 661]) | 9\n",
-      "22707    | torch.Size([1, 1000, 754]) | 9\n",
-      "22708    | torch.Size([1, 1000, 754]) | 9\n",
-      "22709    | torch.Size([1, 1000, 754]) | 9\n",
-      "22710    | torch.Size([1, 1000, 754]) | 9\n",
-      "22711    | torch.Size([1, 1000, 777]) | 9\n",
-      "22712    | torch.Size([1, 1000, 754]) | 9\n",
-      "22713    | torch.Size([1, 1000, 782]) | 9\n",
-      "22714    | torch.Size([1, 1000, 754]) | 9\n",
-      "22715    | torch.Size([1, 1000, 802]) | 9\n",
-      "22716    | torch.Size([1, 1000, 763]) | 9\n",
-      "22717    | torch.Size([1, 1000, 762]) | 9\n",
-      "22718    | torch.Size([1, 1000, 842]) | 9\n",
-      "22719    | torch.Size([1, 1000, 777]) | 9\n",
-      "22720    | torch.Size([1, 1000, 777]) | 9\n",
-      "22721    | torch.Size([1, 1000, 777]) | 9\n",
-      "22722    | torch.Size([1, 1000, 754]) | 9\n",
-      "22723    | torch.Size([1, 1000, 778]) | 9\n",
-      "22724    | torch.Size([1, 1000, 769]) | 9\n",
-      "22725    | torch.Size([1, 1000, 777]) | 9\n",
-      "22726    | torch.Size([1, 1000, 754]) | 9\n",
-      "22727    | torch.Size([1, 1000, 762]) | 9\n",
-      "22728    | torch.Size([1, 1000, 777]) | 9\n",
-      "22729    | torch.Size([1, 1000, 778]) | 9\n",
-      "22730    | torch.Size([1, 1000, 773]) | 9\n",
-      "22731    | torch.Size([1, 1000, 754]) | 9\n",
-      "22732    | torch.Size([1, 1000, 862]) | 9\n",
-      "22733    | torch.Size([1, 1000, 780]) | 9\n",
-      "22734    | torch.Size([1, 1000, 788]) | 9\n",
-      "22735    | torch.Size([1, 1000, 754]) | 9\n",
-      "22736    | torch.Size([1, 1000, 754]) | 9\n",
-      "22737    | torch.Size([1, 1000, 754]) | 9\n",
-      "22738    | torch.Size([1, 1000, 754]) | 9\n",
-      "22739    | torch.Size([1, 1000, 754]) | 9\n",
-      "22740    | torch.Size([1, 1000, 768]) | 9\n",
-      "22741    | torch.Size([1, 1000, 786]) | 9\n",
-      "22742    | torch.Size([1, 1000, 754]) | 9\n",
-      "22743    | torch.Size([1, 1000, 754]) | 9\n",
-      "22744    | torch.Size([1, 1000, 777]) | 9\n",
-      "22745    | torch.Size([1, 1000, 754]) | 9\n",
-      "22746    | torch.Size([1, 1000, 777]) | 9\n",
-      "22747    | torch.Size([1, 1000, 754]) | 9\n",
-      "22748    | torch.Size([1, 1000, 778]) | 9\n",
-      "22749    | torch.Size([1, 1000, 754]) | 9\n",
-      "22750    | torch.Size([1, 1000, 754]) | 9\n",
-      "22751    | torch.Size([1, 1000, 778]) | 9\n",
-      "22752    | torch.Size([1, 1000, 754]) | 9\n",
-      "22753    | torch.Size([1, 1000, 754]) | 9\n",
-      "22754    | torch.Size([1, 1000, 754]) | 9\n",
-      "22755    | torch.Size([1, 1000, 785]) | 9\n",
-      "22756    | torch.Size([1, 1000, 754]) | 9\n",
-      "22757    | torch.Size([1, 1000, 754]) | 9\n",
-      "22758    | torch.Size([1, 1000, 754]) | 9\n",
-      "22759    | torch.Size([1, 1000, 754]) | 9\n",
-      "22760    | torch.Size([1, 1000, 754]) | 9\n",
-      "22761    | torch.Size([1, 1000, 795]) | 9\n",
-      "22762    | torch.Size([1, 1000, 767]) | 9\n",
-      "22763    | torch.Size([1, 1000, 777]) | 9\n",
-      "22764    | torch.Size([1, 1000, 775]) | 9\n",
-      "22765    | torch.Size([1, 1000, 754]) | 9\n",
-      "22766    | torch.Size([1, 1000, 754]) | 9\n",
-      "22767    | torch.Size([1, 1000, 768]) | 9\n",
-      "22768    | torch.Size([1, 1000, 762]) | 9\n",
-      "22769    | torch.Size([1, 1000, 754]) | 9\n",
-      "22770    | torch.Size([1, 1000, 777]) | 9\n",
-      "22771    | torch.Size([1, 1000, 774]) | 9\n",
-      "22772    | torch.Size([1, 1000, 773]) | 9\n",
-      "22773    | torch.Size([1, 1000, 763]) | 9\n",
-      "22774    | torch.Size([1, 1000, 781]) | 9\n",
-      "22775    | torch.Size([1, 1000, 766]) | 9\n",
-      "22776    | torch.Size([1, 1000, 777]) | 9\n",
-      "22777    | torch.Size([1, 1000, 754]) | 9\n",
-      "22778    | torch.Size([1, 1000, 764]) | 9\n",
-      "22779    | torch.Size([1, 1000, 788]) | 9\n",
-      "22780    | torch.Size([1, 1000, 754]) | 9\n",
-      "22781    | torch.Size([1, 1000, 754]) | 9\n",
-      "22782    | torch.Size([1, 1000, 763]) | 9\n",
-      "22783    | torch.Size([1, 1000, 777]) | 9\n",
-      "22784    | torch.Size([1, 1000, 754]) | 9\n",
-      "22785    | torch.Size([1, 1000, 777]) | 9\n",
-      "22786    | torch.Size([1, 1000, 754]) | 9\n",
-      "22787    | torch.Size([1, 1000, 809]) | 9\n",
-      "22788    | torch.Size([1, 1000, 777]) | 9\n",
-      "22789    | torch.Size([1, 1000, 754]) | 9\n",
-      "22790    | torch.Size([1, 1000, 777]) | 9\n",
-      "22791    | torch.Size([1, 1000, 777]) | 9\n",
-      "22792    | torch.Size([1, 1000, 754]) | 9\n",
-      "22793    | torch.Size([1, 1000, 754]) | 9\n",
-      "22794    | torch.Size([1, 1000, 754]) | 9\n",
-      "22795    | torch.Size([1, 1000, 754]) | 9\n",
-      "22796    | torch.Size([1, 1000, 777]) | 9\n",
-      "22797    | torch.Size([1, 1000, 790]) | 9\n",
-      "22798    | torch.Size([1, 1000, 777]) | 9\n",
-      "22799    | torch.Size([1, 1000, 777]) | 9\n",
-      "22800    | torch.Size([1, 1000, 762]) | 9\n",
-      "22801    | torch.Size([1, 1000, 754]) | 9\n",
-      "22802    | torch.Size([1, 1000, 754]) | 9\n",
-      "22803    | torch.Size([1, 1000, 776]) | 9\n",
-      "22804    | torch.Size([1, 1000, 797]) | 9\n",
-      "22805    | torch.Size([1, 1000, 754]) | 9\n",
-      "22806    | torch.Size([1, 1000, 768]) | 9\n",
-      "22807    | torch.Size([1, 1000, 754]) | 9\n",
-      "22808    | torch.Size([1, 1000, 754]) | 9\n",
-      "22809    | torch.Size([1, 1000, 762]) | 9\n",
-      "22810    | torch.Size([1, 1000, 754]) | 9\n",
-      "22811    | torch.Size([1, 1000, 780]) | 9\n",
-      "22812    | torch.Size([1, 1000, 847]) | 9\n",
-      "22813    | torch.Size([1, 1000, 762]) | 9\n",
-      "22814    | torch.Size([1, 1000, 777]) | 9\n",
-      "22815    | torch.Size([1, 1000, 777]) | 9\n",
-      "22816    | torch.Size([1, 1000, 754]) | 9\n",
-      "22817    | torch.Size([1, 1000, 777]) | 9\n",
-      "22818    | torch.Size([1, 1000, 787]) | 9\n",
-      "22819    | torch.Size([1, 1000, 777]) | 9\n",
-      "22820    | torch.Size([1, 1000, 754]) | 9\n",
-      "22821    | torch.Size([1, 1000, 754]) | 9\n",
-      "22822    | torch.Size([1, 1000, 779]) | 9\n",
-      "22823    | torch.Size([1, 1000, 754]) | 9\n",
-      "22824    | torch.Size([1, 1000, 754]) | 9\n",
-      "22825    | torch.Size([1, 1000, 754]) | 9\n",
-      "22826    | torch.Size([1, 1000, 754]) | 9\n",
-      "22827    | torch.Size([1, 1000, 777]) | 9\n",
-      "22828    | torch.Size([1, 1000, 754]) | 9\n",
-      "22829    | torch.Size([1, 1000, 777]) | 9\n",
-      "22830    | torch.Size([1, 1000, 858]) | 9\n",
-      "22831    | torch.Size([1, 1000, 754]) | 9\n",
-      "22832    | torch.Size([1, 1000, 783]) | 9\n",
-      "22833    | torch.Size([1, 1000, 777]) | 9\n",
-      "22834    | torch.Size([1, 1000, 754]) | 9\n",
-      "22835    | torch.Size([1, 1000, 754]) | 9\n",
-      "22836    | torch.Size([1, 1000, 767]) | 9\n",
-      "22837    | torch.Size([1, 1000, 771]) | 9\n",
-      "22838    | torch.Size([1, 1000, 754]) | 9\n",
-      "22839    | torch.Size([1, 1000, 754]) | 9\n",
-      "22840    | torch.Size([1, 1000, 754]) | 9\n",
-      "22841    | torch.Size([1, 1000, 754]) | 9\n",
-      "22842    | torch.Size([1, 1000, 754]) | 9\n",
-      "22843    | torch.Size([1, 1000, 777]) | 9\n",
-      "22844    | torch.Size([1, 1000, 754]) | 9\n",
-      "22845    | torch.Size([1, 1000, 781]) | 9\n",
-      "22846    | torch.Size([1, 1000, 777]) | 9\n",
-      "22847    | torch.Size([1, 1000, 772]) | 9\n",
-      "22848    | torch.Size([1, 1000, 790]) | 9\n",
-      "22849    | torch.Size([1, 1000, 803]) | 9\n",
-      "22850    | torch.Size([1, 1000, 754]) | 9\n",
-      "22851    | torch.Size([1, 1000, 777]) | 9\n",
-      "22852    | torch.Size([1, 1000, 754]) | 9\n",
-      "22853    | torch.Size([1, 1000, 648]) | 9\n",
-      "22854    | torch.Size([1, 1000, 802]) | 9\n",
-      "22855    | torch.Size([1, 1000, 777]) | 9\n",
-      "22856    | torch.Size([1, 1000, 754]) | 9\n",
-      "22857    | torch.Size([1, 1000, 754]) | 9\n",
-      "22858    | torch.Size([1, 1000, 777]) | 9\n",
-      "22859    | torch.Size([1, 1000, 802]) | 9\n",
-      "22860    | torch.Size([1, 1000, 754]) | 9\n",
-      "22861    | torch.Size([1, 1000, 777]) | 9\n",
-      "22862    | torch.Size([1, 1000, 754]) | 9\n",
-      "22863    | torch.Size([1, 1000, 754]) | 9\n",
-      "22864    | torch.Size([1, 1000, 754]) | 9\n",
-      "22865    | torch.Size([1, 1000, 754]) | 9\n",
-      "22866    | torch.Size([1, 1000, 788]) | 9\n",
-      "22867    | torch.Size([1, 1000, 754]) | 9\n",
-      "22868    | torch.Size([1, 1000, 754]) | 9\n",
-      "22869    | torch.Size([1, 1000, 754]) | 9\n",
-      "22870    | torch.Size([1, 1000, 764]) | 9\n",
-      "22871    | torch.Size([1, 1000, 754]) | 9\n",
-      "22872    | torch.Size([1, 1000, 754]) | 9\n",
-      "22873    | torch.Size([1, 1000, 793]) | 9\n",
-      "22874    | torch.Size([1, 1000, 777]) | 9\n",
-      "22875    | torch.Size([1, 1000, 754]) | 9\n",
-      "22876    | torch.Size([1, 1000, 763]) | 9\n",
-      "22877    | torch.Size([1, 1000, 754]) | 9\n",
-      "22878    | torch.Size([1, 1000, 777]) | 9\n",
-      "22879    | torch.Size([1, 1000, 779]) | 9\n",
-      "22880    | torch.Size([1, 1000, 777]) | 9\n",
-      "22881    | torch.Size([1, 1000, 764]) | 9\n",
-      "22882    | torch.Size([1, 1000, 754]) | 9\n",
-      "22883    | torch.Size([1, 1000, 763]) | 9\n",
-      "22884    | torch.Size([1, 1000, 754]) | 9\n",
-      "22885    | torch.Size([1, 1000, 754]) | 9\n",
-      "22886    | torch.Size([1, 1000, 754]) | 9\n",
-      "22887    | torch.Size([1, 1000, 754]) | 9\n",
-      "22888    | torch.Size([1, 1000, 864]) | 9\n",
-      "22889    | torch.Size([1, 1000, 754]) | 9\n",
-      "22890    | torch.Size([1, 1000, 754]) | 9\n",
-      "22891    | torch.Size([1, 1000, 786]) | 9\n",
-      "22892    | torch.Size([1, 1000, 777]) | 9\n",
-      "22893    | torch.Size([1, 1000, 777]) | 9\n",
-      "22894    | torch.Size([1, 1000, 802]) | 9\n",
-      "22895    | torch.Size([1, 1000, 754]) | 9\n",
-      "22896    | torch.Size([1, 1000, 754]) | 9\n",
-      "22897    | torch.Size([1, 1000, 754]) | 9\n",
-      "22898    | torch.Size([1, 1000, 865]) | 9\n",
-      "22899    | torch.Size([1, 1000, 777]) | 9\n",
-      "22900    | torch.Size([1, 1000, 762]) | 9\n",
-      "22901    | torch.Size([1, 1000, 777]) | 9\n",
-      "22902    | torch.Size([1, 1000, 777]) | 9\n",
-      "22903    | torch.Size([1, 1000, 754]) | 9\n",
-      "22904    | torch.Size([1, 1000, 754]) | 9\n",
-      "22905    | torch.Size([1, 1000, 754]) | 9\n",
-      "22906    | torch.Size([1, 1000, 777]) | 9\n",
-      "22907    | torch.Size([1, 1000, 777]) | 9\n",
-      "22908    | torch.Size([1, 1000, 783]) | 9\n",
-      "22909    | torch.Size([1, 1000, 780]) | 9\n",
-      "22910    | torch.Size([1, 1000, 715]) | 9\n",
-      "22911    | torch.Size([1, 1000, 754]) | 9\n",
-      "22912    | torch.Size([1, 1000, 774]) | 9\n",
-      "22913    | torch.Size([1, 1000, 754]) | 9\n",
-      "22914    | torch.Size([1, 1000, 754]) | 9\n",
-      "22915    | torch.Size([1, 1000, 802]) | 9\n",
-      "22916    | torch.Size([1, 1000, 777]) | 9\n",
-      "22917    | torch.Size([1, 1000, 754]) | 9\n",
-      "22918    | torch.Size([1, 1000, 611]) | 9\n",
-      "22919    | torch.Size([1, 1000, 843]) | 9\n",
-      "22920    | torch.Size([1, 1000, 773]) | 9\n",
-      "22921    | torch.Size([1, 1000, 754]) | 9\n",
-      "22922    | torch.Size([1, 1000, 754]) | 9\n",
-      "22923    | torch.Size([1, 1000, 754]) | 9\n",
-      "22924    | torch.Size([1, 1000, 703]) | 9\n",
-      "22925    | torch.Size([1, 1000, 754]) | 9\n",
-      "22926    | torch.Size([1, 1000, 777]) | 9\n",
-      "22927    | torch.Size([1, 1000, 777]) | 9\n",
-      "22928    | torch.Size([1, 1000, 802]) | 9\n",
-      "22929    | torch.Size([1, 1000, 802]) | 9\n",
-      "22930    | torch.Size([1, 1000, 754]) | 9\n",
-      "22931    | torch.Size([1, 1000, 777]) | 9\n",
-      "22932    | torch.Size([1, 1000, 777]) | 9\n",
-      "22933    | torch.Size([1, 1000, 765]) | 9\n",
-      "22934    | torch.Size([1, 1000, 778]) | 9\n",
-      "22935    | torch.Size([1, 1000, 786]) | 9\n",
-      "22936    | torch.Size([1, 1000, 754]) | 9\n",
-      "22937    | torch.Size([1, 1000, 779]) | 9\n",
-      "22938    | torch.Size([1, 1000, 754]) | 9\n",
-      "22939    | torch.Size([1, 1000, 703]) | 9\n",
-      "22940    | torch.Size([1, 1000, 754]) | 9\n",
-      "22941    | torch.Size([1, 1000, 754]) | 9\n",
-      "22942    | torch.Size([1, 1000, 788]) | 9\n",
-      "22943    | torch.Size([1, 1000, 784]) | 9\n",
-      "22944    | torch.Size([1, 1000, 777]) | 9\n",
-      "22945    | torch.Size([1, 1000, 754]) | 9\n",
-      "22946    | torch.Size([1, 1000, 754]) | 9\n",
-      "22947    | torch.Size([1, 1000, 777]) | 9\n",
-      "22948    | torch.Size([1, 1000, 754]) | 9\n",
-      "22949    | torch.Size([1, 1000, 831]) | 9\n",
-      "22950    | torch.Size([1, 1000, 777]) | 9\n",
-      "22951    | torch.Size([1, 1000, 784]) | 9\n",
-      "22952    | torch.Size([1, 1000, 795]) | 9\n",
-      "22953    | torch.Size([1, 1000, 777]) | 9\n",
-      "22954    | torch.Size([1, 1000, 777]) | 9\n",
-      "22955    | torch.Size([1, 1000, 754]) | 9\n",
-      "22956    | torch.Size([1, 1000, 703]) | 9\n",
-      "22957    | torch.Size([1, 1000, 754]) | 9\n",
-      "22958    | torch.Size([1, 1000, 777]) | 9\n",
-      "22959    | torch.Size([1, 1000, 754]) | 9\n",
-      "22960    | torch.Size([1, 1000, 776]) | 9\n",
-      "22961    | torch.Size([1, 1000, 777]) | 9\n",
-      "22962    | torch.Size([1, 1000, 764]) | 9\n",
-      "22963    | torch.Size([1, 1000, 754]) | 9\n",
-      "22964    | torch.Size([1, 1000, 777]) | 9\n",
-      "22965    | torch.Size([1, 1000, 777]) | 9\n",
-      "22966    | torch.Size([1, 1000, 777]) | 9\n",
-      "22967    | torch.Size([1, 1000, 777]) | 9\n",
-      "22968    | torch.Size([1, 1000, 754]) | 9\n",
-      "22969    | torch.Size([1, 1000, 754]) | 9\n",
-      "22970    | torch.Size([1, 1000, 777]) | 9\n",
-      "22971    | torch.Size([1, 1000, 754]) | 9\n",
-      "22972    | torch.Size([1, 1000, 754]) | 9\n",
-      "22973    | torch.Size([1, 1000, 773]) | 9\n",
-      "22974    | torch.Size([1, 1000, 782]) | 9\n",
-      "22975    | torch.Size([1, 1000, 754]) | 9\n",
-      "22976    | torch.Size([1, 1000, 803]) | 9\n",
-      "22977    | torch.Size([1, 1000, 777]) | 9\n",
-      "22978    | torch.Size([1, 1000, 777]) | 9\n",
-      "22979    | torch.Size([1, 1000, 754]) | 9\n",
-      "22980    | torch.Size([1, 1000, 777]) | 9\n",
-      "22981    | torch.Size([1, 1000, 777]) | 9\n",
-      "22982    | torch.Size([1, 1000, 754]) | 9\n",
-      "22983    | torch.Size([1, 1000, 791]) | 9\n",
-      "22984    | torch.Size([1, 1000, 754]) | 9\n",
-      "22985    | torch.Size([1, 1000, 861]) | 9\n",
-      "22986    | torch.Size([1, 1000, 754]) | 9\n",
-      "22987    | torch.Size([1, 1000, 754]) | 9\n",
-      "22988    | torch.Size([1, 1000, 754]) | 9\n",
-      "22989    | torch.Size([1, 1000, 754]) | 9\n",
-      "22990    | torch.Size([1, 1000, 754]) | 9\n",
-      "22991    | torch.Size([1, 1000, 754]) | 9\n",
-      "22992    | torch.Size([1, 1000, 754]) | 9\n",
-      "22993    | torch.Size([1, 1000, 762]) | 9\n",
-      "22994    | torch.Size([1, 1000, 754]) | 9\n",
-      "22995    | torch.Size([1, 1000, 784]) | 9\n",
-      "22996    | torch.Size([1, 1000, 613]) | 9\n",
-      "22997    | torch.Size([1, 1000, 768]) | 9\n",
-      "22998    | torch.Size([1, 1000, 754]) | 9\n",
-      "22999    | torch.Size([1, 1000, 777]) | 9\n",
-      "23000    | torch.Size([1, 1000, 754]) | 9\n",
-      "23001    | torch.Size([1, 1000, 777]) | 9\n",
-      "23002    | torch.Size([1, 1000, 754]) | 9\n",
-      "23003    | torch.Size([1, 1000, 763]) | 9\n",
-      "23004    | torch.Size([1, 1000, 779]) | 9\n",
-      "23005    | torch.Size([1, 1000, 754]) | 9\n",
-      "23006    | torch.Size([1, 1000, 754]) | 9\n",
-      "23007    | torch.Size([1, 1000, 754]) | 9\n",
-      "23008    | torch.Size([1, 1000, 754]) | 9\n",
-      "23009    | torch.Size([1, 1000, 781]) | 9\n",
-      "23010    | torch.Size([1, 1000, 703]) | 9\n",
-      "23011    | torch.Size([1, 1000, 754]) | 9\n",
-      "23012    | torch.Size([1, 1000, 754]) | 9\n",
-      "23013    | torch.Size([1, 1000, 754]) | 9\n",
-      "23014    | torch.Size([1, 1000, 802]) | 9\n",
-      "23015    | torch.Size([1, 1000, 754]) | 9\n",
-      "23016    | torch.Size([1, 1000, 777]) | 9\n",
-      "23017    | torch.Size([1, 1000, 754]) | 9\n",
-      "23018    | torch.Size([1, 1000, 777]) | 9\n",
-      "23019    | torch.Size([1, 1000, 767]) | 9\n",
-      "23020    | torch.Size([1, 1000, 762]) | 9\n",
-      "23021    | torch.Size([1, 1000, 777]) | 9\n",
-      "23022    | torch.Size([1, 1000, 777]) | 9\n",
-      "23023    | torch.Size([1, 1000, 754]) | 9\n",
-      "23024    | torch.Size([1, 1000, 754]) | 9\n",
-      "23025    | torch.Size([1, 1000, 754]) | 9\n",
-      "23026    | torch.Size([1, 1000, 773]) | 9\n",
-      "23027    | torch.Size([1, 1000, 813]) | 9\n",
-      "23028    | torch.Size([1, 1000, 754]) | 9\n",
-      "23029    | torch.Size([1, 1000, 762]) | 9\n",
-      "23030    | torch.Size([1, 1000, 754]) | 9\n",
-      "23031    | torch.Size([1, 1000, 754]) | 9\n",
-      "23032    | torch.Size([1, 1000, 777]) | 9\n",
-      "23033    | torch.Size([1, 1000, 792]) | 9\n",
-      "23034    | torch.Size([1, 1000, 762]) | 9\n",
-      "23035    | torch.Size([1, 1000, 787]) | 9\n",
-      "23036    | torch.Size([1, 1000, 806]) | 9\n",
-      "23037    | torch.Size([1, 1000, 754]) | 9\n",
-      "23038    | torch.Size([1, 1000, 777]) | 9\n",
-      "23039    | torch.Size([1, 1000, 777]) | 9\n",
-      "23040    | torch.Size([1, 1000, 777]) | 9\n",
-      "23041    | torch.Size([1, 1000, 764]) | 9\n",
-      "23042    | torch.Size([1, 1000, 754]) | 9\n",
-      "23043    | torch.Size([1, 1000, 777]) | 9\n",
-      "23044    | torch.Size([1, 1000, 781]) | 9\n",
-      "23045    | torch.Size([1, 1000, 771]) | 9\n",
-      "23046    | torch.Size([1, 1000, 777]) | 9\n",
-      "23047    | torch.Size([1, 1000, 754]) | 9\n",
-      "23048    | torch.Size([1, 1000, 777]) | 9\n",
-      "23049    | torch.Size([1, 1000, 754]) | 9\n",
-      "23050    | torch.Size([1, 1000, 780]) | 9\n",
-      "23051    | torch.Size([1, 1000, 779]) | 9\n",
-      "23052    | torch.Size([1, 1000, 777]) | 9\n",
-      "23053    | torch.Size([1, 1000, 754]) | 9\n",
-      "23054    | torch.Size([1, 1000, 754]) | 9\n",
-      "23055    | torch.Size([1, 1000, 777]) | 9\n",
-      "23056    | torch.Size([1, 1000, 776]) | 9\n",
-      "23057    | torch.Size([1, 1000, 770]) | 9\n",
-      "23058    | torch.Size([1, 1000, 779]) | 9\n",
-      "23059    | torch.Size([1, 1000, 784]) | 9\n",
-      "23060    | torch.Size([1, 1000, 754]) | 9\n",
-      "23061    | torch.Size([1, 1000, 785]) | 9\n",
-      "23062    | torch.Size([1, 1000, 754]) | 9\n",
-      "23063    | torch.Size([1, 1000, 754]) | 9\n",
-      "23064    | torch.Size([1, 1000, 754]) | 9\n",
-      "23065    | torch.Size([1, 1000, 770]) | 9\n",
-      "23066    | torch.Size([1, 1000, 780]) | 9\n",
-      "23067    | torch.Size([1, 1000, 754]) | 9\n",
-      "23068    | torch.Size([1, 1000, 777]) | 9\n",
-      "23069    | torch.Size([1, 1000, 786]) | 9\n",
-      "23070    | torch.Size([1, 1000, 754]) | 9\n",
-      "23071    | torch.Size([1, 1000, 777]) | 9\n",
-      "23072    | torch.Size([1, 1000, 777]) | 9\n",
-      "23073    | torch.Size([1, 1000, 732]) | 9\n",
-      "23074    | torch.Size([1, 1000, 754]) | 9\n",
-      "23075    | torch.Size([1, 1000, 754]) | 9\n",
-      "23076    | torch.Size([1, 1000, 780]) | 9\n",
-      "23077    | torch.Size([1, 1000, 754]) | 9\n",
-      "23078    | torch.Size([1, 1000, 754]) | 9\n",
-      "23079    | torch.Size([1, 1000, 793]) | 9\n",
-      "23080    | torch.Size([1, 1000, 754]) | 9\n",
-      "23081    | torch.Size([1, 1000, 777]) | 9\n",
-      "23082    | torch.Size([1, 1000, 779]) | 9\n",
-      "23083    | torch.Size([1, 1000, 776]) | 9\n",
-      "23084    | torch.Size([1, 1000, 754]) | 9\n",
-      "23085    | torch.Size([1, 1000, 780]) | 9\n",
-      "23086    | torch.Size([1, 1000, 868]) | 9\n",
-      "23087    | torch.Size([1, 1000, 867]) | 9\n",
-      "23088    | torch.Size([1, 1000, 774]) | 9\n",
-      "23089    | torch.Size([1, 1000, 779]) | 9\n",
-      "23090    | torch.Size([1, 1000, 777]) | 9\n",
-      "23091    | torch.Size([1, 1000, 767]) | 9\n",
-      "23092    | torch.Size([1, 1000, 754]) | 9\n",
-      "23093    | torch.Size([1, 1000, 754]) | 9\n",
-      "23094    | torch.Size([1, 1000, 754]) | 9\n",
-      "23095    | torch.Size([1, 1000, 799]) | 9\n",
-      "23096    | torch.Size([1, 1000, 777]) | 9\n",
-      "23097    | torch.Size([1, 1000, 777]) | 9\n",
-      "23098    | torch.Size([1, 1000, 754]) | 9\n",
-      "23099    | torch.Size([1, 1000, 777]) | 9\n",
-      "23100    | torch.Size([1, 1000, 754]) | 9\n",
-      "23101    | torch.Size([1, 1000, 754]) | 9\n",
-      "23102    | torch.Size([1, 1000, 754]) | 9\n",
-      "23103    | torch.Size([1, 1000, 754]) | 9\n",
-      "23104    | torch.Size([1, 1000, 777]) | 9\n",
-      "23105    | torch.Size([1, 1000, 777]) | 9\n",
-      "23106    | torch.Size([1, 1000, 777]) | 9\n",
-      "23107    | torch.Size([1, 1000, 779]) | 9\n",
-      "23108    | torch.Size([1, 1000, 754]) | 9\n",
-      "23109    | torch.Size([1, 1000, 754]) | 9\n",
-      "23110    | torch.Size([1, 1000, 768]) | 9\n",
-      "23111    | torch.Size([1, 1000, 762]) | 9\n",
-      "23112    | torch.Size([1, 1000, 762]) | 9\n",
-      "23113    | torch.Size([1, 1000, 754]) | 9\n",
-      "23114    | torch.Size([1, 1000, 790]) | 9\n",
-      "23115    | torch.Size([1, 1000, 784]) | 9\n",
-      "23116    | torch.Size([1, 1000, 777]) | 9\n",
-      "23117    | torch.Size([1, 1000, 777]) | 9\n",
-      "23118    | torch.Size([1, 1000, 777]) | 9\n",
-      "23119    | torch.Size([1, 1000, 754]) | 9\n",
-      "23120    | torch.Size([1, 1000, 754]) | 9\n",
-      "23121    | torch.Size([1, 1000, 754]) | 9\n",
-      "23122    | torch.Size([1, 1000, 754]) | 9\n",
-      "23123    | torch.Size([1, 1000, 764]) | 9\n",
-      "23124    | torch.Size([1, 1000, 775]) | 9\n",
-      "23125    | torch.Size([1, 1000, 777]) | 9\n",
-      "23126    | torch.Size([1, 1000, 778]) | 9\n",
-      "23127    | torch.Size([1, 1000, 754]) | 9\n",
-      "23128    | torch.Size([1, 1000, 754]) | 9\n",
-      "23129    | torch.Size([1, 1000, 783]) | 9\n",
-      "23130    | torch.Size([1, 1000, 792]) | 9\n",
-      "23131    | torch.Size([1, 1000, 754]) | 9\n",
-      "23132    | torch.Size([1, 1000, 785]) | 9\n",
-      "23133    | torch.Size([1, 1000, 754]) | 9\n",
-      "23134    | torch.Size([1, 1000, 754]) | 9\n",
-      "23135    | torch.Size([1, 1000, 754]) | 9\n",
-      "23136    | torch.Size([1, 1000, 754]) | 9\n",
-      "23137    | torch.Size([1, 1000, 754]) | 9\n",
-      "23138    | torch.Size([1, 1000, 754]) | 9\n",
-      "23139    | torch.Size([1, 1000, 798]) | 9\n",
-      "23140    | torch.Size([1, 1000, 783]) | 9\n",
-      "23141    | torch.Size([1, 1000, 754]) | 9\n",
-      "23142    | torch.Size([1, 1000, 777]) | 9\n",
-      "23143    | torch.Size([1, 1000, 787]) | 9\n",
-      "23144    | torch.Size([1, 1000, 754]) | 9\n",
-      "23145    | torch.Size([1, 1000, 754]) | 9\n",
-      "23146    | torch.Size([1, 1000, 754]) | 9\n",
-      "23147    | torch.Size([1, 1000, 754]) | 9\n",
-      "23148    | torch.Size([1, 1000, 773]) | 9\n",
-      "23149    | torch.Size([1, 1000, 754]) | 9\n",
-      "23150    | torch.Size([1, 1000, 777]) | 9\n",
-      "23151    | torch.Size([1, 1000, 754]) | 9\n",
-      "23152    | torch.Size([1, 1000, 754]) | 9\n",
-      "23153    | torch.Size([1, 1000, 858]) | 9\n",
-      "23154    | torch.Size([1, 1000, 787]) | 9\n",
-      "23155    | torch.Size([1, 1000, 754]) | 9\n",
-      "23156    | torch.Size([1, 1000, 843]) | 9\n",
-      "23157    | torch.Size([1, 1000, 754]) | 9\n",
-      "23158    | torch.Size([1, 1000, 777]) | 9\n",
-      "23159    | torch.Size([1, 1000, 754]) | 9\n",
-      "23160    | torch.Size([1, 1000, 797]) | 9\n",
-      "23161    | torch.Size([1, 1000, 754]) | 9\n",
-      "23162    | torch.Size([1, 1000, 754]) | 9\n",
-      "23163    | torch.Size([1, 1000, 754]) | 9\n",
-      "23164    | torch.Size([1, 1000, 754]) | 9\n",
-      "23165    | torch.Size([1, 1000, 754]) | 9\n",
-      "23166    | torch.Size([1, 1000, 777]) | 9\n",
-      "23167    | torch.Size([1, 1000, 754]) | 9\n",
-      "23168    | torch.Size([1, 1000, 754]) | 9\n",
-      "23169    | torch.Size([1, 1000, 754]) | 9\n",
-      "23170    | torch.Size([1, 1000, 754]) | 9\n",
-      "23171    | torch.Size([1, 1000, 764]) | 9\n",
-      "23172    | torch.Size([1, 1000, 754]) | 9\n",
-      "23173    | torch.Size([1, 1000, 771]) | 9\n",
-      "23174    | torch.Size([1, 1000, 807]) | 9\n",
-      "23175    | torch.Size([1, 1000, 754]) | 9\n",
-      "23176    | torch.Size([1, 1000, 777]) | 9\n",
-      "23177    | torch.Size([1, 1000, 777]) | 9\n",
-      "23178    | torch.Size([1, 1000, 777]) | 9\n",
-      "23179    | torch.Size([1, 1000, 754]) | 9\n",
-      "23180    | torch.Size([1, 1000, 777]) | 9\n",
-      "23181    | torch.Size([1, 1000, 781]) | 9\n",
-      "23182    | torch.Size([1, 1000, 781]) | 9\n",
-      "23183    | torch.Size([1, 1000, 754]) | 9\n",
-      "23184    | torch.Size([1, 1000, 754]) | 9\n",
-      "23185    | torch.Size([1, 1000, 777]) | 9\n",
-      "23186    | torch.Size([1, 1000, 724]) | 9\n",
-      "23187    | torch.Size([1, 1000, 777]) | 9\n",
-      "23188    | torch.Size([1, 1000, 754]) | 9\n",
-      "23189    | torch.Size([1, 1000, 779]) | 9\n",
-      "23190    | torch.Size([1, 1000, 777]) | 9\n",
-      "23191    | torch.Size([1, 1000, 751]) | 9\n",
-      "23192    | torch.Size([1, 1000, 777]) | 9\n",
-      "23193    | torch.Size([1, 1000, 754]) | 9\n",
-      "23194    | torch.Size([1, 1000, 793]) | 9\n",
-      "23195    | torch.Size([1, 1000, 802]) | 9\n",
-      "23196    | torch.Size([1, 1000, 754]) | 9\n",
-      "23197    | torch.Size([1, 1000, 777]) | 9\n",
-      "23198    | torch.Size([1, 1000, 754]) | 9\n",
-      "23199    | torch.Size([1, 1000, 795]) | 9\n",
-      "23200    | torch.Size([1, 1000, 783]) | 9\n",
-      "23201    | torch.Size([1, 1000, 777]) | 9\n",
-      "23202    | torch.Size([1, 1000, 754]) | 9\n",
-      "23203    | torch.Size([1, 1000, 754]) | 9\n",
-      "23204    | torch.Size([1, 1000, 777]) | 9\n",
-      "23205    | torch.Size([1, 1000, 754]) | 9\n",
-      "23206    | torch.Size([1, 1000, 754]) | 9\n",
-      "23207    | torch.Size([1, 1000, 754]) | 9\n",
-      "23208    | torch.Size([1, 1000, 782]) | 9\n",
-      "23209    | torch.Size([1, 1000, 754]) | 9\n",
-      "23210    | torch.Size([1, 1000, 777]) | 9\n",
-      "23211    | torch.Size([1, 1000, 754]) | 9\n",
-      "23212    | torch.Size([1, 1000, 777]) | 9\n",
-      "23213    | torch.Size([1, 1000, 745]) | 9\n",
-      "23214    | torch.Size([1, 1000, 777]) | 9\n",
-      "23215    | torch.Size([1, 1000, 766]) | 9\n",
-      "23216    | torch.Size([1, 1000, 777]) | 9\n",
-      "23217    | torch.Size([1, 1000, 777]) | 9\n",
-      "23218    | torch.Size([1, 1000, 777]) | 9\n",
-      "23219    | torch.Size([1, 1000, 754]) | 9\n",
-      "23220    | torch.Size([1, 1000, 777]) | 9\n",
-      "23221    | torch.Size([1, 1000, 754]) | 9\n",
-      "23222    | torch.Size([1, 1000, 777]) | 9\n",
-      "23223    | torch.Size([1, 1000, 618]) | 9\n",
-      "23224    | torch.Size([1, 1000, 777]) | 9\n",
-      "23225    | torch.Size([1, 1000, 790]) | 9\n",
-      "23226    | torch.Size([1, 1000, 777]) | 9\n",
-      "23227    | torch.Size([1, 1000, 754]) | 9\n",
-      "23228    | torch.Size([1, 1000, 772]) | 9\n",
-      "23229    | torch.Size([1, 1000, 769]) | 9\n",
-      "23230    | torch.Size([1, 1000, 754]) | 9\n",
-      "23231    | torch.Size([1, 1000, 762]) | 9\n",
-      "23232    | torch.Size([1, 1000, 754]) | 9\n",
-      "23233    | torch.Size([1, 1000, 777]) | 9\n",
-      "23234    | torch.Size([1, 1000, 754]) | 9\n",
-      "23235    | torch.Size([1, 1000, 754]) | 9\n",
-      "23236    | torch.Size([1, 1000, 754]) | 9\n",
-      "23237    | torch.Size([1, 1000, 789]) | 9\n",
-      "23238    | torch.Size([1, 1000, 754]) | 9\n",
-      "23239    | torch.Size([1, 1000, 776]) | 9\n",
-      "23240    | torch.Size([1, 1000, 754]) | 9\n",
-      "23241    | torch.Size([1, 1000, 754]) | 9\n",
-      "23242    | torch.Size([1, 1000, 777]) | 9\n",
-      "23243    | torch.Size([1, 1000, 763]) | 9\n",
-      "23244    | torch.Size([1, 1000, 754]) | 9\n",
-      "23245    | torch.Size([1, 1000, 754]) | 9\n",
-      "23246    | torch.Size([1, 1000, 765]) | 9\n",
-      "23247    | torch.Size([1, 1000, 754]) | 9\n",
-      "23248    | torch.Size([1, 1000, 754]) | 9\n",
-      "23249    | torch.Size([1, 1000, 776]) | 9\n",
-      "23250    | torch.Size([1, 1000, 779]) | 9\n",
-      "23251    | torch.Size([1, 1000, 754]) | 9\n",
-      "23252    | torch.Size([1, 1000, 754]) | 9\n",
-      "23253    | torch.Size([1, 1000, 754]) | 9\n",
-      "23254    | torch.Size([1, 1000, 754]) | 9\n",
-      "23255    | torch.Size([1, 1000, 781]) | 9\n",
-      "23256    | torch.Size([1, 1000, 789]) | 9\n",
-      "23257    | torch.Size([1, 1000, 754]) | 9\n",
-      "23258    | torch.Size([1, 1000, 777]) | 9\n",
-      "23259    | torch.Size([1, 1000, 754]) | 9\n",
-      "23260    | torch.Size([1, 1000, 754]) | 9\n",
-      "23261    | torch.Size([1, 1000, 754]) | 9\n",
-      "23262    | torch.Size([1, 1000, 754]) | 9\n",
-      "23263    | torch.Size([1, 1000, 777]) | 9\n",
-      "23264    | torch.Size([1, 1000, 776]) | 9\n",
-      "23265    | torch.Size([1, 1000, 754]) | 9\n",
-      "23266    | torch.Size([1, 1000, 754]) | 9\n",
-      "23267    | torch.Size([1, 1000, 754]) | 9\n",
-      "23268    | torch.Size([1, 1000, 754]) | 9\n",
-      "23269    | torch.Size([1, 1000, 777]) | 9\n",
-      "23270    | torch.Size([1, 1000, 754]) | 9\n",
-      "23271    | torch.Size([1, 1000, 783]) | 9\n",
-      "23272    | torch.Size([1, 1000, 777]) | 9\n",
-      "23273    | torch.Size([1, 1000, 785]) | 9\n",
-      "23274    | torch.Size([1, 1000, 782]) | 9\n",
-      "23275    | torch.Size([1, 1000, 777]) | 9\n",
-      "23276    | torch.Size([1, 1000, 847]) | 9\n",
-      "23277    | torch.Size([1, 1000, 777]) | 9\n",
-      "23278    | torch.Size([1, 1000, 754]) | 9\n",
-      "23279    | torch.Size([1, 1000, 754]) | 9\n",
-      "23280    | torch.Size([1, 1000, 754]) | 9\n",
-      "23281    | torch.Size([1, 1000, 777]) | 9\n",
-      "23282    | torch.Size([1, 1000, 781]) | 9\n",
-      "23283    | torch.Size([1, 1000, 766]) | 9\n",
-      "23284    | torch.Size([1, 1000, 867]) | 9\n",
-      "23285    | torch.Size([1, 1000, 754]) | 9\n",
-      "23286    | torch.Size([1, 1000, 754]) | 9\n",
-      "23287    | torch.Size([1, 1000, 802]) | 9\n",
-      "23288    | torch.Size([1, 1000, 754]) | 9\n",
-      "23289    | torch.Size([1, 1000, 754]) | 9\n",
-      "23290    | torch.Size([1, 1000, 774]) | 9\n",
-      "23291    | torch.Size([1, 1000, 783]) | 9\n",
-      "23292    | torch.Size([1, 1000, 777]) | 9\n",
-      "23293    | torch.Size([1, 1000, 793]) | 9\n",
-      "23294    | torch.Size([1, 1000, 754]) | 9\n",
-      "23295    | torch.Size([1, 1000, 754]) | 9\n",
-      "23296    | torch.Size([1, 1000, 754]) | 9\n",
-      "23297    | torch.Size([1, 1000, 754]) | 9\n",
-      "23298    | torch.Size([1, 1000, 754]) | 9\n",
-      "23299    | torch.Size([1, 1000, 754]) | 9\n",
-      "23300    | torch.Size([1, 1000, 754]) | 9\n",
-      "23301    | torch.Size([1, 1000, 731]) | 9\n",
-      "23302    | torch.Size([1, 1000, 777]) | 9\n",
-      "23303    | torch.Size([1, 1000, 754]) | 9\n",
-      "23304    | torch.Size([1, 1000, 762]) | 9\n",
-      "23305    | torch.Size([1, 1000, 754]) | 9\n",
-      "23306    | torch.Size([1, 1000, 754]) | 9\n",
-      "23307    | torch.Size([1, 1000, 754]) | 9\n",
-      "23308    | torch.Size([1, 1000, 785]) | 9\n",
-      "23309    | torch.Size([1, 1000, 793]) | 9\n",
-      "23310    | torch.Size([1, 1000, 754]) | 9\n",
-      "23311    | torch.Size([1, 1000, 777]) | 9\n",
-      "23312    | torch.Size([1, 1000, 775]) | 9\n",
-      "23313    | torch.Size([1, 1000, 754]) | 9\n",
-      "23314    | torch.Size([1, 1000, 777]) | 9\n",
-      "23315    | torch.Size([1, 1000, 754]) | 9\n",
-      "23316    | torch.Size([1, 1000, 754]) | 9\n",
-      "23317    | torch.Size([1, 1000, 844]) | 9\n",
-      "23318    | torch.Size([1, 1000, 754]) | 9\n",
-      "23319    | torch.Size([1, 1000, 785]) | 9\n",
-      "23320    | torch.Size([1, 1000, 754]) | 9\n",
-      "23321    | torch.Size([1, 1000, 754]) | 9\n",
-      "23322    | torch.Size([1, 1000, 754]) | 9\n",
-      "23323    | torch.Size([1, 1000, 754]) | 9\n",
-      "23324    | torch.Size([1, 1000, 784]) | 9\n",
-      "23325    | torch.Size([1, 1000, 754]) | 9\n",
-      "23326    | torch.Size([1, 1000, 777]) | 9\n",
-      "23327    | torch.Size([1, 1000, 784]) | 9\n",
-      "23328    | torch.Size([1, 1000, 785]) | 9\n",
-      "23329    | torch.Size([1, 1000, 777]) | 9\n",
-      "23330    | torch.Size([1, 1000, 793]) | 9\n",
-      "23331    | torch.Size([1, 1000, 754]) | 9\n",
-      "23332    | torch.Size([1, 1000, 777]) | 9\n",
-      "23333    | torch.Size([1, 1000, 753]) | 9\n",
-      "23334    | torch.Size([1, 1000, 770]) | 9\n",
-      "23335    | torch.Size([1, 1000, 859]) | 9\n",
-      "23336    | torch.Size([1, 1000, 754]) | 9\n",
-      "23337    | torch.Size([1, 1000, 754]) | 9\n",
-      "23338    | torch.Size([1, 1000, 769]) | 9\n",
-      "23339    | torch.Size([1, 1000, 796]) | 9\n",
-      "23340    | torch.Size([1, 1000, 762]) | 9\n",
-      "23341    | torch.Size([1, 1000, 787]) | 9\n",
-      "23342    | torch.Size([1, 1000, 754]) | 9\n",
-      "23343    | torch.Size([1, 1000, 786]) | 9\n",
-      "23344    | torch.Size([1, 1000, 865]) | 9\n",
-      "23345    | torch.Size([1, 1000, 777]) | 9\n",
-      "23346    | torch.Size([1, 1000, 785]) | 9\n",
-      "23347    | torch.Size([1, 1000, 846]) | 9\n",
-      "23348    | torch.Size([1, 1000, 754]) | 9\n",
-      "23349    | torch.Size([1, 1000, 779]) | 9\n",
-      "23350    | torch.Size([1, 1000, 754]) | 9\n",
-      "23351    | torch.Size([1, 1000, 754]) | 9\n",
-      "23352    | torch.Size([1, 1000, 777]) | 9\n",
-      "23353    | torch.Size([1, 1000, 777]) | 9\n",
-      "23354    | torch.Size([1, 1000, 754]) | 9\n",
-      "23355    | torch.Size([1, 1000, 777]) | 9\n",
-      "23356    | torch.Size([1, 1000, 796]) | 9\n",
-      "23357    | torch.Size([1, 1000, 754]) | 9\n",
-      "23358    | torch.Size([1, 1000, 754]) | 9\n",
-      "23359    | torch.Size([1, 1000, 867]) | 9\n",
-      "23360    | torch.Size([1, 1000, 782]) | 9\n",
-      "23361    | torch.Size([1, 1000, 754]) | 9\n",
-      "23362    | torch.Size([1, 1000, 796]) | 9\n",
-      "23363    | torch.Size([1, 1000, 754]) | 9\n",
-      "23364    | torch.Size([1, 1000, 777]) | 9\n",
-      "23365    | torch.Size([1, 1000, 754]) | 9\n",
-      "23366    | torch.Size([1, 1000, 777]) | 9\n",
-      "23367    | torch.Size([1, 1000, 754]) | 9\n",
-      "23368    | torch.Size([1, 1000, 762]) | 9\n",
-      "23369    | torch.Size([1, 1000, 777]) | 9\n",
-      "23370    | torch.Size([1, 1000, 780]) | 9\n",
-      "23371    | torch.Size([1, 1000, 777]) | 9\n",
-      "23372    | torch.Size([1, 1000, 802]) | 9\n",
-      "23373    | torch.Size([1, 1000, 754]) | 9\n",
-      "23374    | torch.Size([1, 1000, 777]) | 9\n",
-      "23375    | torch.Size([1, 1000, 777]) | 9\n",
-      "23376    | torch.Size([1, 1000, 787]) | 9\n",
-      "23377    | torch.Size([1, 1000, 777]) | 9\n",
-      "23378    | torch.Size([1, 1000, 754]) | 9\n",
-      "23379    | torch.Size([1, 1000, 754]) | 9\n",
-      "23380    | torch.Size([1, 1000, 777]) | 9\n",
-      "23381    | torch.Size([1, 1000, 777]) | 9\n",
-      "23382    | torch.Size([1, 1000, 754]) | 9\n",
-      "23383    | torch.Size([1, 1000, 754]) | 9\n",
-      "23384    | torch.Size([1, 1000, 754]) | 9\n",
-      "23385    | torch.Size([1, 1000, 754]) | 9\n",
-      "23386    | torch.Size([1, 1000, 754]) | 9\n",
-      "23387    | torch.Size([1, 1000, 799]) | 9\n",
-      "23388    | torch.Size([1, 1000, 755]) | 9\n",
-      "23389    | torch.Size([1, 1000, 771]) | 9\n",
-      "23390    | torch.Size([1, 1000, 754]) | 9\n",
-      "23391    | torch.Size([1, 1000, 754]) | 9\n",
-      "23392    | torch.Size([1, 1000, 784]) | 9\n",
-      "23393    | torch.Size([1, 1000, 754]) | 9\n",
-      "23394    | torch.Size([1, 1000, 754]) | 9\n",
-      "23395    | torch.Size([1, 1000, 754]) | 9\n",
-      "23396    | torch.Size([1, 1000, 754]) | 9\n",
-      "23397    | torch.Size([1, 1000, 754]) | 9\n",
-      "23398    | torch.Size([1, 1000, 754]) | 9\n",
-      "23399    | torch.Size([1, 1000, 754]) | 9\n",
-      "23400    | torch.Size([1, 1000, 802]) | 9\n",
-      "23401    | torch.Size([1, 1000, 769]) | 9\n",
-      "23402    | torch.Size([1, 1000, 754]) | 9\n",
-      "23403    | torch.Size([1, 1000, 771]) | 9\n",
-      "23404    | torch.Size([1, 1000, 777]) | 9\n",
-      "23405    | torch.Size([1, 1000, 777]) | 9\n",
-      "23406    | torch.Size([1, 1000, 780]) | 9\n",
-      "23407    | torch.Size([1, 1000, 777]) | 9\n",
-      "23408    | torch.Size([1, 1000, 754]) | 9\n",
-      "23409    | torch.Size([1, 1000, 754]) | 9\n",
-      "23410    | torch.Size([1, 1000, 777]) | 9\n",
-      "23411    | torch.Size([1, 1000, 773]) | 9\n",
-      "23412    | torch.Size([1, 1000, 762]) | 9\n",
-      "23413    | torch.Size([1, 1000, 754]) | 9\n",
-      "23414    | torch.Size([1, 1000, 754]) | 9\n",
-      "23415    | torch.Size([1, 1000, 754]) | 9\n",
-      "23416    | torch.Size([1, 1000, 754]) | 9\n",
-      "23417    | torch.Size([1, 1000, 810]) | 9\n",
-      "23418    | torch.Size([1, 1000, 754]) | 9\n",
-      "23419    | torch.Size([1, 1000, 754]) | 9\n",
-      "23420    | torch.Size([1, 1000, 754]) | 9\n",
-      "23421    | torch.Size([1, 1000, 762]) | 9\n",
-      "23422    | torch.Size([1, 1000, 777]) | 9\n",
-      "23423    | torch.Size([1, 1000, 794]) | 9\n",
-      "23424    | torch.Size([1, 1000, 774]) | 9\n",
-      "23425    | torch.Size([1, 1000, 780]) | 9\n",
-      "23426    | torch.Size([1, 1000, 818]) | 9\n",
-      "23427    | torch.Size([1, 1000, 754]) | 9\n",
-      "23428    | torch.Size([1, 1000, 754]) | 9\n",
-      "23429    | torch.Size([1, 1000, 754]) | 9\n",
-      "23430    | torch.Size([1, 1000, 725]) | 9\n",
-      "23431    | torch.Size([1, 1000, 754]) | 9\n",
-      "23432    | torch.Size([1, 1000, 754]) | 9\n",
-      "23433    | torch.Size([1, 1000, 762]) | 9\n",
-      "23434    | torch.Size([1, 1000, 777]) | 9\n",
-      "23435    | torch.Size([1, 1000, 786]) | 9\n",
-      "23436    | torch.Size([1, 1000, 754]) | 9\n",
-      "23437    | torch.Size([1, 1000, 611]) | 9\n",
-      "23438    | torch.Size([1, 1000, 777]) | 9\n",
-      "23439    | torch.Size([1, 1000, 784]) | 9\n",
-      "23440    | torch.Size([1, 1000, 754]) | 9\n",
-      "23441    | torch.Size([1, 1000, 777]) | 9\n",
-      "23442    | torch.Size([1, 1000, 754]) | 9\n",
-      "23443    | torch.Size([1, 1000, 777]) | 9\n",
-      "23444    | torch.Size([1, 1000, 776]) | 9\n",
-      "23445    | torch.Size([1, 1000, 754]) | 9\n",
-      "23446    | torch.Size([1, 1000, 777]) | 9\n",
-      "23447    | torch.Size([1, 1000, 777]) | 9\n",
-      "23448    | torch.Size([1, 1000, 754]) | 9\n",
-      "23449    | torch.Size([1, 1000, 864]) | 9\n",
-      "23450    | torch.Size([1, 1000, 754]) | 9\n",
-      "23451    | torch.Size([1, 1000, 788]) | 9\n",
-      "23452    | torch.Size([1, 1000, 754]) | 9\n",
-      "23453    | torch.Size([1, 1000, 777]) | 9\n",
-      "23454    | torch.Size([1, 1000, 754]) | 9\n",
-      "23455    | torch.Size([1, 1000, 611]) | 9\n",
-      "23456    | torch.Size([1, 1000, 754]) | 9\n",
-      "23457    | torch.Size([1, 1000, 777]) | 9\n",
-      "23458    | torch.Size([1, 1000, 846]) | 9\n",
-      "23459    | torch.Size([1, 1000, 777]) | 9\n",
-      "23460    | torch.Size([1, 1000, 784]) | 9\n",
-      "23461    | torch.Size([1, 1000, 754]) | 9\n",
-      "23462    | torch.Size([1, 1000, 770]) | 9\n",
-      "23463    | torch.Size([1, 1000, 754]) | 9\n",
-      "23464    | torch.Size([1, 1000, 754]) | 9\n",
-      "23465    | torch.Size([1, 1000, 754]) | 9\n",
-      "23466    | torch.Size([1, 1000, 754]) | 9\n",
-      "23467    | torch.Size([1, 1000, 754]) | 9\n",
-      "23468    | torch.Size([1, 1000, 754]) | 9\n",
-      "23469    | torch.Size([1, 1000, 754]) | 9\n",
-      "23470    | torch.Size([1, 1000, 777]) | 9\n",
-      "23471    | torch.Size([1, 1000, 754]) | 9\n",
-      "23472    | torch.Size([1, 1000, 754]) | 9\n",
-      "23473    | torch.Size([1, 1000, 782]) | 9\n",
-      "23474    | torch.Size([1, 1000, 754]) | 9\n",
-      "23475    | torch.Size([1, 1000, 754]) | 9\n",
-      "23476    | torch.Size([1, 1000, 777]) | 9\n",
-      "23477    | torch.Size([1, 1000, 766]) | 9\n",
-      "23478    | torch.Size([1, 1000, 780]) | 9\n",
-      "23479    | torch.Size([1, 1000, 762]) | 9\n",
-      "23480    | torch.Size([1, 1000, 777]) | 9\n",
-      "23481    | torch.Size([1, 1000, 754]) | 9\n",
-      "23482    | torch.Size([1, 1000, 754]) | 9\n",
-      "23483    | torch.Size([1, 1000, 754]) | 9\n",
-      "23484    | torch.Size([1, 1000, 777]) | 9\n",
-      "23485    | torch.Size([1, 1000, 794]) | 9\n",
-      "23486    | torch.Size([1, 1000, 768]) | 9\n",
-      "23487    | torch.Size([1, 1000, 781]) | 9\n",
-      "23488    | torch.Size([1, 1000, 771]) | 9\n",
-      "23489    | torch.Size([1, 1000, 754]) | 9\n",
-      "23490    | torch.Size([1, 1000, 766]) | 9\n",
-      "23491    | torch.Size([1, 1000, 762]) | 9\n",
-      "23492    | torch.Size([1, 1000, 794]) | 9\n",
-      "23493    | torch.Size([1, 1000, 779]) | 9\n",
-      "23494    | torch.Size([1, 1000, 754]) | 9\n",
-      "23495    | torch.Size([1, 1000, 777]) | 9\n",
-      "23496    | torch.Size([1, 1000, 773]) | 9\n",
-      "23497    | torch.Size([1, 1000, 754]) | 9\n",
-      "23498    | torch.Size([1, 1000, 614]) | 9\n",
-      "23499    | torch.Size([1, 1000, 783]) | 9\n",
-      "23500    | torch.Size([1, 1000, 777]) | 9\n",
-      "23501    | torch.Size([1, 1000, 754]) | 9\n",
-      "23502    | torch.Size([1, 1000, 754]) | 9\n",
-      "23503    | torch.Size([1, 1000, 754]) | 9\n",
-      "23504    | torch.Size([1, 1000, 754]) | 9\n",
-      "23505    | torch.Size([1, 1000, 754]) | 9\n",
-      "23506    | torch.Size([1, 1000, 754]) | 9\n",
-      "23507    | torch.Size([1, 1000, 780]) | 9\n",
-      "23508    | torch.Size([1, 1000, 754]) | 9\n",
-      "23509    | torch.Size([1, 1000, 754]) | 9\n",
-      "23510    | torch.Size([1, 1000, 777]) | 9\n",
-      "23511    | torch.Size([1, 1000, 777]) | 9\n",
-      "23512    | torch.Size([1, 1000, 754]) | 9\n",
-      "23513    | torch.Size([1, 1000, 754]) | 9\n",
-      "23514    | torch.Size([1, 1000, 777]) | 9\n",
-      "23515    | torch.Size([1, 1000, 777]) | 9\n",
-      "23516    | torch.Size([1, 1000, 754]) | 9\n",
-      "23517    | torch.Size([1, 1000, 754]) | 9\n",
-      "23518    | torch.Size([1, 1000, 781]) | 9\n",
-      "23519    | torch.Size([1, 1000, 754]) | 9\n",
-      "23520    | torch.Size([1, 1000, 775]) | 9\n",
-      "23521    | torch.Size([1, 1000, 786]) | 9\n",
-      "23522    | torch.Size([1, 1000, 754]) | 9\n",
-      "23523    | torch.Size([1, 1000, 762]) | 9\n",
-      "23524    | torch.Size([1, 1000, 754]) | 9\n",
-      "23525    | torch.Size([1, 1000, 754]) | 9\n",
-      "23526    | torch.Size([1, 1000, 777]) | 9\n",
-      "23527    | torch.Size([1, 1000, 754]) | 9\n",
-      "23528    | torch.Size([1, 1000, 754]) | 9\n",
-      "23529    | torch.Size([1, 1000, 754]) | 9\n",
-      "23530    | torch.Size([1, 1000, 754]) | 9\n",
-      "23531    | torch.Size([1, 1000, 754]) | 9\n",
-      "23532    | torch.Size([1, 1000, 754]) | 9\n",
-      "23533    | torch.Size([1, 1000, 777]) | 9\n",
-      "23534    | torch.Size([1, 1000, 754]) | 9\n",
-      "23535    | torch.Size([1, 1000, 777]) | 9\n",
-      "23536    | torch.Size([1, 1000, 802]) | 9\n",
-      "23537    | torch.Size([1, 1000, 777]) | 9\n",
-      "23538    | torch.Size([1, 1000, 777]) | 9\n",
-      "23539    | torch.Size([1, 1000, 754]) | 9\n",
-      "23540    | torch.Size([1, 1000, 754]) | 9\n",
-      "23541    | torch.Size([1, 1000, 754]) | 9\n",
-      "23542    | torch.Size([1, 1000, 777]) | 9\n",
-      "23543    | torch.Size([1, 1000, 754]) | 9\n",
-      "23544    | torch.Size([1, 1000, 810]) | 9\n",
-      "23545    | torch.Size([1, 1000, 792]) | 9\n",
-      "23546    | torch.Size([1, 1000, 754]) | 9\n",
-      "23547    | torch.Size([1, 1000, 754]) | 9\n",
-      "23548    | torch.Size([1, 1000, 772]) | 9\n",
-      "23549    | torch.Size([1, 1000, 754]) | 9\n",
-      "23550    | torch.Size([1, 1000, 762]) | 9\n",
-      "23551    | torch.Size([1, 1000, 725]) | 9\n",
-      "23552    | torch.Size([1, 1000, 768]) | 9\n",
-      "23553    | torch.Size([1, 1000, 754]) | 9\n",
-      "23554    | torch.Size([1, 1000, 762]) | 9\n",
-      "23555    | torch.Size([1, 1000, 777]) | 9\n",
-      "23556    | torch.Size([1, 1000, 777]) | 9\n",
-      "23557    | torch.Size([1, 1000, 777]) | 9\n",
-      "23558    | torch.Size([1, 1000, 786]) | 9\n",
-      "23559    | torch.Size([1, 1000, 784]) | 9\n",
-      "23560    | torch.Size([1, 1000, 777]) | 9\n",
-      "23561    | torch.Size([1, 1000, 754]) | 9\n",
-      "23562    | torch.Size([1, 1000, 754]) | 9\n",
-      "23563    | torch.Size([1, 1000, 754]) | 9\n",
-      "23564    | torch.Size([1, 1000, 767]) | 9\n",
-      "23565    | torch.Size([1, 1000, 777]) | 9\n",
-      "23566    | torch.Size([1, 1000, 777]) | 9\n",
-      "23567    | torch.Size([1, 1000, 754]) | 9\n",
-      "23568    | torch.Size([1, 1000, 777]) | 9\n",
-      "23569    | torch.Size([1, 1000, 778]) | 9\n",
-      "23570    | torch.Size([1, 1000, 754]) | 9\n",
-      "23571    | torch.Size([1, 1000, 754]) | 9\n",
-      "23572    | torch.Size([1, 1000, 778]) | 9\n",
-      "23573    | torch.Size([1, 1000, 786]) | 9\n",
-      "23574    | torch.Size([1, 1000, 777]) | 9\n",
-      "23575    | torch.Size([1, 1000, 754]) | 9\n",
-      "23576    | torch.Size([1, 1000, 777]) | 9\n",
-      "23577    | torch.Size([1, 1000, 777]) | 9\n",
-      "23578    | torch.Size([1, 1000, 754]) | 9\n",
-      "23579    | torch.Size([1, 1000, 778]) | 9\n",
-      "23580    | torch.Size([1, 1000, 754]) | 9\n",
-      "23581    | torch.Size([1, 1000, 777]) | 9\n",
-      "23582    | torch.Size([1, 1000, 754]) | 9\n",
-      "23583    | torch.Size([1, 1000, 777]) | 9\n",
-      "23584    | torch.Size([1, 1000, 754]) | 9\n",
-      "23585    | torch.Size([1, 1000, 777]) | 9\n",
-      "23586    | torch.Size([1, 1000, 805]) | 9\n",
-      "23587    | torch.Size([1, 1000, 777]) | 9\n",
-      "23588    | torch.Size([1, 1000, 754]) | 9\n",
-      "23589    | torch.Size([1, 1000, 777]) | 9\n",
-      "23590    | torch.Size([1, 1000, 754]) | 9\n",
-      "23591    | torch.Size([1, 1000, 754]) | 9\n",
-      "23592    | torch.Size([1, 1000, 776]) | 9\n",
-      "23593    | torch.Size([1, 1000, 777]) | 9\n",
-      "23594    | torch.Size([1, 1000, 777]) | 9\n",
-      "23595    | torch.Size([1, 1000, 754]) | 9\n",
-      "23596    | torch.Size([1, 1000, 766]) | 9\n",
-      "23597    | torch.Size([1, 1000, 777]) | 9\n",
-      "23598    | torch.Size([1, 1000, 762]) | 9\n",
-      "23599    | torch.Size([1, 1000, 777]) | 9\n",
-      "23600    | torch.Size([1, 1000, 773]) | 9\n",
-      "23601    | torch.Size([1, 1000, 862]) | 9\n",
-      "23602    | torch.Size([1, 1000, 754]) | 9\n",
-      "23603    | torch.Size([1, 1000, 777]) | 9\n",
-      "23604    | torch.Size([1, 1000, 754]) | 9\n",
-      "23605    | torch.Size([1, 1000, 777]) | 9\n",
-      "23606    | torch.Size([1, 1000, 778]) | 9\n",
-      "23607    | torch.Size([1, 1000, 754]) | 9\n",
-      "23608    | torch.Size([1, 1000, 781]) | 9\n",
-      "23609    | torch.Size([1, 1000, 777]) | 9\n",
-      "23610    | torch.Size([1, 1000, 766]) | 9\n",
-      "23611    | torch.Size([1, 1000, 754]) | 9\n",
-      "23612    | torch.Size([1, 1000, 777]) | 9\n",
-      "23613    | torch.Size([1, 1000, 754]) | 9\n",
-      "23614    | torch.Size([1, 1000, 779]) | 9\n",
-      "23615    | torch.Size([1, 1000, 779]) | 9\n",
-      "23616    | torch.Size([1, 1000, 777]) | 9\n",
-      "23617    | torch.Size([1, 1000, 777]) | 9\n",
-      "23618    | torch.Size([1, 1000, 754]) | 9\n",
-      "23619    | torch.Size([1, 1000, 754]) | 9\n",
-      "23620    | torch.Size([1, 1000, 724]) | 9\n",
-      "23621    | torch.Size([1, 1000, 775]) | 9\n",
-      "23622    | torch.Size([1, 1000, 754]) | 9\n",
-      "23623    | torch.Size([1, 1000, 754]) | 9\n",
-      "23624    | torch.Size([1, 1000, 782]) | 9\n",
-      "23625    | torch.Size([1, 1000, 754]) | 9\n",
-      "23626    | torch.Size([1, 1000, 810]) | 9\n",
-      "23627    | torch.Size([1, 1000, 777]) | 9\n",
-      "23628    | torch.Size([1, 1000, 774]) | 9\n",
-      "23629    | torch.Size([1, 1000, 754]) | 9\n",
-      "23630    | torch.Size([1, 1000, 754]) | 9\n",
-      "23631    | torch.Size([1, 1000, 777]) | 9\n",
-      "23632    | torch.Size([1, 1000, 777]) | 9\n",
-      "23633    | torch.Size([1, 1000, 754]) | 9\n",
-      "23634    | torch.Size([1, 1000, 754]) | 9\n",
-      "23635    | torch.Size([1, 1000, 777]) | 9\n",
-      "23636    | torch.Size([1, 1000, 754]) | 9\n",
-      "23637    | torch.Size([1, 1000, 754]) | 9\n",
-      "23638    | torch.Size([1, 1000, 754]) | 9\n",
-      "23639    | torch.Size([1, 1000, 754]) | 9\n",
-      "23640    | torch.Size([1, 1000, 810]) | 9\n",
-      "23641    | torch.Size([1, 1000, 754]) | 9\n",
-      "23642    | torch.Size([1, 1000, 754]) | 9\n",
-      "23643    | torch.Size([1, 1000, 777]) | 9\n",
-      "23644    | torch.Size([1, 1000, 754]) | 9\n",
-      "23645    | torch.Size([1, 1000, 754]) | 9\n",
-      "23646    | torch.Size([1, 1000, 766]) | 9\n",
-      "23647    | torch.Size([1, 1000, 754]) | 9\n",
-      "23648    | torch.Size([1, 1000, 782]) | 9\n",
-      "23649    | torch.Size([1, 1000, 754]) | 9\n",
-      "23650    | torch.Size([1, 1000, 754]) | 9\n",
-      "23651    | torch.Size([1, 1000, 777]) | 9\n",
-      "23652    | torch.Size([1, 1000, 754]) | 9\n",
-      "23653    | torch.Size([1, 1000, 754]) | 9\n",
-      "23654    | torch.Size([1, 1000, 767]) | 9\n",
-      "23655    | torch.Size([1, 1000, 802]) | 9\n",
-      "23656    | torch.Size([1, 1000, 754]) | 9\n",
-      "23657    | torch.Size([1, 1000, 754]) | 9\n",
-      "23658    | torch.Size([1, 1000, 777]) | 9\n",
-      "23659    | torch.Size([1, 1000, 778]) | 9\n",
-      "23660    | torch.Size([1, 1000, 754]) | 9\n",
-      "23661    | torch.Size([1, 1000, 777]) | 9\n",
-      "23662    | torch.Size([1, 1000, 788]) | 9\n",
-      "23663    | torch.Size([1, 1000, 754]) | 9\n",
-      "23664    | torch.Size([1, 1000, 777]) | 9\n",
-      "23665    | torch.Size([1, 1000, 754]) | 9\n",
-      "23666    | torch.Size([1, 1000, 791]) | 9\n",
-      "23667    | torch.Size([1, 1000, 754]) | 9\n",
-      "23668    | torch.Size([1, 1000, 777]) | 9\n",
-      "23669    | torch.Size([1, 1000, 774]) | 9\n",
-      "23670    | torch.Size([1, 1000, 754]) | 9\n",
-      "23671    | torch.Size([1, 1000, 754]) | 9\n",
-      "23672    | torch.Size([1, 1000, 754]) | 9\n",
-      "23673    | torch.Size([1, 1000, 754]) | 9\n",
-      "23674    | torch.Size([1, 1000, 777]) | 9\n",
-      "23675    | torch.Size([1, 1000, 754]) | 9\n",
-      "23676    | torch.Size([1, 1000, 754]) | 9\n",
-      "23677    | torch.Size([1, 1000, 762]) | 9\n",
-      "23678    | torch.Size([1, 1000, 754]) | 9\n",
-      "23679    | torch.Size([1, 1000, 777]) | 9\n",
-      "23680    | torch.Size([1, 1000, 754]) | 9\n",
-      "23681    | torch.Size([1, 1000, 783]) | 9\n",
-      "23682    | torch.Size([1, 1000, 754]) | 9\n",
-      "23683    | torch.Size([1, 1000, 754]) | 9\n",
-      "23684    | torch.Size([1, 1000, 765]) | 9\n",
-      "23685    | torch.Size([1, 1000, 754]) | 9\n",
-      "23686    | torch.Size([1, 1000, 793]) | 9\n",
-      "23687    | torch.Size([1, 1000, 754]) | 9\n",
-      "23688    | torch.Size([1, 1000, 754]) | 9\n",
-      "23689    | torch.Size([1, 1000, 754]) | 9\n",
-      "23690    | torch.Size([1, 1000, 762]) | 9\n",
-      "23691    | torch.Size([1, 1000, 754]) | 9\n",
-      "23692    | torch.Size([1, 1000, 768]) | 9\n",
-      "23693    | torch.Size([1, 1000, 785]) | 9\n",
-      "23694    | torch.Size([1, 1000, 777]) | 9\n",
-      "23695    | torch.Size([1, 1000, 777]) | 9\n",
-      "23696    | torch.Size([1, 1000, 754]) | 9\n",
-      "23697    | torch.Size([1, 1000, 614]) | 9\n",
-      "23698    | torch.Size([1, 1000, 797]) | 9\n",
-      "23699    | torch.Size([1, 1000, 779]) | 9\n",
-      "23700    | torch.Size([1, 1000, 754]) | 9\n",
-      "23701    | torch.Size([1, 1000, 786]) | 9\n",
-      "23702    | torch.Size([1, 1000, 785]) | 9\n",
-      "23703    | torch.Size([1, 1000, 754]) | 9\n",
-      "23704    | torch.Size([1, 1000, 777]) | 9\n",
-      "23705    | torch.Size([1, 1000, 777]) | 9\n",
-      "23706    | torch.Size([1, 1000, 754]) | 9\n",
-      "23707    | torch.Size([1, 1000, 754]) | 9\n",
-      "23708    | torch.Size([1, 1000, 754]) | 9\n",
-      "23709    | torch.Size([1, 1000, 755]) | 9\n",
-      "23710    | torch.Size([1, 1000, 754]) | 9\n",
-      "23711    | torch.Size([1, 1000, 809]) | 9\n",
-      "23712    | torch.Size([1, 1000, 777]) | 9\n",
-      "23713    | torch.Size([1, 1000, 770]) | 9\n",
-      "23714    | torch.Size([1, 1000, 754]) | 9\n",
-      "23715    | torch.Size([1, 1000, 780]) | 9\n",
-      "23716    | torch.Size([1, 1000, 777]) | 9\n",
-      "23717    | torch.Size([1, 1000, 754]) | 9\n",
-      "23718    | torch.Size([1, 1000, 754]) | 9\n",
-      "23719    | torch.Size([1, 1000, 754]) | 9\n",
-      "23720    | torch.Size([1, 1000, 762]) | 9\n",
-      "23721    | torch.Size([1, 1000, 784]) | 9\n",
-      "23722    | torch.Size([1, 1000, 777]) | 9\n",
-      "23723    | torch.Size([1, 1000, 754]) | 9\n",
-      "23724    | torch.Size([1, 1000, 754]) | 9\n",
-      "23725    | torch.Size([1, 1000, 754]) | 9\n",
-      "23726    | torch.Size([1, 1000, 754]) | 9\n",
-      "23727    | torch.Size([1, 1000, 754]) | 9\n",
-      "23728    | torch.Size([1, 1000, 777]) | 9\n",
-      "23729    | torch.Size([1, 1000, 790]) | 9\n",
-      "23730    | torch.Size([1, 1000, 754]) | 9\n",
-      "23731    | torch.Size([1, 1000, 777]) | 9\n",
-      "23732    | torch.Size([1, 1000, 762]) | 9\n",
-      "23733    | torch.Size([1, 1000, 754]) | 9\n",
-      "23734    | torch.Size([1, 1000, 749]) | 9\n",
-      "23735    | torch.Size([1, 1000, 769]) | 9\n",
-      "23736    | torch.Size([1, 1000, 794]) | 9\n",
-      "23737    | torch.Size([1, 1000, 754]) | 9\n",
-      "23738    | torch.Size([1, 1000, 754]) | 9\n",
-      "23739    | torch.Size([1, 1000, 764]) | 9\n",
-      "23740    | torch.Size([1, 1000, 777]) | 9\n",
-      "23741    | torch.Size([1, 1000, 777]) | 9\n",
-      "23742    | torch.Size([1, 1000, 797]) | 9\n",
-      "23743    | torch.Size([1, 1000, 777]) | 9\n",
-      "23744    | torch.Size([1, 1000, 869]) | 9\n",
-      "23745    | torch.Size([1, 1000, 754]) | 9\n",
-      "23746    | torch.Size([1, 1000, 754]) | 9\n",
-      "23747    | torch.Size([1, 1000, 730]) | 9\n",
-      "23748    | torch.Size([1, 1000, 777]) | 9\n",
-      "23749    | torch.Size([1, 1000, 754]) | 9\n",
-      "23750    | torch.Size([1, 1000, 777]) | 9\n",
-      "23751    | torch.Size([1, 1000, 780]) | 9\n",
-      "23752    | torch.Size([1, 1000, 777]) | 9\n",
-      "23753    | torch.Size([1, 1000, 754]) | 9\n",
-      "23754    | torch.Size([1, 1000, 754]) | 9\n",
-      "23755    | torch.Size([1, 1000, 777]) | 9\n",
-      "23756    | torch.Size([1, 1000, 754]) | 9\n",
-      "23757    | torch.Size([1, 1000, 754]) | 9\n",
-      "23758    | torch.Size([1, 1000, 754]) | 9\n",
-      "23759    | torch.Size([1, 1000, 767]) | 9\n",
-      "23760    | torch.Size([1, 1000, 754]) | 9\n",
-      "23761    | torch.Size([1, 1000, 829]) | 9\n",
-      "23762    | torch.Size([1, 1000, 777]) | 9\n",
-      "23763    | torch.Size([1, 1000, 778]) | 9\n",
-      "23764    | torch.Size([1, 1000, 754]) | 9\n",
-      "23765    | torch.Size([1, 1000, 754]) | 9\n",
-      "23766    | torch.Size([1, 1000, 754]) | 9\n",
-      "23767    | torch.Size([1, 1000, 797]) | 9\n",
-      "23768    | torch.Size([1, 1000, 777]) | 9\n",
-      "23769    | torch.Size([1, 1000, 797]) | 9\n",
-      "23770    | torch.Size([1, 1000, 845]) | 9\n",
-      "23771    | torch.Size([1, 1000, 754]) | 9\n",
-      "23772    | torch.Size([1, 1000, 754]) | 9\n",
-      "23773    | torch.Size([1, 1000, 754]) | 9\n",
-      "23774    | torch.Size([1, 1000, 754]) | 9\n",
-      "23775    | torch.Size([1, 1000, 793]) | 9\n",
-      "23776    | torch.Size([1, 1000, 778]) | 9\n",
-      "23777    | torch.Size([1, 1000, 754]) | 9\n",
-      "23778    | torch.Size([1, 1000, 780]) | 9\n",
-      "23779    | torch.Size([1, 1000, 802]) | 9\n",
-      "23780    | torch.Size([1, 1000, 792]) | 9\n",
-      "23781    | torch.Size([1, 1000, 754]) | 9\n",
-      "23782    | torch.Size([1, 1000, 762]) | 9\n",
-      "23783    | torch.Size([1, 1000, 754]) | 9\n",
-      "23784    | torch.Size([1, 1000, 754]) | 9\n",
-      "23785    | torch.Size([1, 1000, 754]) | 9\n",
-      "23786    | torch.Size([1, 1000, 754]) | 9\n",
-      "23787    | torch.Size([1, 1000, 754]) | 9\n",
-      "23788    | torch.Size([1, 1000, 789]) | 9\n",
-      "23789    | torch.Size([1, 1000, 806]) | 9\n",
-      "23790    | torch.Size([1, 1000, 754]) | 9\n",
-      "23791    | torch.Size([1, 1000, 777]) | 9\n",
-      "23792    | torch.Size([1, 1000, 777]) | 9\n",
-      "23793    | torch.Size([1, 1000, 762]) | 9\n",
-      "23794    | torch.Size([1, 1000, 782]) | 9\n",
-      "23795    | torch.Size([1, 1000, 719]) | 9\n",
-      "23796    | torch.Size([1, 1000, 777]) | 9\n",
-      "23797    | torch.Size([1, 1000, 773]) | 9\n",
-      "23798    | torch.Size([1, 1000, 782]) | 9\n",
-      "23799    | torch.Size([1, 1000, 777]) | 9\n",
-      "23800    | torch.Size([1, 1000, 802]) | 9\n",
-      "23801    | torch.Size([1, 1000, 754]) | 9\n",
-      "23802    | torch.Size([1, 1000, 762]) | 9\n",
-      "23803    | torch.Size([1, 1000, 754]) | 9\n",
-      "23804    | torch.Size([1, 1000, 777]) | 9\n",
-      "23805    | torch.Size([1, 1000, 786]) | 9\n",
-      "23806    | torch.Size([1, 1000, 777]) | 9\n",
-      "23807    | torch.Size([1, 1000, 763]) | 9\n",
-      "23808    | torch.Size([1, 1000, 777]) | 9\n",
-      "23809    | torch.Size([1, 1000, 754]) | 9\n",
-      "23810    | torch.Size([1, 1000, 767]) | 9\n",
-      "23811    | torch.Size([1, 1000, 777]) | 9\n",
-      "23812    | torch.Size([1, 1000, 777]) | 9\n",
-      "23813    | torch.Size([1, 1000, 777]) | 9\n",
-      "23814    | torch.Size([1, 1000, 754]) | 9\n",
-      "23815    | torch.Size([1, 1000, 754]) | 9\n",
-      "23816    | torch.Size([1, 1000, 754]) | 9\n",
-      "23817    | torch.Size([1, 1000, 754]) | 9\n",
-      "23818    | torch.Size([1, 1000, 724]) | 9\n",
-      "23819    | torch.Size([1, 1000, 754]) | 9\n",
-      "23820    | torch.Size([1, 1000, 754]) | 9\n",
-      "23821    | torch.Size([1, 1000, 754]) | 9\n",
-      "23822    | torch.Size([1, 1000, 780]) | 9\n",
-      "23823    | torch.Size([1, 1000, 754]) | 9\n",
-      "23824    | torch.Size([1, 1000, 754]) | 9\n",
-      "23825    | torch.Size([1, 1000, 784]) | 9\n",
-      "23826    | torch.Size([1, 1000, 758]) | 9\n",
-      "23827    | torch.Size([1, 1000, 754]) | 9\n",
-      "23828    | torch.Size([1, 1000, 754]) | 9\n",
-      "23829    | torch.Size([1, 1000, 754]) | 9\n",
-      "23830    | torch.Size([1, 1000, 754]) | 9\n",
-      "23831    | torch.Size([1, 1000, 772]) | 9\n",
-      "23832    | torch.Size([1, 1000, 754]) | 9\n",
-      "23833    | torch.Size([1, 1000, 754]) | 9\n",
-      "23834    | torch.Size([1, 1000, 754]) | 9\n",
-      "23835    | torch.Size([1, 1000, 782]) | 9\n",
-      "23836    | torch.Size([1, 1000, 754]) | 9\n",
-      "23837    | torch.Size([1, 1000, 778]) | 9\n",
-      "23838    | torch.Size([1, 1000, 777]) | 9\n",
-      "23839    | torch.Size([1, 1000, 754]) | 9\n",
-      "23840    | torch.Size([1, 1000, 754]) | 9\n",
-      "23841    | torch.Size([1, 1000, 865]) | 9\n",
-      "23842    | torch.Size([1, 1000, 754]) | 9\n",
-      "23843    | torch.Size([1, 1000, 802]) | 9\n",
-      "23844    | torch.Size([1, 1000, 754]) | 9\n",
-      "23845    | torch.Size([1, 1000, 777]) | 9\n",
-      "23846    | torch.Size([1, 1000, 754]) | 9\n",
-      "23847    | torch.Size([1, 1000, 777]) | 9\n",
-      "23848    | torch.Size([1, 1000, 777]) | 9\n",
-      "23849    | torch.Size([1, 1000, 754]) | 9\n",
-      "23850    | torch.Size([1, 1000, 754]) | 9\n",
-      "23851    | torch.Size([1, 1000, 754]) | 9\n",
-      "23852    | torch.Size([1, 1000, 754]) | 9\n",
-      "23853    | torch.Size([1, 1000, 754]) | 9\n",
-      "23854    | torch.Size([1, 1000, 754]) | 9\n",
-      "23855    | torch.Size([1, 1000, 797]) | 9\n",
-      "23856    | torch.Size([1, 1000, 619]) | 9\n",
-      "23857    | torch.Size([1, 1000, 754]) | 9\n",
-      "23858    | torch.Size([1, 1000, 802]) | 9\n",
-      "23859    | torch.Size([1, 1000, 786]) | 9\n",
-      "23860    | torch.Size([1, 1000, 754]) | 9\n",
-      "23861    | torch.Size([1, 1000, 754]) | 9\n",
-      "23862    | torch.Size([1, 1000, 796]) | 9\n",
-      "23863    | torch.Size([1, 1000, 754]) | 9\n",
-      "23864    | torch.Size([1, 1000, 777]) | 9\n",
-      "23865    | torch.Size([1, 1000, 754]) | 9\n",
-      "23866    | torch.Size([1, 1000, 754]) | 9\n",
-      "23867    | torch.Size([1, 1000, 754]) | 9\n",
-      "23868    | torch.Size([1, 1000, 794]) | 9\n",
-      "23869    | torch.Size([1, 1000, 793]) | 9\n",
-      "23870    | torch.Size([1, 1000, 754]) | 9\n",
-      "23871    | torch.Size([1, 1000, 754]) | 9\n",
-      "23872    | torch.Size([1, 1000, 762]) | 9\n",
-      "23873    | torch.Size([1, 1000, 777]) | 9\n",
-      "23874    | torch.Size([1, 1000, 772]) | 9\n",
-      "23875    | torch.Size([1, 1000, 762]) | 9\n",
-      "23876    | torch.Size([1, 1000, 806]) | 9\n",
-      "23877    | torch.Size([1, 1000, 754]) | 9\n",
-      "23878    | torch.Size([1, 1000, 777]) | 9\n",
-      "23879    | torch.Size([1, 1000, 754]) | 9\n",
-      "23880    | torch.Size([1, 1000, 785]) | 9\n",
-      "23881    | torch.Size([1, 1000, 754]) | 9\n",
-      "23882    | torch.Size([1, 1000, 724]) | 9\n",
-      "23883    | torch.Size([1, 1000, 777]) | 9\n",
-      "23884    | torch.Size([1, 1000, 754]) | 9\n",
-      "23885    | torch.Size([1, 1000, 777]) | 9\n",
-      "23886    | torch.Size([1, 1000, 782]) | 9\n",
-      "23887    | torch.Size([1, 1000, 786]) | 9\n",
-      "23888    | torch.Size([1, 1000, 777]) | 9\n",
-      "23889    | torch.Size([1, 1000, 784]) | 9\n",
-      "23890    | torch.Size([1, 1000, 790]) | 9\n",
-      "23891    | torch.Size([1, 1000, 754]) | 9\n",
-      "23892    | torch.Size([1, 1000, 754]) | 9\n",
-      "23893    | torch.Size([1, 1000, 777]) | 9\n",
-      "23894    | torch.Size([1, 1000, 754]) | 9\n",
-      "23895    | torch.Size([1, 1000, 777]) | 9\n",
-      "23896    | torch.Size([1, 1000, 754]) | 9\n",
-      "23897    | torch.Size([1, 1000, 777]) | 9\n",
-      "23898    | torch.Size([1, 1000, 777]) | 9\n",
-      "23899    | torch.Size([1, 1000, 754]) | 9\n",
-      "23900    | torch.Size([1, 1000, 754]) | 9\n",
-      "23901    | torch.Size([1, 1000, 777]) | 9\n",
-      "23902    | torch.Size([1, 1000, 777]) | 9\n",
-      "23903    | torch.Size([1, 1000, 793]) | 9\n",
-      "23904    | torch.Size([1, 1000, 762]) | 9\n",
-      "23905    | torch.Size([1, 1000, 754]) | 9\n",
-      "23906    | torch.Size([1, 1000, 754]) | 9\n",
-      "23907    | torch.Size([1, 1000, 778]) | 9\n",
-      "23908    | torch.Size([1, 1000, 795]) | 9\n",
-      "23909    | torch.Size([1, 1000, 802]) | 9\n",
-      "23910    | torch.Size([1, 1000, 777]) | 9\n",
-      "23911    | torch.Size([1, 1000, 777]) | 9\n",
-      "23912    | torch.Size([1, 1000, 788]) | 9\n",
-      "23913    | torch.Size([1, 1000, 766]) | 9\n",
-      "23914    | torch.Size([1, 1000, 754]) | 9\n",
-      "23915    | torch.Size([1, 1000, 754]) | 9\n",
-      "23916    | torch.Size([1, 1000, 777]) | 9\n",
-      "23917    | torch.Size([1, 1000, 754]) | 9\n",
-      "23918    | torch.Size([1, 1000, 754]) | 9\n",
-      "23919    | torch.Size([1, 1000, 762]) | 9\n",
-      "23920    | torch.Size([1, 1000, 754]) | 9\n",
-      "23921    | torch.Size([1, 1000, 777]) | 9\n",
-      "23922    | torch.Size([1, 1000, 754]) | 9\n",
-      "23923    | torch.Size([1, 1000, 708]) | 9\n",
-      "23924    | torch.Size([1, 1000, 776]) | 9\n",
-      "23925    | torch.Size([1, 1000, 778]) | 9\n",
-      "23926    | torch.Size([1, 1000, 754]) | 9\n",
-      "23927    | torch.Size([1, 1000, 754]) | 9\n",
-      "23928    | torch.Size([1, 1000, 754]) | 9\n",
-      "23929    | torch.Size([1, 1000, 777]) | 9\n",
-      "23930    | torch.Size([1, 1000, 754]) | 9\n",
-      "23931    | torch.Size([1, 1000, 754]) | 9\n",
-      "23932    | torch.Size([1, 1000, 754]) | 9\n",
-      "23933    | torch.Size([1, 1000, 864]) | 9\n",
-      "23934    | torch.Size([1, 1000, 777]) | 9\n",
-      "23935    | torch.Size([1, 1000, 762]) | 9\n",
-      "23936    | torch.Size([1, 1000, 754]) | 9\n",
-      "23937    | torch.Size([1, 1000, 754]) | 9\n",
-      "23938    | torch.Size([1, 1000, 754]) | 9\n",
-      "23939    | torch.Size([1, 1000, 777]) | 9\n",
-      "23940    | torch.Size([1, 1000, 786]) | 9\n",
-      "23941    | torch.Size([1, 1000, 802]) | 9\n",
-      "23942    | torch.Size([1, 1000, 754]) | 9\n",
-      "23943    | torch.Size([1, 1000, 777]) | 9\n",
-      "23944    | torch.Size([1, 1000, 754]) | 9\n",
-      "23945    | torch.Size([1, 1000, 777]) | 9\n",
-      "23946    | torch.Size([1, 1000, 777]) | 9\n",
-      "23947    | torch.Size([1, 1000, 754]) | 9\n",
-      "23948    | torch.Size([1, 1000, 754]) | 9\n",
-      "23949    | torch.Size([1, 1000, 797]) | 9\n",
-      "23950    | torch.Size([1, 1000, 777]) | 9\n",
-      "23951    | torch.Size([1, 1000, 754]) | 9\n",
-      "23952    | torch.Size([1, 1000, 754]) | 9\n",
-      "23953    | torch.Size([1, 1000, 772]) | 9\n",
-      "23954    | torch.Size([1, 1000, 762]) | 9\n",
-      "23955    | torch.Size([1, 1000, 754]) | 9\n",
-      "23956    | torch.Size([1, 1000, 777]) | 9\n",
-      "23957    | torch.Size([1, 1000, 754]) | 9\n",
-      "23958    | torch.Size([1, 1000, 805]) | 9\n",
-      "23959    | torch.Size([1, 1000, 777]) | 9\n",
-      "23960    | torch.Size([1, 1000, 754]) | 9\n",
-      "23961    | torch.Size([1, 1000, 754]) | 9\n",
-      "23962    | torch.Size([1, 1000, 777]) | 9\n",
-      "23963    | torch.Size([1, 1000, 754]) | 9\n",
-      "23964    | torch.Size([1, 1000, 754]) | 9\n",
-      "23965    | torch.Size([1, 1000, 754]) | 9\n",
-      "23966    | torch.Size([1, 1000, 763]) | 9\n",
-      "23967    | torch.Size([1, 1000, 754]) | 9\n",
-      "23968    | torch.Size([1, 1000, 754]) | 9\n",
-      "23969    | torch.Size([1, 1000, 754]) | 9\n",
-      "23970    | torch.Size([1, 1000, 777]) | 9\n",
-      "23971    | torch.Size([1, 1000, 754]) | 9\n",
-      "23972    | torch.Size([1, 1000, 754]) | 9\n",
-      "23973    | torch.Size([1, 1000, 777]) | 9\n",
-      "23974    | torch.Size([1, 1000, 777]) | 9\n",
-      "23975    | torch.Size([1, 1000, 777]) | 9\n",
-      "23976    | torch.Size([1, 1000, 754]) | 9\n",
-      "23977    | torch.Size([1, 1000, 777]) | 9\n",
-      "23978    | torch.Size([1, 1000, 777]) | 9\n",
-      "23979    | torch.Size([1, 1000, 777]) | 9\n",
-      "23980    | torch.Size([1, 1000, 777]) | 9\n",
-      "23981    | torch.Size([1, 1000, 777]) | 9\n",
-      "23982    | torch.Size([1, 1000, 777]) | 9\n",
-      "23983    | torch.Size([1, 1000, 754]) | 9\n",
-      "23984    | torch.Size([1, 1000, 777]) | 9\n",
-      "23985    | torch.Size([1, 1000, 780]) | 9\n",
-      "23986    | torch.Size([1, 1000, 754]) | 9\n",
-      "23987    | torch.Size([1, 1000, 797]) | 9\n",
-      "23988    | torch.Size([1, 1000, 777]) | 9\n",
-      "23989    | torch.Size([1, 1000, 778]) | 9\n",
-      "23990    | torch.Size([1, 1000, 754]) | 9\n",
-      "23991    | torch.Size([1, 1000, 754]) | 9\n",
-      "23992    | torch.Size([1, 1000, 754]) | 9\n",
-      "23993    | torch.Size([1, 1000, 777]) | 9\n",
-      "23994    | torch.Size([1, 1000, 777]) | 9\n",
-      "23995    | torch.Size([1, 1000, 777]) | 9\n",
-      "23996    | torch.Size([1, 1000, 754]) | 9\n",
-      "23997    | torch.Size([1, 1000, 842]) | 9\n",
-      "23998    | torch.Size([1, 1000, 762]) | 9\n",
-      "23999    | torch.Size([1, 1000, 754]) | 9\n",
-      "24000    | torch.Size([1, 1000, 754]) | 9\n",
-      "24001    | torch.Size([1, 1000, 754]) | 9\n",
-      "24002    | torch.Size([1, 1000, 773]) | 9\n",
-      "24003    | torch.Size([1, 1000, 754]) | 9\n",
-      "24004    | torch.Size([1, 1000, 762]) | 9\n",
-      "24005    | torch.Size([1, 1000, 754]) | 9\n",
-      "24006    | torch.Size([1, 1000, 776]) | 9\n",
-      "24007    | torch.Size([1, 1000, 777]) | 9\n",
-      "24008    | torch.Size([1, 1000, 777]) | 9\n",
-      "24009    | torch.Size([1, 1000, 777]) | 9\n",
-      "24010    | torch.Size([1, 1000, 777]) | 9\n",
-      "24011    | torch.Size([1, 1000, 754]) | 9\n",
-      "24012    | torch.Size([1, 1000, 777]) | 9\n",
-      "24013    | torch.Size([1, 1000, 785]) | 9\n",
-      "24014    | torch.Size([1, 1000, 762]) | 9\n",
-      "24015    | torch.Size([1, 1000, 754]) | 9\n",
-      "24016    | torch.Size([1, 1000, 802]) | 9\n",
-      "24017    | torch.Size([1, 1000, 754]) | 9\n",
-      "24018    | torch.Size([1, 1000, 754]) | 9\n",
-      "24019    | torch.Size([1, 1000, 754]) | 9\n",
-      "24020    | torch.Size([1, 1000, 793]) | 9\n",
-      "24021    | torch.Size([1, 1000, 868]) | 9\n",
-      "24022    | torch.Size([1, 1000, 754]) | 9\n",
-      "24023    | torch.Size([1, 1000, 803]) | 9\n",
-      "24024    | torch.Size([1, 1000, 754]) | 9\n",
-      "24025    | torch.Size([1, 1000, 777]) | 9\n",
-      "24026    | torch.Size([1, 1000, 777]) | 9\n",
-      "24027    | torch.Size([1, 1000, 777]) | 9\n",
-      "24028    | torch.Size([1, 1000, 777]) | 9\n",
-      "24029    | torch.Size([1, 1000, 754]) | 9\n",
-      "24030    | torch.Size([1, 1000, 779]) | 9\n",
-      "24031    | torch.Size([1, 1000, 777]) | 9\n",
-      "24032    | torch.Size([1, 1000, 777]) | 9\n",
-      "24033    | torch.Size([1, 1000, 777]) | 9\n",
-      "24034    | torch.Size([1, 1000, 754]) | 9\n",
-      "24035    | torch.Size([1, 1000, 754]) | 9\n",
-      "24036    | torch.Size([1, 1000, 777]) | 9\n",
-      "24037    | torch.Size([1, 1000, 754]) | 9\n",
-      "24038    | torch.Size([1, 1000, 777]) | 9\n",
-      "24039    | torch.Size([1, 1000, 777]) | 9\n",
-      "24040    | torch.Size([1, 1000, 777]) | 9\n",
-      "24041    | torch.Size([1, 1000, 754]) | 9\n",
-      "24042    | torch.Size([1, 1000, 788]) | 9\n",
-      "24043    | torch.Size([1, 1000, 777]) | 9\n",
-      "24044    | torch.Size([1, 1000, 788]) | 9\n",
-      "24045    | torch.Size([1, 1000, 777]) | 9\n",
-      "24046    | torch.Size([1, 1000, 792]) | 9\n",
-      "24047    | torch.Size([1, 1000, 781]) | 9\n",
-      "24048    | torch.Size([1, 1000, 754]) | 9\n",
-      "24049    | torch.Size([1, 1000, 754]) | 9\n",
-      "24050    | torch.Size([1, 1000, 777]) | 9\n",
-      "24051    | torch.Size([1, 1000, 754]) | 9\n",
-      "24052    | torch.Size([1, 1000, 782]) | 9\n",
-      "24053    | torch.Size([1, 1000, 782]) | 9\n",
-      "24054    | torch.Size([1, 1000, 754]) | 9\n",
-      "24055    | torch.Size([1, 1000, 754]) | 9\n",
-      "24056    | torch.Size([1, 1000, 754]) | 9\n",
-      "24057    | torch.Size([1, 1000, 754]) | 9\n",
-      "24058    | torch.Size([1, 1000, 754]) | 9\n",
-      "24059    | torch.Size([1, 1000, 802]) | 9\n",
-      "24060    | torch.Size([1, 1000, 754]) | 9\n",
-      "24061    | torch.Size([1, 1000, 762]) | 9\n",
-      "24062    | torch.Size([1, 1000, 754]) | 9\n",
-      "24063    | torch.Size([1, 1000, 754]) | 9\n",
-      "24064    | torch.Size([1, 1000, 754]) | 9\n",
-      "24065    | torch.Size([1, 1000, 754]) | 9\n",
-      "24066    | torch.Size([1, 1000, 754]) | 9\n",
-      "24067    | torch.Size([1, 1000, 754]) | 9\n",
-      "24068    | torch.Size([1, 1000, 739]) | 9\n",
-      "24069    | torch.Size([1, 1000, 754]) | 9\n",
-      "24070    | torch.Size([1, 1000, 754]) | 9\n",
-      "24071    | torch.Size([1, 1000, 754]) | 9\n",
-      "24072    | torch.Size([1, 1000, 754]) | 9\n",
-      "24073    | torch.Size([1, 1000, 777]) | 9\n",
-      "24074    | torch.Size([1, 1000, 777]) | 9\n",
-      "24075    | torch.Size([1, 1000, 754]) | 9\n",
-      "24076    | torch.Size([1, 1000, 777]) | 9\n",
-      "24077    | torch.Size([1, 1000, 754]) | 9\n",
-      "24078    | torch.Size([1, 1000, 754]) | 9\n",
-      "24079    | torch.Size([1, 1000, 754]) | 9\n",
-      "24080    | torch.Size([1, 1000, 754]) | 9\n",
-      "24081    | torch.Size([1, 1000, 741]) | 9\n",
-      "24082    | torch.Size([1, 1000, 754]) | 9\n",
-      "24083    | torch.Size([1, 1000, 772]) | 9\n",
-      "24084    | torch.Size([1, 1000, 754]) | 9\n",
-      "24085    | torch.Size([1, 1000, 754]) | 9\n",
-      "24086    | torch.Size([1, 1000, 754]) | 9\n",
-      "24087    | torch.Size([1, 1000, 810]) | 9\n",
-      "24088    | torch.Size([1, 1000, 777]) | 9\n",
-      "24089    | torch.Size([1, 1000, 777]) | 9\n",
-      "24090    | torch.Size([1, 1000, 802]) | 9\n",
-      "24091    | torch.Size([1, 1000, 754]) | 9\n",
-      "24092    | torch.Size([1, 1000, 777]) | 9\n",
-      "24093    | torch.Size([1, 1000, 754]) | 9\n",
-      "24094    | torch.Size([1, 1000, 754]) | 9\n",
-      "24095    | torch.Size([1, 1000, 772]) | 9\n",
-      "24096    | torch.Size([1, 1000, 754]) | 9\n",
-      "24097    | torch.Size([1, 1000, 754]) | 9\n",
-      "24098    | torch.Size([1, 1000, 754]) | 9\n",
-      "24099    | torch.Size([1, 1000, 754]) | 9\n",
-      "24100    | torch.Size([1, 1000, 754]) | 9\n",
-      "24101    | torch.Size([1, 1000, 784]) | 9\n",
-      "24102    | torch.Size([1, 1000, 777]) | 9\n",
-      "24103    | torch.Size([1, 1000, 779]) | 9\n",
-      "24104    | torch.Size([1, 1000, 780]) | 9\n",
-      "24105    | torch.Size([1, 1000, 754]) | 9\n",
-      "24106    | torch.Size([1, 1000, 777]) | 9\n",
-      "24107    | torch.Size([1, 1000, 777]) | 9\n",
-      "24108    | torch.Size([1, 1000, 754]) | 9\n",
-      "24109    | torch.Size([1, 1000, 754]) | 9\n",
-      "24110    | torch.Size([1, 1000, 785]) | 9\n",
-      "24111    | torch.Size([1, 1000, 754]) | 9\n",
-      "24112    | torch.Size([1, 1000, 777]) | 9\n",
-      "24113    | torch.Size([1, 1000, 780]) | 9\n",
-      "24114    | torch.Size([1, 1000, 777]) | 9\n",
-      "24115    | torch.Size([1, 1000, 777]) | 9\n",
-      "24116    | torch.Size([1, 1000, 782]) | 9\n",
-      "24117    | torch.Size([1, 1000, 782]) | 9\n",
-      "24118    | torch.Size([1, 1000, 754]) | 9\n",
-      "24119    | torch.Size([1, 1000, 777]) | 9\n",
-      "24120    | torch.Size([1, 1000, 754]) | 9\n",
-      "24121    | torch.Size([1, 1000, 754]) | 9\n",
-      "24122    | torch.Size([1, 1000, 754]) | 9\n",
-      "24123    | torch.Size([1, 1000, 779]) | 9\n",
-      "24124    | torch.Size([1, 1000, 754]) | 9\n",
-      "24125    | torch.Size([1, 1000, 779]) | 9\n",
-      "24126    | torch.Size([1, 1000, 777]) | 9\n",
-      "24127    | torch.Size([1, 1000, 782]) | 9\n",
-      "24128    | torch.Size([1, 1000, 754]) | 9\n",
-      "24129    | torch.Size([1, 1000, 785]) | 9\n",
-      "24130    | torch.Size([1, 1000, 754]) | 9\n",
-      "24131    | torch.Size([1, 1000, 787]) | 9\n",
-      "24132    | torch.Size([1, 1000, 777]) | 9\n",
-      "24133    | torch.Size([1, 1000, 779]) | 9\n",
-      "24134    | torch.Size([1, 1000, 754]) | 9\n",
-      "24135    | torch.Size([1, 1000, 754]) | 9\n",
-      "24136    | torch.Size([1, 1000, 796]) | 9\n",
-      "24137    | torch.Size([1, 1000, 777]) | 9\n",
-      "24138    | torch.Size([1, 1000, 754]) | 9\n",
-      "24139    | torch.Size([1, 1000, 754]) | 9\n",
-      "24140    | torch.Size([1, 1000, 754]) | 9\n",
-      "24141    | torch.Size([1, 1000, 773]) | 9\n",
-      "24142    | torch.Size([1, 1000, 784]) | 9\n",
-      "24143    | torch.Size([1, 1000, 754]) | 9\n",
-      "24144    | torch.Size([1, 1000, 777]) | 9\n",
-      "24145    | torch.Size([1, 1000, 754]) | 9\n",
-      "24146    | torch.Size([1, 1000, 802]) | 9\n",
-      "24147    | torch.Size([1, 1000, 754]) | 9\n",
-      "24148    | torch.Size([1, 1000, 754]) | 9\n",
-      "24149    | torch.Size([1, 1000, 777]) | 9\n",
-      "24150    | torch.Size([1, 1000, 754]) | 9\n",
-      "24151    | torch.Size([1, 1000, 754]) | 9\n",
-      "24152    | torch.Size([1, 1000, 754]) | 9\n",
-      "24153    | torch.Size([1, 1000, 777]) | 9\n",
-      "24154    | torch.Size([1, 1000, 754]) | 9\n",
-      "24155    | torch.Size([1, 1000, 754]) | 9\n",
-      "24156    | torch.Size([1, 1000, 726]) | 9\n",
-      "24157    | torch.Size([1, 1000, 768]) | 9\n",
-      "24158    | torch.Size([1, 1000, 786]) | 9\n",
-      "24159    | torch.Size([1, 1000, 754]) | 9\n",
-      "24160    | torch.Size([1, 1000, 777]) | 9\n",
-      "24161    | torch.Size([1, 1000, 775]) | 9\n",
-      "24162    | torch.Size([1, 1000, 754]) | 9\n",
-      "24163    | torch.Size([1, 1000, 754]) | 9\n",
-      "24164    | torch.Size([1, 1000, 754]) | 9\n",
-      "24165    | torch.Size([1, 1000, 754]) | 9\n",
-      "24166    | torch.Size([1, 1000, 777]) | 9\n",
-      "24167    | torch.Size([1, 1000, 754]) | 9\n",
-      "24168    | torch.Size([1, 1000, 777]) | 9\n",
-      "24169    | torch.Size([1, 1000, 754]) | 9\n",
-      "24170    | torch.Size([1, 1000, 754]) | 9\n",
-      "24171    | torch.Size([1, 1000, 754]) | 9\n",
-      "24172    | torch.Size([1, 1000, 761]) | 9\n",
-      "24173    | torch.Size([1, 1000, 777]) | 9\n",
-      "24174    | torch.Size([1, 1000, 777]) | 9\n",
-      "24175    | torch.Size([1, 1000, 784]) | 9\n",
-      "24176    | torch.Size([1, 1000, 754]) | 9\n",
-      "24177    | torch.Size([1, 1000, 777]) | 9\n",
-      "24178    | torch.Size([1, 1000, 754]) | 9\n",
-      "24179    | torch.Size([1, 1000, 783]) | 9\n",
-      "24180    | torch.Size([1, 1000, 754]) | 9\n",
-      "24181    | torch.Size([1, 1000, 777]) | 9\n",
-      "24182    | torch.Size([1, 1000, 754]) | 9\n",
-      "24183    | torch.Size([1, 1000, 792]) | 9\n",
-      "24184    | torch.Size([1, 1000, 754]) | 9\n",
-      "24185    | torch.Size([1, 1000, 754]) | 9\n",
-      "24186    | torch.Size([1, 1000, 754]) | 9\n",
-      "24187    | torch.Size([1, 1000, 754]) | 9\n",
-      "24188    | torch.Size([1, 1000, 754]) | 9\n",
-      "24189    | torch.Size([1, 1000, 754]) | 9\n",
-      "24190    | torch.Size([1, 1000, 754]) | 9\n",
-      "24191    | torch.Size([1, 1000, 786]) | 9\n",
-      "24192    | torch.Size([1, 1000, 754]) | 9\n",
-      "24193    | torch.Size([1, 1000, 754]) | 9\n",
-      "24194    | torch.Size([1, 1000, 777]) | 9\n",
-      "24195    | torch.Size([1, 1000, 754]) | 9\n",
-      "24196    | torch.Size([1, 1000, 754]) | 9\n",
-      "24197    | torch.Size([1, 1000, 774]) | 9\n",
-      "24198    | torch.Size([1, 1000, 766]) | 9\n",
-      "24199    | torch.Size([1, 1000, 779]) | 9\n",
-      "24200    | torch.Size([1, 1000, 777]) | 9\n",
-      "24201    | torch.Size([1, 1000, 726]) | 9\n",
-      "24202    | torch.Size([1, 1000, 754]) | 9\n",
-      "24203    | torch.Size([1, 1000, 799]) | 9\n",
-      "24204    | torch.Size([1, 1000, 777]) | 9\n",
-      "24205    | torch.Size([1, 1000, 862]) | 9\n",
-      "24206    | torch.Size([1, 1000, 780]) | 9\n",
-      "24207    | torch.Size([1, 1000, 754]) | 9\n",
-      "24208    | torch.Size([1, 1000, 784]) | 9\n",
-      "24209    | torch.Size([1, 1000, 777]) | 9\n",
-      "24210    | torch.Size([1, 1000, 761]) | 9\n",
-      "24211    | torch.Size([1, 1000, 762]) | 9\n",
-      "24212    | torch.Size([1, 1000, 784]) | 9\n",
-      "24213    | torch.Size([1, 1000, 754]) | 9\n",
-      "24214    | torch.Size([1, 1000, 754]) | 9\n",
-      "24215    | torch.Size([1, 1000, 777]) | 9\n",
-      "24216    | torch.Size([1, 1000, 754]) | 9\n",
-      "24217    | torch.Size([1, 1000, 754]) | 9\n",
-      "24218    | torch.Size([1, 1000, 771]) | 9\n",
-      "24219    | torch.Size([1, 1000, 754]) | 9\n",
-      "24220    | torch.Size([1, 1000, 754]) | 9\n",
-      "24221    | torch.Size([1, 1000, 754]) | 9\n",
-      "24222    | torch.Size([1, 1000, 777]) | 9\n",
-      "24223    | torch.Size([1, 1000, 777]) | 9\n",
-      "24224    | torch.Size([1, 1000, 754]) | 9\n",
-      "24225    | torch.Size([1, 1000, 754]) | 9\n",
-      "24226    | torch.Size([1, 1000, 777]) | 9\n",
-      "24227    | torch.Size([1, 1000, 754]) | 9\n",
-      "24228    | torch.Size([1, 1000, 728]) | 9\n",
-      "24229    | torch.Size([1, 1000, 769]) | 9\n",
-      "24230    | torch.Size([1, 1000, 762]) | 9\n",
-      "24231    | torch.Size([1, 1000, 754]) | 9\n",
-      "24232    | torch.Size([1, 1000, 783]) | 9\n",
-      "24233    | torch.Size([1, 1000, 777]) | 9\n",
-      "24234    | torch.Size([1, 1000, 777]) | 9\n",
-      "24235    | torch.Size([1, 1000, 754]) | 9\n",
-      "24236    | torch.Size([1, 1000, 777]) | 9\n",
-      "24237    | torch.Size([1, 1000, 754]) | 9\n",
-      "24238    | torch.Size([1, 1000, 762]) | 9\n",
-      "24239    | torch.Size([1, 1000, 754]) | 9\n",
-      "24240    | torch.Size([1, 1000, 754]) | 9\n",
-      "24241    | torch.Size([1, 1000, 777]) | 9\n",
-      "24242    | torch.Size([1, 1000, 777]) | 9\n",
-      "24243    | torch.Size([1, 1000, 777]) | 9\n",
-      "24244    | torch.Size([1, 1000, 754]) | 9\n",
-      "24245    | torch.Size([1, 1000, 754]) | 9\n",
-      "24246    | torch.Size([1, 1000, 777]) | 9\n",
-      "24247    | torch.Size([1, 1000, 713]) | 9\n",
-      "24248    | torch.Size([1, 1000, 769]) | 9\n",
-      "24249    | torch.Size([1, 1000, 754]) | 9\n",
-      "24250    | torch.Size([1, 1000, 777]) | 9\n",
-      "24251    | torch.Size([1, 1000, 784]) | 9\n",
-      "24252    | torch.Size([1, 1000, 754]) | 9\n",
-      "24253    | torch.Size([1, 1000, 754]) | 9\n",
-      "24254    | torch.Size([1, 1000, 754]) | 9\n",
-      "24255    | torch.Size([1, 1000, 777]) | 9\n",
-      "24256    | torch.Size([1, 1000, 792]) | 9\n",
-      "24257    | torch.Size([1, 1000, 859]) | 9\n",
-      "24258    | torch.Size([1, 1000, 754]) | 9\n",
-      "24259    | torch.Size([1, 1000, 762]) | 9\n",
-      "24260    | torch.Size([1, 1000, 754]) | 9\n",
-      "24261    | torch.Size([1, 1000, 772]) | 9\n",
-      "24262    | torch.Size([1, 1000, 754]) | 9\n",
-      "24263    | torch.Size([1, 1000, 775]) | 9\n",
-      "24264    | torch.Size([1, 1000, 754]) | 9\n",
-      "24265    | torch.Size([1, 1000, 754]) | 9\n",
-      "24266    | torch.Size([1, 1000, 754]) | 9\n",
-      "24267    | torch.Size([1, 1000, 777]) | 9\n",
-      "24268    | torch.Size([1, 1000, 789]) | 9\n",
-      "24269    | torch.Size([1, 1000, 754]) | 9\n",
-      "24270    | torch.Size([1, 1000, 777]) | 9\n",
-      "24271    | torch.Size([1, 1000, 754]) | 9\n",
-      "24272    | torch.Size([1, 1000, 767]) | 9\n",
-      "24273    | torch.Size([1, 1000, 783]) | 9\n",
-      "24274    | torch.Size([1, 1000, 783]) | 9\n",
-      "24275    | torch.Size([1, 1000, 754]) | 9\n",
-      "24276    | torch.Size([1, 1000, 780]) | 9\n",
-      "24277    | torch.Size([1, 1000, 703]) | 9\n",
-      "24278    | torch.Size([1, 1000, 754]) | 9\n",
-      "24279    | torch.Size([1, 1000, 784]) | 9\n",
-      "24280    | torch.Size([1, 1000, 772]) | 9\n",
-      "24281    | torch.Size([1, 1000, 754]) | 9\n",
-      "24282    | torch.Size([1, 1000, 777]) | 9\n",
-      "24283    | torch.Size([1, 1000, 754]) | 9\n",
-      "24284    | torch.Size([1, 1000, 754]) | 9\n",
-      "24285    | torch.Size([1, 1000, 777]) | 9\n",
-      "24286    | torch.Size([1, 1000, 754]) | 9\n",
-      "24287    | torch.Size([1, 1000, 771]) | 9\n",
-      "24288    | torch.Size([1, 1000, 612]) | 9\n",
-      "24289    | torch.Size([1, 1000, 777]) | 9\n",
-      "24290    | torch.Size([1, 1000, 777]) | 9\n",
-      "24291    | torch.Size([1, 1000, 754]) | 9\n",
-      "24292    | torch.Size([1, 1000, 777]) | 9\n",
-      "24293    | torch.Size([1, 1000, 777]) | 9\n",
-      "24294    | torch.Size([1, 1000, 777]) | 9\n",
-      "24295    | torch.Size([1, 1000, 762]) | 9\n",
-      "24296    | torch.Size([1, 1000, 773]) | 9\n",
-      "24297    | torch.Size([1, 1000, 754]) | 9\n",
-      "24298    | torch.Size([1, 1000, 762]) | 9\n",
-      "24299    | torch.Size([1, 1000, 754]) | 9\n",
-      "24300    | torch.Size([1, 1000, 754]) | 9\n",
-      "24301    | torch.Size([1, 1000, 754]) | 9\n",
-      "24302    | torch.Size([1, 1000, 777]) | 9\n",
-      "24303    | torch.Size([1, 1000, 797]) | 9\n",
-      "24304    | torch.Size([1, 1000, 765]) | 9\n",
-      "24305    | torch.Size([1, 1000, 754]) | 9\n",
-      "24306    | torch.Size([1, 1000, 778]) | 9\n",
-      "24307    | torch.Size([1, 1000, 777]) | 9\n",
-      "24308    | torch.Size([1, 1000, 754]) | 9\n",
-      "24309    | torch.Size([1, 1000, 754]) | 9\n",
-      "24310    | torch.Size([1, 1000, 798]) | 9\n",
-      "24311    | torch.Size([1, 1000, 754]) | 9\n",
-      "24312    | torch.Size([1, 1000, 754]) | 9\n",
-      "24313    | torch.Size([1, 1000, 777]) | 9\n",
-      "24314    | torch.Size([1, 1000, 793]) | 9\n",
-      "24315    | torch.Size([1, 1000, 777]) | 9\n",
-      "24316    | torch.Size([1, 1000, 754]) | 9\n",
-      "24317    | torch.Size([1, 1000, 778]) | 9\n",
-      "24318    | torch.Size([1, 1000, 754]) | 9\n",
-      "24319    | torch.Size([1, 1000, 779]) | 9\n",
-      "24320    | torch.Size([1, 1000, 777]) | 9\n",
-      "24321    | torch.Size([1, 1000, 754]) | 9\n",
-      "24322    | torch.Size([1, 1000, 754]) | 9\n",
-      "24323    | torch.Size([1, 1000, 773]) | 9\n",
-      "24324    | torch.Size([1, 1000, 754]) | 9\n",
-      "24325    | torch.Size([1, 1000, 754]) | 9\n",
-      "24326    | torch.Size([1, 1000, 781]) | 9\n",
-      "24327    | torch.Size([1, 1000, 767]) | 9\n",
-      "24328    | torch.Size([1, 1000, 754]) | 9\n",
-      "24329    | torch.Size([1, 1000, 754]) | 9\n",
-      "24330    | torch.Size([1, 1000, 775]) | 9\n",
-      "24331    | torch.Size([1, 1000, 777]) | 9\n",
-      "24332    | torch.Size([1, 1000, 777]) | 9\n",
-      "24333    | torch.Size([1, 1000, 754]) | 9\n",
-      "24334    | torch.Size([1, 1000, 777]) | 9\n",
-      "24335    | torch.Size([1, 1000, 754]) | 9\n",
-      "24336    | torch.Size([1, 1000, 754]) | 9\n",
-      "24337    | torch.Size([1, 1000, 732]) | 9\n",
-      "24338    | torch.Size([1, 1000, 786]) | 9\n",
-      "24339    | torch.Size([1, 1000, 754]) | 9\n",
-      "24340    | torch.Size([1, 1000, 723]) | 9\n",
-      "24341    | torch.Size([1, 1000, 754]) | 9\n",
-      "24342    | torch.Size([1, 1000, 777]) | 9\n",
-      "24343    | torch.Size([1, 1000, 754]) | 9\n",
-      "24344    | torch.Size([1, 1000, 771]) | 9\n",
-      "24345    | torch.Size([1, 1000, 782]) | 9\n",
-      "24346    | torch.Size([1, 1000, 777]) | 9\n",
-      "24347    | torch.Size([1, 1000, 754]) | 9\n",
-      "24348    | torch.Size([1, 1000, 777]) | 9\n",
-      "24349    | torch.Size([1, 1000, 777]) | 9\n",
-      "24350    | torch.Size([1, 1000, 754]) | 9\n",
-      "24351    | torch.Size([1, 1000, 754]) | 9\n",
-      "24352    | torch.Size([1, 1000, 783]) | 9\n",
-      "24353    | torch.Size([1, 1000, 780]) | 9\n",
-      "24354    | torch.Size([1, 1000, 768]) | 9\n",
-      "24355    | torch.Size([1, 1000, 754]) | 9\n",
-      "24356    | torch.Size([1, 1000, 754]) | 9\n",
-      "24357    | torch.Size([1, 1000, 754]) | 9\n",
-      "24358    | torch.Size([1, 1000, 777]) | 9\n",
-      "24359    | torch.Size([1, 1000, 771]) | 9\n",
-      "24360    | torch.Size([1, 1000, 777]) | 9\n",
-      "24361    | torch.Size([1, 1000, 777]) | 9\n",
-      "24362    | torch.Size([1, 1000, 754]) | 9\n",
-      "24363    | torch.Size([1, 1000, 754]) | 9\n",
-      "24364    | torch.Size([1, 1000, 754]) | 9\n",
-      "24365    | torch.Size([1, 1000, 779]) | 9\n",
-      "24366    | torch.Size([1, 1000, 754]) | 9\n",
-      "24367    | torch.Size([1, 1000, 762]) | 9\n",
-      "24368    | torch.Size([1, 1000, 777]) | 9\n",
-      "24369    | torch.Size([1, 1000, 777]) | 9\n",
-      "24370    | torch.Size([1, 1000, 772]) | 9\n",
-      "24371    | torch.Size([1, 1000, 777]) | 9\n",
-      "24372    | torch.Size([1, 1000, 777]) | 9\n",
-      "24373    | torch.Size([1, 1000, 754]) | 9\n",
-      "24374    | torch.Size([1, 1000, 777]) | 9\n",
-      "24375    | torch.Size([1, 1000, 754]) | 9\n",
-      "24376    | torch.Size([1, 1000, 754]) | 9\n",
-      "24377    | torch.Size([1, 1000, 754]) | 9\n",
-      "24378    | torch.Size([1, 1000, 782]) | 9\n",
-      "24379    | torch.Size([1, 1000, 754]) | 9\n",
-      "24380    | torch.Size([1, 1000, 754]) | 9\n",
-      "24381    | torch.Size([1, 1000, 754]) | 9\n",
-      "24382    | torch.Size([1, 1000, 754]) | 9\n",
-      "24383    | torch.Size([1, 1000, 754]) | 9\n",
-      "24384    | torch.Size([1, 1000, 777]) | 9\n",
-      "24385    | torch.Size([1, 1000, 754]) | 9\n",
-      "24386    | torch.Size([1, 1000, 754]) | 9\n",
-      "24387    | torch.Size([1, 1000, 776]) | 9\n",
-      "24388    | torch.Size([1, 1000, 777]) | 9\n",
-      "24389    | torch.Size([1, 1000, 754]) | 9\n",
-      "24390    | torch.Size([1, 1000, 777]) | 9\n",
-      "24391    | torch.Size([1, 1000, 754]) | 9\n",
-      "24392    | torch.Size([1, 1000, 762]) | 9\n",
-      "24393    | torch.Size([1, 1000, 777]) | 9\n",
-      "24394    | torch.Size([1, 1000, 754]) | 9\n",
-      "24395    | torch.Size([1, 1000, 754]) | 9\n",
-      "24396    | torch.Size([1, 1000, 754]) | 9\n",
-      "24397    | torch.Size([1, 1000, 777]) | 9\n",
-      "24398    | torch.Size([1, 1000, 730]) | 9\n",
-      "24399    | torch.Size([1, 1000, 776]) | 9\n",
-      "24400    | torch.Size([1, 1000, 754]) | 9\n",
-      "24401    | torch.Size([1, 1000, 781]) | 9\n",
-      "24402    | torch.Size([1, 1000, 762]) | 9\n",
-      "24403    | torch.Size([1, 1000, 754]) | 9\n",
-      "24404    | torch.Size([1, 1000, 777]) | 9\n",
-      "24405    | torch.Size([1, 1000, 777]) | 9\n",
-      "24406    | torch.Size([1, 1000, 754]) | 9\n",
-      "24407    | torch.Size([1, 1000, 754]) | 9\n",
-      "24408    | torch.Size([1, 1000, 777]) | 9\n",
-      "24409    | torch.Size([1, 1000, 762]) | 9\n",
-      "24410    | torch.Size([1, 1000, 754]) | 9\n",
-      "24411    | torch.Size([1, 1000, 754]) | 9\n",
-      "24412    | torch.Size([1, 1000, 779]) | 9\n",
-      "24413    | torch.Size([1, 1000, 802]) | 9\n",
-      "24414    | torch.Size([1, 1000, 777]) | 9\n",
-      "24415    | torch.Size([1, 1000, 754]) | 9\n",
-      "24416    | torch.Size([1, 1000, 754]) | 9\n",
-      "24417    | torch.Size([1, 1000, 754]) | 9\n",
-      "24418    | torch.Size([1, 1000, 762]) | 9\n",
-      "24419    | torch.Size([1, 1000, 780]) | 9\n",
-      "24420    | torch.Size([1, 1000, 777]) | 9\n",
-      "24421    | torch.Size([1, 1000, 754]) | 9\n",
-      "24422    | torch.Size([1, 1000, 754]) | 9\n",
-      "24423    | torch.Size([1, 1000, 777]) | 9\n",
-      "24424    | torch.Size([1, 1000, 754]) | 9\n",
-      "24425    | torch.Size([1, 1000, 754]) | 9\n",
-      "24426    | torch.Size([1, 1000, 754]) | 9\n",
-      "24427    | torch.Size([1, 1000, 784]) | 9\n",
-      "24428    | torch.Size([1, 1000, 754]) | 9\n",
-      "24429    | torch.Size([1, 1000, 777]) | 9\n",
-      "24430    | torch.Size([1, 1000, 777]) | 9\n",
-      "24431    | torch.Size([1, 1000, 754]) | 9\n",
-      "24432    | torch.Size([1, 1000, 778]) | 9\n",
-      "24433    | torch.Size([1, 1000, 754]) | 9\n",
-      "24434    | torch.Size([1, 1000, 777]) | 9\n",
-      "24435    | torch.Size([1, 1000, 754]) | 9\n",
-      "24436    | torch.Size([1, 1000, 754]) | 9\n",
-      "24437    | torch.Size([1, 1000, 777]) | 9\n",
-      "24438    | torch.Size([1, 1000, 754]) | 9\n",
-      "24439    | torch.Size([1, 1000, 754]) | 9\n",
-      "24440    | torch.Size([1, 1000, 777]) | 9\n",
-      "24441    | torch.Size([1, 1000, 754]) | 9\n",
-      "24442    | torch.Size([1, 1000, 754]) | 9\n",
-      "24443    | torch.Size([1, 1000, 778]) | 9\n",
-      "24444    | torch.Size([1, 1000, 777]) | 9\n",
-      "24445    | torch.Size([1, 1000, 777]) | 9\n",
-      "24446    | torch.Size([1, 1000, 777]) | 9\n",
-      "24447    | torch.Size([1, 1000, 777]) | 9\n",
-      "24448    | torch.Size([1, 1000, 777]) | 9\n",
-      "24449    | torch.Size([1, 1000, 777]) | 9\n",
-      "24450    | torch.Size([1, 1000, 772]) | 9\n",
-      "24451    | torch.Size([1, 1000, 777]) | 9\n",
-      "24452    | torch.Size([1, 1000, 754]) | 9\n",
-      "24453    | torch.Size([1, 1000, 754]) | 9\n",
-      "24454    | torch.Size([1, 1000, 784]) | 9\n",
-      "24455    | torch.Size([1, 1000, 754]) | 9\n",
-      "24456    | torch.Size([1, 1000, 754]) | 9\n",
-      "24457    | torch.Size([1, 1000, 754]) | 9\n",
-      "24458    | torch.Size([1, 1000, 754]) | 9\n",
-      "24459    | torch.Size([1, 1000, 723]) | 9\n",
-      "24460    | torch.Size([1, 1000, 776]) | 9\n",
-      "24461    | torch.Size([1, 1000, 777]) | 9\n",
-      "24462    | torch.Size([1, 1000, 754]) | 9\n",
-      "24463    | torch.Size([1, 1000, 797]) | 9\n",
-      "24464    | torch.Size([1, 1000, 762]) | 9\n",
-      "24465    | torch.Size([1, 1000, 777]) | 9\n",
-      "24466    | torch.Size([1, 1000, 754]) | 9\n",
-      "24467    | torch.Size([1, 1000, 781]) | 9\n",
-      "24468    | torch.Size([1, 1000, 802]) | 9\n",
-      "24469    | torch.Size([1, 1000, 754]) | 9\n",
-      "24470    | torch.Size([1, 1000, 779]) | 9\n",
-      "24471    | torch.Size([1, 1000, 754]) | 9\n",
-      "24472    | torch.Size([1, 1000, 772]) | 9\n",
-      "24473    | torch.Size([1, 1000, 754]) | 9\n",
-      "24474    | torch.Size([1, 1000, 777]) | 9\n",
-      "24475    | torch.Size([1, 1000, 754]) | 9\n",
-      "24476    | torch.Size([1, 1000, 777]) | 9\n",
-      "24477    | torch.Size([1, 1000, 777]) | 9\n",
-      "24478    | torch.Size([1, 1000, 754]) | 9\n",
-      "24479    | torch.Size([1, 1000, 754]) | 9\n",
-      "24480    | torch.Size([1, 1000, 754]) | 9\n",
-      "24481    | torch.Size([1, 1000, 762]) | 9\n",
-      "24482    | torch.Size([1, 1000, 754]) | 9\n",
-      "24483    | torch.Size([1, 1000, 777]) | 9\n",
-      "24484    | torch.Size([1, 1000, 782]) | 9\n",
-      "24485    | torch.Size([1, 1000, 777]) | 9\n",
-      "24486    | torch.Size([1, 1000, 754]) | 9\n",
-      "24487    | torch.Size([1, 1000, 784]) | 9\n",
-      "24488    | torch.Size([1, 1000, 754]) | 9\n",
-      "24489    | torch.Size([1, 1000, 777]) | 9\n",
-      "24490    | torch.Size([1, 1000, 754]) | 9\n",
-      "24491    | torch.Size([1, 1000, 754]) | 9\n",
-      "24492    | torch.Size([1, 1000, 777]) | 9\n",
-      "24493    | torch.Size([1, 1000, 754]) | 9\n",
-      "24494    | torch.Size([1, 1000, 802]) | 9\n",
-      "24495    | torch.Size([1, 1000, 754]) | 9\n",
-      "24496    | torch.Size([1, 1000, 754]) | 9\n",
-      "24497    | torch.Size([1, 1000, 774]) | 9\n",
-      "24498    | torch.Size([1, 1000, 611]) | 9\n",
-      "24499    | torch.Size([1, 1000, 764]) | 9\n",
-      "24500    | torch.Size([1, 1000, 777]) | 9\n",
-      "24501    | torch.Size([1, 1000, 777]) | 9\n",
-      "24502    | torch.Size([1, 1000, 786]) | 9\n",
-      "24503    | torch.Size([1, 1000, 754]) | 9\n",
-      "24504    | torch.Size([1, 1000, 754]) | 9\n",
-      "24505    | torch.Size([1, 1000, 754]) | 9\n",
-      "24506    | torch.Size([1, 1000, 754]) | 9\n",
-      "24507    | torch.Size([1, 1000, 754]) | 9\n",
-      "24508    | torch.Size([1, 1000, 754]) | 9\n",
-      "24509    | torch.Size([1, 1000, 762]) | 9\n",
-      "24510    | torch.Size([1, 1000, 754]) | 9\n",
-      "24511    | torch.Size([1, 1000, 754]) | 9\n",
-      "24512    | torch.Size([1, 1000, 771]) | 9\n",
-      "24513    | torch.Size([1, 1000, 754]) | 9\n",
-      "24514    | torch.Size([1, 1000, 777]) | 9\n",
-      "24515    | torch.Size([1, 1000, 802]) | 9\n",
-      "24516    | torch.Size([1, 1000, 754]) | 9\n",
-      "24517    | torch.Size([1, 1000, 787]) | 9\n",
-      "24518    | torch.Size([1, 1000, 772]) | 9\n",
-      "24519    | torch.Size([1, 1000, 754]) | 9\n",
-      "24520    | torch.Size([1, 1000, 778]) | 9\n",
-      "24521    | torch.Size([1, 1000, 754]) | 9\n",
-      "24522    | torch.Size([1, 1000, 754]) | 9\n",
-      "24523    | torch.Size([1, 1000, 754]) | 9\n",
-      "24524    | torch.Size([1, 1000, 754]) | 9\n",
-      "24525    | torch.Size([1, 1000, 754]) | 9\n",
-      "24526    | torch.Size([1, 1000, 777]) | 9\n",
-      "24527    | torch.Size([1, 1000, 754]) | 9\n",
-      "24528    | torch.Size([1, 1000, 616]) | 9\n",
-      "24529    | torch.Size([1, 1000, 780]) | 9\n",
-      "24530    | torch.Size([1, 1000, 754]) | 9\n",
-      "24531    | torch.Size([1, 1000, 754]) | 9\n",
-      "24532    | torch.Size([1, 1000, 785]) | 9\n",
-      "24533    | torch.Size([1, 1000, 777]) | 9\n",
-      "24534    | torch.Size([1, 1000, 754]) | 9\n",
-      "24535    | torch.Size([1, 1000, 777]) | 9\n",
-      "24536    | torch.Size([1, 1000, 754]) | 9\n",
-      "24537    | torch.Size([1, 1000, 754]) | 9\n",
-      "24538    | torch.Size([1, 1000, 754]) | 9\n",
-      "24539    | torch.Size([1, 1000, 777]) | 9\n",
-      "24540    | torch.Size([1, 1000, 778]) | 9\n",
-      "24541    | torch.Size([1, 1000, 813]) | 9\n",
-      "24542    | torch.Size([1, 1000, 754]) | 9\n",
-      "24543    | torch.Size([1, 1000, 754]) | 9\n",
-      "24544    | torch.Size([1, 1000, 777]) | 9\n",
-      "24545    | torch.Size([1, 1000, 754]) | 9\n",
-      "24546    | torch.Size([1, 1000, 754]) | 9\n",
-      "24547    | torch.Size([1, 1000, 754]) | 9\n",
-      "24548    | torch.Size([1, 1000, 762]) | 9\n",
-      "24549    | torch.Size([1, 1000, 780]) | 9\n",
-      "24550    | torch.Size([1, 1000, 777]) | 9\n",
-      "24551    | torch.Size([1, 1000, 754]) | 9\n",
-      "24552    | torch.Size([1, 1000, 773]) | 9\n",
-      "24553    | torch.Size([1, 1000, 769]) | 9\n",
-      "24554    | torch.Size([1, 1000, 777]) | 9\n",
-      "24555    | torch.Size([1, 1000, 754]) | 9\n",
-      "24556    | torch.Size([1, 1000, 780]) | 9\n",
-      "24557    | torch.Size([1, 1000, 762]) | 9\n",
-      "24558    | torch.Size([1, 1000, 754]) | 9\n",
-      "24559    | torch.Size([1, 1000, 754]) | 9\n",
-      "24560    | torch.Size([1, 1000, 754]) | 9\n",
-      "24561    | torch.Size([1, 1000, 754]) | 9\n",
-      "24562    | torch.Size([1, 1000, 777]) | 9\n",
-      "24563    | torch.Size([1, 1000, 777]) | 9\n",
-      "24564    | torch.Size([1, 1000, 777]) | 9\n",
-      "24565    | torch.Size([1, 1000, 777]) | 9\n",
-      "24566    | torch.Size([1, 1000, 754]) | 9\n",
-      "24567    | torch.Size([1, 1000, 754]) | 9\n",
-      "24568    | torch.Size([1, 1000, 777]) | 9\n",
-      "24569    | torch.Size([1, 1000, 754]) | 9\n",
-      "24570    | torch.Size([1, 1000, 777]) | 9\n",
-      "24571    | torch.Size([1, 1000, 774]) | 9\n",
-      "24572    | torch.Size([1, 1000, 754]) | 9\n",
-      "24573    | torch.Size([1, 1000, 772]) | 9\n",
-      "24574    | torch.Size([1, 1000, 802]) | 9\n",
-      "24575    | torch.Size([1, 1000, 864]) | 9\n",
-      "24576    | torch.Size([1, 1000, 777]) | 9\n",
-      "24577    | torch.Size([1, 1000, 777]) | 9\n",
-      "24578    | torch.Size([1, 1000, 777]) | 9\n",
-      "24579    | torch.Size([1, 1000, 777]) | 9\n",
-      "24580    | torch.Size([1, 1000, 754]) | 9\n",
-      "24581    | torch.Size([1, 1000, 777]) | 9\n",
-      "24582    | torch.Size([1, 1000, 754]) | 9\n",
-      "24583    | torch.Size([1, 1000, 777]) | 9\n",
-      "24584    | torch.Size([1, 1000, 754]) | 9\n",
-      "24585    | torch.Size([1, 1000, 777]) | 9\n",
-      "24586    | torch.Size([1, 1000, 754]) | 9\n",
-      "24587    | torch.Size([1, 1000, 754]) | 9\n",
-      "24588    | torch.Size([1, 1000, 754]) | 9\n",
-      "24589    | torch.Size([1, 1000, 777]) | 9\n",
-      "24590    | torch.Size([1, 1000, 754]) | 9\n",
-      "24591    | torch.Size([1, 1000, 754]) | 9\n",
-      "24592    | torch.Size([1, 1000, 754]) | 9\n",
-      "24593    | torch.Size([1, 1000, 777]) | 9\n",
-      "24594    | torch.Size([1, 1000, 754]) | 9\n",
-      "24595    | torch.Size([1, 1000, 754]) | 9\n",
-      "24596    | torch.Size([1, 1000, 754]) | 9\n",
-      "24597    | torch.Size([1, 1000, 754]) | 9\n",
-      "24598    | torch.Size([1, 1000, 754]) | 9\n",
-      "24599    | torch.Size([1, 1000, 754]) | 9\n",
-      "24600    | torch.Size([1, 1000, 754]) | 9\n",
-      "24601    | torch.Size([1, 1000, 777]) | 9\n",
-      "24602    | torch.Size([1, 1000, 754]) | 9\n",
-      "24603    | torch.Size([1, 1000, 777]) | 9\n",
-      "24604    | torch.Size([1, 1000, 762]) | 9\n",
-      "24605    | torch.Size([1, 1000, 777]) | 9\n",
-      "24606    | torch.Size([1, 1000, 754]) | 9\n",
-      "24607    | torch.Size([1, 1000, 777]) | 9\n",
-      "24608    | torch.Size([1, 1000, 754]) | 9\n",
-      "24609    | torch.Size([1, 1000, 777]) | 9\n",
-      "24610    | torch.Size([1, 1000, 754]) | 9\n",
-      "24611    | torch.Size([1, 1000, 754]) | 9\n",
-      "24612    | torch.Size([1, 1000, 772]) | 9\n",
-      "24613    | torch.Size([1, 1000, 777]) | 9\n",
-      "24614    | torch.Size([1, 1000, 777]) | 9\n",
-      "24615    | torch.Size([1, 1000, 794]) | 9\n",
-      "24616    | torch.Size([1, 1000, 754]) | 9\n",
-      "24617    | torch.Size([1, 1000, 777]) | 9\n",
-      "24618    | torch.Size([1, 1000, 754]) | 9\n",
-      "24619    | torch.Size([1, 1000, 776]) | 9\n",
-      "24620    | torch.Size([1, 1000, 784]) | 9\n",
-      "24621    | torch.Size([1, 1000, 754]) | 9\n",
-      "24622    | torch.Size([1, 1000, 777]) | 9\n",
-      "24623    | torch.Size([1, 1000, 802]) | 9\n",
-      "24624    | torch.Size([1, 1000, 777]) | 9\n",
-      "24625    | torch.Size([1, 1000, 777]) | 9\n",
-      "24626    | torch.Size([1, 1000, 754]) | 9\n",
-      "24627    | torch.Size([1, 1000, 789]) | 9\n",
-      "24628    | torch.Size([1, 1000, 754]) | 9\n",
-      "24629    | torch.Size([1, 1000, 754]) | 9\n",
-      "24630    | torch.Size([1, 1000, 754]) | 9\n",
-      "24631    | torch.Size([1, 1000, 754]) | 9\n",
-      "24632    | torch.Size([1, 1000, 754]) | 9\n",
-      "24633    | torch.Size([1, 1000, 754]) | 9\n",
-      "24634    | torch.Size([1, 1000, 754]) | 9\n",
-      "24635    | torch.Size([1, 1000, 754]) | 9\n",
-      "24636    | torch.Size([1, 1000, 754]) | 9\n",
-      "24637    | torch.Size([1, 1000, 754]) | 9\n",
-      "24638    | torch.Size([1, 1000, 769]) | 9\n",
-      "24639    | torch.Size([1, 1000, 772]) | 9\n",
-      "24640    | torch.Size([1, 1000, 754]) | 9\n",
-      "24641    | torch.Size([1, 1000, 754]) | 9\n",
-      "24642    | torch.Size([1, 1000, 754]) | 9\n",
-      "24643    | torch.Size([1, 1000, 777]) | 9\n",
-      "24644    | torch.Size([1, 1000, 759]) | 9\n",
-      "24645    | torch.Size([1, 1000, 775]) | 9\n",
-      "24646    | torch.Size([1, 1000, 754]) | 9\n",
-      "24647    | torch.Size([1, 1000, 777]) | 9\n",
-      "24648    | torch.Size([1, 1000, 780]) | 9\n",
-      "24649    | torch.Size([1, 1000, 777]) | 9\n",
-      "24650    | torch.Size([1, 1000, 777]) | 9\n",
-      "24651    | torch.Size([1, 1000, 802]) | 9\n",
-      "24652    | torch.Size([1, 1000, 754]) | 9\n",
-      "24653    | torch.Size([1, 1000, 777]) | 9\n",
-      "24654    | torch.Size([1, 1000, 754]) | 9\n",
-      "24655    | torch.Size([1, 1000, 762]) | 9\n",
-      "24656    | torch.Size([1, 1000, 771]) | 9\n",
-      "24657    | torch.Size([1, 1000, 783]) | 9\n",
-      "24658    | torch.Size([1, 1000, 754]) | 9\n",
-      "24659    | torch.Size([1, 1000, 777]) | 9\n",
-      "24660    | torch.Size([1, 1000, 767]) | 9\n",
-      "24661    | torch.Size([1, 1000, 780]) | 9\n",
-      "24662    | torch.Size([1, 1000, 777]) | 9\n",
-      "24663    | torch.Size([1, 1000, 802]) | 9\n",
-      "24664    | torch.Size([1, 1000, 754]) | 9\n",
-      "24665    | torch.Size([1, 1000, 777]) | 9\n",
-      "24666    | torch.Size([1, 1000, 754]) | 9\n",
-      "24667    | torch.Size([1, 1000, 611]) | 9\n",
-      "24668    | torch.Size([1, 1000, 763]) | 9\n",
-      "24669    | torch.Size([1, 1000, 754]) | 9\n",
-      "24670    | torch.Size([1, 1000, 777]) | 9\n",
-      "24671    | torch.Size([1, 1000, 774]) | 9\n",
-      "24672    | torch.Size([1, 1000, 777]) | 9\n",
-      "24673    | torch.Size([1, 1000, 787]) | 9\n",
-      "24674    | torch.Size([1, 1000, 754]) | 9\n",
-      "24675    | torch.Size([1, 1000, 783]) | 9\n",
-      "24676    | torch.Size([1, 1000, 842]) | 9\n",
-      "24677    | torch.Size([1, 1000, 777]) | 9\n",
-      "24678    | torch.Size([1, 1000, 754]) | 9\n",
-      "24679    | torch.Size([1, 1000, 784]) | 9\n",
-      "24680    | torch.Size([1, 1000, 754]) | 9\n",
-      "24681    | torch.Size([1, 1000, 796]) | 9\n",
-      "24682    | torch.Size([1, 1000, 754]) | 9\n",
-      "24683    | torch.Size([1, 1000, 754]) | 9\n",
-      "24684    | torch.Size([1, 1000, 754]) | 9\n",
-      "24685    | torch.Size([1, 1000, 754]) | 9\n",
-      "24686    | torch.Size([1, 1000, 777]) | 9\n",
-      "24687    | torch.Size([1, 1000, 764]) | 9\n",
-      "24688    | torch.Size([1, 1000, 777]) | 9\n",
-      "24689    | torch.Size([1, 1000, 808]) | 9\n",
-      "24690    | torch.Size([1, 1000, 777]) | 9\n",
-      "24691    | torch.Size([1, 1000, 754]) | 9\n",
-      "24692    | torch.Size([1, 1000, 777]) | 9\n",
-      "24693    | torch.Size([1, 1000, 754]) | 9\n",
-      "24694    | torch.Size([1, 1000, 778]) | 9\n",
-      "24695    | torch.Size([1, 1000, 754]) | 9\n",
-      "24696    | torch.Size([1, 1000, 777]) | 9\n",
-      "24697    | torch.Size([1, 1000, 754]) | 9\n",
-      "24698    | torch.Size([1, 1000, 754]) | 9\n",
-      "24699    | torch.Size([1, 1000, 777]) | 9\n",
-      "24700    | torch.Size([1, 1000, 754]) | 9\n",
-      "24701    | torch.Size([1, 1000, 754]) | 9\n",
-      "24702    | torch.Size([1, 1000, 777]) | 9\n",
-      "24703    | torch.Size([1, 1000, 777]) | 9\n",
-      "24704    | torch.Size([1, 1000, 777]) | 9\n",
-      "24705    | torch.Size([1, 1000, 777]) | 9\n",
-      "24706    | torch.Size([1, 1000, 754]) | 9\n",
-      "24707    | torch.Size([1, 1000, 754]) | 9\n",
-      "24708    | torch.Size([1, 1000, 773]) | 9\n",
-      "24709    | torch.Size([1, 1000, 777]) | 9\n",
-      "24710    | torch.Size([1, 1000, 754]) | 9\n",
-      "24711    | torch.Size([1, 1000, 754]) | 9\n",
-      "24712    | torch.Size([1, 1000, 754]) | 9\n",
-      "24713    | torch.Size([1, 1000, 754]) | 9\n",
-      "24714    | torch.Size([1, 1000, 777]) | 9\n",
-      "24715    | torch.Size([1, 1000, 777]) | 9\n",
-      "24716    | torch.Size([1, 1000, 754]) | 9\n",
-      "24717    | torch.Size([1, 1000, 780]) | 9\n",
-      "24718    | torch.Size([1, 1000, 777]) | 9\n",
-      "24719    | torch.Size([1, 1000, 754]) | 9\n",
-      "24720    | torch.Size([1, 1000, 754]) | 9\n",
-      "24721    | torch.Size([1, 1000, 754]) | 9\n",
-      "24722    | torch.Size([1, 1000, 777]) | 9\n",
-      "24723    | torch.Size([1, 1000, 777]) | 9\n",
-      "24724    | torch.Size([1, 1000, 762]) | 9\n",
-      "24725    | torch.Size([1, 1000, 777]) | 9\n",
-      "24726    | torch.Size([1, 1000, 772]) | 9\n",
-      "24727    | torch.Size([1, 1000, 754]) | 9\n",
-      "24728    | torch.Size([1, 1000, 754]) | 9\n",
-      "24729    | torch.Size([1, 1000, 782]) | 9\n",
-      "24730    | torch.Size([1, 1000, 754]) | 9\n",
-      "24731    | torch.Size([1, 1000, 754]) | 9\n",
-      "24732    | torch.Size([1, 1000, 754]) | 9\n",
-      "24733    | torch.Size([1, 1000, 754]) | 9\n",
-      "24734    | torch.Size([1, 1000, 754]) | 9\n",
-      "24735    | torch.Size([1, 1000, 782]) | 9\n",
-      "24736    | torch.Size([1, 1000, 772]) | 9\n",
-      "24737    | torch.Size([1, 1000, 777]) | 9\n",
-      "24738    | torch.Size([1, 1000, 808]) | 9\n",
-      "24739    | torch.Size([1, 1000, 754]) | 9\n",
-      "24740    | torch.Size([1, 1000, 729]) | 9\n",
-      "24741    | torch.Size([1, 1000, 777]) | 9\n",
-      "24742    | torch.Size([1, 1000, 782]) | 9\n",
-      "24743    | torch.Size([1, 1000, 754]) | 9\n",
-      "24744    | torch.Size([1, 1000, 781]) | 9\n",
-      "24745    | torch.Size([1, 1000, 754]) | 9\n",
-      "24746    | torch.Size([1, 1000, 754]) | 9\n",
-      "24747    | torch.Size([1, 1000, 784]) | 9\n",
-      "24748    | torch.Size([1, 1000, 754]) | 9\n",
-      "24749    | torch.Size([1, 1000, 777]) | 9\n",
-      "24750    | torch.Size([1, 1000, 802]) | 9\n",
-      "24751    | torch.Size([1, 1000, 777]) | 9\n",
-      "24752    | torch.Size([1, 1000, 754]) | 9\n",
-      "24753    | torch.Size([1, 1000, 777]) | 9\n",
-      "24754    | torch.Size([1, 1000, 754]) | 9\n",
-      "24755    | torch.Size([1, 1000, 754]) | 9\n",
-      "24756    | torch.Size([1, 1000, 777]) | 9\n",
-      "24757    | torch.Size([1, 1000, 754]) | 9\n",
-      "24758    | torch.Size([1, 1000, 754]) | 9\n",
-      "24759    | torch.Size([1, 1000, 754]) | 9\n",
-      "24760    | torch.Size([1, 1000, 754]) | 9\n",
-      "24761    | torch.Size([1, 1000, 777]) | 9\n",
-      "24762    | torch.Size([1, 1000, 839]) | 9\n",
-      "24763    | torch.Size([1, 1000, 777]) | 9\n",
-      "24764    | torch.Size([1, 1000, 726]) | 9\n",
-      "24765    | torch.Size([1, 1000, 785]) | 9\n",
-      "24766    | torch.Size([1, 1000, 782]) | 9\n",
-      "24767    | torch.Size([1, 1000, 777]) | 9\n",
-      "24768    | torch.Size([1, 1000, 777]) | 9\n",
-      "24769    | torch.Size([1, 1000, 777]) | 9\n",
-      "24770    | torch.Size([1, 1000, 777]) | 9\n",
-      "24771    | torch.Size([1, 1000, 754]) | 9\n",
-      "24772    | torch.Size([1, 1000, 802]) | 9\n",
-      "24773    | torch.Size([1, 1000, 772]) | 9\n",
-      "24774    | torch.Size([1, 1000, 754]) | 9\n",
-      "24775    | torch.Size([1, 1000, 762]) | 9\n",
-      "24776    | torch.Size([1, 1000, 729]) | 9\n",
-      "24777    | torch.Size([1, 1000, 777]) | 9\n",
-      "24778    | torch.Size([1, 1000, 754]) | 9\n",
-      "24779    | torch.Size([1, 1000, 777]) | 9\n",
-      "24780    | torch.Size([1, 1000, 777]) | 9\n",
-      "24781    | torch.Size([1, 1000, 777]) | 9\n",
-      "24782    | torch.Size([1, 1000, 754]) | 9\n",
-      "24783    | torch.Size([1, 1000, 754]) | 9\n",
-      "24784    | torch.Size([1, 1000, 775]) | 9\n",
-      "24785    | torch.Size([1, 1000, 777]) | 9\n",
-      "24786    | torch.Size([1, 1000, 754]) | 9\n",
-      "24787    | torch.Size([1, 1000, 777]) | 9\n",
-      "24788    | torch.Size([1, 1000, 754]) | 9\n",
-      "24789    | torch.Size([1, 1000, 754]) | 9\n",
-      "24790    | torch.Size([1, 1000, 777]) | 9\n",
-      "24791    | torch.Size([1, 1000, 768]) | 9\n",
-      "24792    | torch.Size([1, 1000, 754]) | 9\n",
-      "24793    | torch.Size([1, 1000, 779]) | 9\n",
-      "24794    | torch.Size([1, 1000, 754]) | 9\n",
-      "24795    | torch.Size([1, 1000, 777]) | 9\n",
-      "24796    | torch.Size([1, 1000, 765]) | 9\n",
-      "24797    | torch.Size([1, 1000, 762]) | 9\n",
-      "24798    | torch.Size([1, 1000, 754]) | 9\n",
-      "24799    | torch.Size([1, 1000, 777]) | 9\n",
-      "24800    | torch.Size([1, 1000, 754]) | 9\n",
-      "24801    | torch.Size([1, 1000, 777]) | 9\n",
-      "24802    | torch.Size([1, 1000, 754]) | 9\n",
-      "24803    | torch.Size([1, 1000, 777]) | 9\n",
-      "24804    | torch.Size([1, 1000, 754]) | 9\n",
-      "24805    | torch.Size([1, 1000, 779]) | 9\n",
-      "24806    | torch.Size([1, 1000, 802]) | 9\n",
-      "24807    | torch.Size([1, 1000, 781]) | 9\n",
-      "24808    | torch.Size([1, 1000, 754]) | 9\n",
-      "24809    | torch.Size([1, 1000, 754]) | 9\n",
-      "24810    | torch.Size([1, 1000, 754]) | 9\n",
-      "24811    | torch.Size([1, 1000, 754]) | 9\n",
-      "24812    | torch.Size([1, 1000, 777]) | 9\n",
-      "24813    | torch.Size([1, 1000, 777]) | 9\n",
-      "24814    | torch.Size([1, 1000, 754]) | 9\n",
-      "24815    | torch.Size([1, 1000, 797]) | 9\n",
-      "24816    | torch.Size([1, 1000, 754]) | 9\n",
-      "24817    | torch.Size([1, 1000, 713]) | 9\n",
-      "24818    | torch.Size([1, 1000, 754]) | 9\n",
-      "24819    | torch.Size([1, 1000, 754]) | 9\n",
-      "24820    | torch.Size([1, 1000, 778]) | 9\n",
-      "24821    | torch.Size([1, 1000, 777]) | 9\n",
-      "24822    | torch.Size([1, 1000, 762]) | 9\n",
-      "24823    | torch.Size([1, 1000, 754]) | 9\n",
-      "24824    | torch.Size([1, 1000, 778]) | 9\n",
-      "24825    | torch.Size([1, 1000, 660]) | 9\n",
-      "24826    | torch.Size([1, 1000, 808]) | 9\n",
-      "24827    | torch.Size([1, 1000, 777]) | 9\n",
-      "24828    | torch.Size([1, 1000, 777]) | 9\n",
-      "24829    | torch.Size([1, 1000, 762]) | 9\n",
-      "24830    | torch.Size([1, 1000, 766]) | 9\n",
-      "24831    | torch.Size([1, 1000, 754]) | 9\n",
-      "24832    | torch.Size([1, 1000, 754]) | 9\n",
-      "24833    | torch.Size([1, 1000, 754]) | 9\n",
-      "24834    | torch.Size([1, 1000, 775]) | 9\n",
-      "24835    | torch.Size([1, 1000, 781]) | 9\n",
-      "24836    | torch.Size([1, 1000, 754]) | 9\n",
-      "24837    | torch.Size([1, 1000, 777]) | 9\n",
-      "24838    | torch.Size([1, 1000, 754]) | 9\n",
-      "24839    | torch.Size([1, 1000, 762]) | 9\n",
-      "24840    | torch.Size([1, 1000, 754]) | 9\n",
-      "24841    | torch.Size([1, 1000, 777]) | 9\n",
-      "24842    | torch.Size([1, 1000, 777]) | 9\n",
-      "24843    | torch.Size([1, 1000, 787]) | 9\n",
-      "24844    | torch.Size([1, 1000, 754]) | 9\n",
-      "24845    | torch.Size([1, 1000, 777]) | 9\n",
-      "24846    | torch.Size([1, 1000, 754]) | 9\n",
-      "24847    | torch.Size([1, 1000, 754]) | 9\n",
-      "24848    | torch.Size([1, 1000, 780]) | 9\n",
-      "24849    | torch.Size([1, 1000, 777]) | 9\n",
-      "24850    | torch.Size([1, 1000, 754]) | 9\n",
-      "24851    | torch.Size([1, 1000, 762]) | 9\n",
-      "24852    | torch.Size([1, 1000, 754]) | 9\n",
-      "24853    | torch.Size([1, 1000, 777]) | 9\n",
-      "24854    | torch.Size([1, 1000, 786]) | 9\n",
-      "24855    | torch.Size([1, 1000, 776]) | 9\n",
-      "24856    | torch.Size([1, 1000, 777]) | 9\n",
-      "24857    | torch.Size([1, 1000, 762]) | 9\n",
-      "24858    | torch.Size([1, 1000, 777]) | 9\n",
-      "24859    | torch.Size([1, 1000, 754]) | 9\n",
-      "24860    | torch.Size([1, 1000, 777]) | 9\n",
-      "24861    | torch.Size([1, 1000, 771]) | 9\n",
-      "24862    | torch.Size([1, 1000, 762]) | 9\n",
-      "24863    | torch.Size([1, 1000, 716]) | 9\n",
-      "24864    | torch.Size([1, 1000, 754]) | 9\n",
-      "24865    | torch.Size([1, 1000, 777]) | 9\n",
-      "24866    | torch.Size([1, 1000, 859]) | 9\n",
-      "24867    | torch.Size([1, 1000, 799]) | 9\n",
-      "24868    | torch.Size([1, 1000, 777]) | 9\n",
-      "24869    | torch.Size([1, 1000, 754]) | 9\n",
-      "24870    | torch.Size([1, 1000, 777]) | 9\n",
-      "24871    | torch.Size([1, 1000, 754]) | 9\n",
-      "24872    | torch.Size([1, 1000, 754]) | 9\n",
-      "24873    | torch.Size([1, 1000, 787]) | 9\n",
-      "24874    | torch.Size([1, 1000, 777]) | 9\n",
-      "24875    | torch.Size([1, 1000, 777]) | 9\n",
-      "24876    | torch.Size([1, 1000, 795]) | 9\n",
-      "24877    | torch.Size([1, 1000, 754]) | 9\n",
-      "24878    | torch.Size([1, 1000, 777]) | 9\n",
-      "24879    | torch.Size([1, 1000, 777]) | 9\n",
-      "24880    | torch.Size([1, 1000, 754]) | 9\n",
-      "24881    | torch.Size([1, 1000, 754]) | 9\n",
-      "24882    | torch.Size([1, 1000, 754]) | 9\n",
-      "24883    | torch.Size([1, 1000, 777]) | 9\n",
-      "24884    | torch.Size([1, 1000, 754]) | 9\n",
-      "24885    | torch.Size([1, 1000, 776]) | 9\n",
-      "24886    | torch.Size([1, 1000, 713]) | 9\n",
-      "24887    | torch.Size([1, 1000, 754]) | 9\n",
-      "24888    | torch.Size([1, 1000, 754]) | 9\n",
-      "24889    | torch.Size([1, 1000, 762]) | 9\n",
-      "24890    | torch.Size([1, 1000, 754]) | 9\n",
-      "24891    | torch.Size([1, 1000, 754]) | 9\n",
-      "24892    | torch.Size([1, 1000, 754]) | 9\n",
-      "24893    | torch.Size([1, 1000, 777]) | 9\n",
-      "24894    | torch.Size([1, 1000, 777]) | 9\n",
-      "24895    | torch.Size([1, 1000, 754]) | 9\n",
-      "24896    | torch.Size([1, 1000, 797]) | 9\n",
-      "24897    | torch.Size([1, 1000, 777]) | 9\n",
-      "24898    | torch.Size([1, 1000, 781]) | 9\n",
-      "24899    | torch.Size([1, 1000, 754]) | 9\n",
-      "24900    | torch.Size([1, 1000, 754]) | 9\n",
-      "24901    | torch.Size([1, 1000, 754]) | 9\n",
-      "24902    | torch.Size([1, 1000, 777]) | 9\n",
-      "24903    | torch.Size([1, 1000, 777]) | 9\n",
-      "24904    | torch.Size([1, 1000, 762]) | 9\n",
-      "24905    | torch.Size([1, 1000, 771]) | 9\n",
-      "24906    | torch.Size([1, 1000, 775]) | 9\n",
-      "24907    | torch.Size([1, 1000, 663]) | 9\n",
-      "24908    | torch.Size([1, 1000, 772]) | 9\n",
-      "24909    | torch.Size([1, 1000, 777]) | 9\n",
-      "24910    | torch.Size([1, 1000, 777]) | 9\n",
-      "24911    | torch.Size([1, 1000, 777]) | 9\n",
-      "24912    | torch.Size([1, 1000, 749]) | 9\n",
-      "24913    | torch.Size([1, 1000, 754]) | 9\n",
-      "24914    | torch.Size([1, 1000, 777]) | 9\n",
-      "24915    | torch.Size([1, 1000, 754]) | 9\n",
-      "24916    | torch.Size([1, 1000, 754]) | 9\n",
-      "24917    | torch.Size([1, 1000, 777]) | 9\n",
-      "24918    | torch.Size([1, 1000, 754]) | 9\n",
-      "24919    | torch.Size([1, 1000, 777]) | 9\n",
-      "24920    | torch.Size([1, 1000, 754]) | 9\n",
-      "24921    | torch.Size([1, 1000, 777]) | 9\n",
-      "24922    | torch.Size([1, 1000, 754]) | 9\n",
-      "24923    | torch.Size([1, 1000, 754]) | 9\n",
-      "24924    | torch.Size([1, 1000, 777]) | 9\n",
-      "24925    | torch.Size([1, 1000, 762]) | 9\n",
-      "24926    | torch.Size([1, 1000, 754]) | 9\n",
-      "24927    | torch.Size([1, 1000, 867]) | 9\n",
-      "24928    | torch.Size([1, 1000, 802]) | 9\n",
-      "24929    | torch.Size([1, 1000, 777]) | 9\n",
-      "24930    | torch.Size([1, 1000, 754]) | 9\n",
-      "24931    | torch.Size([1, 1000, 754]) | 9\n",
-      "24932    | torch.Size([1, 1000, 782]) | 9\n",
-      "24933    | torch.Size([1, 1000, 754]) | 9\n",
-      "24934    | torch.Size([1, 1000, 754]) | 9\n",
-      "24935    | torch.Size([1, 1000, 802]) | 9\n",
-      "24936    | torch.Size([1, 1000, 783]) | 9\n",
-      "24937    | torch.Size([1, 1000, 779]) | 9\n",
-      "24938    | torch.Size([1, 1000, 777]) | 9\n",
-      "24939    | torch.Size([1, 1000, 771]) | 9\n",
-      "24940    | torch.Size([1, 1000, 777]) | 9\n",
-      "24941    | torch.Size([1, 1000, 754]) | 9\n",
-      "24942    | torch.Size([1, 1000, 778]) | 9\n",
-      "24943    | torch.Size([1, 1000, 777]) | 9\n",
-      "24944    | torch.Size([1, 1000, 802]) | 9\n",
-      "24945    | torch.Size([1, 1000, 777]) | 9\n",
-      "24946    | torch.Size([1, 1000, 777]) | 9\n",
-      "24947    | torch.Size([1, 1000, 777]) | 9\n",
-      "24948    | torch.Size([1, 1000, 777]) | 9\n",
-      "24949    | torch.Size([1, 1000, 754]) | 9\n",
-      "24950    | torch.Size([1, 1000, 754]) | 9\n",
-      "24951    | torch.Size([1, 1000, 754]) | 9\n",
-      "24952    | torch.Size([1, 1000, 754]) | 9\n",
-      "24953    | torch.Size([1, 1000, 768]) | 9\n",
-      "24954    | torch.Size([1, 1000, 754]) | 9\n",
-      "24955    | torch.Size([1, 1000, 777]) | 9\n",
-      "24956    | torch.Size([1, 1000, 754]) | 9\n",
-      "24957    | torch.Size([1, 1000, 779]) | 9\n",
-      "24958    | torch.Size([1, 1000, 754]) | 9\n",
-      "24959    | torch.Size([1, 1000, 754]) | 9\n",
-      "24960    | torch.Size([1, 1000, 777]) | 9\n",
-      "24961    | torch.Size([1, 1000, 778]) | 9\n",
-      "24962    | torch.Size([1, 1000, 777]) | 9\n",
-      "24963    | torch.Size([1, 1000, 754]) | 9\n",
-      "24964    | torch.Size([1, 1000, 778]) | 9\n",
-      "24965    | torch.Size([1, 1000, 754]) | 9\n",
-      "24966    | torch.Size([1, 1000, 779]) | 9\n",
-      "24967    | torch.Size([1, 1000, 754]) | 9\n",
-      "24968    | torch.Size([1, 1000, 754]) | 9\n",
-      "24969    | torch.Size([1, 1000, 754]) | 9\n",
-      "24970    | torch.Size([1, 1000, 768]) | 9\n",
-      "24971    | torch.Size([1, 1000, 754]) | 9\n",
-      "24972    | torch.Size([1, 1000, 777]) | 9\n",
-      "24973    | torch.Size([1, 1000, 777]) | 9\n",
-      "24974    | torch.Size([1, 1000, 777]) | 9\n",
-      "24975    | torch.Size([1, 1000, 786]) | 9\n",
-      "24976    | torch.Size([1, 1000, 777]) | 9\n",
-      "24977    | torch.Size([1, 1000, 754]) | 9\n",
-      "24978    | torch.Size([1, 1000, 611]) | 9\n",
-      "24979    | torch.Size([1, 1000, 779]) | 9\n",
-      "24980    | torch.Size([1, 1000, 754]) | 9\n",
-      "24981    | torch.Size([1, 1000, 754]) | 9\n",
-      "24982    | torch.Size([1, 1000, 754]) | 9\n",
-      "24983    | torch.Size([1, 1000, 777]) | 9\n",
-      "24984    | torch.Size([1, 1000, 777]) | 9\n",
-      "24985    | torch.Size([1, 1000, 754]) | 9\n",
-      "24986    | torch.Size([1, 1000, 754]) | 9\n",
-      "24987    | torch.Size([1, 1000, 777]) | 9\n",
-      "24988    | torch.Size([1, 1000, 754]) | 9\n",
-      "24989    | torch.Size([1, 1000, 802]) | 9\n",
-      "24990    | torch.Size([1, 1000, 777]) | 9\n",
-      "24991    | torch.Size([1, 1000, 777]) | 9\n",
-      "24992    | torch.Size([1, 1000, 786]) | 9\n",
-      "24993    | torch.Size([1, 1000, 781]) | 9\n",
-      "24994    | torch.Size([1, 1000, 777]) | 9\n",
-      "24995    | torch.Size([1, 1000, 754]) | 9\n",
-      "24996    | torch.Size([1, 1000, 777]) | 9\n",
-      "24997    | torch.Size([1, 1000, 754]) | 10\n",
-      "24998    | torch.Size([1, 1000, 754]) | 10\n",
-      "24999    | torch.Size([1, 1000, 754]) | 10\n",
-      "25000    | torch.Size([1, 1000, 754]) | 10\n",
-      "25001    | torch.Size([1, 1000, 766]) | 10\n",
-      "25002    | torch.Size([1, 1000, 754]) | 10\n",
-      "25003    | torch.Size([1, 1000, 848]) | 10\n",
-      "25004    | torch.Size([1, 1000, 777]) | 10\n",
-      "25005    | torch.Size([1, 1000, 754]) | 10\n",
-      "25006    | torch.Size([1, 1000, 762]) | 10\n",
-      "25007    | torch.Size([1, 1000, 754]) | 10\n",
-      "25008    | torch.Size([1, 1000, 754]) | 10\n",
-      "25009    | torch.Size([1, 1000, 771]) | 10\n",
-      "25010    | torch.Size([1, 1000, 754]) | 10\n",
-      "25011    | torch.Size([1, 1000, 754]) | 10\n",
-      "25012    | torch.Size([1, 1000, 762]) | 10\n",
-      "25013    | torch.Size([1, 1000, 754]) | 10\n",
-      "25014    | torch.Size([1, 1000, 754]) | 10\n",
-      "25015    | torch.Size([1, 1000, 754]) | 10\n",
-      "25016    | torch.Size([1, 1000, 777]) | 10\n",
-      "25017    | torch.Size([1, 1000, 754]) | 10\n",
-      "25018    | torch.Size([1, 1000, 754]) | 10\n",
-      "25019    | torch.Size([1, 1000, 767]) | 10\n",
-      "25020    | torch.Size([1, 1000, 767]) | 10\n",
-      "25021    | torch.Size([1, 1000, 754]) | 10\n",
-      "25022    | torch.Size([1, 1000, 754]) | 10\n",
-      "25023    | torch.Size([1, 1000, 777]) | 10\n",
-      "25024    | torch.Size([1, 1000, 754]) | 10\n",
-      "25025    | torch.Size([1, 1000, 754]) | 10\n",
-      "25026    | torch.Size([1, 1000, 754]) | 10\n",
-      "25027    | torch.Size([1, 1000, 754]) | 10\n",
-      "25028    | torch.Size([1, 1000, 773]) | 10\n",
-      "25029    | torch.Size([1, 1000, 762]) | 10\n",
-      "25030    | torch.Size([1, 1000, 754]) | 10\n",
-      "25031    | torch.Size([1, 1000, 754]) | 10\n",
-      "25032    | torch.Size([1, 1000, 754]) | 10\n",
-      "25033    | torch.Size([1, 1000, 802]) | 10\n",
-      "25034    | torch.Size([1, 1000, 766]) | 10\n",
-      "25035    | torch.Size([1, 1000, 762]) | 10\n",
-      "25036    | torch.Size([1, 1000, 754]) | 10\n",
-      "25037    | torch.Size([1, 1000, 754]) | 10\n",
-      "25038    | torch.Size([1, 1000, 754]) | 10\n",
-      "25039    | torch.Size([1, 1000, 762]) | 10\n",
-      "25040    | torch.Size([1, 1000, 777]) | 10\n",
-      "25041    | torch.Size([1, 1000, 754]) | 10\n",
-      "25042    | torch.Size([1, 1000, 762]) | 10\n",
-      "25043    | torch.Size([1, 1000, 802]) | 10\n",
-      "25044    | torch.Size([1, 1000, 777]) | 10\n",
-      "25045    | torch.Size([1, 1000, 754]) | 10\n",
-      "25046    | torch.Size([1, 1000, 762]) | 10\n",
-      "25047    | torch.Size([1, 1000, 762]) | 10\n",
-      "25048    | torch.Size([1, 1000, 754]) | 10\n",
-      "25049    | torch.Size([1, 1000, 754]) | 10\n",
-      "25050    | torch.Size([1, 1000, 762]) | 10\n",
-      "25051    | torch.Size([1, 1000, 754]) | 10\n",
-      "25052    | torch.Size([1, 1000, 754]) | 10\n",
-      "25053    | torch.Size([1, 1000, 754]) | 10\n",
-      "25054    | torch.Size([1, 1000, 754]) | 10\n",
-      "25055    | torch.Size([1, 1000, 871]) | 10\n",
-      "25056    | torch.Size([1, 1000, 778]) | 10\n",
-      "25057    | torch.Size([1, 1000, 754]) | 10\n",
-      "25058    | torch.Size([1, 1000, 754]) | 10\n",
-      "25059    | torch.Size([1, 1000, 777]) | 10\n",
-      "25060    | torch.Size([1, 1000, 754]) | 10\n",
-      "25061    | torch.Size([1, 1000, 777]) | 10\n",
-      "25062    | torch.Size([1, 1000, 796]) | 10\n",
-      "25063    | torch.Size([1, 1000, 754]) | 10\n",
-      "25064    | torch.Size([1, 1000, 754]) | 10\n",
-      "25065    | torch.Size([1, 1000, 771]) | 10\n",
-      "25066    | torch.Size([1, 1000, 754]) | 10\n",
-      "25067    | torch.Size([1, 1000, 754]) | 10\n",
-      "25068    | torch.Size([1, 1000, 777]) | 10\n",
-      "25069    | torch.Size([1, 1000, 725]) | 10\n",
-      "25070    | torch.Size([1, 1000, 754]) | 10\n",
-      "25071    | torch.Size([1, 1000, 754]) | 10\n",
-      "25072    | torch.Size([1, 1000, 802]) | 10\n",
-      "25073    | torch.Size([1, 1000, 762]) | 10\n",
-      "25074    | torch.Size([1, 1000, 754]) | 10\n",
-      "25075    | torch.Size([1, 1000, 754]) | 10\n",
-      "25076    | torch.Size([1, 1000, 778]) | 10\n",
-      "25077    | torch.Size([1, 1000, 777]) | 10\n",
-      "25078    | torch.Size([1, 1000, 713]) | 10\n",
-      "25079    | torch.Size([1, 1000, 713]) | 10\n",
-      "25080    | torch.Size([1, 1000, 762]) | 10\n",
-      "25081    | torch.Size([1, 1000, 850]) | 10\n",
-      "25082    | torch.Size([1, 1000, 777]) | 10\n",
-      "25083    | torch.Size([1, 1000, 754]) | 10\n",
-      "25084    | torch.Size([1, 1000, 754]) | 10\n",
-      "25085    | torch.Size([1, 1000, 807]) | 10\n",
-      "25086    | torch.Size([1, 1000, 754]) | 10\n",
-      "25087    | torch.Size([1, 1000, 754]) | 10\n",
-      "25088    | torch.Size([1, 1000, 754]) | 10\n",
-      "25089    | torch.Size([1, 1000, 754]) | 10\n",
-      "25090    | torch.Size([1, 1000, 754]) | 10\n",
-      "25091    | torch.Size([1, 1000, 754]) | 10\n",
-      "25092    | torch.Size([1, 1000, 754]) | 10\n",
-      "25093    | torch.Size([1, 1000, 754]) | 10\n",
-      "25094    | torch.Size([1, 1000, 754]) | 10\n",
-      "25095    | torch.Size([1, 1000, 777]) | 10\n",
-      "25096    | torch.Size([1, 1000, 762]) | 10\n",
-      "25097    | torch.Size([1, 1000, 846]) | 10\n",
-      "25098    | torch.Size([1, 1000, 754]) | 10\n",
-      "25099    | torch.Size([1, 1000, 762]) | 10\n",
-      "25100    | torch.Size([1, 1000, 754]) | 10\n",
-      "25101    | torch.Size([1, 1000, 754]) | 10\n",
-      "25102    | torch.Size([1, 1000, 754]) | 10\n",
-      "25103    | torch.Size([1, 1000, 754]) | 10\n",
-      "25104    | torch.Size([1, 1000, 754]) | 10\n",
-      "25105    | torch.Size([1, 1000, 777]) | 10\n",
-      "25106    | torch.Size([1, 1000, 762]) | 10\n",
-      "25107    | torch.Size([1, 1000, 754]) | 10\n",
-      "25108    | torch.Size([1, 1000, 762]) | 10\n",
-      "25109    | torch.Size([1, 1000, 754]) | 10\n",
-      "25110    | torch.Size([1, 1000, 762]) | 10\n",
-      "25111    | torch.Size([1, 1000, 754]) | 10\n",
-      "25112    | torch.Size([1, 1000, 754]) | 10\n",
-      "25113    | torch.Size([1, 1000, 801]) | 10\n",
-      "25114    | torch.Size([1, 1000, 768]) | 10\n",
-      "25115    | torch.Size([1, 1000, 798]) | 10\n",
-      "25116    | torch.Size([1, 1000, 754]) | 10\n",
-      "25117    | torch.Size([1, 1000, 774]) | 10\n",
-      "25118    | torch.Size([1, 1000, 754]) | 10\n",
-      "25119    | torch.Size([1, 1000, 762]) | 10\n",
-      "25120    | torch.Size([1, 1000, 754]) | 10\n",
-      "25121    | torch.Size([1, 1000, 754]) | 10\n",
-      "25122    | torch.Size([1, 1000, 777]) | 10\n",
-      "25123    | torch.Size([1, 1000, 793]) | 10\n",
-      "25124    | torch.Size([1, 1000, 762]) | 10\n",
-      "25125    | torch.Size([1, 1000, 703]) | 10\n",
-      "25126    | torch.Size([1, 1000, 777]) | 10\n",
-      "25127    | torch.Size([1, 1000, 777]) | 10\n",
-      "25128    | torch.Size([1, 1000, 783]) | 10\n",
-      "25129    | torch.Size([1, 1000, 762]) | 10\n",
-      "25130    | torch.Size([1, 1000, 762]) | 10\n",
-      "25131    | torch.Size([1, 1000, 754]) | 10\n",
-      "25132    | torch.Size([1, 1000, 754]) | 10\n",
-      "25133    | torch.Size([1, 1000, 762]) | 10\n",
-      "25134    | torch.Size([1, 1000, 777]) | 10\n",
-      "25135    | torch.Size([1, 1000, 754]) | 10\n",
-      "25136    | torch.Size([1, 1000, 771]) | 10\n",
-      "25137    | torch.Size([1, 1000, 762]) | 10\n",
-      "25138    | torch.Size([1, 1000, 754]) | 10\n",
-      "25139    | torch.Size([1, 1000, 754]) | 10\n",
-      "25140    | torch.Size([1, 1000, 754]) | 10\n",
-      "25141    | torch.Size([1, 1000, 777]) | 10\n",
-      "25142    | torch.Size([1, 1000, 787]) | 10\n",
-      "25143    | torch.Size([1, 1000, 754]) | 10\n",
-      "25144    | torch.Size([1, 1000, 762]) | 10\n",
-      "25145    | torch.Size([1, 1000, 754]) | 10\n",
-      "25146    | torch.Size([1, 1000, 777]) | 10\n",
-      "25147    | torch.Size([1, 1000, 754]) | 10\n",
-      "25148    | torch.Size([1, 1000, 754]) | 10\n",
-      "25149    | torch.Size([1, 1000, 754]) | 10\n",
-      "25150    | torch.Size([1, 1000, 754]) | 10\n",
-      "25151    | torch.Size([1, 1000, 773]) | 10\n",
-      "25152    | torch.Size([1, 1000, 754]) | 10\n",
-      "25153    | torch.Size([1, 1000, 754]) | 10\n",
-      "25154    | torch.Size([1, 1000, 777]) | 10\n",
-      "25155    | torch.Size([1, 1000, 754]) | 10\n",
-      "25156    | torch.Size([1, 1000, 754]) | 10\n",
-      "25157    | torch.Size([1, 1000, 754]) | 10\n",
-      "25158    | torch.Size([1, 1000, 777]) | 10\n",
-      "25159    | torch.Size([1, 1000, 754]) | 10\n",
-      "25160    | torch.Size([1, 1000, 754]) | 10\n",
-      "25161    | torch.Size([1, 1000, 777]) | 10\n",
-      "25162    | torch.Size([1, 1000, 799]) | 10\n",
-      "25163    | torch.Size([1, 1000, 843]) | 10\n",
-      "25164    | torch.Size([1, 1000, 713]) | 10\n",
-      "25165    | torch.Size([1, 1000, 762]) | 10\n",
-      "25166    | torch.Size([1, 1000, 777]) | 10\n",
-      "25167    | torch.Size([1, 1000, 781]) | 10\n",
-      "25168    | torch.Size([1, 1000, 754]) | 10\n",
-      "25169    | torch.Size([1, 1000, 754]) | 10\n",
-      "25170    | torch.Size([1, 1000, 754]) | 10\n",
-      "25171    | torch.Size([1, 1000, 754]) | 10\n",
-      "25172    | torch.Size([1, 1000, 754]) | 10\n",
-      "25173    | torch.Size([1, 1000, 754]) | 10\n",
-      "25174    | torch.Size([1, 1000, 754]) | 10\n",
-      "25175    | torch.Size([1, 1000, 777]) | 10\n",
-      "25176    | torch.Size([1, 1000, 754]) | 10\n",
-      "25177    | torch.Size([1, 1000, 762]) | 10\n",
-      "25178    | torch.Size([1, 1000, 762]) | 10\n",
-      "25179    | torch.Size([1, 1000, 764]) | 10\n",
-      "25180    | torch.Size([1, 1000, 754]) | 10\n",
-      "25181    | torch.Size([1, 1000, 754]) | 10\n",
-      "25182    | torch.Size([1, 1000, 754]) | 10\n",
-      "25183    | torch.Size([1, 1000, 807]) | 10\n",
-      "25184    | torch.Size([1, 1000, 802]) | 10\n",
-      "25185    | torch.Size([1, 1000, 777]) | 10\n",
-      "25186    | torch.Size([1, 1000, 762]) | 10\n",
-      "25187    | torch.Size([1, 1000, 790]) | 10\n",
-      "25188    | torch.Size([1, 1000, 754]) | 10\n",
-      "25189    | torch.Size([1, 1000, 787]) | 10\n",
-      "25190    | torch.Size([1, 1000, 762]) | 10\n",
-      "25191    | torch.Size([1, 1000, 754]) | 10\n",
-      "25192    | torch.Size([1, 1000, 754]) | 10\n",
-      "25193    | torch.Size([1, 1000, 754]) | 10\n",
-      "25194    | torch.Size([1, 1000, 762]) | 10\n",
-      "25195    | torch.Size([1, 1000, 787]) | 10\n",
-      "25196    | torch.Size([1, 1000, 802]) | 10\n",
-      "25197    | torch.Size([1, 1000, 754]) | 10\n",
-      "25198    | torch.Size([1, 1000, 754]) | 10\n",
-      "25199    | torch.Size([1, 1000, 754]) | 10\n",
-      "25200    | torch.Size([1, 1000, 754]) | 10\n",
-      "25201    | torch.Size([1, 1000, 754]) | 10\n",
-      "25202    | torch.Size([1, 1000, 724]) | 10\n",
-      "25203    | torch.Size([1, 1000, 797]) | 10\n",
-      "25204    | torch.Size([1, 1000, 762]) | 10\n",
-      "25205    | torch.Size([1, 1000, 754]) | 10\n",
-      "25206    | torch.Size([1, 1000, 762]) | 10\n",
-      "25207    | torch.Size([1, 1000, 808]) | 10\n",
-      "25208    | torch.Size([1, 1000, 782]) | 10\n",
-      "25209    | torch.Size([1, 1000, 754]) | 10\n",
-      "25210    | torch.Size([1, 1000, 764]) | 10\n",
-      "25211    | torch.Size([1, 1000, 754]) | 10\n",
-      "25212    | torch.Size([1, 1000, 777]) | 10\n",
-      "25213    | torch.Size([1, 1000, 762]) | 10\n",
-      "25214    | torch.Size([1, 1000, 754]) | 10\n",
-      "25215    | torch.Size([1, 1000, 786]) | 10\n",
-      "25216    | torch.Size([1, 1000, 797]) | 10\n",
-      "25217    | torch.Size([1, 1000, 754]) | 10\n",
-      "25218    | torch.Size([1, 1000, 754]) | 10\n",
-      "25219    | torch.Size([1, 1000, 754]) | 10\n",
-      "25220    | torch.Size([1, 1000, 754]) | 10\n",
-      "25221    | torch.Size([1, 1000, 777]) | 10\n",
-      "25222    | torch.Size([1, 1000, 762]) | 10\n",
-      "25223    | torch.Size([1, 1000, 754]) | 10\n",
-      "25224    | torch.Size([1, 1000, 777]) | 10\n",
-      "25225    | torch.Size([1, 1000, 762]) | 10\n",
-      "25226    | torch.Size([1, 1000, 754]) | 10\n",
-      "25227    | torch.Size([1, 1000, 754]) | 10\n",
-      "25228    | torch.Size([1, 1000, 802]) | 10\n",
-      "25229    | torch.Size([1, 1000, 762]) | 10\n",
-      "25230    | torch.Size([1, 1000, 754]) | 10\n",
-      "25231    | torch.Size([1, 1000, 780]) | 10\n",
-      "25232    | torch.Size([1, 1000, 714]) | 10\n",
-      "25233    | torch.Size([1, 1000, 762]) | 10\n",
-      "25234    | torch.Size([1, 1000, 771]) | 10\n",
-      "25235    | torch.Size([1, 1000, 754]) | 10\n",
-      "25236    | torch.Size([1, 1000, 754]) | 10\n",
-      "25237    | torch.Size([1, 1000, 754]) | 10\n",
-      "25238    | torch.Size([1, 1000, 754]) | 10\n",
-      "25239    | torch.Size([1, 1000, 754]) | 10\n",
-      "25240    | torch.Size([1, 1000, 754]) | 10\n",
-      "25241    | torch.Size([1, 1000, 768]) | 10\n",
-      "25242    | torch.Size([1, 1000, 754]) | 10\n",
-      "25243    | torch.Size([1, 1000, 777]) | 10\n",
-      "25244    | torch.Size([1, 1000, 754]) | 10\n",
-      "25245    | torch.Size([1, 1000, 754]) | 10\n",
-      "25246    | torch.Size([1, 1000, 777]) | 10\n",
-      "25247    | torch.Size([1, 1000, 754]) | 10\n",
-      "25248    | torch.Size([1, 1000, 762]) | 10\n",
-      "25249    | torch.Size([1, 1000, 812]) | 10\n",
-      "25250    | torch.Size([1, 1000, 754]) | 10\n",
-      "25251    | torch.Size([1, 1000, 754]) | 10\n",
-      "25252    | torch.Size([1, 1000, 777]) | 10\n",
-      "25253    | torch.Size([1, 1000, 789]) | 10\n",
-      "25254    | torch.Size([1, 1000, 802]) | 10\n",
-      "25255    | torch.Size([1, 1000, 776]) | 10\n",
-      "25256    | torch.Size([1, 1000, 762]) | 10\n",
-      "25257    | torch.Size([1, 1000, 762]) | 10\n",
-      "25258    | torch.Size([1, 1000, 762]) | 10\n",
-      "25259    | torch.Size([1, 1000, 754]) | 10\n",
-      "25260    | torch.Size([1, 1000, 754]) | 10\n",
-      "25261    | torch.Size([1, 1000, 754]) | 10\n",
-      "25262    | torch.Size([1, 1000, 754]) | 10\n",
-      "25263    | torch.Size([1, 1000, 762]) | 10\n",
-      "25264    | torch.Size([1, 1000, 762]) | 10\n",
-      "25265    | torch.Size([1, 1000, 794]) | 10\n",
-      "25266    | torch.Size([1, 1000, 777]) | 10\n",
-      "25267    | torch.Size([1, 1000, 762]) | 10\n",
-      "25268    | torch.Size([1, 1000, 754]) | 10\n",
-      "25269    | torch.Size([1, 1000, 754]) | 10\n",
-      "25270    | torch.Size([1, 1000, 754]) | 10\n",
-      "25271    | torch.Size([1, 1000, 777]) | 10\n",
-      "25272    | torch.Size([1, 1000, 809]) | 10\n",
-      "25273    | torch.Size([1, 1000, 754]) | 10\n",
-      "25274    | torch.Size([1, 1000, 793]) | 10\n",
-      "25275    | torch.Size([1, 1000, 754]) | 10\n",
-      "25276    | torch.Size([1, 1000, 777]) | 10\n",
-      "25277    | torch.Size([1, 1000, 762]) | 10\n",
-      "25278    | torch.Size([1, 1000, 754]) | 10\n",
-      "25279    | torch.Size([1, 1000, 727]) | 10\n",
-      "25280    | torch.Size([1, 1000, 775]) | 10\n",
-      "25281    | torch.Size([1, 1000, 754]) | 10\n",
-      "25282    | torch.Size([1, 1000, 789]) | 10\n",
-      "25283    | torch.Size([1, 1000, 781]) | 10\n",
-      "25284    | torch.Size([1, 1000, 754]) | 10\n",
-      "25285    | torch.Size([1, 1000, 754]) | 10\n",
-      "25286    | torch.Size([1, 1000, 762]) | 10\n",
-      "25287    | torch.Size([1, 1000, 762]) | 10\n",
-      "25288    | torch.Size([1, 1000, 771]) | 10\n",
-      "25289    | torch.Size([1, 1000, 777]) | 10\n",
-      "25290    | torch.Size([1, 1000, 754]) | 10\n",
-      "25291    | torch.Size([1, 1000, 754]) | 10\n",
-      "25292    | torch.Size([1, 1000, 754]) | 10\n",
-      "25293    | torch.Size([1, 1000, 762]) | 10\n",
-      "25294    | torch.Size([1, 1000, 754]) | 10\n",
-      "25295    | torch.Size([1, 1000, 802]) | 10\n",
-      "25296    | torch.Size([1, 1000, 754]) | 10\n",
-      "25297    | torch.Size([1, 1000, 777]) | 10\n",
-      "25298    | torch.Size([1, 1000, 754]) | 10\n",
-      "25299    | torch.Size([1, 1000, 802]) | 10\n",
-      "25300    | torch.Size([1, 1000, 754]) | 10\n",
-      "25301    | torch.Size([1, 1000, 777]) | 10\n",
-      "25302    | torch.Size([1, 1000, 762]) | 10\n",
-      "25303    | torch.Size([1, 1000, 754]) | 10\n",
-      "25304    | torch.Size([1, 1000, 765]) | 10\n",
-      "25305    | torch.Size([1, 1000, 724]) | 10\n",
-      "25306    | torch.Size([1, 1000, 754]) | 10\n",
-      "25307    | torch.Size([1, 1000, 802]) | 10\n",
-      "25308    | torch.Size([1, 1000, 754]) | 10\n",
-      "25309    | torch.Size([1, 1000, 754]) | 10\n",
-      "25310    | torch.Size([1, 1000, 754]) | 10\n",
-      "25311    | torch.Size([1, 1000, 754]) | 10\n",
-      "25312    | torch.Size([1, 1000, 754]) | 10\n",
-      "25313    | torch.Size([1, 1000, 754]) | 10\n",
-      "25314    | torch.Size([1, 1000, 777]) | 10\n",
-      "25315    | torch.Size([1, 1000, 754]) | 10\n",
-      "25316    | torch.Size([1, 1000, 754]) | 10\n",
-      "25317    | torch.Size([1, 1000, 783]) | 10\n",
-      "25318    | torch.Size([1, 1000, 754]) | 10\n",
-      "25319    | torch.Size([1, 1000, 794]) | 10\n",
-      "25320    | torch.Size([1, 1000, 782]) | 10\n",
-      "25321    | torch.Size([1, 1000, 754]) | 10\n",
-      "25322    | torch.Size([1, 1000, 773]) | 10\n",
-      "25323    | torch.Size([1, 1000, 754]) | 10\n",
-      "25324    | torch.Size([1, 1000, 754]) | 10\n",
-      "25325    | torch.Size([1, 1000, 754]) | 10\n",
-      "25326    | torch.Size([1, 1000, 770]) | 10\n",
-      "25327    | torch.Size([1, 1000, 754]) | 10\n",
-      "25328    | torch.Size([1, 1000, 754]) | 10\n",
-      "25329    | torch.Size([1, 1000, 762]) | 10\n",
-      "25330    | torch.Size([1, 1000, 713]) | 10\n",
-      "25331    | torch.Size([1, 1000, 754]) | 10\n",
-      "25332    | torch.Size([1, 1000, 754]) | 10\n",
-      "25333    | torch.Size([1, 1000, 802]) | 10\n",
-      "25334    | torch.Size([1, 1000, 754]) | 10\n",
-      "25335    | torch.Size([1, 1000, 765]) | 10\n",
-      "25336    | torch.Size([1, 1000, 754]) | 10\n",
-      "25337    | torch.Size([1, 1000, 754]) | 10\n",
-      "25338    | torch.Size([1, 1000, 754]) | 10\n",
-      "25339    | torch.Size([1, 1000, 754]) | 10\n",
-      "25340    | torch.Size([1, 1000, 754]) | 10\n",
-      "25341    | torch.Size([1, 1000, 754]) | 10\n",
-      "25342    | torch.Size([1, 1000, 775]) | 10\n",
-      "25343    | torch.Size([1, 1000, 762]) | 10\n",
-      "25344    | torch.Size([1, 1000, 762]) | 10\n",
-      "25345    | torch.Size([1, 1000, 754]) | 10\n",
-      "25346    | torch.Size([1, 1000, 754]) | 10\n",
-      "25347    | torch.Size([1, 1000, 777]) | 10\n",
-      "25348    | torch.Size([1, 1000, 754]) | 10\n",
-      "25349    | torch.Size([1, 1000, 777]) | 10\n",
-      "25350    | torch.Size([1, 1000, 754]) | 10\n",
-      "25351    | torch.Size([1, 1000, 762]) | 10\n",
-      "25352    | torch.Size([1, 1000, 762]) | 10\n",
-      "25353    | torch.Size([1, 1000, 754]) | 10\n",
-      "25354    | torch.Size([1, 1000, 763]) | 10\n",
-      "25355    | torch.Size([1, 1000, 754]) | 10\n",
-      "25356    | torch.Size([1, 1000, 811]) | 10\n",
-      "25357    | torch.Size([1, 1000, 777]) | 10\n",
-      "25358    | torch.Size([1, 1000, 754]) | 10\n",
-      "25359    | torch.Size([1, 1000, 754]) | 10\n",
-      "25360    | torch.Size([1, 1000, 754]) | 10\n",
-      "25361    | torch.Size([1, 1000, 754]) | 10\n",
-      "25362    | torch.Size([1, 1000, 754]) | 10\n",
-      "25363    | torch.Size([1, 1000, 762]) | 10\n",
-      "25364    | torch.Size([1, 1000, 777]) | 10\n",
-      "25365    | torch.Size([1, 1000, 775]) | 10\n",
-      "25366    | torch.Size([1, 1000, 754]) | 10\n",
-      "25367    | torch.Size([1, 1000, 802]) | 10\n",
-      "25368    | torch.Size([1, 1000, 766]) | 10\n",
-      "25369    | torch.Size([1, 1000, 777]) | 10\n",
-      "25370    | torch.Size([1, 1000, 779]) | 10\n",
-      "25371    | torch.Size([1, 1000, 754]) | 10\n",
-      "25372    | torch.Size([1, 1000, 754]) | 10\n",
-      "25373    | torch.Size([1, 1000, 754]) | 10\n",
-      "25374    | torch.Size([1, 1000, 754]) | 10\n",
-      "25375    | torch.Size([1, 1000, 762]) | 10\n",
-      "25376    | torch.Size([1, 1000, 754]) | 10\n",
-      "25377    | torch.Size([1, 1000, 819]) | 10\n",
-      "25378    | torch.Size([1, 1000, 754]) | 10\n",
-      "25379    | torch.Size([1, 1000, 754]) | 10\n",
-      "25380    | torch.Size([1, 1000, 754]) | 10\n",
-      "25381    | torch.Size([1, 1000, 762]) | 10\n",
-      "25382    | torch.Size([1, 1000, 762]) | 10\n",
-      "25383    | torch.Size([1, 1000, 777]) | 10\n",
-      "25384    | torch.Size([1, 1000, 754]) | 10\n",
-      "25385    | torch.Size([1, 1000, 797]) | 10\n",
-      "25386    | torch.Size([1, 1000, 754]) | 10\n",
-      "25387    | torch.Size([1, 1000, 781]) | 10\n",
-      "25388    | torch.Size([1, 1000, 754]) | 10\n",
-      "25389    | torch.Size([1, 1000, 783]) | 10\n",
-      "25390    | torch.Size([1, 1000, 754]) | 10\n",
-      "25391    | torch.Size([1, 1000, 754]) | 10\n",
-      "25392    | torch.Size([1, 1000, 754]) | 10\n",
-      "25393    | torch.Size([1, 1000, 770]) | 10\n",
-      "25394    | torch.Size([1, 1000, 754]) | 10\n",
-      "25395    | torch.Size([1, 1000, 789]) | 10\n",
-      "25396    | torch.Size([1, 1000, 777]) | 10\n",
-      "25397    | torch.Size([1, 1000, 783]) | 10\n",
-      "25398    | torch.Size([1, 1000, 754]) | 10\n",
-      "25399    | torch.Size([1, 1000, 754]) | 10\n",
-      "25400    | torch.Size([1, 1000, 777]) | 10\n",
-      "25401    | torch.Size([1, 1000, 754]) | 10\n",
-      "25402    | torch.Size([1, 1000, 754]) | 10\n",
-      "25403    | torch.Size([1, 1000, 762]) | 10\n",
-      "25404    | torch.Size([1, 1000, 754]) | 10\n",
-      "25405    | torch.Size([1, 1000, 777]) | 10\n",
-      "25406    | torch.Size([1, 1000, 754]) | 10\n",
-      "25407    | torch.Size([1, 1000, 802]) | 10\n",
-      "25408    | torch.Size([1, 1000, 777]) | 10\n",
-      "25409    | torch.Size([1, 1000, 754]) | 10\n",
-      "25410    | torch.Size([1, 1000, 754]) | 10\n",
-      "25411    | torch.Size([1, 1000, 754]) | 10\n",
-      "25412    | torch.Size([1, 1000, 792]) | 10\n",
-      "25413    | torch.Size([1, 1000, 754]) | 10\n",
-      "25414    | torch.Size([1, 1000, 754]) | 10\n",
-      "25415    | torch.Size([1, 1000, 754]) | 10\n",
-      "25416    | torch.Size([1, 1000, 754]) | 10\n",
-      "25417    | torch.Size([1, 1000, 754]) | 10\n",
-      "25418    | torch.Size([1, 1000, 754]) | 10\n",
-      "25419    | torch.Size([1, 1000, 754]) | 10\n",
-      "25420    | torch.Size([1, 1000, 777]) | 10\n",
-      "25421    | torch.Size([1, 1000, 754]) | 10\n",
-      "25422    | torch.Size([1, 1000, 773]) | 10\n",
-      "25423    | torch.Size([1, 1000, 754]) | 10\n",
-      "25424    | torch.Size([1, 1000, 787]) | 10\n",
-      "25425    | torch.Size([1, 1000, 754]) | 10\n",
-      "25426    | torch.Size([1, 1000, 777]) | 10\n",
-      "25427    | torch.Size([1, 1000, 798]) | 10\n",
-      "25428    | torch.Size([1, 1000, 777]) | 10\n",
-      "25429    | torch.Size([1, 1000, 754]) | 10\n",
-      "25430    | torch.Size([1, 1000, 754]) | 10\n",
-      "25431    | torch.Size([1, 1000, 777]) | 10\n",
-      "25432    | torch.Size([1, 1000, 754]) | 10\n",
-      "25433    | torch.Size([1, 1000, 777]) | 10\n",
-      "25434    | torch.Size([1, 1000, 778]) | 10\n",
-      "25435    | torch.Size([1, 1000, 754]) | 10\n",
-      "25436    | torch.Size([1, 1000, 754]) | 10\n",
-      "25437    | torch.Size([1, 1000, 777]) | 10\n",
-      "25438    | torch.Size([1, 1000, 777]) | 10\n",
-      "25439    | torch.Size([1, 1000, 754]) | 10\n",
-      "25440    | torch.Size([1, 1000, 802]) | 10\n",
-      "25441    | torch.Size([1, 1000, 754]) | 10\n",
-      "25442    | torch.Size([1, 1000, 754]) | 10\n",
-      "25443    | torch.Size([1, 1000, 754]) | 10\n",
-      "25444    | torch.Size([1, 1000, 787]) | 10\n",
-      "25445    | torch.Size([1, 1000, 754]) | 10\n",
-      "25446    | torch.Size([1, 1000, 777]) | 10\n",
-      "25447    | torch.Size([1, 1000, 754]) | 10\n",
-      "25448    | torch.Size([1, 1000, 754]) | 10\n",
-      "25449    | torch.Size([1, 1000, 754]) | 10\n",
-      "25450    | torch.Size([1, 1000, 783]) | 10\n",
-      "25451    | torch.Size([1, 1000, 777]) | 10\n",
-      "25452    | torch.Size([1, 1000, 762]) | 10\n",
-      "25453    | torch.Size([1, 1000, 754]) | 10\n",
-      "25454    | torch.Size([1, 1000, 754]) | 10\n",
-      "25455    | torch.Size([1, 1000, 771]) | 10\n",
-      "25456    | torch.Size([1, 1000, 762]) | 10\n",
-      "25457    | torch.Size([1, 1000, 777]) | 10\n",
-      "25458    | torch.Size([1, 1000, 786]) | 10\n",
-      "25459    | torch.Size([1, 1000, 804]) | 10\n",
-      "25460    | torch.Size([1, 1000, 777]) | 10\n",
-      "25461    | torch.Size([1, 1000, 754]) | 10\n",
-      "25462    | torch.Size([1, 1000, 754]) | 10\n",
-      "25463    | torch.Size([1, 1000, 754]) | 10\n",
-      "25464    | torch.Size([1, 1000, 777]) | 10\n",
-      "25465    | torch.Size([1, 1000, 754]) | 10\n",
-      "25466    | torch.Size([1, 1000, 754]) | 10\n",
-      "25467    | torch.Size([1, 1000, 754]) | 10\n",
-      "25468    | torch.Size([1, 1000, 797]) | 10\n",
-      "25469    | torch.Size([1, 1000, 778]) | 10\n",
-      "25470    | torch.Size([1, 1000, 828]) | 10\n",
-      "25471    | torch.Size([1, 1000, 754]) | 10\n",
-      "25472    | torch.Size([1, 1000, 766]) | 10\n",
-      "25473    | torch.Size([1, 1000, 754]) | 10\n",
-      "25474    | torch.Size([1, 1000, 754]) | 10\n",
-      "25475    | torch.Size([1, 1000, 754]) | 10\n",
-      "25476    | torch.Size([1, 1000, 754]) | 10\n",
-      "25477    | torch.Size([1, 1000, 754]) | 10\n",
-      "25478    | torch.Size([1, 1000, 777]) | 10\n",
-      "25479    | torch.Size([1, 1000, 777]) | 10\n",
-      "25480    | torch.Size([1, 1000, 754]) | 10\n",
-      "25481    | torch.Size([1, 1000, 762]) | 10\n",
-      "25482    | torch.Size([1, 1000, 779]) | 10\n",
-      "25483    | torch.Size([1, 1000, 792]) | 10\n",
-      "25484    | torch.Size([1, 1000, 754]) | 10\n",
-      "25485    | torch.Size([1, 1000, 802]) | 10\n",
-      "25486    | torch.Size([1, 1000, 786]) | 10\n",
-      "25487    | torch.Size([1, 1000, 807]) | 10\n",
-      "25488    | torch.Size([1, 1000, 762]) | 10\n",
-      "25489    | torch.Size([1, 1000, 754]) | 10\n",
-      "25490    | torch.Size([1, 1000, 784]) | 10\n",
-      "25491    | torch.Size([1, 1000, 767]) | 10\n",
-      "25492    | torch.Size([1, 1000, 724]) | 10\n",
-      "25493    | torch.Size([1, 1000, 762]) | 10\n",
-      "25494    | torch.Size([1, 1000, 795]) | 10\n",
-      "25495    | torch.Size([1, 1000, 795]) | 10\n",
-      "25496    | torch.Size([1, 1000, 771]) | 10\n",
-      "25497    | torch.Size([1, 1000, 802]) | 10\n",
-      "25498    | torch.Size([1, 1000, 777]) | 10\n",
-      "25499    | torch.Size([1, 1000, 754]) | 10\n",
-      "25500    | torch.Size([1, 1000, 754]) | 10\n",
-      "25501    | torch.Size([1, 1000, 786]) | 10\n",
-      "25502    | torch.Size([1, 1000, 754]) | 10\n",
-      "25503    | torch.Size([1, 1000, 754]) | 10\n",
-      "25504    | torch.Size([1, 1000, 802]) | 10\n",
-      "25505    | torch.Size([1, 1000, 754]) | 10\n",
-      "25506    | torch.Size([1, 1000, 754]) | 10\n",
-      "25507    | torch.Size([1, 1000, 754]) | 10\n",
-      "25508    | torch.Size([1, 1000, 809]) | 10\n",
-      "25509    | torch.Size([1, 1000, 754]) | 10\n",
-      "25510    | torch.Size([1, 1000, 777]) | 10\n",
-      "25511    | torch.Size([1, 1000, 783]) | 10\n",
-      "25512    | torch.Size([1, 1000, 766]) | 10\n",
-      "25513    | torch.Size([1, 1000, 786]) | 10\n",
-      "25514    | torch.Size([1, 1000, 754]) | 10\n",
-      "25515    | torch.Size([1, 1000, 777]) | 10\n",
-      "25516    | torch.Size([1, 1000, 762]) | 10\n",
-      "25517    | torch.Size([1, 1000, 846]) | 10\n",
-      "25518    | torch.Size([1, 1000, 777]) | 10\n",
-      "25519    | torch.Size([1, 1000, 754]) | 10\n",
-      "25520    | torch.Size([1, 1000, 754]) | 10\n",
-      "25521    | torch.Size([1, 1000, 754]) | 10\n",
-      "25522    | torch.Size([1, 1000, 754]) | 10\n",
-      "25523    | torch.Size([1, 1000, 802]) | 10\n",
-      "25524    | torch.Size([1, 1000, 754]) | 10\n",
-      "25525    | torch.Size([1, 1000, 754]) | 10\n",
-      "25526    | torch.Size([1, 1000, 754]) | 10\n",
-      "25527    | torch.Size([1, 1000, 777]) | 10\n",
-      "25528    | torch.Size([1, 1000, 771]) | 10\n",
-      "25529    | torch.Size([1, 1000, 754]) | 10\n",
-      "25530    | torch.Size([1, 1000, 754]) | 10\n",
-      "25531    | torch.Size([1, 1000, 754]) | 10\n",
-      "25532    | torch.Size([1, 1000, 777]) | 10\n",
-      "25533    | torch.Size([1, 1000, 802]) | 10\n",
-      "25534    | torch.Size([1, 1000, 777]) | 10\n",
-      "25535    | torch.Size([1, 1000, 762]) | 10\n",
-      "25536    | torch.Size([1, 1000, 754]) | 10\n",
-      "25537    | torch.Size([1, 1000, 776]) | 10\n",
-      "25538    | torch.Size([1, 1000, 754]) | 10\n",
-      "25539    | torch.Size([1, 1000, 754]) | 10\n",
-      "25540    | torch.Size([1, 1000, 754]) | 10\n",
-      "25541    | torch.Size([1, 1000, 762]) | 10\n",
-      "25542    | torch.Size([1, 1000, 801]) | 10\n",
-      "25543    | torch.Size([1, 1000, 754]) | 10\n",
-      "25544    | torch.Size([1, 1000, 754]) | 10\n",
-      "25545    | torch.Size([1, 1000, 754]) | 10\n",
-      "25546    | torch.Size([1, 1000, 754]) | 10\n",
-      "25547    | torch.Size([1, 1000, 754]) | 10\n",
-      "25548    | torch.Size([1, 1000, 762]) | 10\n",
-      "25549    | torch.Size([1, 1000, 777]) | 10\n",
-      "25550    | torch.Size([1, 1000, 754]) | 10\n",
-      "25551    | torch.Size([1, 1000, 754]) | 10\n",
-      "25552    | torch.Size([1, 1000, 805]) | 10\n",
-      "25553    | torch.Size([1, 1000, 754]) | 10\n",
-      "25554    | torch.Size([1, 1000, 762]) | 10\n",
-      "25555    | torch.Size([1, 1000, 762]) | 10\n",
-      "25556    | torch.Size([1, 1000, 802]) | 10\n",
-      "25557    | torch.Size([1, 1000, 762]) | 10\n",
-      "25558    | torch.Size([1, 1000, 754]) | 10\n",
-      "25559    | torch.Size([1, 1000, 754]) | 10\n",
-      "25560    | torch.Size([1, 1000, 728]) | 10\n",
-      "25561    | torch.Size([1, 1000, 779]) | 10\n",
-      "25562    | torch.Size([1, 1000, 777]) | 10\n",
-      "25563    | torch.Size([1, 1000, 762]) | 10\n",
-      "25564    | torch.Size([1, 1000, 777]) | 10\n",
-      "25565    | torch.Size([1, 1000, 754]) | 10\n",
-      "25566    | torch.Size([1, 1000, 754]) | 10\n",
-      "25567    | torch.Size([1, 1000, 754]) | 10\n",
-      "25568    | torch.Size([1, 1000, 802]) | 10\n",
-      "25569    | torch.Size([1, 1000, 777]) | 10\n",
-      "25570    | torch.Size([1, 1000, 754]) | 10\n",
-      "25571    | torch.Size([1, 1000, 754]) | 10\n",
-      "25572    | torch.Size([1, 1000, 754]) | 10\n",
-      "25573    | torch.Size([1, 1000, 754]) | 10\n",
-      "25574    | torch.Size([1, 1000, 802]) | 10\n",
-      "25575    | torch.Size([1, 1000, 781]) | 10\n",
-      "25576    | torch.Size([1, 1000, 754]) | 10\n",
-      "25577    | torch.Size([1, 1000, 754]) | 10\n",
-      "25578    | torch.Size([1, 1000, 754]) | 10\n",
-      "25579    | torch.Size([1, 1000, 871]) | 10\n",
-      "25580    | torch.Size([1, 1000, 777]) | 10\n",
-      "25581    | torch.Size([1, 1000, 754]) | 10\n",
-      "25582    | torch.Size([1, 1000, 780]) | 10\n",
-      "25583    | torch.Size([1, 1000, 777]) | 10\n",
-      "25584    | torch.Size([1, 1000, 769]) | 10\n",
-      "25585    | torch.Size([1, 1000, 772]) | 10\n",
-      "25586    | torch.Size([1, 1000, 762]) | 10\n",
-      "25587    | torch.Size([1, 1000, 754]) | 10\n",
-      "25588    | torch.Size([1, 1000, 754]) | 10\n",
-      "25589    | torch.Size([1, 1000, 762]) | 10\n",
-      "25590    | torch.Size([1, 1000, 754]) | 10\n",
-      "25591    | torch.Size([1, 1000, 784]) | 10\n",
-      "25592    | torch.Size([1, 1000, 754]) | 10\n",
-      "25593    | torch.Size([1, 1000, 762]) | 10\n",
-      "25594    | torch.Size([1, 1000, 754]) | 10\n",
-      "25595    | torch.Size([1, 1000, 762]) | 10\n",
-      "25596    | torch.Size([1, 1000, 754]) | 10\n",
-      "25597    | torch.Size([1, 1000, 762]) | 10\n",
-      "25598    | torch.Size([1, 1000, 754]) | 10\n",
-      "25599    | torch.Size([1, 1000, 762]) | 10\n",
-      "25600    | torch.Size([1, 1000, 762]) | 10\n",
-      "25601    | torch.Size([1, 1000, 775]) | 10\n",
-      "25602    | torch.Size([1, 1000, 754]) | 10\n",
-      "25603    | torch.Size([1, 1000, 762]) | 10\n",
-      "25604    | torch.Size([1, 1000, 797]) | 10\n",
-      "25605    | torch.Size([1, 1000, 754]) | 10\n",
-      "25606    | torch.Size([1, 1000, 773]) | 10\n",
-      "25607    | torch.Size([1, 1000, 815]) | 10\n",
-      "25608    | torch.Size([1, 1000, 754]) | 10\n",
-      "25609    | torch.Size([1, 1000, 762]) | 10\n",
-      "25610    | torch.Size([1, 1000, 754]) | 10\n",
-      "25611    | torch.Size([1, 1000, 754]) | 10\n",
-      "25612    | torch.Size([1, 1000, 780]) | 10\n",
-      "25613    | torch.Size([1, 1000, 754]) | 10\n",
-      "25614    | torch.Size([1, 1000, 777]) | 10\n",
-      "25615    | torch.Size([1, 1000, 754]) | 10\n",
-      "25616    | torch.Size([1, 1000, 762]) | 10\n",
-      "25617    | torch.Size([1, 1000, 754]) | 10\n",
-      "25618    | torch.Size([1, 1000, 777]) | 10\n",
-      "25619    | torch.Size([1, 1000, 849]) | 10\n",
-      "25620    | torch.Size([1, 1000, 754]) | 10\n",
-      "25621    | torch.Size([1, 1000, 754]) | 10\n",
-      "25622    | torch.Size([1, 1000, 754]) | 10\n",
-      "25623    | torch.Size([1, 1000, 802]) | 10\n",
-      "25624    | torch.Size([1, 1000, 754]) | 10\n",
-      "25625    | torch.Size([1, 1000, 847]) | 10\n",
-      "25626    | torch.Size([1, 1000, 754]) | 10\n",
-      "25627    | torch.Size([1, 1000, 777]) | 10\n",
-      "25628    | torch.Size([1, 1000, 754]) | 10\n",
-      "25629    | torch.Size([1, 1000, 762]) | 10\n",
-      "25630    | torch.Size([1, 1000, 786]) | 10\n",
-      "25631    | torch.Size([1, 1000, 762]) | 10\n",
-      "25632    | torch.Size([1, 1000, 777]) | 10\n",
-      "25633    | torch.Size([1, 1000, 754]) | 10\n",
-      "25634    | torch.Size([1, 1000, 754]) | 10\n",
-      "25635    | torch.Size([1, 1000, 762]) | 10\n",
-      "25636    | torch.Size([1, 1000, 754]) | 10\n",
-      "25637    | torch.Size([1, 1000, 728]) | 10\n",
-      "25638    | torch.Size([1, 1000, 762]) | 10\n",
-      "25639    | torch.Size([1, 1000, 762]) | 10\n",
-      "25640    | torch.Size([1, 1000, 754]) | 10\n",
-      "25641    | torch.Size([1, 1000, 781]) | 10\n",
-      "25642    | torch.Size([1, 1000, 754]) | 10\n",
-      "25643    | torch.Size([1, 1000, 754]) | 10\n",
-      "25644    | torch.Size([1, 1000, 754]) | 10\n",
-      "25645    | torch.Size([1, 1000, 762]) | 10\n",
-      "25646    | torch.Size([1, 1000, 762]) | 10\n",
-      "25647    | torch.Size([1, 1000, 754]) | 10\n",
-      "25648    | torch.Size([1, 1000, 754]) | 10\n",
-      "25649    | torch.Size([1, 1000, 762]) | 10\n",
-      "25650    | torch.Size([1, 1000, 777]) | 10\n",
-      "25651    | torch.Size([1, 1000, 754]) | 10\n",
-      "25652    | torch.Size([1, 1000, 754]) | 10\n",
-      "25653    | torch.Size([1, 1000, 777]) | 10\n",
-      "25654    | torch.Size([1, 1000, 779]) | 10\n",
-      "25655    | torch.Size([1, 1000, 777]) | 10\n",
-      "25656    | torch.Size([1, 1000, 754]) | 10\n",
-      "25657    | torch.Size([1, 1000, 754]) | 10\n",
-      "25658    | torch.Size([1, 1000, 754]) | 10\n",
-      "25659    | torch.Size([1, 1000, 754]) | 10\n",
-      "25660    | torch.Size([1, 1000, 802]) | 10\n",
-      "25661    | torch.Size([1, 1000, 777]) | 10\n",
-      "25662    | torch.Size([1, 1000, 788]) | 10\n",
-      "25663    | torch.Size([1, 1000, 754]) | 10\n",
-      "25664    | torch.Size([1, 1000, 803]) | 10\n",
-      "25665    | torch.Size([1, 1000, 762]) | 10\n",
-      "25666    | torch.Size([1, 1000, 754]) | 10\n",
-      "25667    | torch.Size([1, 1000, 847]) | 10\n",
-      "25668    | torch.Size([1, 1000, 754]) | 10\n",
-      "25669    | torch.Size([1, 1000, 754]) | 10\n",
-      "25670    | torch.Size([1, 1000, 807]) | 10\n",
-      "25671    | torch.Size([1, 1000, 865]) | 10\n",
-      "25672    | torch.Size([1, 1000, 754]) | 10\n",
-      "25673    | torch.Size([1, 1000, 762]) | 10\n",
-      "25674    | torch.Size([1, 1000, 771]) | 10\n",
-      "25675    | torch.Size([1, 1000, 762]) | 10\n",
-      "25676    | torch.Size([1, 1000, 754]) | 10\n",
-      "25677    | torch.Size([1, 1000, 754]) | 10\n",
-      "25678    | torch.Size([1, 1000, 754]) | 10\n",
-      "25679    | torch.Size([1, 1000, 754]) | 10\n",
-      "25680    | torch.Size([1, 1000, 754]) | 10\n",
-      "25681    | torch.Size([1, 1000, 754]) | 10\n",
-      "25682    | torch.Size([1, 1000, 777]) | 10\n",
-      "25683    | torch.Size([1, 1000, 762]) | 10\n",
-      "25684    | torch.Size([1, 1000, 762]) | 10\n",
-      "25685    | torch.Size([1, 1000, 754]) | 10\n",
-      "25686    | torch.Size([1, 1000, 754]) | 10\n",
-      "25687    | torch.Size([1, 1000, 777]) | 10\n",
-      "25688    | torch.Size([1, 1000, 842]) | 10\n",
-      "25689    | torch.Size([1, 1000, 754]) | 10\n",
-      "25690    | torch.Size([1, 1000, 762]) | 10\n",
-      "25691    | torch.Size([1, 1000, 754]) | 10\n",
-      "25692    | torch.Size([1, 1000, 762]) | 10\n",
-      "25693    | torch.Size([1, 1000, 754]) | 10\n",
-      "25694    | torch.Size([1, 1000, 762]) | 10\n",
-      "25695    | torch.Size([1, 1000, 833]) | 10\n",
-      "25696    | torch.Size([1, 1000, 719]) | 10\n",
-      "25697    | torch.Size([1, 1000, 715]) | 10\n",
-      "25698    | torch.Size([1, 1000, 777]) | 10\n",
-      "25699    | torch.Size([1, 1000, 754]) | 10\n",
-      "25700    | torch.Size([1, 1000, 802]) | 10\n",
-      "25701    | torch.Size([1, 1000, 762]) | 10\n",
-      "25702    | torch.Size([1, 1000, 754]) | 10\n",
-      "25703    | torch.Size([1, 1000, 754]) | 10\n",
-      "25704    | torch.Size([1, 1000, 797]) | 10\n",
-      "25705    | torch.Size([1, 1000, 762]) | 10\n",
-      "25706    | torch.Size([1, 1000, 611]) | 10\n",
-      "25707    | torch.Size([1, 1000, 762]) | 10\n",
-      "25708    | torch.Size([1, 1000, 777]) | 10\n",
-      "25709    | torch.Size([1, 1000, 762]) | 10\n",
-      "25710    | torch.Size([1, 1000, 762]) | 10\n",
-      "25711    | torch.Size([1, 1000, 789]) | 10\n",
-      "25712    | torch.Size([1, 1000, 776]) | 10\n",
-      "25713    | torch.Size([1, 1000, 754]) | 10\n",
-      "25714    | torch.Size([1, 1000, 785]) | 10\n",
-      "25715    | torch.Size([1, 1000, 810]) | 10\n",
-      "25716    | torch.Size([1, 1000, 763]) | 10\n",
-      "25717    | torch.Size([1, 1000, 780]) | 10\n",
-      "25718    | torch.Size([1, 1000, 762]) | 10\n",
-      "25719    | torch.Size([1, 1000, 754]) | 10\n",
-      "25720    | torch.Size([1, 1000, 762]) | 10\n",
-      "25721    | torch.Size([1, 1000, 754]) | 10\n",
-      "25722    | torch.Size([1, 1000, 754]) | 10\n",
-      "25723    | torch.Size([1, 1000, 754]) | 10\n",
-      "25724    | torch.Size([1, 1000, 762]) | 10\n",
-      "25725    | torch.Size([1, 1000, 754]) | 10\n",
-      "25726    | torch.Size([1, 1000, 754]) | 10\n",
-      "25727    | torch.Size([1, 1000, 777]) | 10\n",
-      "25728    | torch.Size([1, 1000, 754]) | 10\n",
-      "25729    | torch.Size([1, 1000, 762]) | 10\n",
-      "25730    | torch.Size([1, 1000, 754]) | 10\n",
-      "25731    | torch.Size([1, 1000, 777]) | 10\n",
-      "25732    | torch.Size([1, 1000, 754]) | 10\n",
-      "25733    | torch.Size([1, 1000, 754]) | 10\n",
-      "25734    | torch.Size([1, 1000, 754]) | 10\n",
-      "25735    | torch.Size([1, 1000, 754]) | 10\n",
-      "25736    | torch.Size([1, 1000, 754]) | 10\n",
-      "25737    | torch.Size([1, 1000, 754]) | 10\n",
-      "25738    | torch.Size([1, 1000, 777]) | 10\n",
-      "25739    | torch.Size([1, 1000, 754]) | 10\n",
-      "25740    | torch.Size([1, 1000, 754]) | 10\n",
-      "25741    | torch.Size([1, 1000, 754]) | 10\n",
-      "25742    | torch.Size([1, 1000, 777]) | 10\n",
-      "25743    | torch.Size([1, 1000, 781]) | 10\n",
-      "25744    | torch.Size([1, 1000, 762]) | 10\n",
-      "25745    | torch.Size([1, 1000, 779]) | 10\n",
-      "25746    | torch.Size([1, 1000, 754]) | 10\n",
-      "25747    | torch.Size([1, 1000, 777]) | 10\n",
-      "25748    | torch.Size([1, 1000, 754]) | 10\n",
-      "25749    | torch.Size([1, 1000, 762]) | 10\n",
-      "25750    | torch.Size([1, 1000, 754]) | 10\n",
-      "25751    | torch.Size([1, 1000, 781]) | 10\n",
-      "25752    | torch.Size([1, 1000, 754]) | 10\n",
-      "25753    | torch.Size([1, 1000, 754]) | 10\n",
-      "25754    | torch.Size([1, 1000, 722]) | 10\n",
-      "25755    | torch.Size([1, 1000, 754]) | 10\n",
-      "25756    | torch.Size([1, 1000, 757]) | 10\n",
-      "25757    | torch.Size([1, 1000, 754]) | 10\n",
-      "25758    | torch.Size([1, 1000, 777]) | 10\n",
-      "25759    | torch.Size([1, 1000, 754]) | 10\n",
-      "25760    | torch.Size([1, 1000, 768]) | 10\n",
-      "25761    | torch.Size([1, 1000, 777]) | 10\n",
-      "25762    | torch.Size([1, 1000, 754]) | 10\n",
-      "25763    | torch.Size([1, 1000, 754]) | 10\n",
-      "25764    | torch.Size([1, 1000, 777]) | 10\n",
-      "25765    | torch.Size([1, 1000, 754]) | 10\n",
-      "25766    | torch.Size([1, 1000, 754]) | 10\n",
-      "25767    | torch.Size([1, 1000, 794]) | 10\n",
-      "25768    | torch.Size([1, 1000, 802]) | 10\n",
-      "25769    | torch.Size([1, 1000, 777]) | 10\n",
-      "25770    | torch.Size([1, 1000, 762]) | 10\n",
-      "25771    | torch.Size([1, 1000, 762]) | 10\n",
-      "25772    | torch.Size([1, 1000, 813]) | 10\n",
-      "25773    | torch.Size([1, 1000, 754]) | 10\n",
-      "25774    | torch.Size([1, 1000, 754]) | 10\n",
-      "25775    | torch.Size([1, 1000, 754]) | 10\n",
-      "25776    | torch.Size([1, 1000, 754]) | 10\n",
-      "25777    | torch.Size([1, 1000, 802]) | 10\n",
-      "25778    | torch.Size([1, 1000, 762]) | 10\n",
-      "25779    | torch.Size([1, 1000, 762]) | 10\n",
-      "25780    | torch.Size([1, 1000, 785]) | 10\n",
-      "25781    | torch.Size([1, 1000, 782]) | 10\n",
-      "25782    | torch.Size([1, 1000, 762]) | 10\n",
-      "25783    | torch.Size([1, 1000, 777]) | 10\n",
-      "25784    | torch.Size([1, 1000, 777]) | 10\n",
-      "25785    | torch.Size([1, 1000, 774]) | 10\n",
-      "25786    | torch.Size([1, 1000, 754]) | 10\n",
-      "25787    | torch.Size([1, 1000, 762]) | 10\n",
-      "25788    | torch.Size([1, 1000, 777]) | 10\n",
-      "25789    | torch.Size([1, 1000, 754]) | 10\n",
-      "25790    | torch.Size([1, 1000, 777]) | 10\n",
-      "25791    | torch.Size([1, 1000, 754]) | 10\n",
-      "25792    | torch.Size([1, 1000, 754]) | 10\n",
-      "25793    | torch.Size([1, 1000, 754]) | 10\n",
-      "25794    | torch.Size([1, 1000, 754]) | 10\n",
-      "25795    | torch.Size([1, 1000, 754]) | 10\n",
-      "25796    | torch.Size([1, 1000, 762]) | 10\n",
-      "25797    | torch.Size([1, 1000, 802]) | 10\n",
-      "25798    | torch.Size([1, 1000, 754]) | 10\n",
-      "25799    | torch.Size([1, 1000, 778]) | 10\n",
-      "25800    | torch.Size([1, 1000, 754]) | 10\n",
-      "25801    | torch.Size([1, 1000, 754]) | 10\n",
-      "25802    | torch.Size([1, 1000, 754]) | 10\n",
-      "25803    | torch.Size([1, 1000, 754]) | 10\n",
-      "25804    | torch.Size([1, 1000, 781]) | 10\n",
-      "25805    | torch.Size([1, 1000, 762]) | 10\n",
-      "25806    | torch.Size([1, 1000, 754]) | 10\n",
-      "25807    | torch.Size([1, 1000, 802]) | 10\n",
-      "25808    | torch.Size([1, 1000, 754]) | 10\n",
-      "25809    | torch.Size([1, 1000, 779]) | 10\n",
-      "25810    | torch.Size([1, 1000, 754]) | 10\n",
-      "25811    | torch.Size([1, 1000, 611]) | 10\n",
-      "25812    | torch.Size([1, 1000, 754]) | 10\n",
-      "25813    | torch.Size([1, 1000, 754]) | 10\n",
-      "25814    | torch.Size([1, 1000, 754]) | 10\n",
-      "25815    | torch.Size([1, 1000, 783]) | 10\n",
-      "25816    | torch.Size([1, 1000, 762]) | 10\n",
-      "25817    | torch.Size([1, 1000, 754]) | 10\n",
-      "25818    | torch.Size([1, 1000, 754]) | 10\n",
-      "25819    | torch.Size([1, 1000, 777]) | 10\n",
-      "25820    | torch.Size([1, 1000, 777]) | 10\n",
-      "25821    | torch.Size([1, 1000, 754]) | 10\n",
-      "25822    | torch.Size([1, 1000, 754]) | 10\n",
-      "25823    | torch.Size([1, 1000, 777]) | 10\n",
-      "25824    | torch.Size([1, 1000, 802]) | 10\n",
-      "25825    | torch.Size([1, 1000, 802]) | 10\n",
-      "25826    | torch.Size([1, 1000, 754]) | 10\n",
-      "25827    | torch.Size([1, 1000, 754]) | 10\n",
-      "25828    | torch.Size([1, 1000, 762]) | 10\n",
-      "25829    | torch.Size([1, 1000, 754]) | 10\n",
-      "25830    | torch.Size([1, 1000, 810]) | 10\n",
-      "25831    | torch.Size([1, 1000, 754]) | 10\n",
-      "25832    | torch.Size([1, 1000, 782]) | 10\n",
-      "25833    | torch.Size([1, 1000, 754]) | 10\n",
-      "25834    | torch.Size([1, 1000, 777]) | 10\n",
-      "25835    | torch.Size([1, 1000, 777]) | 10\n",
-      "25836    | torch.Size([1, 1000, 754]) | 10\n",
-      "25837    | torch.Size([1, 1000, 754]) | 10\n",
-      "25838    | torch.Size([1, 1000, 754]) | 10\n",
-      "25839    | torch.Size([1, 1000, 754]) | 10\n",
-      "25840    | torch.Size([1, 1000, 754]) | 10\n",
-      "25841    | torch.Size([1, 1000, 769]) | 10\n",
-      "25842    | torch.Size([1, 1000, 777]) | 10\n",
-      "25843    | torch.Size([1, 1000, 754]) | 10\n",
-      "25844    | torch.Size([1, 1000, 754]) | 10\n",
-      "25845    | torch.Size([1, 1000, 777]) | 10\n",
-      "25846    | torch.Size([1, 1000, 754]) | 10\n",
-      "25847    | torch.Size([1, 1000, 754]) | 10\n",
-      "25848    | torch.Size([1, 1000, 840]) | 10\n",
-      "25849    | torch.Size([1, 1000, 754]) | 10\n",
-      "25850    | torch.Size([1, 1000, 754]) | 10\n",
-      "25851    | torch.Size([1, 1000, 754]) | 10\n",
-      "25852    | torch.Size([1, 1000, 796]) | 10\n",
-      "25853    | torch.Size([1, 1000, 777]) | 10\n",
-      "25854    | torch.Size([1, 1000, 723]) | 10\n",
-      "25855    | torch.Size([1, 1000, 754]) | 10\n",
-      "25856    | torch.Size([1, 1000, 782]) | 10\n",
-      "25857    | torch.Size([1, 1000, 747]) | 10\n",
-      "25858    | torch.Size([1, 1000, 769]) | 10\n",
-      "25859    | torch.Size([1, 1000, 793]) | 10\n",
-      "25860    | torch.Size([1, 1000, 754]) | 10\n",
-      "25861    | torch.Size([1, 1000, 775]) | 10\n",
-      "25862    | torch.Size([1, 1000, 786]) | 10\n",
-      "25863    | torch.Size([1, 1000, 777]) | 10\n",
-      "25864    | torch.Size([1, 1000, 777]) | 10\n",
-      "25865    | torch.Size([1, 1000, 754]) | 10\n",
-      "25866    | torch.Size([1, 1000, 804]) | 10\n",
-      "25867    | torch.Size([1, 1000, 754]) | 10\n",
-      "25868    | torch.Size([1, 1000, 777]) | 10\n",
-      "25869    | torch.Size([1, 1000, 795]) | 10\n",
-      "25870    | torch.Size([1, 1000, 754]) | 10\n",
-      "25871    | torch.Size([1, 1000, 754]) | 10\n",
-      "25872    | torch.Size([1, 1000, 777]) | 10\n",
-      "25873    | torch.Size([1, 1000, 754]) | 10\n",
-      "25874    | torch.Size([1, 1000, 777]) | 10\n",
-      "25875    | torch.Size([1, 1000, 847]) | 10\n",
-      "25876    | torch.Size([1, 1000, 754]) | 10\n",
-      "25877    | torch.Size([1, 1000, 754]) | 10\n",
-      "25878    | torch.Size([1, 1000, 827]) | 10\n",
-      "25879    | torch.Size([1, 1000, 781]) | 10\n",
-      "25880    | torch.Size([1, 1000, 754]) | 10\n",
-      "25881    | torch.Size([1, 1000, 762]) | 10\n",
-      "25882    | torch.Size([1, 1000, 754]) | 10\n",
-      "25883    | torch.Size([1, 1000, 777]) | 10\n",
-      "25884    | torch.Size([1, 1000, 802]) | 10\n",
-      "25885    | torch.Size([1, 1000, 762]) | 10\n",
-      "25886    | torch.Size([1, 1000, 754]) | 10\n",
-      "25887    | torch.Size([1, 1000, 777]) | 10\n",
-      "25888    | torch.Size([1, 1000, 754]) | 10\n",
-      "25889    | torch.Size([1, 1000, 754]) | 10\n",
-      "25890    | torch.Size([1, 1000, 767]) | 10\n",
-      "25891    | torch.Size([1, 1000, 754]) | 10\n",
-      "25892    | torch.Size([1, 1000, 754]) | 10\n",
-      "25893    | torch.Size([1, 1000, 793]) | 10\n",
-      "25894    | torch.Size([1, 1000, 802]) | 10\n",
-      "25895    | torch.Size([1, 1000, 754]) | 10\n",
-      "25896    | torch.Size([1, 1000, 762]) | 10\n",
-      "25897    | torch.Size([1, 1000, 765]) | 10\n",
-      "25898    | torch.Size([1, 1000, 754]) | 10\n",
-      "25899    | torch.Size([1, 1000, 754]) | 10\n",
-      "25900    | torch.Size([1, 1000, 781]) | 10\n",
-      "25901    | torch.Size([1, 1000, 754]) | 10\n",
-      "25902    | torch.Size([1, 1000, 754]) | 10\n",
-      "25903    | torch.Size([1, 1000, 762]) | 10\n",
-      "25904    | torch.Size([1, 1000, 754]) | 10\n",
-      "25905    | torch.Size([1, 1000, 754]) | 10\n",
-      "25906    | torch.Size([1, 1000, 754]) | 10\n",
-      "25907    | torch.Size([1, 1000, 754]) | 10\n",
-      "25908    | torch.Size([1, 1000, 802]) | 10\n",
-      "25909    | torch.Size([1, 1000, 754]) | 10\n",
-      "25910    | torch.Size([1, 1000, 777]) | 10\n",
-      "25911    | torch.Size([1, 1000, 738]) | 10\n",
-      "25912    | torch.Size([1, 1000, 754]) | 10\n",
-      "25913    | torch.Size([1, 1000, 754]) | 10\n",
-      "25914    | torch.Size([1, 1000, 754]) | 10\n",
-      "25915    | torch.Size([1, 1000, 754]) | 10\n",
-      "25916    | torch.Size([1, 1000, 754]) | 10\n",
-      "25917    | torch.Size([1, 1000, 802]) | 10\n",
-      "25918    | torch.Size([1, 1000, 754]) | 10\n",
-      "25919    | torch.Size([1, 1000, 754]) | 10\n",
-      "25920    | torch.Size([1, 1000, 802]) | 10\n",
-      "25921    | torch.Size([1, 1000, 777]) | 10\n",
-      "25922    | torch.Size([1, 1000, 754]) | 10\n",
-      "25923    | torch.Size([1, 1000, 802]) | 10\n",
-      "25924    | torch.Size([1, 1000, 768]) | 10\n",
-      "25925    | torch.Size([1, 1000, 754]) | 10\n",
-      "25926    | torch.Size([1, 1000, 754]) | 10\n",
-      "25927    | torch.Size([1, 1000, 782]) | 10\n",
-      "25928    | torch.Size([1, 1000, 754]) | 10\n",
-      "25929    | torch.Size([1, 1000, 802]) | 10\n",
-      "25930    | torch.Size([1, 1000, 762]) | 10\n",
-      "25931    | torch.Size([1, 1000, 754]) | 10\n",
-      "25932    | torch.Size([1, 1000, 845]) | 10\n",
-      "25933    | torch.Size([1, 1000, 774]) | 10\n",
-      "25934    | torch.Size([1, 1000, 754]) | 10\n",
-      "25935    | torch.Size([1, 1000, 754]) | 10\n",
-      "25936    | torch.Size([1, 1000, 754]) | 10\n",
-      "25937    | torch.Size([1, 1000, 754]) | 10\n",
-      "25938    | torch.Size([1, 1000, 754]) | 10\n",
-      "25939    | torch.Size([1, 1000, 802]) | 10\n",
-      "25940    | torch.Size([1, 1000, 762]) | 10\n",
-      "25941    | torch.Size([1, 1000, 778]) | 10\n",
-      "25942    | torch.Size([1, 1000, 754]) | 10\n",
-      "25943    | torch.Size([1, 1000, 754]) | 10\n",
-      "25944    | torch.Size([1, 1000, 754]) | 10\n",
-      "25945    | torch.Size([1, 1000, 754]) | 10\n",
-      "25946    | torch.Size([1, 1000, 713]) | 10\n",
-      "25947    | torch.Size([1, 1000, 754]) | 10\n",
-      "25948    | torch.Size([1, 1000, 762]) | 10\n",
-      "25949    | torch.Size([1, 1000, 754]) | 10\n",
-      "25950    | torch.Size([1, 1000, 762]) | 10\n",
-      "25951    | torch.Size([1, 1000, 754]) | 10\n",
-      "25952    | torch.Size([1, 1000, 762]) | 10\n",
-      "25953    | torch.Size([1, 1000, 754]) | 10\n",
-      "25954    | torch.Size([1, 1000, 762]) | 10\n",
-      "25955    | torch.Size([1, 1000, 754]) | 10\n",
-      "25956    | torch.Size([1, 1000, 786]) | 10\n",
-      "25957    | torch.Size([1, 1000, 754]) | 10\n",
-      "25958    | torch.Size([1, 1000, 754]) | 10\n",
-      "25959    | torch.Size([1, 1000, 754]) | 10\n",
-      "25960    | torch.Size([1, 1000, 754]) | 10\n",
-      "25961    | torch.Size([1, 1000, 754]) | 10\n",
-      "25962    | torch.Size([1, 1000, 754]) | 10\n",
-      "25963    | torch.Size([1, 1000, 754]) | 10\n",
-      "25964    | torch.Size([1, 1000, 776]) | 10\n",
-      "25965    | torch.Size([1, 1000, 754]) | 10\n",
-      "25966    | torch.Size([1, 1000, 754]) | 10\n",
-      "25967    | torch.Size([1, 1000, 754]) | 10\n",
-      "25968    | torch.Size([1, 1000, 799]) | 10\n",
-      "25969    | torch.Size([1, 1000, 754]) | 10\n",
-      "25970    | torch.Size([1, 1000, 785]) | 10\n",
-      "25971    | torch.Size([1, 1000, 754]) | 10\n",
-      "25972    | torch.Size([1, 1000, 777]) | 10\n",
-      "25973    | torch.Size([1, 1000, 776]) | 10\n",
-      "25974    | torch.Size([1, 1000, 762]) | 10\n",
-      "25975    | torch.Size([1, 1000, 754]) | 10\n",
-      "25976    | torch.Size([1, 1000, 754]) | 10\n",
-      "25977    | torch.Size([1, 1000, 777]) | 10\n",
-      "25978    | torch.Size([1, 1000, 798]) | 10\n",
-      "25979    | torch.Size([1, 1000, 754]) | 10\n",
-      "25980    | torch.Size([1, 1000, 802]) | 10\n",
-      "25981    | torch.Size([1, 1000, 754]) | 10\n",
-      "25982    | torch.Size([1, 1000, 754]) | 10\n",
-      "25983    | torch.Size([1, 1000, 762]) | 10\n",
-      "25984    | torch.Size([1, 1000, 762]) | 10\n",
-      "25985    | torch.Size([1, 1000, 777]) | 10\n",
-      "25986    | torch.Size([1, 1000, 754]) | 10\n",
-      "25987    | torch.Size([1, 1000, 754]) | 10\n",
-      "25988    | torch.Size([1, 1000, 754]) | 10\n",
-      "25989    | torch.Size([1, 1000, 754]) | 10\n",
-      "25990    | torch.Size([1, 1000, 769]) | 10\n",
-      "25991    | torch.Size([1, 1000, 754]) | 10\n",
-      "25992    | torch.Size([1, 1000, 777]) | 10\n",
-      "25993    | torch.Size([1, 1000, 777]) | 10\n",
-      "25994    | torch.Size([1, 1000, 762]) | 10\n",
-      "25995    | torch.Size([1, 1000, 781]) | 10\n",
-      "25996    | torch.Size([1, 1000, 776]) | 10\n",
-      "25997    | torch.Size([1, 1000, 754]) | 10\n",
-      "25998    | torch.Size([1, 1000, 754]) | 10\n",
-      "25999    | torch.Size([1, 1000, 713]) | 10\n",
-      "26000    | torch.Size([1, 1000, 777]) | 10\n",
-      "26001    | torch.Size([1, 1000, 777]) | 10\n",
-      "26002    | torch.Size([1, 1000, 762]) | 10\n",
-      "26003    | torch.Size([1, 1000, 780]) | 10\n",
-      "26004    | torch.Size([1, 1000, 754]) | 10\n",
-      "26005    | torch.Size([1, 1000, 754]) | 10\n",
-      "26006    | torch.Size([1, 1000, 754]) | 10\n",
-      "26007    | torch.Size([1, 1000, 762]) | 10\n",
-      "26008    | torch.Size([1, 1000, 754]) | 10\n",
-      "26009    | torch.Size([1, 1000, 785]) | 10\n",
-      "26010    | torch.Size([1, 1000, 802]) | 10\n",
-      "26011    | torch.Size([1, 1000, 773]) | 10\n",
-      "26012    | torch.Size([1, 1000, 754]) | 10\n",
-      "26013    | torch.Size([1, 1000, 754]) | 10\n",
-      "26014    | torch.Size([1, 1000, 754]) | 10\n",
-      "26015    | torch.Size([1, 1000, 768]) | 10\n",
-      "26016    | torch.Size([1, 1000, 754]) | 10\n",
-      "26017    | torch.Size([1, 1000, 786]) | 10\n",
-      "26018    | torch.Size([1, 1000, 775]) | 10\n",
-      "26019    | torch.Size([1, 1000, 777]) | 10\n",
-      "26020    | torch.Size([1, 1000, 762]) | 10\n",
-      "26021    | torch.Size([1, 1000, 788]) | 10\n",
-      "26022    | torch.Size([1, 1000, 777]) | 10\n",
-      "26023    | torch.Size([1, 1000, 788]) | 10\n",
-      "26024    | torch.Size([1, 1000, 754]) | 10\n",
-      "26025    | torch.Size([1, 1000, 762]) | 10\n",
-      "26026    | torch.Size([1, 1000, 754]) | 10\n",
-      "26027    | torch.Size([1, 1000, 754]) | 10\n",
-      "26028    | torch.Size([1, 1000, 754]) | 10\n",
-      "26029    | torch.Size([1, 1000, 754]) | 10\n",
-      "26030    | torch.Size([1, 1000, 762]) | 10\n",
-      "26031    | torch.Size([1, 1000, 766]) | 10\n",
-      "26032    | torch.Size([1, 1000, 754]) | 10\n",
-      "26033    | torch.Size([1, 1000, 754]) | 10\n",
-      "26034    | torch.Size([1, 1000, 764]) | 10\n",
-      "26035    | torch.Size([1, 1000, 754]) | 10\n",
-      "26036    | torch.Size([1, 1000, 754]) | 10\n",
-      "26037    | torch.Size([1, 1000, 777]) | 10\n",
-      "26038    | torch.Size([1, 1000, 777]) | 10\n",
-      "26039    | torch.Size([1, 1000, 777]) | 10\n",
-      "26040    | torch.Size([1, 1000, 762]) | 10\n",
-      "26041    | torch.Size([1, 1000, 754]) | 10\n",
-      "26042    | torch.Size([1, 1000, 802]) | 10\n",
-      "26043    | torch.Size([1, 1000, 754]) | 10\n",
-      "26044    | torch.Size([1, 1000, 777]) | 10\n",
-      "26045    | torch.Size([1, 1000, 778]) | 10\n",
-      "26046    | torch.Size([1, 1000, 754]) | 10\n",
-      "26047    | torch.Size([1, 1000, 802]) | 10\n",
-      "26048    | torch.Size([1, 1000, 754]) | 10\n",
-      "26049    | torch.Size([1, 1000, 792]) | 10\n",
-      "26050    | torch.Size([1, 1000, 782]) | 10\n",
-      "26051    | torch.Size([1, 1000, 809]) | 10\n",
-      "26052    | torch.Size([1, 1000, 762]) | 10\n",
-      "26053    | torch.Size([1, 1000, 754]) | 10\n",
-      "26054    | torch.Size([1, 1000, 754]) | 10\n",
-      "26055    | torch.Size([1, 1000, 790]) | 10\n",
-      "26056    | torch.Size([1, 1000, 754]) | 10\n",
-      "26057    | torch.Size([1, 1000, 754]) | 10\n",
-      "26058    | torch.Size([1, 1000, 802]) | 10\n",
-      "26059    | torch.Size([1, 1000, 763]) | 10\n",
-      "26060    | torch.Size([1, 1000, 754]) | 10\n",
-      "26061    | torch.Size([1, 1000, 764]) | 10\n",
-      "26062    | torch.Size([1, 1000, 754]) | 10\n",
-      "26063    | torch.Size([1, 1000, 773]) | 10\n",
-      "26064    | torch.Size([1, 1000, 754]) | 10\n",
-      "26065    | torch.Size([1, 1000, 762]) | 10\n",
-      "26066    | torch.Size([1, 1000, 754]) | 10\n",
-      "26067    | torch.Size([1, 1000, 762]) | 10\n",
-      "26068    | torch.Size([1, 1000, 754]) | 10\n",
-      "26069    | torch.Size([1, 1000, 754]) | 10\n",
-      "26070    | torch.Size([1, 1000, 754]) | 10\n",
-      "26071    | torch.Size([1, 1000, 754]) | 10\n",
-      "26072    | torch.Size([1, 1000, 777]) | 10\n",
-      "26073    | torch.Size([1, 1000, 754]) | 10\n",
-      "26074    | torch.Size([1, 1000, 784]) | 10\n",
-      "26075    | torch.Size([1, 1000, 792]) | 10\n",
-      "26076    | torch.Size([1, 1000, 802]) | 10\n",
-      "26077    | torch.Size([1, 1000, 775]) | 10\n",
-      "26078    | torch.Size([1, 1000, 754]) | 10\n",
-      "26079    | torch.Size([1, 1000, 754]) | 10\n",
-      "26080    | torch.Size([1, 1000, 762]) | 10\n",
-      "26081    | torch.Size([1, 1000, 777]) | 10\n",
-      "26082    | torch.Size([1, 1000, 754]) | 10\n",
-      "26083    | torch.Size([1, 1000, 802]) | 10\n",
-      "26084    | torch.Size([1, 1000, 754]) | 10\n",
-      "26085    | torch.Size([1, 1000, 754]) | 10\n",
-      "26086    | torch.Size([1, 1000, 777]) | 10\n",
-      "26087    | torch.Size([1, 1000, 789]) | 10\n",
-      "26088    | torch.Size([1, 1000, 777]) | 10\n",
-      "26089    | torch.Size([1, 1000, 754]) | 10\n",
-      "26090    | torch.Size([1, 1000, 777]) | 10\n",
-      "26091    | torch.Size([1, 1000, 777]) | 10\n",
-      "26092    | torch.Size([1, 1000, 754]) | 10\n",
-      "26093    | torch.Size([1, 1000, 754]) | 10\n",
-      "26094    | torch.Size([1, 1000, 754]) | 10\n",
-      "26095    | torch.Size([1, 1000, 762]) | 10\n",
-      "26096    | torch.Size([1, 1000, 777]) | 10\n",
-      "26097    | torch.Size([1, 1000, 777]) | 10\n",
-      "26098    | torch.Size([1, 1000, 762]) | 10\n",
-      "26099    | torch.Size([1, 1000, 754]) | 10\n",
-      "26100    | torch.Size([1, 1000, 754]) | 10\n",
-      "26101    | torch.Size([1, 1000, 754]) | 10\n",
-      "26102    | torch.Size([1, 1000, 762]) | 10\n",
-      "26103    | torch.Size([1, 1000, 754]) | 10\n",
-      "26104    | torch.Size([1, 1000, 802]) | 10\n",
-      "26105    | torch.Size([1, 1000, 754]) | 10\n",
-      "26106    | torch.Size([1, 1000, 796]) | 10\n",
-      "26107    | torch.Size([1, 1000, 784]) | 10\n",
-      "26108    | torch.Size([1, 1000, 777]) | 10\n",
-      "26109    | torch.Size([1, 1000, 784]) | 10\n",
-      "26110    | torch.Size([1, 1000, 777]) | 10\n",
-      "26111    | torch.Size([1, 1000, 754]) | 10\n",
-      "26112    | torch.Size([1, 1000, 762]) | 10\n",
-      "26113    | torch.Size([1, 1000, 762]) | 10\n",
-      "26114    | torch.Size([1, 1000, 802]) | 10\n",
-      "26115    | torch.Size([1, 1000, 777]) | 10\n",
-      "26116    | torch.Size([1, 1000, 754]) | 10\n",
-      "26117    | torch.Size([1, 1000, 754]) | 10\n",
-      "26118    | torch.Size([1, 1000, 792]) | 10\n",
-      "26119    | torch.Size([1, 1000, 754]) | 10\n",
-      "26120    | torch.Size([1, 1000, 754]) | 10\n",
-      "26121    | torch.Size([1, 1000, 762]) | 10\n",
-      "26122    | torch.Size([1, 1000, 754]) | 10\n",
-      "26123    | torch.Size([1, 1000, 762]) | 10\n",
-      "26124    | torch.Size([1, 1000, 777]) | 10\n",
-      "26125    | torch.Size([1, 1000, 802]) | 10\n",
-      "26126    | torch.Size([1, 1000, 754]) | 10\n",
-      "26127    | torch.Size([1, 1000, 754]) | 10\n",
-      "26128    | torch.Size([1, 1000, 754]) | 10\n",
-      "26129    | torch.Size([1, 1000, 781]) | 10\n",
-      "26130    | torch.Size([1, 1000, 754]) | 10\n",
-      "26131    | torch.Size([1, 1000, 754]) | 10\n",
-      "26132    | torch.Size([1, 1000, 754]) | 10\n",
-      "26133    | torch.Size([1, 1000, 754]) | 10\n",
-      "26134    | torch.Size([1, 1000, 780]) | 10\n",
-      "26135    | torch.Size([1, 1000, 777]) | 10\n",
-      "26136    | torch.Size([1, 1000, 835]) | 10\n",
-      "26137    | torch.Size([1, 1000, 762]) | 10\n",
-      "26138    | torch.Size([1, 1000, 754]) | 10\n",
-      "26139    | torch.Size([1, 1000, 754]) | 10\n",
-      "26140    | torch.Size([1, 1000, 754]) | 10\n",
-      "26141    | torch.Size([1, 1000, 762]) | 10\n",
-      "26142    | torch.Size([1, 1000, 754]) | 10\n",
-      "26143    | torch.Size([1, 1000, 754]) | 10\n",
-      "26144    | torch.Size([1, 1000, 754]) | 10\n",
-      "26145    | torch.Size([1, 1000, 754]) | 10\n",
-      "26146    | torch.Size([1, 1000, 771]) | 10\n",
-      "26147    | torch.Size([1, 1000, 754]) | 10\n",
-      "26148    | torch.Size([1, 1000, 762]) | 10\n",
-      "26149    | torch.Size([1, 1000, 754]) | 10\n",
-      "26150    | torch.Size([1, 1000, 802]) | 10\n",
-      "26151    | torch.Size([1, 1000, 754]) | 10\n",
-      "26152    | torch.Size([1, 1000, 802]) | 10\n",
-      "26153    | torch.Size([1, 1000, 781]) | 10\n",
-      "26154    | torch.Size([1, 1000, 762]) | 10\n",
-      "26155    | torch.Size([1, 1000, 754]) | 10\n",
-      "26156    | torch.Size([1, 1000, 754]) | 10\n",
-      "26157    | torch.Size([1, 1000, 754]) | 10\n",
-      "26158    | torch.Size([1, 1000, 754]) | 10\n",
-      "26159    | torch.Size([1, 1000, 754]) | 10\n",
-      "26160    | torch.Size([1, 1000, 754]) | 10\n",
-      "26161    | torch.Size([1, 1000, 846]) | 10\n",
-      "26162    | torch.Size([1, 1000, 762]) | 10\n",
-      "26163    | torch.Size([1, 1000, 777]) | 10\n",
-      "26164    | torch.Size([1, 1000, 754]) | 10\n",
-      "26165    | torch.Size([1, 1000, 777]) | 10\n",
-      "26166    | torch.Size([1, 1000, 754]) | 10\n",
-      "26167    | torch.Size([1, 1000, 754]) | 10\n",
-      "26168    | torch.Size([1, 1000, 754]) | 10\n",
-      "26169    | torch.Size([1, 1000, 754]) | 10\n",
-      "26170    | torch.Size([1, 1000, 777]) | 10\n",
-      "26171    | torch.Size([1, 1000, 754]) | 10\n",
-      "26172    | torch.Size([1, 1000, 754]) | 10\n",
-      "26173    | torch.Size([1, 1000, 754]) | 10\n",
-      "26174    | torch.Size([1, 1000, 754]) | 10\n",
-      "26175    | torch.Size([1, 1000, 754]) | 10\n",
-      "26176    | torch.Size([1, 1000, 767]) | 10\n",
-      "26177    | torch.Size([1, 1000, 754]) | 10\n",
-      "26178    | torch.Size([1, 1000, 754]) | 10\n",
-      "26179    | torch.Size([1, 1000, 754]) | 10\n",
-      "26180    | torch.Size([1, 1000, 809]) | 10\n",
-      "26181    | torch.Size([1, 1000, 777]) | 10\n",
-      "26182    | torch.Size([1, 1000, 754]) | 10\n",
-      "26183    | torch.Size([1, 1000, 754]) | 10\n",
-      "26184    | torch.Size([1, 1000, 754]) | 10\n",
-      "26185    | torch.Size([1, 1000, 762]) | 10\n",
-      "26186    | torch.Size([1, 1000, 754]) | 10\n",
-      "26187    | torch.Size([1, 1000, 762]) | 10\n",
-      "26188    | torch.Size([1, 1000, 754]) | 10\n",
-      "26189    | torch.Size([1, 1000, 754]) | 10\n",
-      "26190    | torch.Size([1, 1000, 754]) | 10\n",
-      "26191    | torch.Size([1, 1000, 754]) | 10\n",
-      "26192    | torch.Size([1, 1000, 754]) | 10\n",
-      "26193    | torch.Size([1, 1000, 762]) | 10\n",
-      "26194    | torch.Size([1, 1000, 754]) | 10\n",
-      "26195    | torch.Size([1, 1000, 762]) | 10\n",
-      "26196    | torch.Size([1, 1000, 818]) | 10\n",
-      "26197    | torch.Size([1, 1000, 802]) | 10\n",
-      "26198    | torch.Size([1, 1000, 777]) | 10\n",
-      "26199    | torch.Size([1, 1000, 778]) | 10\n",
-      "26200    | torch.Size([1, 1000, 766]) | 10\n",
-      "26201    | torch.Size([1, 1000, 720]) | 10\n",
-      "26202    | torch.Size([1, 1000, 807]) | 10\n",
-      "26203    | torch.Size([1, 1000, 754]) | 10\n",
-      "26204    | torch.Size([1, 1000, 754]) | 10\n",
-      "26205    | torch.Size([1, 1000, 762]) | 10\n",
-      "26206    | torch.Size([1, 1000, 777]) | 10\n",
-      "26207    | torch.Size([1, 1000, 754]) | 10\n",
-      "26208    | torch.Size([1, 1000, 754]) | 10\n",
-      "26209    | torch.Size([1, 1000, 754]) | 10\n",
-      "26210    | torch.Size([1, 1000, 754]) | 10\n",
-      "26211    | torch.Size([1, 1000, 754]) | 10\n",
-      "26212    | torch.Size([1, 1000, 754]) | 10\n",
-      "26213    | torch.Size([1, 1000, 754]) | 10\n",
-      "26214    | torch.Size([1, 1000, 845]) | 10\n",
-      "26215    | torch.Size([1, 1000, 754]) | 10\n",
-      "26216    | torch.Size([1, 1000, 754]) | 10\n",
-      "26217    | torch.Size([1, 1000, 777]) | 10\n",
-      "26218    | torch.Size([1, 1000, 754]) | 10\n",
-      "26219    | torch.Size([1, 1000, 754]) | 10\n",
-      "26220    | torch.Size([1, 1000, 754]) | 10\n",
-      "26221    | torch.Size([1, 1000, 777]) | 10\n",
-      "26222    | torch.Size([1, 1000, 754]) | 10\n",
-      "26223    | torch.Size([1, 1000, 777]) | 10\n",
-      "26224    | torch.Size([1, 1000, 754]) | 10\n",
-      "26225    | torch.Size([1, 1000, 785]) | 10\n",
-      "26226    | torch.Size([1, 1000, 754]) | 10\n",
-      "26227    | torch.Size([1, 1000, 762]) | 10\n",
-      "26228    | torch.Size([1, 1000, 754]) | 10\n",
-      "26229    | torch.Size([1, 1000, 777]) | 10\n",
-      "26230    | torch.Size([1, 1000, 777]) | 10\n",
-      "26231    | torch.Size([1, 1000, 754]) | 10\n",
-      "26232    | torch.Size([1, 1000, 776]) | 10\n",
-      "26233    | torch.Size([1, 1000, 777]) | 10\n",
-      "26234    | torch.Size([1, 1000, 815]) | 10\n",
-      "26235    | torch.Size([1, 1000, 777]) | 10\n",
-      "26236    | torch.Size([1, 1000, 754]) | 10\n",
-      "26237    | torch.Size([1, 1000, 778]) | 10\n",
-      "26238    | torch.Size([1, 1000, 754]) | 10\n",
-      "26239    | torch.Size([1, 1000, 783]) | 10\n",
-      "26240    | torch.Size([1, 1000, 864]) | 10\n",
-      "26241    | torch.Size([1, 1000, 786]) | 10\n",
-      "26242    | torch.Size([1, 1000, 802]) | 10\n",
-      "26243    | torch.Size([1, 1000, 754]) | 10\n",
-      "26244    | torch.Size([1, 1000, 754]) | 10\n",
-      "26245    | torch.Size([1, 1000, 754]) | 10\n",
-      "26246    | torch.Size([1, 1000, 678]) | 10\n",
-      "26247    | torch.Size([1, 1000, 762]) | 10\n",
-      "26248    | torch.Size([1, 1000, 777]) | 10\n",
-      "26249    | torch.Size([1, 1000, 754]) | 10\n",
-      "26250    | torch.Size([1, 1000, 777]) | 10\n",
-      "26251    | torch.Size([1, 1000, 754]) | 10\n",
-      "26252    | torch.Size([1, 1000, 754]) | 10\n",
-      "26253    | torch.Size([1, 1000, 846]) | 10\n",
-      "26254    | torch.Size([1, 1000, 772]) | 10\n",
-      "26255    | torch.Size([1, 1000, 762]) | 10\n",
-      "26256    | torch.Size([1, 1000, 777]) | 10\n",
-      "26257    | torch.Size([1, 1000, 754]) | 10\n",
-      "26258    | torch.Size([1, 1000, 777]) | 10\n",
-      "26259    | torch.Size([1, 1000, 754]) | 10\n",
-      "26260    | torch.Size([1, 1000, 754]) | 10\n",
-      "26261    | torch.Size([1, 1000, 762]) | 10\n",
-      "26262    | torch.Size([1, 1000, 754]) | 10\n",
-      "26263    | torch.Size([1, 1000, 754]) | 10\n",
-      "26264    | torch.Size([1, 1000, 814]) | 10\n",
-      "26265    | torch.Size([1, 1000, 754]) | 10\n",
-      "26266    | torch.Size([1, 1000, 754]) | 10\n",
-      "26267    | torch.Size([1, 1000, 754]) | 10\n",
-      "26268    | torch.Size([1, 1000, 754]) | 10\n",
-      "26269    | torch.Size([1, 1000, 776]) | 10\n",
-      "26270    | torch.Size([1, 1000, 802]) | 10\n",
-      "26271    | torch.Size([1, 1000, 777]) | 10\n",
-      "26272    | torch.Size([1, 1000, 754]) | 10\n",
-      "26273    | torch.Size([1, 1000, 762]) | 10\n",
-      "26274    | torch.Size([1, 1000, 754]) | 10\n",
-      "26275    | torch.Size([1, 1000, 754]) | 10\n",
-      "26276    | torch.Size([1, 1000, 754]) | 10\n",
-      "26277    | torch.Size([1, 1000, 777]) | 10\n",
-      "26278    | torch.Size([1, 1000, 843]) | 10\n",
-      "26279    | torch.Size([1, 1000, 754]) | 10\n",
-      "26280    | torch.Size([1, 1000, 785]) | 10\n",
-      "26281    | torch.Size([1, 1000, 754]) | 10\n",
-      "26282    | torch.Size([1, 1000, 762]) | 10\n",
-      "26283    | torch.Size([1, 1000, 754]) | 10\n",
-      "26284    | torch.Size([1, 1000, 766]) | 10\n",
-      "26285    | torch.Size([1, 1000, 768]) | 10\n",
-      "26286    | torch.Size([1, 1000, 762]) | 10\n",
-      "26287    | torch.Size([1, 1000, 778]) | 10\n",
-      "26288    | torch.Size([1, 1000, 793]) | 10\n",
-      "26289    | torch.Size([1, 1000, 777]) | 10\n",
-      "26290    | torch.Size([1, 1000, 754]) | 10\n",
-      "26291    | torch.Size([1, 1000, 754]) | 10\n",
-      "26292    | torch.Size([1, 1000, 762]) | 10\n",
-      "26293    | torch.Size([1, 1000, 754]) | 10\n",
-      "26294    | torch.Size([1, 1000, 754]) | 10\n",
-      "26295    | torch.Size([1, 1000, 785]) | 10\n",
-      "26296    | torch.Size([1, 1000, 762]) | 10\n",
-      "26297    | torch.Size([1, 1000, 797]) | 10\n",
-      "26298    | torch.Size([1, 1000, 785]) | 10\n",
-      "26299    | torch.Size([1, 1000, 754]) | 10\n",
-      "26300    | torch.Size([1, 1000, 777]) | 10\n",
-      "26301    | torch.Size([1, 1000, 754]) | 10\n",
-      "26302    | torch.Size([1, 1000, 780]) | 10\n",
-      "26303    | torch.Size([1, 1000, 754]) | 10\n",
-      "26304    | torch.Size([1, 1000, 754]) | 10\n",
-      "26305    | torch.Size([1, 1000, 754]) | 10\n",
-      "26306    | torch.Size([1, 1000, 754]) | 10\n",
-      "26307    | torch.Size([1, 1000, 754]) | 10\n",
-      "26308    | torch.Size([1, 1000, 730]) | 10\n",
-      "26309    | torch.Size([1, 1000, 799]) | 10\n",
-      "26310    | torch.Size([1, 1000, 754]) | 10\n",
-      "26311    | torch.Size([1, 1000, 754]) | 10\n",
-      "26312    | torch.Size([1, 1000, 781]) | 10\n",
-      "26313    | torch.Size([1, 1000, 769]) | 10\n",
-      "26314    | torch.Size([1, 1000, 754]) | 10\n",
-      "26315    | torch.Size([1, 1000, 754]) | 10\n",
-      "26316    | torch.Size([1, 1000, 754]) | 10\n",
-      "26317    | torch.Size([1, 1000, 777]) | 10\n",
-      "26318    | torch.Size([1, 1000, 762]) | 10\n",
-      "26319    | torch.Size([1, 1000, 754]) | 10\n",
-      "26320    | torch.Size([1, 1000, 783]) | 10\n",
-      "26321    | torch.Size([1, 1000, 762]) | 10\n",
-      "26322    | torch.Size([1, 1000, 802]) | 10\n",
-      "26323    | torch.Size([1, 1000, 762]) | 10\n",
-      "26324    | torch.Size([1, 1000, 792]) | 10\n",
-      "26325    | torch.Size([1, 1000, 754]) | 10\n",
-      "26326    | torch.Size([1, 1000, 795]) | 10\n",
-      "26327    | torch.Size([1, 1000, 777]) | 10\n",
-      "26328    | torch.Size([1, 1000, 777]) | 10\n",
-      "26329    | torch.Size([1, 1000, 762]) | 10\n",
-      "26330    | torch.Size([1, 1000, 808]) | 10\n",
-      "26331    | torch.Size([1, 1000, 754]) | 10\n",
-      "26332    | torch.Size([1, 1000, 802]) | 10\n",
-      "26333    | torch.Size([1, 1000, 796]) | 10\n",
-      "26334    | torch.Size([1, 1000, 754]) | 10\n",
-      "26335    | torch.Size([1, 1000, 754]) | 10\n",
-      "26336    | torch.Size([1, 1000, 754]) | 10\n",
-      "26337    | torch.Size([1, 1000, 754]) | 10\n",
-      "26338    | torch.Size([1, 1000, 796]) | 10\n",
-      "26339    | torch.Size([1, 1000, 780]) | 10\n",
-      "26340    | torch.Size([1, 1000, 773]) | 10\n",
-      "26341    | torch.Size([1, 1000, 754]) | 10\n",
-      "26342    | torch.Size([1, 1000, 762]) | 10\n",
-      "26343    | torch.Size([1, 1000, 754]) | 10\n",
-      "26344    | torch.Size([1, 1000, 778]) | 10\n",
-      "26345    | torch.Size([1, 1000, 754]) | 10\n",
-      "26346    | torch.Size([1, 1000, 754]) | 10\n",
-      "26347    | torch.Size([1, 1000, 762]) | 10\n",
-      "26348    | torch.Size([1, 1000, 754]) | 10\n",
-      "26349    | torch.Size([1, 1000, 754]) | 10\n",
-      "26350    | torch.Size([1, 1000, 804]) | 10\n",
-      "26351    | torch.Size([1, 1000, 802]) | 10\n",
-      "26352    | torch.Size([1, 1000, 773]) | 10\n",
-      "26353    | torch.Size([1, 1000, 762]) | 10\n",
-      "26354    | torch.Size([1, 1000, 842]) | 10\n",
-      "26355    | torch.Size([1, 1000, 777]) | 10\n",
-      "26356    | torch.Size([1, 1000, 762]) | 10\n",
-      "26357    | torch.Size([1, 1000, 771]) | 10\n",
-      "26358    | torch.Size([1, 1000, 763]) | 10\n",
-      "26359    | torch.Size([1, 1000, 754]) | 10\n",
-      "26360    | torch.Size([1, 1000, 754]) | 10\n",
-      "26361    | torch.Size([1, 1000, 754]) | 10\n",
-      "26362    | torch.Size([1, 1000, 762]) | 10\n",
-      "26363    | torch.Size([1, 1000, 777]) | 10\n",
-      "26364    | torch.Size([1, 1000, 754]) | 10\n",
-      "26365    | torch.Size([1, 1000, 848]) | 10\n",
-      "26366    | torch.Size([1, 1000, 777]) | 10\n",
-      "26367    | torch.Size([1, 1000, 762]) | 10\n",
-      "26368    | torch.Size([1, 1000, 754]) | 10\n",
-      "26369    | torch.Size([1, 1000, 777]) | 10\n",
-      "26370    | torch.Size([1, 1000, 754]) | 10\n",
-      "26371    | torch.Size([1, 1000, 767]) | 10\n",
-      "26372    | torch.Size([1, 1000, 754]) | 10\n",
-      "26373    | torch.Size([1, 1000, 754]) | 10\n",
-      "26374    | torch.Size([1, 1000, 775]) | 10\n",
-      "26375    | torch.Size([1, 1000, 777]) | 10\n",
-      "26376    | torch.Size([1, 1000, 754]) | 10\n",
-      "26377    | torch.Size([1, 1000, 754]) | 10\n",
-      "26378    | torch.Size([1, 1000, 774]) | 10\n",
-      "26379    | torch.Size([1, 1000, 762]) | 10\n",
-      "26380    | torch.Size([1, 1000, 754]) | 10\n",
-      "26381    | torch.Size([1, 1000, 777]) | 10\n",
-      "26382    | torch.Size([1, 1000, 754]) | 10\n",
-      "26383    | torch.Size([1, 1000, 754]) | 10\n",
-      "26384    | torch.Size([1, 1000, 754]) | 10\n",
-      "26385    | torch.Size([1, 1000, 754]) | 10\n",
-      "26386    | torch.Size([1, 1000, 802]) | 10\n",
-      "26387    | torch.Size([1, 1000, 802]) | 10\n",
-      "26388    | torch.Size([1, 1000, 797]) | 10\n",
-      "26389    | torch.Size([1, 1000, 754]) | 10\n",
-      "26390    | torch.Size([1, 1000, 762]) | 10\n",
-      "26391    | torch.Size([1, 1000, 777]) | 10\n",
-      "26392    | torch.Size([1, 1000, 611]) | 10\n",
-      "26393    | torch.Size([1, 1000, 777]) | 10\n",
-      "26394    | torch.Size([1, 1000, 777]) | 10\n",
-      "26395    | torch.Size([1, 1000, 754]) | 10\n",
-      "26396    | torch.Size([1, 1000, 802]) | 10\n",
-      "26397    | torch.Size([1, 1000, 802]) | 10\n",
-      "26398    | torch.Size([1, 1000, 768]) | 10\n",
-      "26399    | torch.Size([1, 1000, 777]) | 10\n",
-      "26400    | torch.Size([1, 1000, 777]) | 10\n",
-      "26401    | torch.Size([1, 1000, 754]) | 10\n",
-      "26402    | torch.Size([1, 1000, 772]) | 10\n",
-      "26403    | torch.Size([1, 1000, 723]) | 10\n",
-      "26404    | torch.Size([1, 1000, 802]) | 10\n",
-      "26405    | torch.Size([1, 1000, 754]) | 10\n",
-      "26406    | torch.Size([1, 1000, 754]) | 10\n",
-      "26407    | torch.Size([1, 1000, 754]) | 10\n",
-      "26408    | torch.Size([1, 1000, 777]) | 10\n",
-      "26409    | torch.Size([1, 1000, 754]) | 10\n",
-      "26410    | torch.Size([1, 1000, 777]) | 10\n",
-      "26411    | torch.Size([1, 1000, 773]) | 10\n",
-      "26412    | torch.Size([1, 1000, 754]) | 10\n",
-      "26413    | torch.Size([1, 1000, 762]) | 10\n",
-      "26414    | torch.Size([1, 1000, 754]) | 10\n",
-      "26415    | torch.Size([1, 1000, 772]) | 10\n",
-      "26416    | torch.Size([1, 1000, 754]) | 10\n",
-      "26417    | torch.Size([1, 1000, 754]) | 10\n",
-      "26418    | torch.Size([1, 1000, 754]) | 10\n",
-      "26419    | torch.Size([1, 1000, 773]) | 10\n",
-      "26420    | torch.Size([1, 1000, 754]) | 10\n",
-      "26421    | torch.Size([1, 1000, 799]) | 10\n",
-      "26422    | torch.Size([1, 1000, 754]) | 10\n",
-      "26423    | torch.Size([1, 1000, 754]) | 10\n",
-      "26424    | torch.Size([1, 1000, 754]) | 10\n",
-      "26425    | torch.Size([1, 1000, 777]) | 10\n",
-      "26426    | torch.Size([1, 1000, 789]) | 10\n",
-      "26427    | torch.Size([1, 1000, 802]) | 10\n",
-      "26428    | torch.Size([1, 1000, 754]) | 10\n",
-      "26429    | torch.Size([1, 1000, 777]) | 10\n",
-      "26430    | torch.Size([1, 1000, 777]) | 10\n",
-      "26431    | torch.Size([1, 1000, 776]) | 10\n",
-      "26432    | torch.Size([1, 1000, 812]) | 10\n",
-      "26433    | torch.Size([1, 1000, 809]) | 10\n",
-      "26434    | torch.Size([1, 1000, 754]) | 10\n",
-      "26435    | torch.Size([1, 1000, 754]) | 10\n",
-      "26436    | torch.Size([1, 1000, 777]) | 10\n",
-      "26437    | torch.Size([1, 1000, 754]) | 10\n",
-      "26438    | torch.Size([1, 1000, 754]) | 10\n",
-      "26439    | torch.Size([1, 1000, 754]) | 10\n",
-      "26440    | torch.Size([1, 1000, 762]) | 10\n",
-      "26441    | torch.Size([1, 1000, 789]) | 10\n",
-      "26442    | torch.Size([1, 1000, 802]) | 10\n",
-      "26443    | torch.Size([1, 1000, 754]) | 10\n",
-      "26444    | torch.Size([1, 1000, 787]) | 10\n",
-      "26445    | torch.Size([1, 1000, 754]) | 10\n",
-      "26446    | torch.Size([1, 1000, 754]) | 10\n",
-      "26447    | torch.Size([1, 1000, 754]) | 10\n",
-      "26448    | torch.Size([1, 1000, 754]) | 10\n",
-      "26449    | torch.Size([1, 1000, 754]) | 10\n",
-      "26450    | torch.Size([1, 1000, 754]) | 10\n",
-      "26451    | torch.Size([1, 1000, 762]) | 10\n",
-      "26452    | torch.Size([1, 1000, 784]) | 10\n",
-      "26453    | torch.Size([1, 1000, 754]) | 10\n",
-      "26454    | torch.Size([1, 1000, 754]) | 10\n",
-      "26455    | torch.Size([1, 1000, 774]) | 10\n",
-      "26456    | torch.Size([1, 1000, 754]) | 10\n",
-      "26457    | torch.Size([1, 1000, 754]) | 10\n",
-      "26458    | torch.Size([1, 1000, 754]) | 10\n",
-      "26459    | torch.Size([1, 1000, 754]) | 10\n",
-      "26460    | torch.Size([1, 1000, 860]) | 10\n",
-      "26461    | torch.Size([1, 1000, 754]) | 10\n",
-      "26462    | torch.Size([1, 1000, 777]) | 10\n",
-      "26463    | torch.Size([1, 1000, 754]) | 10\n",
-      "26464    | torch.Size([1, 1000, 762]) | 10\n",
-      "26465    | torch.Size([1, 1000, 754]) | 10\n",
-      "26466    | torch.Size([1, 1000, 754]) | 10\n",
-      "26467    | torch.Size([1, 1000, 762]) | 10\n",
-      "26468    | torch.Size([1, 1000, 754]) | 10\n",
-      "26469    | torch.Size([1, 1000, 754]) | 10\n",
-      "26470    | torch.Size([1, 1000, 762]) | 10\n",
-      "26471    | torch.Size([1, 1000, 754]) | 10\n",
-      "26472    | torch.Size([1, 1000, 754]) | 10\n",
-      "26473    | torch.Size([1, 1000, 754]) | 10\n",
-      "26474    | torch.Size([1, 1000, 754]) | 10\n",
-      "26475    | torch.Size([1, 1000, 754]) | 10\n",
-      "26476    | torch.Size([1, 1000, 754]) | 10\n",
-      "26477    | torch.Size([1, 1000, 754]) | 10\n",
-      "26478    | torch.Size([1, 1000, 754]) | 10\n",
-      "26479    | torch.Size([1, 1000, 754]) | 10\n",
-      "26480    | torch.Size([1, 1000, 754]) | 10\n",
-      "26481    | torch.Size([1, 1000, 754]) | 10\n",
-      "26482    | torch.Size([1, 1000, 777]) | 10\n",
-      "26483    | torch.Size([1, 1000, 777]) | 10\n",
-      "26484    | torch.Size([1, 1000, 754]) | 10\n",
-      "26485    | torch.Size([1, 1000, 754]) | 10\n",
-      "26486    | torch.Size([1, 1000, 682]) | 10\n",
-      "26487    | torch.Size([1, 1000, 777]) | 10\n",
-      "26488    | torch.Size([1, 1000, 754]) | 10\n",
-      "26489    | torch.Size([1, 1000, 777]) | 10\n",
-      "26490    | torch.Size([1, 1000, 783]) | 10\n",
-      "26491    | torch.Size([1, 1000, 778]) | 10\n",
-      "26492    | torch.Size([1, 1000, 781]) | 10\n",
-      "26493    | torch.Size([1, 1000, 754]) | 10\n",
-      "26494    | torch.Size([1, 1000, 754]) | 10\n",
-      "26495    | torch.Size([1, 1000, 754]) | 10\n",
-      "26496    | torch.Size([1, 1000, 754]) | 10\n",
-      "26497    | torch.Size([1, 1000, 754]) | 10\n",
-      "26498    | torch.Size([1, 1000, 787]) | 10\n",
-      "26499    | torch.Size([1, 1000, 754]) | 10\n",
-      "26500    | torch.Size([1, 1000, 840]) | 10\n",
-      "26501    | torch.Size([1, 1000, 771]) | 10\n",
-      "26502    | torch.Size([1, 1000, 754]) | 10\n",
-      "26503    | torch.Size([1, 1000, 754]) | 10\n",
-      "26504    | torch.Size([1, 1000, 754]) | 10\n",
-      "26505    | torch.Size([1, 1000, 780]) | 10\n",
-      "26506    | torch.Size([1, 1000, 754]) | 10\n",
-      "26507    | torch.Size([1, 1000, 754]) | 10\n",
-      "26508    | torch.Size([1, 1000, 777]) | 10\n",
-      "26509    | torch.Size([1, 1000, 754]) | 10\n",
-      "26510    | torch.Size([1, 1000, 788]) | 10\n",
-      "26511    | torch.Size([1, 1000, 783]) | 10\n",
-      "26512    | torch.Size([1, 1000, 783]) | 10\n",
-      "26513    | torch.Size([1, 1000, 754]) | 10\n",
-      "26514    | torch.Size([1, 1000, 777]) | 10\n",
-      "26515    | torch.Size([1, 1000, 754]) | 10\n",
-      "26516    | torch.Size([1, 1000, 754]) | 10\n",
-      "26517    | torch.Size([1, 1000, 754]) | 10\n",
-      "26518    | torch.Size([1, 1000, 754]) | 10\n",
-      "26519    | torch.Size([1, 1000, 754]) | 10\n",
-      "26520    | torch.Size([1, 1000, 801]) | 10\n",
-      "26521    | torch.Size([1, 1000, 793]) | 10\n",
-      "26522    | torch.Size([1, 1000, 723]) | 10\n",
-      "26523    | torch.Size([1, 1000, 754]) | 10\n",
-      "26524    | torch.Size([1, 1000, 754]) | 10\n",
-      "26525    | torch.Size([1, 1000, 754]) | 10\n",
-      "26526    | torch.Size([1, 1000, 805]) | 10\n",
-      "26527    | torch.Size([1, 1000, 754]) | 10\n",
-      "26528    | torch.Size([1, 1000, 762]) | 10\n",
-      "26529    | torch.Size([1, 1000, 754]) | 10\n",
-      "26530    | torch.Size([1, 1000, 801]) | 10\n",
-      "26531    | torch.Size([1, 1000, 777]) | 10\n",
-      "26532    | torch.Size([1, 1000, 772]) | 10\n",
-      "26533    | torch.Size([1, 1000, 754]) | 10\n",
-      "26534    | torch.Size([1, 1000, 754]) | 10\n",
-      "26535    | torch.Size([1, 1000, 754]) | 10\n",
-      "26536    | torch.Size([1, 1000, 802]) | 10\n",
-      "26537    | torch.Size([1, 1000, 754]) | 10\n",
-      "26538    | torch.Size([1, 1000, 754]) | 10\n",
-      "26539    | torch.Size([1, 1000, 754]) | 10\n",
-      "26540    | torch.Size([1, 1000, 754]) | 10\n",
-      "26541    | torch.Size([1, 1000, 754]) | 10\n",
-      "26542    | torch.Size([1, 1000, 754]) | 10\n",
-      "26543    | torch.Size([1, 1000, 754]) | 10\n",
-      "26544    | torch.Size([1, 1000, 754]) | 10\n",
-      "26545    | torch.Size([1, 1000, 754]) | 10\n",
-      "26546    | torch.Size([1, 1000, 754]) | 10\n",
-      "26547    | torch.Size([1, 1000, 754]) | 10\n",
-      "26548    | torch.Size([1, 1000, 754]) | 10\n",
-      "26549    | torch.Size([1, 1000, 771]) | 10\n",
-      "26550    | torch.Size([1, 1000, 754]) | 10\n",
-      "26551    | torch.Size([1, 1000, 754]) | 10\n",
-      "26552    | torch.Size([1, 1000, 802]) | 10\n",
-      "26553    | torch.Size([1, 1000, 763]) | 10\n",
-      "26554    | torch.Size([1, 1000, 754]) | 10\n",
-      "26555    | torch.Size([1, 1000, 773]) | 10\n",
-      "26556    | torch.Size([1, 1000, 754]) | 10\n",
-      "26557    | torch.Size([1, 1000, 777]) | 10\n",
-      "26558    | torch.Size([1, 1000, 754]) | 10\n",
-      "26559    | torch.Size([1, 1000, 762]) | 10\n",
-      "26560    | torch.Size([1, 1000, 754]) | 10\n",
-      "26561    | torch.Size([1, 1000, 777]) | 10\n",
-      "26562    | torch.Size([1, 1000, 771]) | 10\n",
-      "26563    | torch.Size([1, 1000, 754]) | 10\n",
-      "26564    | torch.Size([1, 1000, 777]) | 10\n",
-      "26565    | torch.Size([1, 1000, 802]) | 10\n",
-      "26566    | torch.Size([1, 1000, 777]) | 10\n",
-      "26567    | torch.Size([1, 1000, 754]) | 10\n",
-      "26568    | torch.Size([1, 1000, 771]) | 10\n",
-      "26569    | torch.Size([1, 1000, 754]) | 10\n",
-      "26570    | torch.Size([1, 1000, 777]) | 10\n",
-      "26571    | torch.Size([1, 1000, 754]) | 10\n",
-      "26572    | torch.Size([1, 1000, 762]) | 10\n",
-      "26573    | torch.Size([1, 1000, 762]) | 10\n",
-      "26574    | torch.Size([1, 1000, 777]) | 10\n",
-      "26575    | torch.Size([1, 1000, 754]) | 10\n",
-      "26576    | torch.Size([1, 1000, 754]) | 10\n",
-      "26577    | torch.Size([1, 1000, 762]) | 10\n",
-      "26578    | torch.Size([1, 1000, 762]) | 10\n",
-      "26579    | torch.Size([1, 1000, 767]) | 10\n",
-      "26580    | torch.Size([1, 1000, 777]) | 10\n",
-      "26581    | torch.Size([1, 1000, 796]) | 10\n",
-      "26582    | torch.Size([1, 1000, 754]) | 10\n",
-      "26583    | torch.Size([1, 1000, 802]) | 10\n",
-      "26584    | torch.Size([1, 1000, 754]) | 10\n",
-      "26585    | torch.Size([1, 1000, 753]) | 10\n",
-      "26586    | torch.Size([1, 1000, 762]) | 10\n",
-      "26587    | torch.Size([1, 1000, 777]) | 10\n",
-      "26588    | torch.Size([1, 1000, 762]) | 10\n",
-      "26589    | torch.Size([1, 1000, 754]) | 10\n",
-      "26590    | torch.Size([1, 1000, 766]) | 10\n",
-      "26591    | torch.Size([1, 1000, 777]) | 10\n",
-      "26592    | torch.Size([1, 1000, 762]) | 10\n",
-      "26593    | torch.Size([1, 1000, 754]) | 10\n",
-      "26594    | torch.Size([1, 1000, 754]) | 10\n",
-      "26595    | torch.Size([1, 1000, 771]) | 10\n",
-      "26596    | torch.Size([1, 1000, 754]) | 10\n",
-      "26597    | torch.Size([1, 1000, 754]) | 10\n",
-      "26598    | torch.Size([1, 1000, 790]) | 10\n",
-      "26599    | torch.Size([1, 1000, 764]) | 10\n",
-      "26600    | torch.Size([1, 1000, 754]) | 10\n",
-      "26601    | torch.Size([1, 1000, 754]) | 10\n",
-      "26602    | torch.Size([1, 1000, 777]) | 10\n",
-      "26603    | torch.Size([1, 1000, 754]) | 10\n",
-      "26604    | torch.Size([1, 1000, 777]) | 10\n",
-      "26605    | torch.Size([1, 1000, 754]) | 10\n",
-      "26606    | torch.Size([1, 1000, 777]) | 10\n",
-      "26607    | torch.Size([1, 1000, 754]) | 10\n",
-      "26608    | torch.Size([1, 1000, 754]) | 10\n",
-      "26609    | torch.Size([1, 1000, 762]) | 10\n",
-      "26610    | torch.Size([1, 1000, 754]) | 10\n",
-      "26611    | torch.Size([1, 1000, 762]) | 10\n",
-      "26612    | torch.Size([1, 1000, 762]) | 10\n",
-      "26613    | torch.Size([1, 1000, 754]) | 10\n",
-      "26614    | torch.Size([1, 1000, 777]) | 10\n",
-      "26615    | torch.Size([1, 1000, 847]) | 10\n",
-      "26616    | torch.Size([1, 1000, 754]) | 10\n",
-      "26617    | torch.Size([1, 1000, 754]) | 10\n",
-      "26618    | torch.Size([1, 1000, 754]) | 10\n",
-      "26619    | torch.Size([1, 1000, 754]) | 10\n",
-      "26620    | torch.Size([1, 1000, 762]) | 10\n",
-      "26621    | torch.Size([1, 1000, 754]) | 10\n",
-      "26622    | torch.Size([1, 1000, 754]) | 10\n",
-      "26623    | torch.Size([1, 1000, 811]) | 10\n",
-      "26624    | torch.Size([1, 1000, 754]) | 10\n",
-      "26625    | torch.Size([1, 1000, 754]) | 10\n",
-      "26626    | torch.Size([1, 1000, 762]) | 10\n",
-      "26627    | torch.Size([1, 1000, 754]) | 10\n",
-      "26628    | torch.Size([1, 1000, 754]) | 10\n",
-      "26629    | torch.Size([1, 1000, 754]) | 10\n",
-      "26630    | torch.Size([1, 1000, 754]) | 10\n",
-      "26631    | torch.Size([1, 1000, 762]) | 10\n",
-      "26632    | torch.Size([1, 1000, 776]) | 10\n",
-      "26633    | torch.Size([1, 1000, 754]) | 10\n",
-      "26634    | torch.Size([1, 1000, 777]) | 10\n",
-      "26635    | torch.Size([1, 1000, 754]) | 10\n",
-      "26636    | torch.Size([1, 1000, 771]) | 10\n",
-      "26637    | torch.Size([1, 1000, 754]) | 10\n",
-      "26638    | torch.Size([1, 1000, 754]) | 10\n",
-      "26639    | torch.Size([1, 1000, 754]) | 10\n",
-      "26640    | torch.Size([1, 1000, 732]) | 10\n",
-      "26641    | torch.Size([1, 1000, 754]) | 10\n",
-      "26642    | torch.Size([1, 1000, 802]) | 10\n",
-      "26643    | torch.Size([1, 1000, 754]) | 10\n",
-      "26644    | torch.Size([1, 1000, 762]) | 10\n",
-      "26645    | torch.Size([1, 1000, 754]) | 10\n",
-      "26646    | torch.Size([1, 1000, 754]) | 10\n",
-      "26647    | torch.Size([1, 1000, 754]) | 10\n",
-      "26648    | torch.Size([1, 1000, 754]) | 10\n",
-      "26649    | torch.Size([1, 1000, 754]) | 10\n",
-      "26650    | torch.Size([1, 1000, 754]) | 10\n",
-      "26651    | torch.Size([1, 1000, 784]) | 10\n",
-      "26652    | torch.Size([1, 1000, 754]) | 10\n",
-      "26653    | torch.Size([1, 1000, 754]) | 10\n",
-      "26654    | torch.Size([1, 1000, 844]) | 10\n",
-      "26655    | torch.Size([1, 1000, 762]) | 10\n",
-      "26656    | torch.Size([1, 1000, 754]) | 10\n",
-      "26657    | torch.Size([1, 1000, 802]) | 10\n",
-      "26658    | torch.Size([1, 1000, 754]) | 10\n",
-      "26659    | torch.Size([1, 1000, 754]) | 10\n",
-      "26660    | torch.Size([1, 1000, 777]) | 10\n",
-      "26661    | torch.Size([1, 1000, 777]) | 10\n",
-      "26662    | torch.Size([1, 1000, 754]) | 10\n",
-      "26663    | torch.Size([1, 1000, 796]) | 10\n",
-      "26664    | torch.Size([1, 1000, 781]) | 10\n",
-      "26665    | torch.Size([1, 1000, 754]) | 10\n",
-      "26666    | torch.Size([1, 1000, 754]) | 10\n",
-      "26667    | torch.Size([1, 1000, 754]) | 10\n",
-      "26668    | torch.Size([1, 1000, 777]) | 10\n",
-      "26669    | torch.Size([1, 1000, 770]) | 10\n",
-      "26670    | torch.Size([1, 1000, 773]) | 10\n",
-      "26671    | torch.Size([1, 1000, 846]) | 10\n",
-      "26672    | torch.Size([1, 1000, 754]) | 10\n",
-      "26673    | torch.Size([1, 1000, 754]) | 10\n",
-      "26674    | torch.Size([1, 1000, 771]) | 10\n",
-      "26675    | torch.Size([1, 1000, 810]) | 10\n",
-      "26676    | torch.Size([1, 1000, 777]) | 10\n",
-      "26677    | torch.Size([1, 1000, 777]) | 10\n",
-      "26678    | torch.Size([1, 1000, 754]) | 10\n",
-      "26679    | torch.Size([1, 1000, 762]) | 10\n",
-      "26680    | torch.Size([1, 1000, 754]) | 10\n",
-      "26681    | torch.Size([1, 1000, 777]) | 10\n",
-      "26682    | torch.Size([1, 1000, 773]) | 10\n",
-      "26683    | torch.Size([1, 1000, 768]) | 10\n",
-      "26684    | torch.Size([1, 1000, 762]) | 10\n",
-      "26685    | torch.Size([1, 1000, 762]) | 10\n",
-      "26686    | torch.Size([1, 1000, 777]) | 10\n",
-      "26687    | torch.Size([1, 1000, 754]) | 10\n",
-      "26688    | torch.Size([1, 1000, 776]) | 10\n",
-      "26689    | torch.Size([1, 1000, 762]) | 10\n",
-      "26690    | torch.Size([1, 1000, 754]) | 10\n",
-      "26691    | torch.Size([1, 1000, 777]) | 10\n",
-      "26692    | torch.Size([1, 1000, 777]) | 10\n",
-      "26693    | torch.Size([1, 1000, 762]) | 10\n",
-      "26694    | torch.Size([1, 1000, 769]) | 10\n",
-      "26695    | torch.Size([1, 1000, 754]) | 10\n",
-      "26696    | torch.Size([1, 1000, 762]) | 10\n",
-      "26697    | torch.Size([1, 1000, 762]) | 10\n",
-      "26698    | torch.Size([1, 1000, 754]) | 10\n",
-      "26699    | torch.Size([1, 1000, 754]) | 10\n",
-      "26700    | torch.Size([1, 1000, 780]) | 10\n",
-      "26701    | torch.Size([1, 1000, 777]) | 10\n",
-      "26702    | torch.Size([1, 1000, 795]) | 10\n",
-      "26703    | torch.Size([1, 1000, 754]) | 10\n",
-      "26704    | torch.Size([1, 1000, 754]) | 10\n",
-      "26705    | torch.Size([1, 1000, 754]) | 10\n",
-      "26706    | torch.Size([1, 1000, 754]) | 10\n",
-      "26707    | torch.Size([1, 1000, 777]) | 10\n",
-      "26708    | torch.Size([1, 1000, 785]) | 10\n",
-      "26709    | torch.Size([1, 1000, 754]) | 10\n",
-      "26710    | torch.Size([1, 1000, 850]) | 10\n",
-      "26711    | torch.Size([1, 1000, 777]) | 10\n",
-      "26712    | torch.Size([1, 1000, 754]) | 10\n",
-      "26713    | torch.Size([1, 1000, 754]) | 10\n",
-      "26714    | torch.Size([1, 1000, 777]) | 10\n",
-      "26715    | torch.Size([1, 1000, 783]) | 10\n",
-      "26716    | torch.Size([1, 1000, 754]) | 10\n",
-      "26717    | torch.Size([1, 1000, 754]) | 10\n",
-      "26718    | torch.Size([1, 1000, 782]) | 10\n",
-      "26719    | torch.Size([1, 1000, 772]) | 10\n",
-      "26720    | torch.Size([1, 1000, 777]) | 10\n",
-      "26721    | torch.Size([1, 1000, 798]) | 10\n",
-      "26722    | torch.Size([1, 1000, 754]) | 10\n",
-      "26723    | torch.Size([1, 1000, 788]) | 10\n",
-      "26724    | torch.Size([1, 1000, 848]) | 10\n",
-      "26725    | torch.Size([1, 1000, 754]) | 10\n",
-      "26726    | torch.Size([1, 1000, 781]) | 10\n",
-      "26727    | torch.Size([1, 1000, 762]) | 10\n",
-      "26728    | torch.Size([1, 1000, 765]) | 10\n",
-      "26729    | torch.Size([1, 1000, 754]) | 10\n",
-      "26730    | torch.Size([1, 1000, 782]) | 10\n",
-      "26731    | torch.Size([1, 1000, 754]) | 10\n",
-      "26732    | torch.Size([1, 1000, 725]) | 10\n",
-      "26733    | torch.Size([1, 1000, 777]) | 10\n",
-      "26734    | torch.Size([1, 1000, 754]) | 10\n",
-      "26735    | torch.Size([1, 1000, 842]) | 10\n",
-      "26736    | torch.Size([1, 1000, 754]) | 10\n",
-      "26737    | torch.Size([1, 1000, 754]) | 10\n",
-      "26738    | torch.Size([1, 1000, 754]) | 10\n",
-      "26739    | torch.Size([1, 1000, 754]) | 10\n",
-      "26740    | torch.Size([1, 1000, 754]) | 10\n",
-      "26741    | torch.Size([1, 1000, 777]) | 10\n",
-      "26742    | torch.Size([1, 1000, 782]) | 10\n",
-      "26743    | torch.Size([1, 1000, 754]) | 10\n",
-      "26744    | torch.Size([1, 1000, 777]) | 10\n",
-      "26745    | torch.Size([1, 1000, 754]) | 10\n",
-      "26746    | torch.Size([1, 1000, 802]) | 10\n",
-      "26747    | torch.Size([1, 1000, 754]) | 10\n",
-      "26748    | torch.Size([1, 1000, 754]) | 10\n",
-      "26749    | torch.Size([1, 1000, 754]) | 10\n",
-      "26750    | torch.Size([1, 1000, 789]) | 10\n",
-      "26751    | torch.Size([1, 1000, 775]) | 10\n",
-      "26752    | torch.Size([1, 1000, 754]) | 10\n",
-      "26753    | torch.Size([1, 1000, 777]) | 10\n",
-      "26754    | torch.Size([1, 1000, 777]) | 10\n",
-      "26755    | torch.Size([1, 1000, 754]) | 10\n",
-      "26756    | torch.Size([1, 1000, 762]) | 10\n",
-      "26757    | torch.Size([1, 1000, 754]) | 10\n",
-      "26758    | torch.Size([1, 1000, 754]) | 10\n",
-      "26759    | torch.Size([1, 1000, 754]) | 10\n",
-      "26760    | torch.Size([1, 1000, 777]) | 10\n",
-      "26761    | torch.Size([1, 1000, 754]) | 10\n",
-      "26762    | torch.Size([1, 1000, 802]) | 10\n",
-      "26763    | torch.Size([1, 1000, 754]) | 10\n",
-      "26764    | torch.Size([1, 1000, 777]) | 10\n",
-      "26765    | torch.Size([1, 1000, 754]) | 10\n",
-      "26766    | torch.Size([1, 1000, 762]) | 10\n",
-      "26767    | torch.Size([1, 1000, 777]) | 10\n",
-      "26768    | torch.Size([1, 1000, 754]) | 10\n",
-      "26769    | torch.Size([1, 1000, 786]) | 10\n",
-      "26770    | torch.Size([1, 1000, 762]) | 10\n",
-      "26771    | torch.Size([1, 1000, 754]) | 10\n",
-      "26772    | torch.Size([1, 1000, 777]) | 10\n",
-      "26773    | torch.Size([1, 1000, 754]) | 10\n",
-      "26774    | torch.Size([1, 1000, 777]) | 10\n",
-      "26775    | torch.Size([1, 1000, 777]) | 10\n",
-      "26776    | torch.Size([1, 1000, 723]) | 10\n",
-      "26777    | torch.Size([1, 1000, 777]) | 10\n",
-      "26778    | torch.Size([1, 1000, 754]) | 10\n",
-      "26779    | torch.Size([1, 1000, 754]) | 10\n",
-      "26780    | torch.Size([1, 1000, 789]) | 10\n",
-      "26781    | torch.Size([1, 1000, 777]) | 10\n",
-      "26782    | torch.Size([1, 1000, 777]) | 10\n",
-      "26783    | torch.Size([1, 1000, 754]) | 10\n",
-      "26784    | torch.Size([1, 1000, 754]) | 10\n",
-      "26785    | torch.Size([1, 1000, 853]) | 10\n",
-      "26786    | torch.Size([1, 1000, 767]) | 10\n",
-      "26787    | torch.Size([1, 1000, 777]) | 10\n",
-      "26788    | torch.Size([1, 1000, 762]) | 10\n",
-      "26789    | torch.Size([1, 1000, 845]) | 10\n",
-      "26790    | torch.Size([1, 1000, 762]) | 10\n",
-      "26791    | torch.Size([1, 1000, 754]) | 10\n",
-      "26792    | torch.Size([1, 1000, 754]) | 10\n",
-      "26793    | torch.Size([1, 1000, 782]) | 10\n",
-      "26794    | torch.Size([1, 1000, 754]) | 10\n",
-      "26795    | torch.Size([1, 1000, 762]) | 10\n",
-      "26796    | torch.Size([1, 1000, 777]) | 10\n",
-      "26797    | torch.Size([1, 1000, 754]) | 10\n",
-      "26798    | torch.Size([1, 1000, 754]) | 10\n",
-      "26799    | torch.Size([1, 1000, 754]) | 10\n",
-      "26800    | torch.Size([1, 1000, 777]) | 10\n",
-      "26801    | torch.Size([1, 1000, 754]) | 10\n",
-      "26802    | torch.Size([1, 1000, 762]) | 10\n",
-      "26803    | torch.Size([1, 1000, 754]) | 10\n",
-      "26804    | torch.Size([1, 1000, 754]) | 10\n",
-      "26805    | torch.Size([1, 1000, 754]) | 10\n",
-      "26806    | torch.Size([1, 1000, 754]) | 10\n",
-      "26807    | torch.Size([1, 1000, 781]) | 10\n",
-      "26808    | torch.Size([1, 1000, 777]) | 10\n",
-      "26809    | torch.Size([1, 1000, 754]) | 10\n",
-      "26810    | torch.Size([1, 1000, 802]) | 10\n",
-      "26811    | torch.Size([1, 1000, 773]) | 10\n",
-      "26812    | torch.Size([1, 1000, 754]) | 10\n",
-      "26813    | torch.Size([1, 1000, 795]) | 10\n",
-      "26814    | torch.Size([1, 1000, 754]) | 10\n",
-      "26815    | torch.Size([1, 1000, 777]) | 10\n",
-      "26816    | torch.Size([1, 1000, 754]) | 10\n",
-      "26817    | torch.Size([1, 1000, 777]) | 10\n",
-      "26818    | torch.Size([1, 1000, 786]) | 10\n",
-      "26819    | torch.Size([1, 1000, 777]) | 10\n",
-      "26820    | torch.Size([1, 1000, 777]) | 10\n",
-      "26821    | torch.Size([1, 1000, 777]) | 10\n",
-      "26822    | torch.Size([1, 1000, 754]) | 10\n",
-      "26823    | torch.Size([1, 1000, 754]) | 10\n",
-      "26824    | torch.Size([1, 1000, 754]) | 10\n",
-      "26825    | torch.Size([1, 1000, 754]) | 10\n",
-      "26826    | torch.Size([1, 1000, 777]) | 10\n",
-      "26827    | torch.Size([1, 1000, 777]) | 10\n",
-      "26828    | torch.Size([1, 1000, 762]) | 10\n",
-      "26829    | torch.Size([1, 1000, 782]) | 10\n",
-      "26830    | torch.Size([1, 1000, 777]) | 10\n",
-      "26831    | torch.Size([1, 1000, 802]) | 10\n",
-      "26832    | torch.Size([1, 1000, 762]) | 10\n",
-      "26833    | torch.Size([1, 1000, 777]) | 10\n",
-      "26834    | torch.Size([1, 1000, 799]) | 10\n",
-      "26835    | torch.Size([1, 1000, 754]) | 10\n",
-      "26836    | torch.Size([1, 1000, 754]) | 10\n",
-      "26837    | torch.Size([1, 1000, 762]) | 10\n",
-      "26838    | torch.Size([1, 1000, 754]) | 10\n",
-      "26839    | torch.Size([1, 1000, 777]) | 10\n",
-      "26840    | torch.Size([1, 1000, 848]) | 10\n",
-      "26841    | torch.Size([1, 1000, 772]) | 10\n",
-      "26842    | torch.Size([1, 1000, 762]) | 10\n",
-      "26843    | torch.Size([1, 1000, 754]) | 10\n",
-      "26844    | torch.Size([1, 1000, 777]) | 10\n",
-      "26845    | torch.Size([1, 1000, 762]) | 10\n",
-      "26846    | torch.Size([1, 1000, 754]) | 10\n",
-      "26847    | torch.Size([1, 1000, 754]) | 10\n",
-      "26848    | torch.Size([1, 1000, 754]) | 10\n",
-      "26849    | torch.Size([1, 1000, 754]) | 10\n",
-      "26850    | torch.Size([1, 1000, 754]) | 10\n",
-      "26851    | torch.Size([1, 1000, 762]) | 10\n",
-      "26852    | torch.Size([1, 1000, 771]) | 10\n",
-      "26853    | torch.Size([1, 1000, 773]) | 10\n",
-      "26854    | torch.Size([1, 1000, 785]) | 10\n",
-      "26855    | torch.Size([1, 1000, 754]) | 10\n",
-      "26856    | torch.Size([1, 1000, 754]) | 10\n",
-      "26857    | torch.Size([1, 1000, 754]) | 10\n",
-      "26858    | torch.Size([1, 1000, 762]) | 10\n",
-      "26859    | torch.Size([1, 1000, 802]) | 10\n",
-      "26860    | torch.Size([1, 1000, 776]) | 10\n",
-      "26861    | torch.Size([1, 1000, 754]) | 10\n",
-      "26862    | torch.Size([1, 1000, 804]) | 10\n",
-      "26863    | torch.Size([1, 1000, 754]) | 10\n",
-      "26864    | torch.Size([1, 1000, 754]) | 10\n",
-      "26865    | torch.Size([1, 1000, 754]) | 10\n",
-      "26866    | torch.Size([1, 1000, 754]) | 10\n",
-      "26867    | torch.Size([1, 1000, 754]) | 10\n",
-      "26868    | torch.Size([1, 1000, 754]) | 10\n",
-      "26869    | torch.Size([1, 1000, 780]) | 10\n",
-      "26870    | torch.Size([1, 1000, 777]) | 10\n",
-      "26871    | torch.Size([1, 1000, 754]) | 10\n",
-      "26872    | torch.Size([1, 1000, 762]) | 10\n",
-      "26873    | torch.Size([1, 1000, 754]) | 10\n",
-      "26874    | torch.Size([1, 1000, 784]) | 10\n",
-      "26875    | torch.Size([1, 1000, 754]) | 10\n",
-      "26876    | torch.Size([1, 1000, 754]) | 10\n",
-      "26877    | torch.Size([1, 1000, 786]) | 10\n",
-      "26878    | torch.Size([1, 1000, 754]) | 10\n",
-      "26879    | torch.Size([1, 1000, 762]) | 10\n",
-      "26880    | torch.Size([1, 1000, 754]) | 10\n",
-      "26881    | torch.Size([1, 1000, 807]) | 10\n",
-      "26882    | torch.Size([1, 1000, 777]) | 10\n",
-      "26883    | torch.Size([1, 1000, 754]) | 10\n",
-      "26884    | torch.Size([1, 1000, 754]) | 10\n",
-      "26885    | torch.Size([1, 1000, 799]) | 10\n",
-      "26886    | torch.Size([1, 1000, 754]) | 10\n",
-      "26887    | torch.Size([1, 1000, 777]) | 10\n",
-      "26888    | torch.Size([1, 1000, 754]) | 10\n",
-      "26889    | torch.Size([1, 1000, 777]) | 10\n",
-      "26890    | torch.Size([1, 1000, 754]) | 10\n",
-      "26891    | torch.Size([1, 1000, 762]) | 10\n",
-      "26892    | torch.Size([1, 1000, 754]) | 10\n",
-      "26893    | torch.Size([1, 1000, 754]) | 10\n",
-      "26894    | torch.Size([1, 1000, 812]) | 10\n",
-      "26895    | torch.Size([1, 1000, 809]) | 10\n",
-      "26896    | torch.Size([1, 1000, 754]) | 10\n",
-      "26897    | torch.Size([1, 1000, 777]) | 10\n",
-      "26898    | torch.Size([1, 1000, 762]) | 10\n",
-      "26899    | torch.Size([1, 1000, 754]) | 10\n",
-      "26900    | torch.Size([1, 1000, 777]) | 10\n",
-      "26901    | torch.Size([1, 1000, 754]) | 10\n",
-      "26902    | torch.Size([1, 1000, 754]) | 10\n",
-      "26903    | torch.Size([1, 1000, 777]) | 10\n",
-      "26904    | torch.Size([1, 1000, 754]) | 10\n",
-      "26905    | torch.Size([1, 1000, 775]) | 10\n",
-      "26906    | torch.Size([1, 1000, 777]) | 10\n",
-      "26907    | torch.Size([1, 1000, 777]) | 10\n",
-      "26908    | torch.Size([1, 1000, 797]) | 10\n",
-      "26909    | torch.Size([1, 1000, 777]) | 10\n",
-      "26910    | torch.Size([1, 1000, 777]) | 10\n",
-      "26911    | torch.Size([1, 1000, 777]) | 10\n",
-      "26912    | torch.Size([1, 1000, 754]) | 10\n",
-      "26913    | torch.Size([1, 1000, 795]) | 10\n",
-      "26914    | torch.Size([1, 1000, 754]) | 10\n",
-      "26915    | torch.Size([1, 1000, 754]) | 10\n",
-      "26916    | torch.Size([1, 1000, 795]) | 10\n",
-      "26917    | torch.Size([1, 1000, 777]) | 10\n",
-      "26918    | torch.Size([1, 1000, 778]) | 10\n",
-      "26919    | torch.Size([1, 1000, 754]) | 10\n",
-      "26920    | torch.Size([1, 1000, 791]) | 10\n",
-      "26921    | torch.Size([1, 1000, 771]) | 10\n",
-      "26922    | torch.Size([1, 1000, 754]) | 10\n",
-      "26923    | torch.Size([1, 1000, 754]) | 10\n",
-      "26924    | torch.Size([1, 1000, 754]) | 10\n",
-      "26925    | torch.Size([1, 1000, 762]) | 10\n",
-      "26926    | torch.Size([1, 1000, 762]) | 10\n",
-      "26927    | torch.Size([1, 1000, 754]) | 10\n",
-      "26928    | torch.Size([1, 1000, 754]) | 10\n",
-      "26929    | torch.Size([1, 1000, 754]) | 10\n",
-      "26930    | torch.Size([1, 1000, 754]) | 10\n",
-      "26931    | torch.Size([1, 1000, 777]) | 10\n",
-      "26932    | torch.Size([1, 1000, 766]) | 10\n",
-      "26933    | torch.Size([1, 1000, 802]) | 10\n",
-      "26934    | torch.Size([1, 1000, 754]) | 10\n",
-      "26935    | torch.Size([1, 1000, 771]) | 10\n",
-      "26936    | torch.Size([1, 1000, 774]) | 10\n",
-      "26937    | torch.Size([1, 1000, 754]) | 10\n",
-      "26938    | torch.Size([1, 1000, 754]) | 10\n",
-      "26939    | torch.Size([1, 1000, 754]) | 10\n",
-      "26940    | torch.Size([1, 1000, 754]) | 10\n",
-      "26941    | torch.Size([1, 1000, 762]) | 10\n",
-      "26942    | torch.Size([1, 1000, 773]) | 10\n",
-      "26943    | torch.Size([1, 1000, 754]) | 10\n",
-      "26944    | torch.Size([1, 1000, 762]) | 10\n",
-      "26945    | torch.Size([1, 1000, 754]) | 10\n",
-      "26946    | torch.Size([1, 1000, 754]) | 10\n",
-      "26947    | torch.Size([1, 1000, 754]) | 10\n",
-      "26948    | torch.Size([1, 1000, 777]) | 10\n",
-      "26949    | torch.Size([1, 1000, 754]) | 10\n",
-      "26950    | torch.Size([1, 1000, 754]) | 10\n",
-      "26951    | torch.Size([1, 1000, 754]) | 10\n",
-      "26952    | torch.Size([1, 1000, 762]) | 10\n",
-      "26953    | torch.Size([1, 1000, 762]) | 10\n",
-      "26954    | torch.Size([1, 1000, 762]) | 10\n",
-      "26955    | torch.Size([1, 1000, 762]) | 10\n",
-      "26956    | torch.Size([1, 1000, 754]) | 10\n",
-      "26957    | torch.Size([1, 1000, 770]) | 10\n",
-      "26958    | torch.Size([1, 1000, 802]) | 10\n",
-      "26959    | torch.Size([1, 1000, 845]) | 10\n",
-      "26960    | torch.Size([1, 1000, 774]) | 10\n",
-      "26961    | torch.Size([1, 1000, 754]) | 10\n",
-      "26962    | torch.Size([1, 1000, 762]) | 10\n",
-      "26963    | torch.Size([1, 1000, 754]) | 10\n",
-      "26964    | torch.Size([1, 1000, 754]) | 10\n",
-      "26965    | torch.Size([1, 1000, 754]) | 10\n",
-      "26966    | torch.Size([1, 1000, 777]) | 10\n",
-      "26967    | torch.Size([1, 1000, 754]) | 10\n",
-      "26968    | torch.Size([1, 1000, 772]) | 10\n",
-      "26969    | torch.Size([1, 1000, 777]) | 10\n",
-      "26970    | torch.Size([1, 1000, 754]) | 10\n",
-      "26971    | torch.Size([1, 1000, 754]) | 10\n",
-      "26972    | torch.Size([1, 1000, 754]) | 10\n",
-      "26973    | torch.Size([1, 1000, 772]) | 10\n",
-      "26974    | torch.Size([1, 1000, 777]) | 10\n",
-      "26975    | torch.Size([1, 1000, 794]) | 10\n",
-      "26976    | torch.Size([1, 1000, 754]) | 10\n",
-      "26977    | torch.Size([1, 1000, 754]) | 10\n",
-      "26978    | torch.Size([1, 1000, 762]) | 10\n",
-      "26979    | torch.Size([1, 1000, 777]) | 10\n",
-      "26980    | torch.Size([1, 1000, 774]) | 10\n",
-      "26981    | torch.Size([1, 1000, 740]) | 10\n",
-      "26982    | torch.Size([1, 1000, 754]) | 10\n",
-      "26983    | torch.Size([1, 1000, 754]) | 10\n",
-      "26984    | torch.Size([1, 1000, 754]) | 10\n",
-      "26985    | torch.Size([1, 1000, 754]) | 10\n",
-      "26986    | torch.Size([1, 1000, 762]) | 10\n",
-      "26987    | torch.Size([1, 1000, 777]) | 10\n",
-      "26988    | torch.Size([1, 1000, 754]) | 10\n",
-      "26989    | torch.Size([1, 1000, 754]) | 10\n",
-      "26990    | torch.Size([1, 1000, 777]) | 10\n",
-      "26991    | torch.Size([1, 1000, 789]) | 10\n",
-      "26992    | torch.Size([1, 1000, 809]) | 10\n",
-      "26993    | torch.Size([1, 1000, 776]) | 10\n",
-      "26994    | torch.Size([1, 1000, 754]) | 10\n",
-      "26995    | torch.Size([1, 1000, 754]) | 10\n",
-      "26996    | torch.Size([1, 1000, 771]) | 10\n",
-      "26997    | torch.Size([1, 1000, 802]) | 10\n",
-      "26998    | torch.Size([1, 1000, 754]) | 10\n",
-      "26999    | torch.Size([1, 1000, 754]) | 10\n",
-      "27000    | torch.Size([1, 1000, 777]) | 10\n",
-      "27001    | torch.Size([1, 1000, 789]) | 10\n",
-      "27002    | torch.Size([1, 1000, 754]) | 10\n",
-      "27003    | torch.Size([1, 1000, 754]) | 10\n",
-      "27004    | torch.Size([1, 1000, 754]) | 10\n",
-      "27005    | torch.Size([1, 1000, 754]) | 10\n",
-      "27006    | torch.Size([1, 1000, 754]) | 10\n",
-      "27007    | torch.Size([1, 1000, 762]) | 10\n",
-      "27008    | torch.Size([1, 1000, 811]) | 10\n",
-      "27009    | torch.Size([1, 1000, 754]) | 10\n",
-      "27010    | torch.Size([1, 1000, 786]) | 10\n",
-      "27011    | torch.Size([1, 1000, 754]) | 10\n",
-      "27012    | torch.Size([1, 1000, 777]) | 10\n",
-      "27013    | torch.Size([1, 1000, 754]) | 10\n",
-      "27014    | torch.Size([1, 1000, 754]) | 10\n",
-      "27015    | torch.Size([1, 1000, 754]) | 10\n",
-      "27016    | torch.Size([1, 1000, 754]) | 10\n",
-      "27017    | torch.Size([1, 1000, 762]) | 10\n",
-      "27018    | torch.Size([1, 1000, 796]) | 10\n",
-      "27019    | torch.Size([1, 1000, 788]) | 10\n",
-      "27020    | torch.Size([1, 1000, 762]) | 10\n",
-      "27021    | torch.Size([1, 1000, 802]) | 10\n",
-      "27022    | torch.Size([1, 1000, 754]) | 10\n",
-      "27023    | torch.Size([1, 1000, 772]) | 10\n",
-      "27024    | torch.Size([1, 1000, 788]) | 10\n",
-      "27025    | torch.Size([1, 1000, 754]) | 10\n",
-      "27026    | torch.Size([1, 1000, 785]) | 10\n",
-      "27027    | torch.Size([1, 1000, 713]) | 10\n",
-      "27028    | torch.Size([1, 1000, 754]) | 10\n",
-      "27029    | torch.Size([1, 1000, 754]) | 10\n",
-      "27030    | torch.Size([1, 1000, 777]) | 10\n",
-      "27031    | torch.Size([1, 1000, 754]) | 10\n",
-      "27032    | torch.Size([1, 1000, 754]) | 10\n",
-      "27033    | torch.Size([1, 1000, 841]) | 10\n",
-      "27034    | torch.Size([1, 1000, 802]) | 10\n",
-      "27035    | torch.Size([1, 1000, 754]) | 10\n",
-      "27036    | torch.Size([1, 1000, 780]) | 10\n",
-      "27037    | torch.Size([1, 1000, 767]) | 10\n",
-      "27038    | torch.Size([1, 1000, 777]) | 10\n",
-      "27039    | torch.Size([1, 1000, 754]) | 10\n",
-      "27040    | torch.Size([1, 1000, 762]) | 10\n",
-      "27041    | torch.Size([1, 1000, 754]) | 10\n",
-      "27042    | torch.Size([1, 1000, 777]) | 10\n",
-      "27043    | torch.Size([1, 1000, 784]) | 10\n",
-      "27044    | torch.Size([1, 1000, 796]) | 10\n",
-      "27045    | torch.Size([1, 1000, 754]) | 10\n",
-      "27046    | torch.Size([1, 1000, 762]) | 10\n",
-      "27047    | torch.Size([1, 1000, 764]) | 10\n",
-      "27048    | torch.Size([1, 1000, 771]) | 10\n",
-      "27049    | torch.Size([1, 1000, 754]) | 10\n",
-      "27050    | torch.Size([1, 1000, 754]) | 10\n",
-      "27051    | torch.Size([1, 1000, 754]) | 10\n",
-      "27052    | torch.Size([1, 1000, 754]) | 10\n",
-      "27053    | torch.Size([1, 1000, 771]) | 10\n",
-      "27054    | torch.Size([1, 1000, 754]) | 10\n",
-      "27055    | torch.Size([1, 1000, 754]) | 10\n",
-      "27056    | torch.Size([1, 1000, 762]) | 10\n",
-      "27057    | torch.Size([1, 1000, 754]) | 10\n",
-      "27058    | torch.Size([1, 1000, 754]) | 10\n",
-      "27059    | torch.Size([1, 1000, 754]) | 10\n",
-      "27060    | torch.Size([1, 1000, 754]) | 10\n",
-      "27061    | torch.Size([1, 1000, 754]) | 10\n",
-      "27062    | torch.Size([1, 1000, 787]) | 10\n",
-      "27063    | torch.Size([1, 1000, 777]) | 10\n",
-      "27064    | torch.Size([1, 1000, 754]) | 10\n",
-      "27065    | torch.Size([1, 1000, 777]) | 10\n",
-      "27066    | torch.Size([1, 1000, 781]) | 10\n",
-      "27067    | torch.Size([1, 1000, 754]) | 10\n",
-      "27068    | torch.Size([1, 1000, 777]) | 10\n",
-      "27069    | torch.Size([1, 1000, 754]) | 10\n",
-      "27070    | torch.Size([1, 1000, 777]) | 10\n",
-      "27071    | torch.Size([1, 1000, 754]) | 10\n",
-      "27072    | torch.Size([1, 1000, 801]) | 10\n",
-      "27073    | torch.Size([1, 1000, 775]) | 10\n",
-      "27074    | torch.Size([1, 1000, 754]) | 10\n",
-      "27075    | torch.Size([1, 1000, 754]) | 10\n",
-      "27076    | torch.Size([1, 1000, 762]) | 10\n",
-      "27077    | torch.Size([1, 1000, 768]) | 10\n",
-      "27078    | torch.Size([1, 1000, 754]) | 10\n",
-      "27079    | torch.Size([1, 1000, 772]) | 10\n",
-      "27080    | torch.Size([1, 1000, 713]) | 10\n",
-      "27081    | torch.Size([1, 1000, 754]) | 10\n",
-      "27082    | torch.Size([1, 1000, 762]) | 10\n",
-      "27083    | torch.Size([1, 1000, 754]) | 10\n",
-      "27084    | torch.Size([1, 1000, 768]) | 10\n",
-      "27085    | torch.Size([1, 1000, 723]) | 10\n",
-      "27086    | torch.Size([1, 1000, 754]) | 10\n",
-      "27087    | torch.Size([1, 1000, 802]) | 10\n",
-      "27088    | torch.Size([1, 1000, 754]) | 10\n",
-      "27089    | torch.Size([1, 1000, 802]) | 10\n",
-      "27090    | torch.Size([1, 1000, 754]) | 10\n",
-      "27091    | torch.Size([1, 1000, 777]) | 10\n",
-      "27092    | torch.Size([1, 1000, 777]) | 10\n",
-      "27093    | torch.Size([1, 1000, 754]) | 10\n",
-      "27094    | torch.Size([1, 1000, 754]) | 10\n",
-      "27095    | torch.Size([1, 1000, 762]) | 10\n",
-      "27096    | torch.Size([1, 1000, 754]) | 10\n",
-      "27097    | torch.Size([1, 1000, 762]) | 10\n",
-      "27098    | torch.Size([1, 1000, 762]) | 10\n",
-      "27099    | torch.Size([1, 1000, 777]) | 10\n",
-      "27100    | torch.Size([1, 1000, 780]) | 10\n",
-      "27101    | torch.Size([1, 1000, 754]) | 10\n",
-      "27102    | torch.Size([1, 1000, 754]) | 10\n",
-      "27103    | torch.Size([1, 1000, 777]) | 10\n",
-      "27104    | torch.Size([1, 1000, 754]) | 10\n",
-      "27105    | torch.Size([1, 1000, 777]) | 10\n",
-      "27106    | torch.Size([1, 1000, 762]) | 10\n",
-      "27107    | torch.Size([1, 1000, 762]) | 10\n",
-      "27108    | torch.Size([1, 1000, 754]) | 10\n",
-      "27109    | torch.Size([1, 1000, 762]) | 10\n",
-      "27110    | torch.Size([1, 1000, 754]) | 10\n",
-      "27111    | torch.Size([1, 1000, 762]) | 10\n",
-      "27112    | torch.Size([1, 1000, 777]) | 10\n",
-      "27113    | torch.Size([1, 1000, 780]) | 10\n",
-      "27114    | torch.Size([1, 1000, 777]) | 10\n",
-      "27115    | torch.Size([1, 1000, 754]) | 10\n",
-      "27116    | torch.Size([1, 1000, 762]) | 10\n",
-      "27117    | torch.Size([1, 1000, 754]) | 10\n",
-      "27118    | torch.Size([1, 1000, 754]) | 10\n",
-      "27119    | torch.Size([1, 1000, 777]) | 10\n",
-      "27120    | torch.Size([1, 1000, 754]) | 10\n",
-      "27121    | torch.Size([1, 1000, 754]) | 10\n",
-      "27122    | torch.Size([1, 1000, 754]) | 10\n",
-      "27123    | torch.Size([1, 1000, 754]) | 10\n",
-      "27124    | torch.Size([1, 1000, 777]) | 10\n",
-      "27125    | torch.Size([1, 1000, 754]) | 10\n",
-      "27126    | torch.Size([1, 1000, 783]) | 10\n",
-      "27127    | torch.Size([1, 1000, 781]) | 10\n",
-      "27128    | torch.Size([1, 1000, 797]) | 10\n",
-      "27129    | torch.Size([1, 1000, 754]) | 10\n",
-      "27130    | torch.Size([1, 1000, 777]) | 10\n",
-      "27131    | torch.Size([1, 1000, 754]) | 10\n",
-      "27132    | torch.Size([1, 1000, 754]) | 10\n",
-      "27133    | torch.Size([1, 1000, 754]) | 10\n",
-      "27134    | torch.Size([1, 1000, 754]) | 10\n",
-      "27135    | torch.Size([1, 1000, 782]) | 10\n",
-      "27136    | torch.Size([1, 1000, 753]) | 10\n",
-      "27137    | torch.Size([1, 1000, 786]) | 10\n",
-      "27138    | torch.Size([1, 1000, 777]) | 10\n",
-      "27139    | torch.Size([1, 1000, 777]) | 10\n",
-      "27140    | torch.Size([1, 1000, 792]) | 10\n",
-      "27141    | torch.Size([1, 1000, 754]) | 10\n",
-      "27142    | torch.Size([1, 1000, 754]) | 10\n",
-      "27143    | torch.Size([1, 1000, 754]) | 10\n",
-      "27144    | torch.Size([1, 1000, 754]) | 10\n",
-      "27145    | torch.Size([1, 1000, 754]) | 10\n",
-      "27146    | torch.Size([1, 1000, 754]) | 10\n",
-      "27147    | torch.Size([1, 1000, 776]) | 10\n",
-      "27148    | torch.Size([1, 1000, 725]) | 10\n",
-      "27149    | torch.Size([1, 1000, 762]) | 10\n",
-      "27150    | torch.Size([1, 1000, 754]) | 10\n",
-      "27151    | torch.Size([1, 1000, 777]) | 10\n",
-      "27152    | torch.Size([1, 1000, 754]) | 10\n",
-      "27153    | torch.Size([1, 1000, 754]) | 10\n",
-      "27154    | torch.Size([1, 1000, 764]) | 10\n",
-      "27155    | torch.Size([1, 1000, 754]) | 10\n",
-      "27156    | torch.Size([1, 1000, 754]) | 10\n",
-      "27157    | torch.Size([1, 1000, 754]) | 10\n",
-      "27158    | torch.Size([1, 1000, 754]) | 10\n",
-      "27159    | torch.Size([1, 1000, 754]) | 10\n",
-      "27160    | torch.Size([1, 1000, 784]) | 10\n",
-      "27161    | torch.Size([1, 1000, 777]) | 10\n",
-      "27162    | torch.Size([1, 1000, 754]) | 10\n",
-      "27163    | torch.Size([1, 1000, 776]) | 10\n",
-      "27164    | torch.Size([1, 1000, 754]) | 10\n",
-      "27165    | torch.Size([1, 1000, 754]) | 10\n",
-      "27166    | torch.Size([1, 1000, 777]) | 10\n",
-      "27167    | torch.Size([1, 1000, 754]) | 10\n",
-      "27168    | torch.Size([1, 1000, 754]) | 10\n",
-      "27169    | torch.Size([1, 1000, 762]) | 10\n",
-      "27170    | torch.Size([1, 1000, 754]) | 10\n",
-      "27171    | torch.Size([1, 1000, 754]) | 10\n",
-      "27172    | torch.Size([1, 1000, 777]) | 10\n",
-      "27173    | torch.Size([1, 1000, 787]) | 10\n",
-      "27174    | torch.Size([1, 1000, 777]) | 10\n",
-      "27175    | torch.Size([1, 1000, 777]) | 10\n",
-      "27176    | torch.Size([1, 1000, 754]) | 10\n",
-      "27177    | torch.Size([1, 1000, 754]) | 10\n",
-      "27178    | torch.Size([1, 1000, 754]) | 10\n",
-      "27179    | torch.Size([1, 1000, 777]) | 10\n",
-      "27180    | torch.Size([1, 1000, 754]) | 10\n",
-      "27181    | torch.Size([1, 1000, 802]) | 10\n",
-      "27182    | torch.Size([1, 1000, 754]) | 10\n",
-      "27183    | torch.Size([1, 1000, 754]) | 10\n",
-      "27184    | torch.Size([1, 1000, 754]) | 10\n",
-      "27185    | torch.Size([1, 1000, 792]) | 10\n",
-      "27186    | torch.Size([1, 1000, 762]) | 10\n",
-      "27187    | torch.Size([1, 1000, 807]) | 10\n",
-      "27188    | torch.Size([1, 1000, 802]) | 10\n",
-      "27189    | torch.Size([1, 1000, 777]) | 10\n",
-      "27190    | torch.Size([1, 1000, 754]) | 10\n",
-      "27191    | torch.Size([1, 1000, 754]) | 10\n",
-      "27192    | torch.Size([1, 1000, 762]) | 10\n",
-      "27193    | torch.Size([1, 1000, 754]) | 10\n",
-      "27194    | torch.Size([1, 1000, 810]) | 10\n",
-      "27195    | torch.Size([1, 1000, 754]) | 10\n",
-      "27196    | torch.Size([1, 1000, 754]) | 10\n",
-      "27197    | torch.Size([1, 1000, 777]) | 10\n",
-      "27198    | torch.Size([1, 1000, 762]) | 10\n",
-      "27199    | torch.Size([1, 1000, 762]) | 10\n",
-      "27200    | torch.Size([1, 1000, 811]) | 10\n",
-      "27201    | torch.Size([1, 1000, 762]) | 10\n",
-      "27202    | torch.Size([1, 1000, 754]) | 10\n",
-      "27203    | torch.Size([1, 1000, 754]) | 10\n",
-      "27204    | torch.Size([1, 1000, 754]) | 10\n",
-      "27205    | torch.Size([1, 1000, 754]) | 10\n",
-      "27206    | torch.Size([1, 1000, 777]) | 10\n",
-      "27207    | torch.Size([1, 1000, 777]) | 10\n",
-      "27208    | torch.Size([1, 1000, 762]) | 10\n",
-      "27209    | torch.Size([1, 1000, 754]) | 10\n",
-      "27210    | torch.Size([1, 1000, 754]) | 10\n",
-      "27211    | torch.Size([1, 1000, 780]) | 10\n",
-      "27212    | torch.Size([1, 1000, 777]) | 10\n",
-      "27213    | torch.Size([1, 1000, 762]) | 10\n",
-      "27214    | torch.Size([1, 1000, 754]) | 10\n",
-      "27215    | torch.Size([1, 1000, 777]) | 10\n",
-      "27216    | torch.Size([1, 1000, 777]) | 10\n",
-      "27217    | torch.Size([1, 1000, 777]) | 10\n",
-      "27218    | torch.Size([1, 1000, 754]) | 10\n",
-      "27219    | torch.Size([1, 1000, 754]) | 10\n",
-      "27220    | torch.Size([1, 1000, 762]) | 10\n",
-      "27221    | torch.Size([1, 1000, 777]) | 10\n",
-      "27222    | torch.Size([1, 1000, 775]) | 10\n",
-      "27223    | torch.Size([1, 1000, 778]) | 10\n",
-      "27224    | torch.Size([1, 1000, 754]) | 10\n",
-      "27225    | torch.Size([1, 1000, 777]) | 10\n",
-      "27226    | torch.Size([1, 1000, 777]) | 10\n",
-      "27227    | torch.Size([1, 1000, 754]) | 10\n",
-      "27228    | torch.Size([1, 1000, 754]) | 10\n",
-      "27229    | torch.Size([1, 1000, 721]) | 10\n",
-      "27230    | torch.Size([1, 1000, 811]) | 10\n",
-      "27231    | torch.Size([1, 1000, 796]) | 10\n",
-      "27232    | torch.Size([1, 1000, 783]) | 10\n",
-      "27233    | torch.Size([1, 1000, 771]) | 10\n",
-      "27234    | torch.Size([1, 1000, 754]) | 10\n",
-      "27235    | torch.Size([1, 1000, 777]) | 10\n",
-      "27236    | torch.Size([1, 1000, 772]) | 10\n",
-      "27237    | torch.Size([1, 1000, 762]) | 10\n",
-      "27238    | torch.Size([1, 1000, 754]) | 10\n",
-      "27239    | torch.Size([1, 1000, 779]) | 10\n",
-      "27240    | torch.Size([1, 1000, 762]) | 10\n",
-      "27241    | torch.Size([1, 1000, 777]) | 10\n",
-      "27242    | torch.Size([1, 1000, 754]) | 10\n",
-      "27243    | torch.Size([1, 1000, 703]) | 10\n",
-      "27244    | torch.Size([1, 1000, 777]) | 10\n",
-      "27245    | torch.Size([1, 1000, 777]) | 10\n",
-      "27246    | torch.Size([1, 1000, 754]) | 10\n",
-      "27247    | torch.Size([1, 1000, 754]) | 10\n",
-      "27248    | torch.Size([1, 1000, 754]) | 10\n",
-      "27249    | torch.Size([1, 1000, 802]) | 10\n",
-      "27250    | torch.Size([1, 1000, 754]) | 10\n",
-      "27251    | torch.Size([1, 1000, 802]) | 10\n",
-      "27252    | torch.Size([1, 1000, 754]) | 10\n",
-      "27253    | torch.Size([1, 1000, 754]) | 10\n",
-      "27254    | torch.Size([1, 1000, 802]) | 10\n",
-      "27255    | torch.Size([1, 1000, 762]) | 10\n",
-      "27256    | torch.Size([1, 1000, 754]) | 10\n",
-      "27257    | torch.Size([1, 1000, 762]) | 10\n",
-      "27258    | torch.Size([1, 1000, 769]) | 10\n",
-      "27259    | torch.Size([1, 1000, 799]) | 10\n",
-      "27260    | torch.Size([1, 1000, 762]) | 10\n",
-      "27261    | torch.Size([1, 1000, 754]) | 10\n",
-      "27262    | torch.Size([1, 1000, 769]) | 10\n",
-      "27263    | torch.Size([1, 1000, 754]) | 10\n",
-      "27264    | torch.Size([1, 1000, 754]) | 10\n",
-      "27265    | torch.Size([1, 1000, 754]) | 10\n",
-      "27266    | torch.Size([1, 1000, 754]) | 10\n",
-      "27267    | torch.Size([1, 1000, 754]) | 10\n",
-      "27268    | torch.Size([1, 1000, 754]) | 10\n",
-      "27269    | torch.Size([1, 1000, 777]) | 10\n",
-      "27270    | torch.Size([1, 1000, 754]) | 10\n",
-      "27271    | torch.Size([1, 1000, 762]) | 10\n",
-      "27272    | torch.Size([1, 1000, 762]) | 10\n",
-      "27273    | torch.Size([1, 1000, 754]) | 10\n",
-      "27274    | torch.Size([1, 1000, 762]) | 10\n",
-      "27275    | torch.Size([1, 1000, 754]) | 10\n",
-      "27276    | torch.Size([1, 1000, 754]) | 10\n",
-      "27277    | torch.Size([1, 1000, 754]) | 10\n",
-      "27278    | torch.Size([1, 1000, 754]) | 10\n",
-      "27279    | torch.Size([1, 1000, 784]) | 10\n",
-      "27280    | torch.Size([1, 1000, 754]) | 10\n",
-      "27281    | torch.Size([1, 1000, 762]) | 10\n",
-      "27282    | torch.Size([1, 1000, 754]) | 10\n",
-      "27283    | torch.Size([1, 1000, 754]) | 10\n",
-      "27284    | torch.Size([1, 1000, 762]) | 10\n",
-      "27285    | torch.Size([1, 1000, 754]) | 10\n",
-      "27286    | torch.Size([1, 1000, 777]) | 10\n",
-      "27287    | torch.Size([1, 1000, 803]) | 10\n",
-      "27288    | torch.Size([1, 1000, 772]) | 10\n",
-      "27289    | torch.Size([1, 1000, 762]) | 10\n",
-      "27290    | torch.Size([1, 1000, 754]) | 10\n",
-      "27291    | torch.Size([1, 1000, 780]) | 10\n",
-      "27292    | torch.Size([1, 1000, 803]) | 10\n",
-      "27293    | torch.Size([1, 1000, 777]) | 10\n",
-      "27294    | torch.Size([1, 1000, 754]) | 10\n",
-      "27295    | torch.Size([1, 1000, 754]) | 10\n",
-      "27296    | torch.Size([1, 1000, 750]) | 10\n",
-      "27297    | torch.Size([1, 1000, 754]) | 10\n",
-      "27298    | torch.Size([1, 1000, 754]) | 10\n",
-      "27299    | torch.Size([1, 1000, 762]) | 10\n",
-      "27300    | torch.Size([1, 1000, 777]) | 10\n",
-      "27301    | torch.Size([1, 1000, 762]) | 10\n",
-      "27302    | torch.Size([1, 1000, 754]) | 10\n",
-      "27303    | torch.Size([1, 1000, 762]) | 10\n",
-      "27304    | torch.Size([1, 1000, 754]) | 10\n",
-      "27305    | torch.Size([1, 1000, 754]) | 10\n",
-      "27306    | torch.Size([1, 1000, 777]) | 10\n",
-      "27307    | torch.Size([1, 1000, 762]) | 10\n",
-      "27308    | torch.Size([1, 1000, 777]) | 10\n",
-      "27309    | torch.Size([1, 1000, 754]) | 10\n",
-      "27310    | torch.Size([1, 1000, 802]) | 10\n",
-      "27311    | torch.Size([1, 1000, 754]) | 10\n",
-      "27312    | torch.Size([1, 1000, 754]) | 10\n",
-      "27313    | torch.Size([1, 1000, 754]) | 10\n",
-      "27314    | torch.Size([1, 1000, 754]) | 10\n",
-      "27315    | torch.Size([1, 1000, 777]) | 10\n",
-      "27316    | torch.Size([1, 1000, 778]) | 10\n",
-      "27317    | torch.Size([1, 1000, 754]) | 10\n",
-      "27318    | torch.Size([1, 1000, 754]) | 10\n",
-      "27319    | torch.Size([1, 1000, 762]) | 10\n",
-      "27320    | torch.Size([1, 1000, 774]) | 10\n",
-      "27321    | torch.Size([1, 1000, 777]) | 10\n",
-      "27322    | torch.Size([1, 1000, 754]) | 10\n",
-      "27323    | torch.Size([1, 1000, 777]) | 10\n",
-      "27324    | torch.Size([1, 1000, 754]) | 10\n",
-      "27325    | torch.Size([1, 1000, 777]) | 10\n",
-      "27326    | torch.Size([1, 1000, 762]) | 10\n",
-      "27327    | torch.Size([1, 1000, 781]) | 10\n",
-      "27328    | torch.Size([1, 1000, 754]) | 10\n",
-      "27329    | torch.Size([1, 1000, 773]) | 10\n",
-      "27330    | torch.Size([1, 1000, 611]) | 10\n",
-      "27331    | torch.Size([1, 1000, 754]) | 10\n",
-      "27332    | torch.Size([1, 1000, 777]) | 10\n",
-      "27333    | torch.Size([1, 1000, 754]) | 10\n",
-      "27334    | torch.Size([1, 1000, 777]) | 10\n",
-      "27335    | torch.Size([1, 1000, 754]) | 10\n",
-      "27336    | torch.Size([1, 1000, 777]) | 10\n",
-      "27337    | torch.Size([1, 1000, 754]) | 10\n",
-      "27338    | torch.Size([1, 1000, 779]) | 10\n",
-      "27339    | torch.Size([1, 1000, 754]) | 10\n",
-      "27340    | torch.Size([1, 1000, 754]) | 10\n",
-      "27341    | torch.Size([1, 1000, 754]) | 10\n",
-      "27342    | torch.Size([1, 1000, 754]) | 10\n",
-      "27343    | torch.Size([1, 1000, 754]) | 10\n",
-      "27344    | torch.Size([1, 1000, 754]) | 10\n",
-      "27345    | torch.Size([1, 1000, 777]) | 10\n",
-      "27346    | torch.Size([1, 1000, 754]) | 10\n",
-      "27347    | torch.Size([1, 1000, 754]) | 10\n",
-      "27348    | torch.Size([1, 1000, 754]) | 10\n",
-      "27349    | torch.Size([1, 1000, 754]) | 10\n",
-      "27350    | torch.Size([1, 1000, 778]) | 10\n",
-      "27351    | torch.Size([1, 1000, 754]) | 10\n",
-      "27352    | torch.Size([1, 1000, 754]) | 10\n",
-      "27353    | torch.Size([1, 1000, 754]) | 10\n",
-      "27354    | torch.Size([1, 1000, 762]) | 10\n",
-      "27355    | torch.Size([1, 1000, 786]) | 10\n",
-      "27356    | torch.Size([1, 1000, 754]) | 10\n",
-      "27357    | torch.Size([1, 1000, 754]) | 10\n",
-      "27358    | torch.Size([1, 1000, 754]) | 10\n",
-      "27359    | torch.Size([1, 1000, 754]) | 10\n",
-      "27360    | torch.Size([1, 1000, 810]) | 10\n",
-      "27361    | torch.Size([1, 1000, 762]) | 10\n",
-      "27362    | torch.Size([1, 1000, 777]) | 10\n",
-      "27363    | torch.Size([1, 1000, 754]) | 10\n",
-      "27364    | torch.Size([1, 1000, 754]) | 10\n",
-      "27365    | torch.Size([1, 1000, 791]) | 10\n",
-      "27366    | torch.Size([1, 1000, 802]) | 10\n",
-      "27367    | torch.Size([1, 1000, 754]) | 10\n",
-      "27368    | torch.Size([1, 1000, 754]) | 10\n",
-      "27369    | torch.Size([1, 1000, 762]) | 10\n",
-      "27370    | torch.Size([1, 1000, 802]) | 10\n",
-      "27371    | torch.Size([1, 1000, 754]) | 10\n",
-      "27372    | torch.Size([1, 1000, 777]) | 10\n",
-      "27373    | torch.Size([1, 1000, 762]) | 10\n",
-      "27374    | torch.Size([1, 1000, 754]) | 10\n",
-      "27375    | torch.Size([1, 1000, 782]) | 10\n",
-      "27376    | torch.Size([1, 1000, 777]) | 10\n",
-      "27377    | torch.Size([1, 1000, 754]) | 10\n",
-      "27378    | torch.Size([1, 1000, 777]) | 10\n",
-      "27379    | torch.Size([1, 1000, 754]) | 10\n",
-      "27380    | torch.Size([1, 1000, 754]) | 10\n",
-      "27381    | torch.Size([1, 1000, 802]) | 10\n",
-      "27382    | torch.Size([1, 1000, 754]) | 10\n",
-      "27383    | torch.Size([1, 1000, 754]) | 10\n",
-      "27384    | torch.Size([1, 1000, 762]) | 10\n",
-      "27385    | torch.Size([1, 1000, 754]) | 10\n",
-      "27386    | torch.Size([1, 1000, 777]) | 10\n",
-      "27387    | torch.Size([1, 1000, 777]) | 10\n",
-      "27388    | torch.Size([1, 1000, 754]) | 10\n",
-      "27389    | torch.Size([1, 1000, 802]) | 10\n",
-      "27390    | torch.Size([1, 1000, 754]) | 10\n",
-      "27391    | torch.Size([1, 1000, 754]) | 10\n",
-      "27392    | torch.Size([1, 1000, 754]) | 10\n",
-      "27393    | torch.Size([1, 1000, 754]) | 10\n",
-      "27394    | torch.Size([1, 1000, 754]) | 10\n",
-      "27395    | torch.Size([1, 1000, 754]) | 10\n",
-      "27396    | torch.Size([1, 1000, 754]) | 10\n",
-      "27397    | torch.Size([1, 1000, 754]) | 10\n",
-      "27398    | torch.Size([1, 1000, 867]) | 10\n",
-      "27399    | torch.Size([1, 1000, 762]) | 10\n",
-      "27400    | torch.Size([1, 1000, 754]) | 10\n",
-      "27401    | torch.Size([1, 1000, 848]) | 10\n",
-      "27402    | torch.Size([1, 1000, 778]) | 10\n",
-      "27403    | torch.Size([1, 1000, 754]) | 10\n",
-      "27404    | torch.Size([1, 1000, 754]) | 10\n",
-      "27405    | torch.Size([1, 1000, 754]) | 10\n",
-      "27406    | torch.Size([1, 1000, 762]) | 10\n",
-      "27407    | torch.Size([1, 1000, 777]) | 10\n",
-      "27408    | torch.Size([1, 1000, 777]) | 10\n",
-      "27409    | torch.Size([1, 1000, 754]) | 10\n",
-      "27410    | torch.Size([1, 1000, 754]) | 10\n",
-      "27411    | torch.Size([1, 1000, 768]) | 10\n",
-      "27412    | torch.Size([1, 1000, 754]) | 10\n",
-      "27413    | torch.Size([1, 1000, 754]) | 10\n",
-      "27414    | torch.Size([1, 1000, 788]) | 10\n",
-      "27415    | torch.Size([1, 1000, 754]) | 10\n",
-      "27416    | torch.Size([1, 1000, 754]) | 10\n",
-      "27417    | torch.Size([1, 1000, 777]) | 10\n",
-      "27418    | torch.Size([1, 1000, 762]) | 10\n",
-      "27419    | torch.Size([1, 1000, 754]) | 10\n",
-      "27420    | torch.Size([1, 1000, 772]) | 10\n",
-      "27421    | torch.Size([1, 1000, 754]) | 10\n",
-      "27422    | torch.Size([1, 1000, 754]) | 10\n",
-      "27423    | torch.Size([1, 1000, 754]) | 10\n",
-      "27424    | torch.Size([1, 1000, 809]) | 10\n",
-      "27425    | torch.Size([1, 1000, 777]) | 10\n",
-      "27426    | torch.Size([1, 1000, 754]) | 10\n",
-      "27427    | torch.Size([1, 1000, 768]) | 10\n",
-      "27428    | torch.Size([1, 1000, 754]) | 10\n",
-      "27429    | torch.Size([1, 1000, 754]) | 10\n",
-      "27430    | torch.Size([1, 1000, 777]) | 10\n",
-      "27431    | torch.Size([1, 1000, 774]) | 10\n",
-      "27432    | torch.Size([1, 1000, 754]) | 10\n",
-      "27433    | torch.Size([1, 1000, 783]) | 10\n",
-      "27434    | torch.Size([1, 1000, 754]) | 10\n",
-      "27435    | torch.Size([1, 1000, 754]) | 10\n",
-      "27436    | torch.Size([1, 1000, 754]) | 10\n",
-      "27437    | torch.Size([1, 1000, 754]) | 10\n",
-      "27438    | torch.Size([1, 1000, 754]) | 10\n",
-      "27439    | torch.Size([1, 1000, 754]) | 10\n",
-      "27440    | torch.Size([1, 1000, 784]) | 10\n",
-      "27441    | torch.Size([1, 1000, 773]) | 10\n",
-      "27442    | torch.Size([1, 1000, 843]) | 10\n",
-      "27443    | torch.Size([1, 1000, 754]) | 10\n",
-      "27444    | torch.Size([1, 1000, 754]) | 10\n",
-      "27445    | torch.Size([1, 1000, 754]) | 10\n",
-      "27446    | torch.Size([1, 1000, 754]) | 10\n",
-      "27447    | torch.Size([1, 1000, 754]) | 10\n",
-      "27448    | torch.Size([1, 1000, 754]) | 10\n",
-      "27449    | torch.Size([1, 1000, 754]) | 10\n",
-      "27450    | torch.Size([1, 1000, 754]) | 10\n",
-      "27451    | torch.Size([1, 1000, 754]) | 10\n",
-      "27452    | torch.Size([1, 1000, 777]) | 10\n",
-      "27453    | torch.Size([1, 1000, 843]) | 10\n",
-      "27454    | torch.Size([1, 1000, 754]) | 10\n",
-      "27455    | torch.Size([1, 1000, 754]) | 10\n",
-      "27456    | torch.Size([1, 1000, 754]) | 10\n",
-      "27457    | torch.Size([1, 1000, 762]) | 10\n",
-      "27458    | torch.Size([1, 1000, 754]) | 10\n",
-      "27459    | torch.Size([1, 1000, 777]) | 10\n",
-      "27460    | torch.Size([1, 1000, 754]) | 10\n",
-      "27461    | torch.Size([1, 1000, 754]) | 10\n",
-      "27462    | torch.Size([1, 1000, 754]) | 10\n",
-      "27463    | torch.Size([1, 1000, 754]) | 10\n",
-      "27464    | torch.Size([1, 1000, 754]) | 10\n",
-      "27465    | torch.Size([1, 1000, 764]) | 10\n",
-      "27466    | torch.Size([1, 1000, 777]) | 10\n",
-      "27467    | torch.Size([1, 1000, 754]) | 10\n",
-      "27468    | torch.Size([1, 1000, 775]) | 10\n",
-      "27469    | torch.Size([1, 1000, 762]) | 10\n",
-      "27470    | torch.Size([1, 1000, 777]) | 10\n",
-      "27471    | torch.Size([1, 1000, 762]) | 10\n",
-      "27472    | torch.Size([1, 1000, 754]) | 10\n",
-      "27473    | torch.Size([1, 1000, 781]) | 10\n",
-      "27474    | torch.Size([1, 1000, 754]) | 10\n",
-      "27475    | torch.Size([1, 1000, 754]) | 10\n",
-      "27476    | torch.Size([1, 1000, 762]) | 10\n",
-      "27477    | torch.Size([1, 1000, 777]) | 10\n",
-      "27478    | torch.Size([1, 1000, 762]) | 10\n",
-      "27479    | torch.Size([1, 1000, 754]) | 10\n",
-      "27480    | torch.Size([1, 1000, 779]) | 10\n",
-      "27481    | torch.Size([1, 1000, 777]) | 10\n",
-      "27482    | torch.Size([1, 1000, 799]) | 10\n",
-      "27483    | torch.Size([1, 1000, 777]) | 10\n",
-      "27484    | torch.Size([1, 1000, 777]) | 10\n",
-      "27485    | torch.Size([1, 1000, 794]) | 10\n",
-      "27486    | torch.Size([1, 1000, 773]) | 11\n",
-      "27487    | torch.Size([1, 1000, 777]) | 11\n",
-      "27488    | torch.Size([1, 1000, 777]) | 11\n",
-      "27489    | torch.Size([1, 1000, 848]) | 11\n",
-      "27490    | torch.Size([1, 1000, 767]) | 11\n",
-      "27491    | torch.Size([1, 1000, 786]) | 11\n",
-      "27492    | torch.Size([1, 1000, 754]) | 11\n",
-      "27493    | torch.Size([1, 1000, 754]) | 11\n",
-      "27494    | torch.Size([1, 1000, 762]) | 11\n",
-      "27495    | torch.Size([1, 1000, 762]) | 11\n",
-      "27496    | torch.Size([1, 1000, 754]) | 11\n",
-      "27497    | torch.Size([1, 1000, 645]) | 11\n",
-      "27498    | torch.Size([1, 1000, 846]) | 11\n",
-      "27499    | torch.Size([1, 1000, 800]) | 11\n",
-      "27500    | torch.Size([1, 1000, 773]) | 11\n",
-      "27501    | torch.Size([1, 1000, 762]) | 11\n",
-      "27502    | torch.Size([1, 1000, 802]) | 11\n",
-      "27503    | torch.Size([1, 1000, 762]) | 11\n",
-      "27504    | torch.Size([1, 1000, 773]) | 11\n",
-      "27505    | torch.Size([1, 1000, 771]) | 11\n",
-      "27506    | torch.Size([1, 1000, 796]) | 11\n",
-      "27507    | torch.Size([1, 1000, 754]) | 11\n",
-      "27508    | torch.Size([1, 1000, 762]) | 11\n",
-      "27509    | torch.Size([1, 1000, 777]) | 11\n",
-      "27510    | torch.Size([1, 1000, 767]) | 11\n",
-      "27511    | torch.Size([1, 1000, 762]) | 11\n",
-      "27512    | torch.Size([1, 1000, 713]) | 11\n",
-      "27513    | torch.Size([1, 1000, 762]) | 11\n",
-      "27514    | torch.Size([1, 1000, 754]) | 11\n",
-      "27515    | torch.Size([1, 1000, 762]) | 11\n",
-      "27516    | torch.Size([1, 1000, 754]) | 11\n",
-      "27517    | torch.Size([1, 1000, 762]) | 11\n",
-      "27518    | torch.Size([1, 1000, 778]) | 11\n",
-      "27519    | torch.Size([1, 1000, 754]) | 11\n",
-      "27520    | torch.Size([1, 1000, 754]) | 11\n",
-      "27521    | torch.Size([1, 1000, 777]) | 11\n",
-      "27522    | torch.Size([1, 1000, 777]) | 11\n",
-      "27523    | torch.Size([1, 1000, 762]) | 11\n",
-      "27524    | torch.Size([1, 1000, 772]) | 11\n",
-      "27525    | torch.Size([1, 1000, 777]) | 11\n",
-      "27526    | torch.Size([1, 1000, 754]) | 11\n",
-      "27527    | torch.Size([1, 1000, 803]) | 11\n",
-      "27528    | torch.Size([1, 1000, 852]) | 11\n",
-      "27529    | torch.Size([1, 1000, 782]) | 11\n",
-      "27530    | torch.Size([1, 1000, 754]) | 11\n",
-      "27531    | torch.Size([1, 1000, 754]) | 11\n",
-      "27532    | torch.Size([1, 1000, 754]) | 11\n",
-      "27533    | torch.Size([1, 1000, 754]) | 11\n",
-      "27534    | torch.Size([1, 1000, 762]) | 11\n",
-      "27535    | torch.Size([1, 1000, 803]) | 11\n",
-      "27536    | torch.Size([1, 1000, 754]) | 11\n",
-      "27537    | torch.Size([1, 1000, 779]) | 11\n",
-      "27538    | torch.Size([1, 1000, 754]) | 11\n",
-      "27539    | torch.Size([1, 1000, 754]) | 11\n",
-      "27540    | torch.Size([1, 1000, 754]) | 11\n",
-      "27541    | torch.Size([1, 1000, 777]) | 11\n",
-      "27542    | torch.Size([1, 1000, 789]) | 11\n",
-      "27543    | torch.Size([1, 1000, 762]) | 11\n",
-      "27544    | torch.Size([1, 1000, 754]) | 11\n",
-      "27545    | torch.Size([1, 1000, 847]) | 11\n",
-      "27546    | torch.Size([1, 1000, 754]) | 11\n",
-      "27547    | torch.Size([1, 1000, 783]) | 11\n",
-      "27548    | torch.Size([1, 1000, 773]) | 11\n",
-      "27549    | torch.Size([1, 1000, 796]) | 11\n",
-      "27550    | torch.Size([1, 1000, 783]) | 11\n",
-      "27551    | torch.Size([1, 1000, 754]) | 11\n",
-      "27552    | torch.Size([1, 1000, 762]) | 11\n",
-      "27553    | torch.Size([1, 1000, 810]) | 11\n",
-      "27554    | torch.Size([1, 1000, 762]) | 11\n",
-      "27555    | torch.Size([1, 1000, 802]) | 11\n",
-      "27556    | torch.Size([1, 1000, 798]) | 11\n",
-      "27557    | torch.Size([1, 1000, 761]) | 11\n",
-      "27558    | torch.Size([1, 1000, 777]) | 11\n",
-      "27559    | torch.Size([1, 1000, 762]) | 11\n",
-      "27560    | torch.Size([1, 1000, 762]) | 11\n",
-      "27561    | torch.Size([1, 1000, 754]) | 11\n",
-      "27562    | torch.Size([1, 1000, 797]) | 11\n",
-      "27563    | torch.Size([1, 1000, 762]) | 11\n",
-      "27564    | torch.Size([1, 1000, 777]) | 11\n",
-      "27565    | torch.Size([1, 1000, 754]) | 11\n",
-      "27566    | torch.Size([1, 1000, 754]) | 11\n",
-      "27567    | torch.Size([1, 1000, 859]) | 11\n",
-      "27568    | torch.Size([1, 1000, 775]) | 11\n",
-      "27569    | torch.Size([1, 1000, 662]) | 11\n",
-      "27570    | torch.Size([1, 1000, 796]) | 11\n",
-      "27571    | torch.Size([1, 1000, 778]) | 11\n",
-      "27572    | torch.Size([1, 1000, 780]) | 11\n",
-      "27573    | torch.Size([1, 1000, 780]) | 11\n",
-      "27574    | torch.Size([1, 1000, 762]) | 11\n",
-      "27575    | torch.Size([1, 1000, 754]) | 11\n",
-      "27576    | torch.Size([1, 1000, 774]) | 11\n",
-      "27577    | torch.Size([1, 1000, 778]) | 11\n",
-      "27578    | torch.Size([1, 1000, 762]) | 11\n",
-      "27579    | torch.Size([1, 1000, 771]) | 11\n",
-      "27580    | torch.Size([1, 1000, 777]) | 11\n",
-      "27581    | torch.Size([1, 1000, 754]) | 11\n",
-      "27582    | torch.Size([1, 1000, 614]) | 11\n",
-      "27583    | torch.Size([1, 1000, 719]) | 11\n",
-      "27584    | torch.Size([1, 1000, 777]) | 11\n",
-      "27585    | torch.Size([1, 1000, 777]) | 11\n",
-      "27586    | torch.Size([1, 1000, 634]) | 11\n",
-      "27587    | torch.Size([1, 1000, 802]) | 11\n",
-      "27588    | torch.Size([1, 1000, 754]) | 11\n",
-      "27589    | torch.Size([1, 1000, 762]) | 11\n",
-      "27590    | torch.Size([1, 1000, 776]) | 11\n",
-      "27591    | torch.Size([1, 1000, 838]) | 11\n",
-      "27592    | torch.Size([1, 1000, 772]) | 11\n",
-      "27593    | torch.Size([1, 1000, 754]) | 11\n",
-      "27594    | torch.Size([1, 1000, 754]) | 11\n",
-      "27595    | torch.Size([1, 1000, 762]) | 11\n",
-      "27596    | torch.Size([1, 1000, 757]) | 11\n",
-      "27597    | torch.Size([1, 1000, 754]) | 11\n",
-      "27598    | torch.Size([1, 1000, 754]) | 11\n",
-      "27599    | torch.Size([1, 1000, 777]) | 11\n",
-      "27600    | torch.Size([1, 1000, 762]) | 11\n",
-      "27601    | torch.Size([1, 1000, 754]) | 11\n",
-      "27602    | torch.Size([1, 1000, 776]) | 11\n",
-      "27603    | torch.Size([1, 1000, 617]) | 11\n",
-      "27604    | torch.Size([1, 1000, 777]) | 11\n",
-      "27605    | torch.Size([1, 1000, 777]) | 11\n",
-      "27606    | torch.Size([1, 1000, 754]) | 11\n",
-      "27607    | torch.Size([1, 1000, 762]) | 11\n",
-      "27608    | torch.Size([1, 1000, 645]) | 11\n",
-      "27609    | torch.Size([1, 1000, 754]) | 11\n",
-      "27610    | torch.Size([1, 1000, 754]) | 11\n",
-      "27611    | torch.Size([1, 1000, 754]) | 11\n",
-      "27612    | torch.Size([1, 1000, 754]) | 11\n",
-      "27613    | torch.Size([1, 1000, 777]) | 11\n",
-      "27614    | torch.Size([1, 1000, 754]) | 11\n",
-      "27615    | torch.Size([1, 1000, 754]) | 11\n",
-      "27616    | torch.Size([1, 1000, 772]) | 11\n",
-      "27617    | torch.Size([1, 1000, 775]) | 11\n",
-      "27618    | torch.Size([1, 1000, 754]) | 11\n",
-      "27619    | torch.Size([1, 1000, 767]) | 11\n",
-      "27620    | torch.Size([1, 1000, 762]) | 11\n",
-      "27621    | torch.Size([1, 1000, 777]) | 11\n",
-      "27622    | torch.Size([1, 1000, 762]) | 11\n",
-      "27623    | torch.Size([1, 1000, 777]) | 11\n",
-      "27624    | torch.Size([1, 1000, 754]) | 11\n",
-      "27625    | torch.Size([1, 1000, 754]) | 11\n",
-      "27626    | torch.Size([1, 1000, 762]) | 11\n",
-      "27627    | torch.Size([1, 1000, 765]) | 11\n",
-      "27628    | torch.Size([1, 1000, 777]) | 11\n",
-      "27629    | torch.Size([1, 1000, 810]) | 11\n",
-      "27630    | torch.Size([1, 1000, 782]) | 11\n",
-      "27631    | torch.Size([1, 1000, 762]) | 11\n",
-      "27632    | torch.Size([1, 1000, 797]) | 11\n",
-      "27633    | torch.Size([1, 1000, 754]) | 11\n",
-      "27634    | torch.Size([1, 1000, 777]) | 11\n",
-      "27635    | torch.Size([1, 1000, 762]) | 11\n",
-      "27636    | torch.Size([1, 1000, 613]) | 11\n",
-      "27637    | torch.Size([1, 1000, 762]) | 11\n",
-      "27638    | torch.Size([1, 1000, 762]) | 11\n",
-      "27639    | torch.Size([1, 1000, 754]) | 11\n",
-      "27640    | torch.Size([1, 1000, 777]) | 11\n",
-      "27641    | torch.Size([1, 1000, 764]) | 11\n",
-      "27642    | torch.Size([1, 1000, 762]) | 11\n",
-      "27643    | torch.Size([1, 1000, 769]) | 11\n",
-      "27644    | torch.Size([1, 1000, 777]) | 11\n",
-      "27645    | torch.Size([1, 1000, 754]) | 11\n",
-      "27646    | torch.Size([1, 1000, 777]) | 11\n",
-      "27647    | torch.Size([1, 1000, 741]) | 11\n",
-      "27648    | torch.Size([1, 1000, 774]) | 11\n",
-      "27649    | torch.Size([1, 1000, 754]) | 11\n",
-      "27650    | torch.Size([1, 1000, 754]) | 11\n",
-      "27651    | torch.Size([1, 1000, 754]) | 11\n",
-      "27652    | torch.Size([1, 1000, 762]) | 11\n",
-      "27653    | torch.Size([1, 1000, 772]) | 11\n",
-      "27654    | torch.Size([1, 1000, 611]) | 11\n",
-      "27655    | torch.Size([1, 1000, 777]) | 11\n",
-      "27656    | torch.Size([1, 1000, 754]) | 11\n",
-      "27657    | torch.Size([1, 1000, 762]) | 11\n",
-      "27658    | torch.Size([1, 1000, 777]) | 11\n",
-      "27659    | torch.Size([1, 1000, 754]) | 11\n",
-      "27660    | torch.Size([1, 1000, 774]) | 11\n",
-      "27661    | torch.Size([1, 1000, 754]) | 11\n",
-      "27662    | torch.Size([1, 1000, 763]) | 11\n",
-      "27663    | torch.Size([1, 1000, 754]) | 11\n",
-      "27664    | torch.Size([1, 1000, 792]) | 11\n",
-      "27665    | torch.Size([1, 1000, 754]) | 11\n",
-      "27666    | torch.Size([1, 1000, 754]) | 11\n",
-      "27667    | torch.Size([1, 1000, 764]) | 11\n",
-      "27668    | torch.Size([1, 1000, 762]) | 11\n",
-      "27669    | torch.Size([1, 1000, 614]) | 11\n",
-      "27670    | torch.Size([1, 1000, 754]) | 11\n",
-      "27671    | torch.Size([1, 1000, 754]) | 11\n",
-      "27672    | torch.Size([1, 1000, 848]) | 11\n",
-      "27673    | torch.Size([1, 1000, 754]) | 11\n",
-      "27674    | torch.Size([1, 1000, 802]) | 11\n",
-      "27675    | torch.Size([1, 1000, 754]) | 11\n",
-      "27676    | torch.Size([1, 1000, 779]) | 11\n",
-      "27677    | torch.Size([1, 1000, 754]) | 11\n",
-      "27678    | torch.Size([1, 1000, 754]) | 11\n",
-      "27679    | torch.Size([1, 1000, 813]) | 11\n",
-      "27680    | torch.Size([1, 1000, 769]) | 11\n",
-      "27681    | torch.Size([1, 1000, 802]) | 11\n",
-      "27682    | torch.Size([1, 1000, 754]) | 11\n",
-      "27683    | torch.Size([1, 1000, 777]) | 11\n",
-      "27684    | torch.Size([1, 1000, 754]) | 11\n",
-      "27685    | torch.Size([1, 1000, 754]) | 11\n",
-      "27686    | torch.Size([1, 1000, 802]) | 11\n",
-      "27687    | torch.Size([1, 1000, 754]) | 11\n",
-      "27688    | torch.Size([1, 1000, 754]) | 11\n",
-      "27689    | torch.Size([1, 1000, 796]) | 11\n",
-      "27690    | torch.Size([1, 1000, 754]) | 11\n",
-      "27691    | torch.Size([1, 1000, 754]) | 11\n",
-      "27692    | torch.Size([1, 1000, 786]) | 11\n",
-      "27693    | torch.Size([1, 1000, 783]) | 11\n",
-      "27694    | torch.Size([1, 1000, 802]) | 11\n",
-      "27695    | torch.Size([1, 1000, 754]) | 11\n",
-      "27696    | torch.Size([1, 1000, 780]) | 11\n",
-      "27697    | torch.Size([1, 1000, 762]) | 11\n",
-      "27698    | torch.Size([1, 1000, 769]) | 11\n",
-      "27699    | torch.Size([1, 1000, 762]) | 11\n",
-      "27700    | torch.Size([1, 1000, 777]) | 11\n",
-      "27701    | torch.Size([1, 1000, 795]) | 11\n",
-      "27702    | torch.Size([1, 1000, 742]) | 11\n",
-      "27703    | torch.Size([1, 1000, 779]) | 11\n",
-      "27704    | torch.Size([1, 1000, 772]) | 11\n",
-      "27705    | torch.Size([1, 1000, 754]) | 11\n",
-      "27706    | torch.Size([1, 1000, 761]) | 11\n",
-      "27707    | torch.Size([1, 1000, 754]) | 11\n",
-      "27708    | torch.Size([1, 1000, 797]) | 11\n",
-      "27709    | torch.Size([1, 1000, 842]) | 11\n",
-      "27710    | torch.Size([1, 1000, 762]) | 11\n",
-      "27711    | torch.Size([1, 1000, 764]) | 11\n",
-      "27712    | torch.Size([1, 1000, 754]) | 11\n",
-      "27713    | torch.Size([1, 1000, 754]) | 11\n",
-      "27714    | torch.Size([1, 1000, 762]) | 11\n",
-      "27715    | torch.Size([1, 1000, 754]) | 11\n",
-      "27716    | torch.Size([1, 1000, 754]) | 11\n",
-      "27717    | torch.Size([1, 1000, 617]) | 11\n",
-      "27718    | torch.Size([1, 1000, 754]) | 11\n",
-      "27719    | torch.Size([1, 1000, 773]) | 11\n",
-      "27720    | torch.Size([1, 1000, 777]) | 11\n",
-      "27721    | torch.Size([1, 1000, 754]) | 11\n",
-      "27722    | torch.Size([1, 1000, 778]) | 11\n",
-      "27723    | torch.Size([1, 1000, 754]) | 11\n",
-      "27724    | torch.Size([1, 1000, 762]) | 11\n",
-      "27725    | torch.Size([1, 1000, 754]) | 11\n",
-      "27726    | torch.Size([1, 1000, 762]) | 11\n",
-      "27727    | torch.Size([1, 1000, 762]) | 11\n",
-      "27728    | torch.Size([1, 1000, 807]) | 11\n",
-      "27729    | torch.Size([1, 1000, 786]) | 11\n",
-      "27730    | torch.Size([1, 1000, 785]) | 11\n",
-      "27731    | torch.Size([1, 1000, 754]) | 11\n",
-      "27732    | torch.Size([1, 1000, 785]) | 11\n",
-      "27733    | torch.Size([1, 1000, 754]) | 11\n",
-      "27734    | torch.Size([1, 1000, 762]) | 11\n",
-      "27735    | torch.Size([1, 1000, 754]) | 11\n",
-      "27736    | torch.Size([1, 1000, 777]) | 11\n",
-      "27737    | torch.Size([1, 1000, 754]) | 11\n",
-      "27738    | torch.Size([1, 1000, 762]) | 11\n",
-      "27739    | torch.Size([1, 1000, 762]) | 11\n",
-      "27740    | torch.Size([1, 1000, 762]) | 11\n",
-      "27741    | torch.Size([1, 1000, 762]) | 11\n",
-      "27742    | torch.Size([1, 1000, 782]) | 11\n",
-      "27743    | torch.Size([1, 1000, 754]) | 11\n",
-      "27744    | torch.Size([1, 1000, 777]) | 11\n",
-      "27745    | torch.Size([1, 1000, 754]) | 11\n",
-      "27746    | torch.Size([1, 1000, 762]) | 11\n",
-      "27747    | torch.Size([1, 1000, 796]) | 11\n",
-      "27748    | torch.Size([1, 1000, 613]) | 11\n",
-      "27749    | torch.Size([1, 1000, 754]) | 11\n",
-      "27750    | torch.Size([1, 1000, 762]) | 11\n",
-      "27751    | torch.Size([1, 1000, 808]) | 11\n",
-      "27752    | torch.Size([1, 1000, 762]) | 11\n",
-      "27753    | torch.Size([1, 1000, 754]) | 11\n",
-      "27754    | torch.Size([1, 1000, 772]) | 11\n",
-      "27755    | torch.Size([1, 1000, 856]) | 11\n",
-      "27756    | torch.Size([1, 1000, 808]) | 11\n",
-      "27757    | torch.Size([1, 1000, 777]) | 11\n",
-      "27758    | torch.Size([1, 1000, 777]) | 11\n",
-      "27759    | torch.Size([1, 1000, 802]) | 11\n",
-      "27760    | torch.Size([1, 1000, 762]) | 11\n",
-      "27761    | torch.Size([1, 1000, 792]) | 11\n",
-      "27762    | torch.Size([1, 1000, 754]) | 11\n",
-      "27763    | torch.Size([1, 1000, 802]) | 11\n",
-      "27764    | torch.Size([1, 1000, 762]) | 11\n",
-      "27765    | torch.Size([1, 1000, 777]) | 11\n",
-      "27766    | torch.Size([1, 1000, 762]) | 11\n",
-      "27767    | torch.Size([1, 1000, 754]) | 11\n",
-      "27768    | torch.Size([1, 1000, 803]) | 11\n",
-      "27769    | torch.Size([1, 1000, 762]) | 11\n",
-      "27770    | torch.Size([1, 1000, 671]) | 11\n",
-      "27771    | torch.Size([1, 1000, 762]) | 11\n",
-      "27772    | torch.Size([1, 1000, 621]) | 11\n",
-      "27773    | torch.Size([1, 1000, 777]) | 11\n",
-      "27774    | torch.Size([1, 1000, 754]) | 11\n",
-      "27775    | torch.Size([1, 1000, 741]) | 11\n",
-      "27776    | torch.Size([1, 1000, 754]) | 11\n",
-      "27777    | torch.Size([1, 1000, 754]) | 11\n",
-      "27778    | torch.Size([1, 1000, 762]) | 11\n",
-      "27779    | torch.Size([1, 1000, 834]) | 11\n",
-      "27780    | torch.Size([1, 1000, 611]) | 11\n",
-      "27781    | torch.Size([1, 1000, 754]) | 11\n",
-      "27782    | torch.Size([1, 1000, 742]) | 11\n",
-      "27783    | torch.Size([1, 1000, 754]) | 11\n",
-      "27784    | torch.Size([1, 1000, 611]) | 11\n",
-      "27785    | torch.Size([1, 1000, 800]) | 11\n",
-      "27786    | torch.Size([1, 1000, 765]) | 11\n",
-      "27787    | torch.Size([1, 1000, 754]) | 11\n",
-      "27788    | torch.Size([1, 1000, 783]) | 11\n",
-      "27789    | torch.Size([1, 1000, 754]) | 11\n",
-      "27790    | torch.Size([1, 1000, 777]) | 11\n",
-      "27791    | torch.Size([1, 1000, 762]) | 11\n",
-      "27792    | torch.Size([1, 1000, 762]) | 11\n",
-      "27793    | torch.Size([1, 1000, 792]) | 11\n",
-      "27794    | torch.Size([1, 1000, 764]) | 11\n",
-      "27795    | torch.Size([1, 1000, 778]) | 11\n",
-      "27796    | torch.Size([1, 1000, 754]) | 11\n",
-      "27797    | torch.Size([1, 1000, 743]) | 11\n",
-      "27798    | torch.Size([1, 1000, 762]) | 11\n",
-      "27799    | torch.Size([1, 1000, 778]) | 11\n",
-      "27800    | torch.Size([1, 1000, 777]) | 11\n",
-      "27801    | torch.Size([1, 1000, 788]) | 11\n",
-      "27802    | torch.Size([1, 1000, 781]) | 11\n",
-      "27803    | torch.Size([1, 1000, 765]) | 11\n",
-      "27804    | torch.Size([1, 1000, 762]) | 11\n",
-      "27805    | torch.Size([1, 1000, 762]) | 11\n",
-      "27806    | torch.Size([1, 1000, 762]) | 11\n",
-      "27807    | torch.Size([1, 1000, 779]) | 11\n",
-      "27808    | torch.Size([1, 1000, 762]) | 11\n",
-      "27809    | torch.Size([1, 1000, 778]) | 11\n",
-      "27810    | torch.Size([1, 1000, 754]) | 11\n",
-      "27811    | torch.Size([1, 1000, 762]) | 11\n",
-      "27812    | torch.Size([1, 1000, 754]) | 11\n",
-      "27813    | torch.Size([1, 1000, 777]) | 11\n",
-      "27814    | torch.Size([1, 1000, 762]) | 11\n",
-      "27815    | torch.Size([1, 1000, 762]) | 11\n",
-      "27816    | torch.Size([1, 1000, 754]) | 11\n",
-      "27817    | torch.Size([1, 1000, 613]) | 11\n",
-      "27818    | torch.Size([1, 1000, 745]) | 11\n",
-      "27819    | torch.Size([1, 1000, 754]) | 11\n",
-      "27820    | torch.Size([1, 1000, 765]) | 11\n",
-      "27821    | torch.Size([1, 1000, 754]) | 11\n",
-      "27822    | torch.Size([1, 1000, 773]) | 11\n",
-      "27823    | torch.Size([1, 1000, 754]) | 11\n",
-      "27824    | torch.Size([1, 1000, 778]) | 11\n",
-      "27825    | torch.Size([1, 1000, 773]) | 11\n",
-      "27826    | torch.Size([1, 1000, 762]) | 11\n",
-      "27827    | torch.Size([1, 1000, 762]) | 11\n",
-      "27828    | torch.Size([1, 1000, 762]) | 11\n",
-      "27829    | torch.Size([1, 1000, 777]) | 11\n",
-      "27830    | torch.Size([1, 1000, 762]) | 11\n",
-      "27831    | torch.Size([1, 1000, 611]) | 11\n",
-      "27832    | torch.Size([1, 1000, 771]) | 11\n",
-      "27833    | torch.Size([1, 1000, 754]) | 11\n",
-      "27834    | torch.Size([1, 1000, 802]) | 11\n",
-      "27835    | torch.Size([1, 1000, 646]) | 11\n",
-      "27836    | torch.Size([1, 1000, 754]) | 11\n",
-      "27837    | torch.Size([1, 1000, 796]) | 11\n",
-      "27838    | torch.Size([1, 1000, 777]) | 11\n",
-      "27839    | torch.Size([1, 1000, 772]) | 11\n",
-      "27840    | torch.Size([1, 1000, 762]) | 11\n",
-      "27841    | torch.Size([1, 1000, 777]) | 11\n",
-      "27842    | torch.Size([1, 1000, 771]) | 11\n",
-      "27843    | torch.Size([1, 1000, 797]) | 11\n",
-      "27844    | torch.Size([1, 1000, 754]) | 11\n",
-      "27845    | torch.Size([1, 1000, 792]) | 11\n",
-      "27846    | torch.Size([1, 1000, 754]) | 11\n",
-      "27847    | torch.Size([1, 1000, 803]) | 11\n",
-      "27848    | torch.Size([1, 1000, 754]) | 11\n",
-      "27849    | torch.Size([1, 1000, 762]) | 11\n",
-      "27850    | torch.Size([1, 1000, 754]) | 11\n",
-      "27851    | torch.Size([1, 1000, 769]) | 11\n",
-      "27852    | torch.Size([1, 1000, 762]) | 11\n",
-      "27853    | torch.Size([1, 1000, 786]) | 11\n",
-      "27854    | torch.Size([1, 1000, 762]) | 11\n",
-      "27855    | torch.Size([1, 1000, 773]) | 11\n",
-      "27856    | torch.Size([1, 1000, 754]) | 11\n",
-      "27857    | torch.Size([1, 1000, 754]) | 11\n",
-      "27858    | torch.Size([1, 1000, 771]) | 11\n",
-      "27859    | torch.Size([1, 1000, 754]) | 11\n",
-      "27860    | torch.Size([1, 1000, 770]) | 11\n",
-      "27861    | torch.Size([1, 1000, 648]) | 11\n",
-      "27862    | torch.Size([1, 1000, 762]) | 11\n",
-      "27863    | torch.Size([1, 1000, 754]) | 11\n",
-      "27864    | torch.Size([1, 1000, 636]) | 11\n",
-      "27865    | torch.Size([1, 1000, 780]) | 11\n",
-      "27866    | torch.Size([1, 1000, 611]) | 11\n",
-      "27867    | torch.Size([1, 1000, 800]) | 11\n",
-      "27868    | torch.Size([1, 1000, 777]) | 11\n",
-      "27869    | torch.Size([1, 1000, 762]) | 11\n",
-      "27870    | torch.Size([1, 1000, 767]) | 11\n",
-      "27871    | torch.Size([1, 1000, 802]) | 11\n",
-      "27872    | torch.Size([1, 1000, 762]) | 11\n",
-      "27873    | torch.Size([1, 1000, 762]) | 11\n",
-      "27874    | torch.Size([1, 1000, 771]) | 11\n",
-      "27875    | torch.Size([1, 1000, 771]) | 11\n",
-      "27876    | torch.Size([1, 1000, 777]) | 11\n",
-      "27877    | torch.Size([1, 1000, 754]) | 11\n",
-      "27878    | torch.Size([1, 1000, 754]) | 11\n",
-      "27879    | torch.Size([1, 1000, 774]) | 11\n",
-      "27880    | torch.Size([1, 1000, 777]) | 11\n",
-      "27881    | torch.Size([1, 1000, 762]) | 11\n",
-      "27882    | torch.Size([1, 1000, 767]) | 11\n",
-      "27883    | torch.Size([1, 1000, 754]) | 11\n",
-      "27884    | torch.Size([1, 1000, 762]) | 11\n",
-      "27885    | torch.Size([1, 1000, 810]) | 11\n",
-      "27886    | torch.Size([1, 1000, 771]) | 11\n",
-      "27887    | torch.Size([1, 1000, 811]) | 11\n",
-      "27888    | torch.Size([1, 1000, 847]) | 11\n",
-      "27889    | torch.Size([1, 1000, 809]) | 11\n",
-      "27890    | torch.Size([1, 1000, 762]) | 11\n",
-      "27891    | torch.Size([1, 1000, 810]) | 11\n",
-      "27892    | torch.Size([1, 1000, 781]) | 11\n",
-      "27893    | torch.Size([1, 1000, 754]) | 11\n",
-      "27894    | torch.Size([1, 1000, 824]) | 11\n",
-      "27895    | torch.Size([1, 1000, 782]) | 11\n",
-      "27896    | torch.Size([1, 1000, 768]) | 11\n",
-      "27897    | torch.Size([1, 1000, 778]) | 11\n",
-      "27898    | torch.Size([1, 1000, 754]) | 11\n",
-      "27899    | torch.Size([1, 1000, 754]) | 11\n",
-      "27900    | torch.Size([1, 1000, 637]) | 11\n",
-      "27901    | torch.Size([1, 1000, 762]) | 11\n",
-      "27902    | torch.Size([1, 1000, 783]) | 11\n",
-      "27903    | torch.Size([1, 1000, 754]) | 11\n",
-      "27904    | torch.Size([1, 1000, 790]) | 11\n",
-      "27905    | torch.Size([1, 1000, 779]) | 11\n",
-      "27906    | torch.Size([1, 1000, 754]) | 11\n",
-      "27907    | torch.Size([1, 1000, 754]) | 11\n",
-      "27908    | torch.Size([1, 1000, 762]) | 11\n",
-      "27909    | torch.Size([1, 1000, 754]) | 11\n",
-      "27910    | torch.Size([1, 1000, 811]) | 11\n",
-      "27911    | torch.Size([1, 1000, 754]) | 11\n",
-      "27912    | torch.Size([1, 1000, 762]) | 11\n",
-      "27913    | torch.Size([1, 1000, 811]) | 11\n",
-      "27914    | torch.Size([1, 1000, 843]) | 11\n",
-      "27915    | torch.Size([1, 1000, 793]) | 11\n",
-      "27916    | torch.Size([1, 1000, 741]) | 11\n",
-      "27917    | torch.Size([1, 1000, 762]) | 11\n",
-      "27918    | torch.Size([1, 1000, 623]) | 11\n",
-      "27919    | torch.Size([1, 1000, 754]) | 11\n",
-      "27920    | torch.Size([1, 1000, 775]) | 11\n",
-      "27921    | torch.Size([1, 1000, 762]) | 11\n",
-      "27922    | torch.Size([1, 1000, 767]) | 11\n",
-      "27923    | torch.Size([1, 1000, 754]) | 11\n",
-      "27924    | torch.Size([1, 1000, 754]) | 11\n",
-      "27925    | torch.Size([1, 1000, 754]) | 11\n",
-      "27926    | torch.Size([1, 1000, 754]) | 11\n",
-      "27927    | torch.Size([1, 1000, 762]) | 11\n",
-      "27928    | torch.Size([1, 1000, 786]) | 11\n",
-      "27929    | torch.Size([1, 1000, 754]) | 11\n",
-      "27930    | torch.Size([1, 1000, 754]) | 11\n",
-      "27931    | torch.Size([1, 1000, 773]) | 11\n",
-      "27932    | torch.Size([1, 1000, 782]) | 11\n",
-      "27933    | torch.Size([1, 1000, 754]) | 11\n",
-      "27934    | torch.Size([1, 1000, 777]) | 11\n",
-      "27935    | torch.Size([1, 1000, 780]) | 11\n",
-      "27936    | torch.Size([1, 1000, 662]) | 11\n",
-      "27937    | torch.Size([1, 1000, 789]) | 11\n",
-      "27938    | torch.Size([1, 1000, 754]) | 11\n",
-      "27939    | torch.Size([1, 1000, 754]) | 11\n",
-      "27940    | torch.Size([1, 1000, 764]) | 11\n",
-      "27941    | torch.Size([1, 1000, 777]) | 11\n",
-      "27942    | torch.Size([1, 1000, 611]) | 11\n",
-      "27943    | torch.Size([1, 1000, 777]) | 11\n",
-      "27944    | torch.Size([1, 1000, 754]) | 11\n",
-      "27945    | torch.Size([1, 1000, 754]) | 11\n",
-      "27946    | torch.Size([1, 1000, 771]) | 11\n",
-      "27947    | torch.Size([1, 1000, 754]) | 11\n",
-      "27948    | torch.Size([1, 1000, 762]) | 11\n",
-      "27949    | torch.Size([1, 1000, 616]) | 11\n",
-      "27950    | torch.Size([1, 1000, 801]) | 11\n",
-      "27951    | torch.Size([1, 1000, 754]) | 11\n",
-      "27952    | torch.Size([1, 1000, 611]) | 11\n",
-      "27953    | torch.Size([1, 1000, 762]) | 11\n",
-      "27954    | torch.Size([1, 1000, 754]) | 11\n",
-      "27955    | torch.Size([1, 1000, 617]) | 11\n",
-      "27956    | torch.Size([1, 1000, 808]) | 11\n",
-      "27957    | torch.Size([1, 1000, 777]) | 11\n",
-      "27958    | torch.Size([1, 1000, 791]) | 11\n",
-      "27959    | torch.Size([1, 1000, 773]) | 11\n",
-      "27960    | torch.Size([1, 1000, 781]) | 11\n",
-      "27961    | torch.Size([1, 1000, 783]) | 11\n",
-      "27962    | torch.Size([1, 1000, 780]) | 11\n",
-      "27963    | torch.Size([1, 1000, 762]) | 11\n",
-      "27964    | torch.Size([1, 1000, 611]) | 11\n",
-      "27965    | torch.Size([1, 1000, 791]) | 11\n",
-      "27966    | torch.Size([1, 1000, 802]) | 11\n",
-      "27967    | torch.Size([1, 1000, 754]) | 11\n",
-      "27968    | torch.Size([1, 1000, 778]) | 11\n",
-      "27969    | torch.Size([1, 1000, 754]) | 11\n",
-      "27970    | torch.Size([1, 1000, 762]) | 11\n",
-      "27971    | torch.Size([1, 1000, 767]) | 11\n",
-      "27972    | torch.Size([1, 1000, 771]) | 11\n",
-      "27973    | torch.Size([1, 1000, 754]) | 11\n",
-      "27974    | torch.Size([1, 1000, 754]) | 11\n",
-      "27975    | torch.Size([1, 1000, 766]) | 11\n",
-      "27976    | torch.Size([1, 1000, 754]) | 11\n",
-      "27977    | torch.Size([1, 1000, 792]) | 11\n",
-      "27978    | torch.Size([1, 1000, 770]) | 11\n",
-      "27979    | torch.Size([1, 1000, 611]) | 11\n",
-      "27980    | torch.Size([1, 1000, 762]) | 11\n",
-      "27981    | torch.Size([1, 1000, 754]) | 11\n",
-      "27982    | torch.Size([1, 1000, 782]) | 11\n",
-      "27983    | torch.Size([1, 1000, 777]) | 11\n",
-      "27984    | torch.Size([1, 1000, 773]) | 11\n",
-      "27985    | torch.Size([1, 1000, 754]) | 11\n",
-      "27986    | torch.Size([1, 1000, 797]) | 11\n",
-      "27987    | torch.Size([1, 1000, 754]) | 11\n",
-      "27988    | torch.Size([1, 1000, 762]) | 11\n",
-      "27989    | torch.Size([1, 1000, 825]) | 11\n",
-      "27990    | torch.Size([1, 1000, 754]) | 11\n",
-      "27991    | torch.Size([1, 1000, 754]) | 11\n",
-      "27992    | torch.Size([1, 1000, 777]) | 11\n",
-      "27993    | torch.Size([1, 1000, 762]) | 11\n",
-      "27994    | torch.Size([1, 1000, 754]) | 11\n",
-      "27995    | torch.Size([1, 1000, 775]) | 11\n",
-      "27996    | torch.Size([1, 1000, 773]) | 11\n",
-      "27997    | torch.Size([1, 1000, 777]) | 11\n",
-      "27998    | torch.Size([1, 1000, 777]) | 11\n",
-      "27999    | torch.Size([1, 1000, 777]) | 11\n",
-      "28000    | torch.Size([1, 1000, 777]) | 11\n",
-      "28001    | torch.Size([1, 1000, 762]) | 11\n",
-      "28002    | torch.Size([1, 1000, 754]) | 11\n",
-      "28003    | torch.Size([1, 1000, 762]) | 11\n",
-      "28004    | torch.Size([1, 1000, 754]) | 11\n",
-      "28005    | torch.Size([1, 1000, 833]) | 11\n",
-      "28006    | torch.Size([1, 1000, 779]) | 11\n",
-      "28007    | torch.Size([1, 1000, 754]) | 11\n",
-      "28008    | torch.Size([1, 1000, 781]) | 11\n",
-      "28009    | torch.Size([1, 1000, 766]) | 11\n",
-      "28010    | torch.Size([1, 1000, 637]) | 11\n",
-      "28011    | torch.Size([1, 1000, 754]) | 11\n",
-      "28012    | torch.Size([1, 1000, 777]) | 11\n",
-      "28013    | torch.Size([1, 1000, 762]) | 11\n",
-      "28014    | torch.Size([1, 1000, 777]) | 11\n",
-      "28015    | torch.Size([1, 1000, 798]) | 11\n",
-      "28016    | torch.Size([1, 1000, 766]) | 11\n",
-      "28017    | torch.Size([1, 1000, 754]) | 11\n",
-      "28018    | torch.Size([1, 1000, 811]) | 11\n",
-      "28019    | torch.Size([1, 1000, 777]) | 11\n",
-      "28020    | torch.Size([1, 1000, 754]) | 11\n",
-      "28021    | torch.Size([1, 1000, 774]) | 11\n",
-      "28022    | torch.Size([1, 1000, 754]) | 11\n",
-      "28023    | torch.Size([1, 1000, 754]) | 11\n",
-      "28024    | torch.Size([1, 1000, 847]) | 11\n",
-      "28025    | torch.Size([1, 1000, 773]) | 11\n",
-      "28026    | torch.Size([1, 1000, 754]) | 11\n",
-      "28027    | torch.Size([1, 1000, 754]) | 11\n",
-      "28028    | torch.Size([1, 1000, 739]) | 11\n",
-      "28029    | torch.Size([1, 1000, 800]) | 11\n",
-      "28030    | torch.Size([1, 1000, 754]) | 11\n",
-      "28031    | torch.Size([1, 1000, 754]) | 11\n",
-      "28032    | torch.Size([1, 1000, 754]) | 11\n",
-      "28033    | torch.Size([1, 1000, 754]) | 11\n",
-      "28034    | torch.Size([1, 1000, 787]) | 11\n",
-      "28035    | torch.Size([1, 1000, 771]) | 11\n",
-      "28036    | torch.Size([1, 1000, 795]) | 11\n",
-      "28037    | torch.Size([1, 1000, 778]) | 11\n",
-      "28038    | torch.Size([1, 1000, 762]) | 11\n",
-      "28039    | torch.Size([1, 1000, 762]) | 11\n",
-      "28040    | torch.Size([1, 1000, 772]) | 11\n",
-      "28041    | torch.Size([1, 1000, 773]) | 11\n",
-      "28042    | torch.Size([1, 1000, 777]) | 11\n",
-      "28043    | torch.Size([1, 1000, 766]) | 11\n",
-      "28044    | torch.Size([1, 1000, 754]) | 11\n",
-      "28045    | torch.Size([1, 1000, 777]) | 11\n",
-      "28046    | torch.Size([1, 1000, 754]) | 11\n",
-      "28047    | torch.Size([1, 1000, 754]) | 11\n",
-      "28048    | torch.Size([1, 1000, 786]) | 11\n",
-      "28049    | torch.Size([1, 1000, 762]) | 11\n",
-      "28050    | torch.Size([1, 1000, 754]) | 11\n",
-      "28051    | torch.Size([1, 1000, 777]) | 11\n",
-      "28052    | torch.Size([1, 1000, 777]) | 11\n",
-      "28053    | torch.Size([1, 1000, 802]) | 11\n",
-      "28054    | torch.Size([1, 1000, 754]) | 11\n",
-      "28055    | torch.Size([1, 1000, 762]) | 11\n",
-      "28056    | torch.Size([1, 1000, 771]) | 11\n",
-      "28057    | torch.Size([1, 1000, 762]) | 11\n",
-      "28058    | torch.Size([1, 1000, 762]) | 11\n",
-      "28059    | torch.Size([1, 1000, 754]) | 11\n",
-      "28060    | torch.Size([1, 1000, 754]) | 11\n",
-      "28061    | torch.Size([1, 1000, 754]) | 11\n",
-      "28062    | torch.Size([1, 1000, 762]) | 11\n",
-      "28063    | torch.Size([1, 1000, 754]) | 11\n",
-      "28064    | torch.Size([1, 1000, 754]) | 11\n",
-      "28065    | torch.Size([1, 1000, 754]) | 11\n",
-      "28066    | torch.Size([1, 1000, 754]) | 11\n",
-      "28067    | torch.Size([1, 1000, 769]) | 11\n",
-      "28068    | torch.Size([1, 1000, 754]) | 11\n",
-      "28069    | torch.Size([1, 1000, 864]) | 11\n",
-      "28070    | torch.Size([1, 1000, 800]) | 11\n",
-      "28071    | torch.Size([1, 1000, 754]) | 11\n",
-      "28072    | torch.Size([1, 1000, 762]) | 11\n",
-      "28073    | torch.Size([1, 1000, 777]) | 11\n",
-      "28074    | torch.Size([1, 1000, 754]) | 11\n",
-      "28075    | torch.Size([1, 1000, 754]) | 11\n",
-      "28076    | torch.Size([1, 1000, 762]) | 11\n",
-      "28077    | torch.Size([1, 1000, 777]) | 11\n",
-      "28078    | torch.Size([1, 1000, 762]) | 11\n",
-      "28079    | torch.Size([1, 1000, 765]) | 11\n",
-      "28080    | torch.Size([1, 1000, 774]) | 11\n",
-      "28081    | torch.Size([1, 1000, 769]) | 11\n",
-      "28082    | torch.Size([1, 1000, 754]) | 11\n",
-      "28083    | torch.Size([1, 1000, 783]) | 11\n",
-      "28084    | torch.Size([1, 1000, 762]) | 11\n",
-      "28085    | torch.Size([1, 1000, 754]) | 11\n",
-      "28086    | torch.Size([1, 1000, 762]) | 11\n",
-      "28087    | torch.Size([1, 1000, 767]) | 11\n",
-      "28088    | torch.Size([1, 1000, 800]) | 11\n",
-      "28089    | torch.Size([1, 1000, 754]) | 11\n",
-      "28090    | torch.Size([1, 1000, 762]) | 11\n",
-      "28091    | torch.Size([1, 1000, 727]) | 11\n",
-      "28092    | torch.Size([1, 1000, 777]) | 11\n",
-      "28093    | torch.Size([1, 1000, 778]) | 11\n",
-      "28094    | torch.Size([1, 1000, 762]) | 11\n",
-      "28095    | torch.Size([1, 1000, 762]) | 11\n",
-      "28096    | torch.Size([1, 1000, 777]) | 11\n",
-      "28097    | torch.Size([1, 1000, 856]) | 11\n",
-      "28098    | torch.Size([1, 1000, 754]) | 11\n",
-      "28099    | torch.Size([1, 1000, 846]) | 11\n",
-      "28100    | torch.Size([1, 1000, 754]) | 11\n",
-      "28101    | torch.Size([1, 1000, 754]) | 11\n",
-      "28102    | torch.Size([1, 1000, 762]) | 11\n",
-      "28103    | torch.Size([1, 1000, 771]) | 11\n",
-      "28104    | torch.Size([1, 1000, 754]) | 11\n",
-      "28105    | torch.Size([1, 1000, 780]) | 11\n",
-      "28106    | torch.Size([1, 1000, 776]) | 11\n",
-      "28107    | torch.Size([1, 1000, 762]) | 11\n",
-      "28108    | torch.Size([1, 1000, 754]) | 11\n",
-      "28109    | torch.Size([1, 1000, 801]) | 11\n",
-      "28110    | torch.Size([1, 1000, 777]) | 11\n",
-      "28111    | torch.Size([1, 1000, 777]) | 11\n",
-      "28112    | torch.Size([1, 1000, 754]) | 11\n",
-      "28113    | torch.Size([1, 1000, 777]) | 11\n",
-      "28114    | torch.Size([1, 1000, 754]) | 11\n",
-      "28115    | torch.Size([1, 1000, 754]) | 11\n",
-      "28116    | torch.Size([1, 1000, 611]) | 11\n",
-      "28117    | torch.Size([1, 1000, 754]) | 11\n",
-      "28118    | torch.Size([1, 1000, 754]) | 11\n",
-      "28119    | torch.Size([1, 1000, 777]) | 11\n",
-      "28120    | torch.Size([1, 1000, 771]) | 11\n",
-      "28121    | torch.Size([1, 1000, 754]) | 11\n",
-      "28122    | torch.Size([1, 1000, 771]) | 11\n",
-      "28123    | torch.Size([1, 1000, 831]) | 11\n",
-      "28124    | torch.Size([1, 1000, 762]) | 11\n",
-      "28125    | torch.Size([1, 1000, 808]) | 11\n",
-      "28126    | torch.Size([1, 1000, 769]) | 11\n",
-      "28127    | torch.Size([1, 1000, 846]) | 11\n",
-      "28128    | torch.Size([1, 1000, 762]) | 11\n",
-      "28129    | torch.Size([1, 1000, 777]) | 11\n",
-      "28130    | torch.Size([1, 1000, 773]) | 11\n",
-      "28131    | torch.Size([1, 1000, 762]) | 11\n",
-      "28132    | torch.Size([1, 1000, 814]) | 11\n",
-      "28133    | torch.Size([1, 1000, 777]) | 11\n",
-      "28134    | torch.Size([1, 1000, 785]) | 11\n",
-      "28135    | torch.Size([1, 1000, 754]) | 11\n",
-      "28136    | torch.Size([1, 1000, 792]) | 11\n",
-      "28137    | torch.Size([1, 1000, 754]) | 11\n",
-      "28138    | torch.Size([1, 1000, 762]) | 11\n",
-      "28139    | torch.Size([1, 1000, 765]) | 11\n",
-      "28140    | torch.Size([1, 1000, 781]) | 11\n",
-      "28141    | torch.Size([1, 1000, 793]) | 11\n",
-      "28142    | torch.Size([1, 1000, 771]) | 11\n",
-      "28143    | torch.Size([1, 1000, 754]) | 11\n",
-      "28144    | torch.Size([1, 1000, 778]) | 11\n",
-      "28145    | torch.Size([1, 1000, 676]) | 11\n",
-      "28146    | torch.Size([1, 1000, 787]) | 11\n",
-      "28147    | torch.Size([1, 1000, 644]) | 11\n",
-      "28148    | torch.Size([1, 1000, 762]) | 11\n",
-      "28149    | torch.Size([1, 1000, 770]) | 11\n",
-      "28150    | torch.Size([1, 1000, 791]) | 11\n",
-      "28151    | torch.Size([1, 1000, 771]) | 11\n",
-      "28152    | torch.Size([1, 1000, 802]) | 11\n",
-      "28153    | torch.Size([1, 1000, 754]) | 11\n",
-      "28154    | torch.Size([1, 1000, 807]) | 11\n",
-      "28155    | torch.Size([1, 1000, 634]) | 11\n",
-      "28156    | torch.Size([1, 1000, 788]) | 11\n",
-      "28157    | torch.Size([1, 1000, 754]) | 11\n",
-      "28158    | torch.Size([1, 1000, 762]) | 11\n",
-      "28159    | torch.Size([1, 1000, 776]) | 11\n",
-      "28160    | torch.Size([1, 1000, 762]) | 11\n",
-      "28161    | torch.Size([1, 1000, 762]) | 11\n",
-      "28162    | torch.Size([1, 1000, 762]) | 11\n",
-      "28163    | torch.Size([1, 1000, 754]) | 11\n",
-      "28164    | torch.Size([1, 1000, 778]) | 11\n",
-      "28165    | torch.Size([1, 1000, 754]) | 11\n",
-      "28166    | torch.Size([1, 1000, 777]) | 11\n",
-      "28167    | torch.Size([1, 1000, 762]) | 11\n",
-      "28168    | torch.Size([1, 1000, 762]) | 11\n",
-      "28169    | torch.Size([1, 1000, 796]) | 11\n",
-      "28170    | torch.Size([1, 1000, 777]) | 11\n",
-      "28171    | torch.Size([1, 1000, 754]) | 11\n",
-      "28172    | torch.Size([1, 1000, 754]) | 11\n",
-      "28173    | torch.Size([1, 1000, 754]) | 11\n",
-      "28174    | torch.Size([1, 1000, 777]) | 11\n",
-      "28175    | torch.Size([1, 1000, 754]) | 11\n",
-      "28176    | torch.Size([1, 1000, 754]) | 11\n",
-      "28177    | torch.Size([1, 1000, 773]) | 11\n",
-      "28178    | torch.Size([1, 1000, 762]) | 11\n",
-      "28179    | torch.Size([1, 1000, 771]) | 11\n",
-      "28180    | torch.Size([1, 1000, 778]) | 11\n",
-      "28181    | torch.Size([1, 1000, 754]) | 11\n",
-      "28182    | torch.Size([1, 1000, 754]) | 11\n",
-      "28183    | torch.Size([1, 1000, 777]) | 11\n",
-      "28184    | torch.Size([1, 1000, 763]) | 11\n",
-      "28185    | torch.Size([1, 1000, 754]) | 11\n",
-      "28186    | torch.Size([1, 1000, 777]) | 11\n",
-      "28187    | torch.Size([1, 1000, 768]) | 11\n",
-      "28188    | torch.Size([1, 1000, 777]) | 11\n",
-      "28189    | torch.Size([1, 1000, 762]) | 11\n",
-      "28190    | torch.Size([1, 1000, 771]) | 11\n",
-      "28191    | torch.Size([1, 1000, 766]) | 11\n",
-      "28192    | torch.Size([1, 1000, 777]) | 11\n",
-      "28193    | torch.Size([1, 1000, 824]) | 11\n",
-      "28194    | torch.Size([1, 1000, 611]) | 11\n",
-      "28195    | torch.Size([1, 1000, 663]) | 11\n",
-      "28196    | torch.Size([1, 1000, 754]) | 11\n",
-      "28197    | torch.Size([1, 1000, 791]) | 11\n",
-      "28198    | torch.Size([1, 1000, 754]) | 11\n",
-      "28199    | torch.Size([1, 1000, 782]) | 11\n",
-      "28200    | torch.Size([1, 1000, 778]) | 11\n",
-      "28201    | torch.Size([1, 1000, 777]) | 11\n",
-      "28202    | torch.Size([1, 1000, 742]) | 11\n",
-      "28203    | torch.Size([1, 1000, 754]) | 11\n",
-      "28204    | torch.Size([1, 1000, 777]) | 11\n",
-      "28205    | torch.Size([1, 1000, 762]) | 11\n",
-      "28206    | torch.Size([1, 1000, 762]) | 11\n",
-      "28207    | torch.Size([1, 1000, 809]) | 11\n",
-      "28208    | torch.Size([1, 1000, 777]) | 11\n",
-      "28209    | torch.Size([1, 1000, 754]) | 11\n",
-      "28210    | torch.Size([1, 1000, 762]) | 11\n",
-      "28211    | torch.Size([1, 1000, 754]) | 11\n",
-      "28212    | torch.Size([1, 1000, 777]) | 11\n",
-      "28213    | torch.Size([1, 1000, 762]) | 11\n",
-      "28214    | torch.Size([1, 1000, 777]) | 11\n",
-      "28215    | torch.Size([1, 1000, 766]) | 11\n",
-      "28216    | torch.Size([1, 1000, 754]) | 11\n",
-      "28217    | torch.Size([1, 1000, 762]) | 11\n",
-      "28218    | torch.Size([1, 1000, 762]) | 11\n",
-      "28219    | torch.Size([1, 1000, 762]) | 11\n",
-      "28220    | torch.Size([1, 1000, 766]) | 11\n",
-      "28221    | torch.Size([1, 1000, 782]) | 11\n",
-      "28222    | torch.Size([1, 1000, 782]) | 11\n",
-      "28223    | torch.Size([1, 1000, 754]) | 11\n",
-      "28224    | torch.Size([1, 1000, 754]) | 11\n",
-      "28225    | torch.Size([1, 1000, 754]) | 11\n",
-      "28226    | torch.Size([1, 1000, 802]) | 11\n",
-      "28227    | torch.Size([1, 1000, 771]) | 11\n",
-      "28228    | torch.Size([1, 1000, 766]) | 11\n",
-      "28229    | torch.Size([1, 1000, 764]) | 11\n",
-      "28230    | torch.Size([1, 1000, 808]) | 11\n",
-      "28231    | torch.Size([1, 1000, 762]) | 11\n",
-      "28232    | torch.Size([1, 1000, 777]) | 11\n",
-      "28233    | torch.Size([1, 1000, 754]) | 11\n",
-      "28234    | torch.Size([1, 1000, 754]) | 11\n",
-      "28235    | torch.Size([1, 1000, 777]) | 11\n",
-      "28236    | torch.Size([1, 1000, 762]) | 11\n",
-      "28237    | torch.Size([1, 1000, 777]) | 11\n",
-      "28238    | torch.Size([1, 1000, 762]) | 11\n",
-      "28239    | torch.Size([1, 1000, 754]) | 11\n",
-      "28240    | torch.Size([1, 1000, 804]) | 11\n",
-      "28241    | torch.Size([1, 1000, 754]) | 11\n",
-      "28242    | torch.Size([1, 1000, 769]) | 11\n",
-      "28243    | torch.Size([1, 1000, 772]) | 11\n",
-      "28244    | torch.Size([1, 1000, 847]) | 11\n",
-      "28245    | torch.Size([1, 1000, 762]) | 11\n",
-      "28246    | torch.Size([1, 1000, 777]) | 11\n",
-      "28247    | torch.Size([1, 1000, 762]) | 11\n",
-      "28248    | torch.Size([1, 1000, 754]) | 11\n",
-      "28249    | torch.Size([1, 1000, 796]) | 11\n",
-      "28250    | torch.Size([1, 1000, 771]) | 11\n",
-      "28251    | torch.Size([1, 1000, 754]) | 11\n",
-      "28252    | torch.Size([1, 1000, 754]) | 11\n",
-      "28253    | torch.Size([1, 1000, 777]) | 11\n",
-      "28254    | torch.Size([1, 1000, 784]) | 11\n",
-      "28255    | torch.Size([1, 1000, 774]) | 11\n",
-      "28256    | torch.Size([1, 1000, 802]) | 11\n",
-      "28257    | torch.Size([1, 1000, 754]) | 11\n",
-      "28258    | torch.Size([1, 1000, 754]) | 11\n",
-      "28259    | torch.Size([1, 1000, 777]) | 11\n",
-      "28260    | torch.Size([1, 1000, 754]) | 11\n",
-      "28261    | torch.Size([1, 1000, 766]) | 11\n",
-      "28262    | torch.Size([1, 1000, 754]) | 11\n",
-      "28263    | torch.Size([1, 1000, 843]) | 11\n",
-      "28264    | torch.Size([1, 1000, 754]) | 11\n",
-      "28265    | torch.Size([1, 1000, 777]) | 11\n",
-      "28266    | torch.Size([1, 1000, 754]) | 11\n",
-      "28267    | torch.Size([1, 1000, 754]) | 11\n",
-      "28268    | torch.Size([1, 1000, 754]) | 11\n",
-      "28269    | torch.Size([1, 1000, 777]) | 11\n",
-      "28270    | torch.Size([1, 1000, 762]) | 11\n",
-      "28271    | torch.Size([1, 1000, 762]) | 11\n",
-      "28272    | torch.Size([1, 1000, 781]) | 11\n",
-      "28273    | torch.Size([1, 1000, 777]) | 11\n",
-      "28274    | torch.Size([1, 1000, 622]) | 11\n",
-      "28275    | torch.Size([1, 1000, 762]) | 11\n",
-      "28276    | torch.Size([1, 1000, 776]) | 11\n",
-      "28277    | torch.Size([1, 1000, 797]) | 11\n",
-      "28278    | torch.Size([1, 1000, 754]) | 11\n",
-      "28279    | torch.Size([1, 1000, 850]) | 11\n",
-      "28280    | torch.Size([1, 1000, 847]) | 11\n",
-      "28281    | torch.Size([1, 1000, 771]) | 11\n",
-      "28282    | torch.Size([1, 1000, 624]) | 11\n",
-      "28283    | torch.Size([1, 1000, 782]) | 11\n",
-      "28284    | torch.Size([1, 1000, 754]) | 11\n",
-      "28285    | torch.Size([1, 1000, 793]) | 11\n",
-      "28286    | torch.Size([1, 1000, 839]) | 11\n",
-      "28287    | torch.Size([1, 1000, 762]) | 11\n",
-      "28288    | torch.Size([1, 1000, 762]) | 11\n",
-      "28289    | torch.Size([1, 1000, 762]) | 11\n",
-      "28290    | torch.Size([1, 1000, 771]) | 11\n",
-      "28291    | torch.Size([1, 1000, 762]) | 11\n",
-      "28292    | torch.Size([1, 1000, 754]) | 11\n",
-      "28293    | torch.Size([1, 1000, 754]) | 11\n",
-      "28294    | torch.Size([1, 1000, 762]) | 11\n",
-      "28295    | torch.Size([1, 1000, 762]) | 11\n",
-      "28296    | torch.Size([1, 1000, 791]) | 11\n",
-      "28297    | torch.Size([1, 1000, 614]) | 11\n",
-      "28298    | torch.Size([1, 1000, 754]) | 11\n",
-      "28299    | torch.Size([1, 1000, 778]) | 11\n",
-      "28300    | torch.Size([1, 1000, 754]) | 11\n",
-      "28301    | torch.Size([1, 1000, 783]) | 11\n",
-      "28302    | torch.Size([1, 1000, 754]) | 11\n",
-      "28303    | torch.Size([1, 1000, 777]) | 11\n",
-      "28304    | torch.Size([1, 1000, 754]) | 11\n",
-      "28305    | torch.Size([1, 1000, 780]) | 11\n",
-      "28306    | torch.Size([1, 1000, 754]) | 11\n",
-      "28307    | torch.Size([1, 1000, 754]) | 11\n",
-      "28308    | torch.Size([1, 1000, 764]) | 11\n",
-      "28309    | torch.Size([1, 1000, 754]) | 11\n",
-      "28310    | torch.Size([1, 1000, 802]) | 11\n",
-      "28311    | torch.Size([1, 1000, 782]) | 11\n",
-      "28312    | torch.Size([1, 1000, 649]) | 11\n",
-      "28313    | torch.Size([1, 1000, 762]) | 11\n",
-      "28314    | torch.Size([1, 1000, 789]) | 11\n",
-      "28315    | torch.Size([1, 1000, 754]) | 11\n",
-      "28316    | torch.Size([1, 1000, 781]) | 11\n",
-      "28317    | torch.Size([1, 1000, 800]) | 11\n",
-      "28318    | torch.Size([1, 1000, 849]) | 11\n",
-      "28319    | torch.Size([1, 1000, 762]) | 11\n",
-      "28320    | torch.Size([1, 1000, 754]) | 11\n",
-      "28321    | torch.Size([1, 1000, 776]) | 11\n",
-      "28322    | torch.Size([1, 1000, 754]) | 11\n",
-      "28323    | torch.Size([1, 1000, 781]) | 11\n",
-      "28324    | torch.Size([1, 1000, 799]) | 11\n",
-      "28325    | torch.Size([1, 1000, 754]) | 11\n",
-      "28326    | torch.Size([1, 1000, 754]) | 11\n",
-      "28327    | torch.Size([1, 1000, 754]) | 11\n",
-      "28328    | torch.Size([1, 1000, 777]) | 11\n",
-      "28329    | torch.Size([1, 1000, 754]) | 11\n",
-      "28330    | torch.Size([1, 1000, 803]) | 11\n",
-      "28331    | torch.Size([1, 1000, 762]) | 11\n",
-      "28332    | torch.Size([1, 1000, 783]) | 11\n",
-      "28333    | torch.Size([1, 1000, 766]) | 11\n",
-      "28334    | torch.Size([1, 1000, 754]) | 11\n",
-      "28335    | torch.Size([1, 1000, 762]) | 11\n",
-      "28336    | torch.Size([1, 1000, 762]) | 11\n",
-      "28337    | torch.Size([1, 1000, 762]) | 11\n",
-      "28338    | torch.Size([1, 1000, 762]) | 11\n",
-      "28339    | torch.Size([1, 1000, 754]) | 11\n",
-      "28340    | torch.Size([1, 1000, 754]) | 11\n",
-      "28341    | torch.Size([1, 1000, 759]) | 11\n",
-      "28342    | torch.Size([1, 1000, 754]) | 11\n",
-      "28343    | torch.Size([1, 1000, 808]) | 11\n",
-      "28344    | torch.Size([1, 1000, 754]) | 11\n",
-      "28345    | torch.Size([1, 1000, 754]) | 11\n",
-      "28346    | torch.Size([1, 1000, 754]) | 11\n",
-      "28347    | torch.Size([1, 1000, 800]) | 11\n",
-      "28348    | torch.Size([1, 1000, 754]) | 11\n",
-      "28349    | torch.Size([1, 1000, 754]) | 11\n",
-      "28350    | torch.Size([1, 1000, 754]) | 11\n",
-      "28351    | torch.Size([1, 1000, 754]) | 11\n",
-      "28352    | torch.Size([1, 1000, 777]) | 11\n",
-      "28353    | torch.Size([1, 1000, 841]) | 11\n",
-      "28354    | torch.Size([1, 1000, 762]) | 11\n",
-      "28355    | torch.Size([1, 1000, 777]) | 11\n",
-      "28356    | torch.Size([1, 1000, 762]) | 11\n",
-      "28357    | torch.Size([1, 1000, 762]) | 11\n",
-      "28358    | torch.Size([1, 1000, 754]) | 11\n",
-      "28359    | torch.Size([1, 1000, 777]) | 11\n",
-      "28360    | torch.Size([1, 1000, 754]) | 11\n",
-      "28361    | torch.Size([1, 1000, 807]) | 11\n",
-      "28362    | torch.Size([1, 1000, 762]) | 11\n",
-      "28363    | torch.Size([1, 1000, 754]) | 11\n",
-      "28364    | torch.Size([1, 1000, 762]) | 11\n",
-      "28365    | torch.Size([1, 1000, 777]) | 11\n",
-      "28366    | torch.Size([1, 1000, 765]) | 11\n",
-      "28367    | torch.Size([1, 1000, 611]) | 11\n",
-      "28368    | torch.Size([1, 1000, 787]) | 11\n",
-      "28369    | torch.Size([1, 1000, 785]) | 11\n",
-      "28370    | torch.Size([1, 1000, 762]) | 11\n",
-      "28371    | torch.Size([1, 1000, 762]) | 11\n",
-      "28372    | torch.Size([1, 1000, 762]) | 11\n",
-      "28373    | torch.Size([1, 1000, 777]) | 11\n",
-      "28374    | torch.Size([1, 1000, 615]) | 11\n",
-      "28375    | torch.Size([1, 1000, 754]) | 11\n",
-      "28376    | torch.Size([1, 1000, 777]) | 11\n",
-      "28377    | torch.Size([1, 1000, 777]) | 11\n",
-      "28378    | torch.Size([1, 1000, 754]) | 11\n",
-      "28379    | torch.Size([1, 1000, 625]) | 11\n",
-      "28380    | torch.Size([1, 1000, 754]) | 11\n",
-      "28381    | torch.Size([1, 1000, 754]) | 11\n",
-      "28382    | torch.Size([1, 1000, 638]) | 11\n",
-      "28383    | torch.Size([1, 1000, 762]) | 11\n",
-      "28384    | torch.Size([1, 1000, 785]) | 11\n",
-      "28385    | torch.Size([1, 1000, 777]) | 11\n",
-      "28386    | torch.Size([1, 1000, 802]) | 11\n",
-      "28387    | torch.Size([1, 1000, 779]) | 11\n",
-      "28388    | torch.Size([1, 1000, 754]) | 11\n",
-      "28389    | torch.Size([1, 1000, 754]) | 11\n",
-      "28390    | torch.Size([1, 1000, 801]) | 11\n",
-      "28391    | torch.Size([1, 1000, 754]) | 11\n",
-      "28392    | torch.Size([1, 1000, 781]) | 11\n",
-      "28393    | torch.Size([1, 1000, 801]) | 11\n",
-      "28394    | torch.Size([1, 1000, 762]) | 11\n",
-      "28395    | torch.Size([1, 1000, 754]) | 11\n",
-      "28396    | torch.Size([1, 1000, 762]) | 11\n",
-      "28397    | torch.Size([1, 1000, 772]) | 11\n",
-      "28398    | torch.Size([1, 1000, 778]) | 11\n",
-      "28399    | torch.Size([1, 1000, 765]) | 11\n",
-      "28400    | torch.Size([1, 1000, 754]) | 11\n",
-      "28401    | torch.Size([1, 1000, 762]) | 11\n",
-      "28402    | torch.Size([1, 1000, 783]) | 11\n",
-      "28403    | torch.Size([1, 1000, 762]) | 11\n",
-      "28404    | torch.Size([1, 1000, 762]) | 11\n",
-      "28405    | torch.Size([1, 1000, 762]) | 11\n",
-      "28406    | torch.Size([1, 1000, 754]) | 11\n",
-      "28407    | torch.Size([1, 1000, 786]) | 11\n",
-      "28408    | torch.Size([1, 1000, 762]) | 11\n",
-      "28409    | torch.Size([1, 1000, 789]) | 11\n",
-      "28410    | torch.Size([1, 1000, 767]) | 11\n",
-      "28411    | torch.Size([1, 1000, 754]) | 11\n",
-      "28412    | torch.Size([1, 1000, 754]) | 11\n",
-      "28413    | torch.Size([1, 1000, 777]) | 11\n",
-      "28414    | torch.Size([1, 1000, 771]) | 11\n",
-      "28415    | torch.Size([1, 1000, 762]) | 11\n",
-      "28416    | torch.Size([1, 1000, 776]) | 11\n",
-      "28417    | torch.Size([1, 1000, 754]) | 11\n",
-      "28418    | torch.Size([1, 1000, 780]) | 11\n",
-      "28419    | torch.Size([1, 1000, 777]) | 11\n",
-      "28420    | torch.Size([1, 1000, 764]) | 11\n",
-      "28421    | torch.Size([1, 1000, 812]) | 11\n",
-      "28422    | torch.Size([1, 1000, 754]) | 11\n",
-      "28423    | torch.Size([1, 1000, 787]) | 11\n",
-      "28424    | torch.Size([1, 1000, 777]) | 11\n",
-      "28425    | torch.Size([1, 1000, 762]) | 11\n",
-      "28426    | torch.Size([1, 1000, 772]) | 11\n",
-      "28427    | torch.Size([1, 1000, 825]) | 11\n",
-      "28428    | torch.Size([1, 1000, 754]) | 11\n",
-      "28429    | torch.Size([1, 1000, 754]) | 11\n",
-      "28430    | torch.Size([1, 1000, 781]) | 11\n",
-      "28431    | torch.Size([1, 1000, 754]) | 11\n",
-      "28432    | torch.Size([1, 1000, 762]) | 11\n",
-      "28433    | torch.Size([1, 1000, 754]) | 11\n",
-      "28434    | torch.Size([1, 1000, 777]) | 11\n",
-      "28435    | torch.Size([1, 1000, 762]) | 11\n",
-      "28436    | torch.Size([1, 1000, 785]) | 11\n",
-      "28437    | torch.Size([1, 1000, 754]) | 11\n",
-      "28438    | torch.Size([1, 1000, 776]) | 11\n",
-      "28439    | torch.Size([1, 1000, 780]) | 11\n",
-      "28440    | torch.Size([1, 1000, 762]) | 11\n",
-      "28441    | torch.Size([1, 1000, 762]) | 11\n",
-      "28442    | torch.Size([1, 1000, 762]) | 11\n",
-      "28443    | torch.Size([1, 1000, 762]) | 11\n",
-      "28444    | torch.Size([1, 1000, 754]) | 11\n",
-      "28445    | torch.Size([1, 1000, 754]) | 11\n",
-      "28446    | torch.Size([1, 1000, 771]) | 11\n",
-      "28447    | torch.Size([1, 1000, 754]) | 11\n",
-      "28448    | torch.Size([1, 1000, 754]) | 11\n",
-      "28449    | torch.Size([1, 1000, 762]) | 11\n",
-      "28450    | torch.Size([1, 1000, 765]) | 11\n",
-      "28451    | torch.Size([1, 1000, 754]) | 11\n",
-      "28452    | torch.Size([1, 1000, 780]) | 11\n",
-      "28453    | torch.Size([1, 1000, 754]) | 11\n",
-      "28454    | torch.Size([1, 1000, 762]) | 11\n",
-      "28455    | torch.Size([1, 1000, 777]) | 11\n",
-      "28456    | torch.Size([1, 1000, 810]) | 11\n",
-      "28457    | torch.Size([1, 1000, 754]) | 11\n",
-      "28458    | torch.Size([1, 1000, 772]) | 11\n",
-      "28459    | torch.Size([1, 1000, 777]) | 11\n",
-      "28460    | torch.Size([1, 1000, 754]) | 11\n",
-      "28461    | torch.Size([1, 1000, 742]) | 11\n",
-      "28462    | torch.Size([1, 1000, 754]) | 11\n",
-      "28463    | torch.Size([1, 1000, 776]) | 11\n",
-      "28464    | torch.Size([1, 1000, 766]) | 11\n",
-      "28465    | torch.Size([1, 1000, 754]) | 11\n",
-      "28466    | torch.Size([1, 1000, 729]) | 11\n",
-      "28467    | torch.Size([1, 1000, 754]) | 11\n",
-      "28468    | torch.Size([1, 1000, 754]) | 11\n",
-      "28469    | torch.Size([1, 1000, 762]) | 11\n",
-      "28470    | torch.Size([1, 1000, 777]) | 11\n",
-      "28471    | torch.Size([1, 1000, 779]) | 11\n",
-      "28472    | torch.Size([1, 1000, 754]) | 11\n",
-      "28473    | torch.Size([1, 1000, 754]) | 11\n",
-      "28474    | torch.Size([1, 1000, 754]) | 11\n",
-      "28475    | torch.Size([1, 1000, 810]) | 11\n",
-      "28476    | torch.Size([1, 1000, 754]) | 11\n",
-      "28477    | torch.Size([1, 1000, 783]) | 11\n",
-      "28478    | torch.Size([1, 1000, 754]) | 11\n",
-      "28479    | torch.Size([1, 1000, 754]) | 11\n",
-      "28480    | torch.Size([1, 1000, 764]) | 11\n",
-      "28481    | torch.Size([1, 1000, 770]) | 11\n",
-      "28482    | torch.Size([1, 1000, 754]) | 11\n",
-      "28483    | torch.Size([1, 1000, 766]) | 11\n",
-      "28484    | torch.Size([1, 1000, 754]) | 11\n",
-      "28485    | torch.Size([1, 1000, 754]) | 11\n",
-      "28486    | torch.Size([1, 1000, 762]) | 11\n",
-      "28487    | torch.Size([1, 1000, 754]) | 11\n",
-      "28488    | torch.Size([1, 1000, 754]) | 11\n",
-      "28489    | torch.Size([1, 1000, 754]) | 11\n",
-      "28490    | torch.Size([1, 1000, 762]) | 11\n",
-      "28491    | torch.Size([1, 1000, 777]) | 11\n",
-      "28492    | torch.Size([1, 1000, 762]) | 11\n",
-      "28493    | torch.Size([1, 1000, 762]) | 11\n",
-      "28494    | torch.Size([1, 1000, 802]) | 11\n",
-      "28495    | torch.Size([1, 1000, 637]) | 11\n",
-      "28496    | torch.Size([1, 1000, 762]) | 11\n",
-      "28497    | torch.Size([1, 1000, 614]) | 11\n",
-      "28498    | torch.Size([1, 1000, 762]) | 11\n",
-      "28499    | torch.Size([1, 1000, 762]) | 11\n",
-      "28500    | torch.Size([1, 1000, 723]) | 11\n",
-      "28501    | torch.Size([1, 1000, 754]) | 11\n",
-      "28502    | torch.Size([1, 1000, 754]) | 11\n",
-      "28503    | torch.Size([1, 1000, 754]) | 11\n",
-      "28504    | torch.Size([1, 1000, 754]) | 11\n",
-      "28505    | torch.Size([1, 1000, 796]) | 11\n",
-      "28506    | torch.Size([1, 1000, 784]) | 11\n",
-      "28507    | torch.Size([1, 1000, 762]) | 11\n",
-      "28508    | torch.Size([1, 1000, 754]) | 11\n",
-      "28509    | torch.Size([1, 1000, 777]) | 11\n",
-      "28510    | torch.Size([1, 1000, 789]) | 11\n",
-      "28511    | torch.Size([1, 1000, 777]) | 11\n",
-      "28512    | torch.Size([1, 1000, 772]) | 11\n",
-      "28513    | torch.Size([1, 1000, 754]) | 11\n",
-      "28514    | torch.Size([1, 1000, 771]) | 11\n",
-      "28515    | torch.Size([1, 1000, 809]) | 11\n",
-      "28516    | torch.Size([1, 1000, 754]) | 11\n",
-      "28517    | torch.Size([1, 1000, 762]) | 11\n",
-      "28518    | torch.Size([1, 1000, 754]) | 11\n",
-      "28519    | torch.Size([1, 1000, 742]) | 11\n",
-      "28520    | torch.Size([1, 1000, 754]) | 11\n",
-      "28521    | torch.Size([1, 1000, 762]) | 11\n",
-      "28522    | torch.Size([1, 1000, 762]) | 11\n",
-      "28523    | torch.Size([1, 1000, 847]) | 11\n",
-      "28524    | torch.Size([1, 1000, 795]) | 11\n",
-      "28525    | torch.Size([1, 1000, 754]) | 11\n",
-      "28526    | torch.Size([1, 1000, 786]) | 11\n",
-      "28527    | torch.Size([1, 1000, 754]) | 11\n",
-      "28528    | torch.Size([1, 1000, 754]) | 11\n",
-      "28529    | torch.Size([1, 1000, 762]) | 11\n",
-      "28530    | torch.Size([1, 1000, 765]) | 11\n",
-      "28531    | torch.Size([1, 1000, 777]) | 11\n",
-      "28532    | torch.Size([1, 1000, 762]) | 11\n",
-      "28533    | torch.Size([1, 1000, 754]) | 11\n",
-      "28534    | torch.Size([1, 1000, 723]) | 11\n",
-      "28535    | torch.Size([1, 1000, 846]) | 11\n",
-      "28536    | torch.Size([1, 1000, 802]) | 11\n",
-      "28537    | torch.Size([1, 1000, 803]) | 11\n",
-      "28538    | torch.Size([1, 1000, 762]) | 11\n",
-      "28539    | torch.Size([1, 1000, 842]) | 11\n",
-      "28540    | torch.Size([1, 1000, 754]) | 11\n",
-      "28541    | torch.Size([1, 1000, 754]) | 11\n",
-      "28542    | torch.Size([1, 1000, 762]) | 11\n",
-      "28543    | torch.Size([1, 1000, 754]) | 11\n",
-      "28544    | torch.Size([1, 1000, 777]) | 11\n",
-      "28545    | torch.Size([1, 1000, 786]) | 11\n",
-      "28546    | torch.Size([1, 1000, 754]) | 11\n",
-      "28547    | torch.Size([1, 1000, 754]) | 11\n",
-      "28548    | torch.Size([1, 1000, 802]) | 11\n",
-      "28549    | torch.Size([1, 1000, 771]) | 11\n",
-      "28550    | torch.Size([1, 1000, 762]) | 11\n",
-      "28551    | torch.Size([1, 1000, 754]) | 11\n",
-      "28552    | torch.Size([1, 1000, 754]) | 11\n",
-      "28553    | torch.Size([1, 1000, 754]) | 11\n",
-      "28554    | torch.Size([1, 1000, 762]) | 11\n",
-      "28555    | torch.Size([1, 1000, 754]) | 11\n",
-      "28556    | torch.Size([1, 1000, 754]) | 11\n",
-      "28557    | torch.Size([1, 1000, 754]) | 11\n",
-      "28558    | torch.Size([1, 1000, 790]) | 11\n",
-      "28559    | torch.Size([1, 1000, 740]) | 11\n",
-      "28560    | torch.Size([1, 1000, 754]) | 11\n",
-      "28561    | torch.Size([1, 1000, 769]) | 11\n",
-      "28562    | torch.Size([1, 1000, 794]) | 11\n",
-      "28563    | torch.Size([1, 1000, 768]) | 11\n",
-      "28564    | torch.Size([1, 1000, 754]) | 11\n",
-      "28565    | torch.Size([1, 1000, 759]) | 11\n",
-      "28566    | torch.Size([1, 1000, 777]) | 11\n",
-      "28567    | torch.Size([1, 1000, 754]) | 11\n",
-      "28568    | torch.Size([1, 1000, 777]) | 11\n",
-      "28569    | torch.Size([1, 1000, 796]) | 11\n",
-      "28570    | torch.Size([1, 1000, 762]) | 11\n",
-      "28571    | torch.Size([1, 1000, 779]) | 11\n",
-      "28572    | torch.Size([1, 1000, 762]) | 11\n",
-      "28573    | torch.Size([1, 1000, 742]) | 11\n",
-      "28574    | torch.Size([1, 1000, 830]) | 11\n",
-      "28575    | torch.Size([1, 1000, 653]) | 11\n",
-      "28576    | torch.Size([1, 1000, 635]) | 11\n",
-      "28577    | torch.Size([1, 1000, 771]) | 11\n",
-      "28578    | torch.Size([1, 1000, 784]) | 11\n",
-      "28579    | torch.Size([1, 1000, 842]) | 11\n",
-      "28580    | torch.Size([1, 1000, 777]) | 11\n",
-      "28581    | torch.Size([1, 1000, 862]) | 11\n",
-      "28582    | torch.Size([1, 1000, 777]) | 11\n",
-      "28583    | torch.Size([1, 1000, 802]) | 11\n",
-      "28584    | torch.Size([1, 1000, 762]) | 11\n",
-      "28585    | torch.Size([1, 1000, 766]) | 11\n",
-      "28586    | torch.Size([1, 1000, 754]) | 11\n",
-      "28587    | torch.Size([1, 1000, 780]) | 11\n",
-      "28588    | torch.Size([1, 1000, 754]) | 11\n",
-      "28589    | torch.Size([1, 1000, 788]) | 11\n",
-      "28590    | torch.Size([1, 1000, 807]) | 11\n",
-      "28591    | torch.Size([1, 1000, 777]) | 11\n",
-      "28592    | torch.Size([1, 1000, 754]) | 11\n",
-      "28593    | torch.Size([1, 1000, 754]) | 11\n",
-      "28594    | torch.Size([1, 1000, 777]) | 11\n",
-      "28595    | torch.Size([1, 1000, 754]) | 11\n",
-      "28596    | torch.Size([1, 1000, 785]) | 11\n",
-      "28597    | torch.Size([1, 1000, 615]) | 11\n",
-      "28598    | torch.Size([1, 1000, 754]) | 11\n",
-      "28599    | torch.Size([1, 1000, 762]) | 11\n",
-      "28600    | torch.Size([1, 1000, 805]) | 11\n",
-      "28601    | torch.Size([1, 1000, 777]) | 11\n",
-      "28602    | torch.Size([1, 1000, 754]) | 11\n",
-      "28603    | torch.Size([1, 1000, 765]) | 11\n",
-      "28604    | torch.Size([1, 1000, 762]) | 11\n",
-      "28605    | torch.Size([1, 1000, 771]) | 11\n",
-      "28606    | torch.Size([1, 1000, 777]) | 11\n",
-      "28607    | torch.Size([1, 1000, 662]) | 11\n",
-      "28608    | torch.Size([1, 1000, 754]) | 11\n",
-      "28609    | torch.Size([1, 1000, 802]) | 11\n",
-      "28610    | torch.Size([1, 1000, 777]) | 11\n",
-      "28611    | torch.Size([1, 1000, 754]) | 11\n",
-      "28612    | torch.Size([1, 1000, 754]) | 11\n",
-      "28613    | torch.Size([1, 1000, 662]) | 11\n",
-      "28614    | torch.Size([1, 1000, 762]) | 11\n",
-      "28615    | torch.Size([1, 1000, 762]) | 11\n",
-      "28616    | torch.Size([1, 1000, 786]) | 11\n",
-      "28617    | torch.Size([1, 1000, 754]) | 11\n",
-      "28618    | torch.Size([1, 1000, 754]) | 11\n",
-      "28619    | torch.Size([1, 1000, 754]) | 11\n",
-      "28620    | torch.Size([1, 1000, 754]) | 11\n",
-      "28621    | torch.Size([1, 1000, 754]) | 11\n",
-      "28622    | torch.Size([1, 1000, 792]) | 11\n",
-      "28623    | torch.Size([1, 1000, 762]) | 11\n",
-      "28624    | torch.Size([1, 1000, 802]) | 11\n",
-      "28625    | torch.Size([1, 1000, 771]) | 11\n",
-      "28626    | torch.Size([1, 1000, 802]) | 11\n",
-      "28627    | torch.Size([1, 1000, 754]) | 11\n",
-      "28628    | torch.Size([1, 1000, 762]) | 11\n",
-      "28629    | torch.Size([1, 1000, 754]) | 11\n",
-      "28630    | torch.Size([1, 1000, 777]) | 11\n",
-      "28631    | torch.Size([1, 1000, 754]) | 11\n",
-      "28632    | torch.Size([1, 1000, 771]) | 11\n",
-      "28633    | torch.Size([1, 1000, 814]) | 11\n",
-      "28634    | torch.Size([1, 1000, 754]) | 11\n",
-      "28635    | torch.Size([1, 1000, 762]) | 11\n",
-      "28636    | torch.Size([1, 1000, 754]) | 11\n",
-      "28637    | torch.Size([1, 1000, 762]) | 11\n",
-      "28638    | torch.Size([1, 1000, 762]) | 11\n",
-      "28639    | torch.Size([1, 1000, 777]) | 11\n",
-      "28640    | torch.Size([1, 1000, 762]) | 11\n",
-      "28641    | torch.Size([1, 1000, 754]) | 11\n",
-      "28642    | torch.Size([1, 1000, 754]) | 11\n",
-      "28643    | torch.Size([1, 1000, 772]) | 11\n",
-      "28644    | torch.Size([1, 1000, 645]) | 11\n",
-      "28645    | torch.Size([1, 1000, 762]) | 11\n",
-      "28646    | torch.Size([1, 1000, 762]) | 11\n",
-      "28647    | torch.Size([1, 1000, 777]) | 11\n",
-      "28648    | torch.Size([1, 1000, 777]) | 11\n",
-      "28649    | torch.Size([1, 1000, 758]) | 11\n",
-      "28650    | torch.Size([1, 1000, 848]) | 11\n",
-      "28651    | torch.Size([1, 1000, 777]) | 11\n",
-      "28652    | torch.Size([1, 1000, 754]) | 11\n",
-      "28653    | torch.Size([1, 1000, 762]) | 11\n",
-      "28654    | torch.Size([1, 1000, 770]) | 11\n",
-      "28655    | torch.Size([1, 1000, 754]) | 11\n",
-      "28656    | torch.Size([1, 1000, 754]) | 11\n",
-      "28657    | torch.Size([1, 1000, 781]) | 11\n",
-      "28658    | torch.Size([1, 1000, 796]) | 11\n",
-      "28659    | torch.Size([1, 1000, 777]) | 11\n",
-      "28660    | torch.Size([1, 1000, 798]) | 11\n",
-      "28661    | torch.Size([1, 1000, 754]) | 11\n",
-      "28662    | torch.Size([1, 1000, 785]) | 11\n",
-      "28663    | torch.Size([1, 1000, 754]) | 11\n",
-      "28664    | torch.Size([1, 1000, 777]) | 11\n",
-      "28665    | torch.Size([1, 1000, 783]) | 11\n",
-      "28666    | torch.Size([1, 1000, 777]) | 11\n",
-      "28667    | torch.Size([1, 1000, 754]) | 11\n",
-      "28668    | torch.Size([1, 1000, 777]) | 11\n",
-      "28669    | torch.Size([1, 1000, 777]) | 11\n",
-      "28670    | torch.Size([1, 1000, 754]) | 11\n",
-      "28671    | torch.Size([1, 1000, 666]) | 11\n",
-      "28672    | torch.Size([1, 1000, 780]) | 11\n",
-      "28673    | torch.Size([1, 1000, 762]) | 11\n",
-      "28674    | torch.Size([1, 1000, 754]) | 11\n",
-      "28675    | torch.Size([1, 1000, 767]) | 11\n",
-      "28676    | torch.Size([1, 1000, 782]) | 11\n",
-      "28677    | torch.Size([1, 1000, 754]) | 11\n",
-      "28678    | torch.Size([1, 1000, 766]) | 11\n",
-      "28679    | torch.Size([1, 1000, 754]) | 11\n",
-      "28680    | torch.Size([1, 1000, 754]) | 11\n",
-      "28681    | torch.Size([1, 1000, 754]) | 11\n",
-      "28682    | torch.Size([1, 1000, 762]) | 11\n",
-      "28683    | torch.Size([1, 1000, 762]) | 11\n",
-      "28684    | torch.Size([1, 1000, 777]) | 11\n",
-      "28685    | torch.Size([1, 1000, 762]) | 11\n",
-      "28686    | torch.Size([1, 1000, 762]) | 11\n",
-      "28687    | torch.Size([1, 1000, 762]) | 11\n",
-      "28688    | torch.Size([1, 1000, 762]) | 11\n",
-      "28689    | torch.Size([1, 1000, 781]) | 11\n",
-      "28690    | torch.Size([1, 1000, 754]) | 11\n",
-      "28691    | torch.Size([1, 1000, 777]) | 11\n",
-      "28692    | torch.Size([1, 1000, 754]) | 11\n",
-      "28693    | torch.Size([1, 1000, 802]) | 11\n",
-      "28694    | torch.Size([1, 1000, 758]) | 11\n",
-      "28695    | torch.Size([1, 1000, 754]) | 11\n",
-      "28696    | torch.Size([1, 1000, 754]) | 11\n",
-      "28697    | torch.Size([1, 1000, 777]) | 11\n",
-      "28698    | torch.Size([1, 1000, 635]) | 11\n",
-      "28699    | torch.Size([1, 1000, 786]) | 11\n",
-      "28700    | torch.Size([1, 1000, 770]) | 11\n",
-      "28701    | torch.Size([1, 1000, 777]) | 11\n",
-      "28702    | torch.Size([1, 1000, 779]) | 11\n",
-      "28703    | torch.Size([1, 1000, 846]) | 11\n",
-      "28704    | torch.Size([1, 1000, 793]) | 11\n",
-      "28705    | torch.Size([1, 1000, 767]) | 11\n",
-      "28706    | torch.Size([1, 1000, 651]) | 11\n",
-      "28707    | torch.Size([1, 1000, 662]) | 11\n",
-      "28708    | torch.Size([1, 1000, 843]) | 11\n",
-      "28709    | torch.Size([1, 1000, 780]) | 11\n",
-      "28710    | torch.Size([1, 1000, 754]) | 11\n",
-      "28711    | torch.Size([1, 1000, 762]) | 11\n",
-      "28712    | torch.Size([1, 1000, 754]) | 11\n",
-      "28713    | torch.Size([1, 1000, 754]) | 11\n",
-      "28714    | torch.Size([1, 1000, 754]) | 11\n",
-      "28715    | torch.Size([1, 1000, 754]) | 11\n",
-      "28716    | torch.Size([1, 1000, 754]) | 11\n",
-      "28717    | torch.Size([1, 1000, 779]) | 11\n",
-      "28718    | torch.Size([1, 1000, 787]) | 11\n",
-      "28719    | torch.Size([1, 1000, 754]) | 11\n",
-      "28720    | torch.Size([1, 1000, 754]) | 11\n",
-      "28721    | torch.Size([1, 1000, 754]) | 11\n",
-      "28722    | torch.Size([1, 1000, 754]) | 11\n",
-      "28723    | torch.Size([1, 1000, 777]) | 11\n",
-      "28724    | torch.Size([1, 1000, 613]) | 11\n",
-      "28725    | torch.Size([1, 1000, 754]) | 11\n",
-      "28726    | torch.Size([1, 1000, 623]) | 11\n",
-      "28727    | torch.Size([1, 1000, 754]) | 11\n",
-      "28728    | torch.Size([1, 1000, 762]) | 11\n",
-      "28729    | torch.Size([1, 1000, 792]) | 11\n",
-      "28730    | torch.Size([1, 1000, 754]) | 11\n",
-      "28731    | torch.Size([1, 1000, 762]) | 11\n",
-      "28732    | torch.Size([1, 1000, 771]) | 11\n",
-      "28733    | torch.Size([1, 1000, 777]) | 11\n",
-      "28734    | torch.Size([1, 1000, 754]) | 11\n",
-      "28735    | torch.Size([1, 1000, 762]) | 11\n",
-      "28736    | torch.Size([1, 1000, 762]) | 11\n",
-      "28737    | torch.Size([1, 1000, 779]) | 11\n",
-      "28738    | torch.Size([1, 1000, 754]) | 11\n",
-      "28739    | torch.Size([1, 1000, 778]) | 11\n",
-      "28740    | torch.Size([1, 1000, 831]) | 11\n",
-      "28741    | torch.Size([1, 1000, 754]) | 11\n",
-      "28742    | torch.Size([1, 1000, 754]) | 11\n",
-      "28743    | torch.Size([1, 1000, 754]) | 11\n",
-      "28744    | torch.Size([1, 1000, 771]) | 11\n",
-      "28745    | torch.Size([1, 1000, 762]) | 11\n",
-      "28746    | torch.Size([1, 1000, 781]) | 11\n",
-      "28747    | torch.Size([1, 1000, 754]) | 11\n",
-      "28748    | torch.Size([1, 1000, 771]) | 11\n",
-      "28749    | torch.Size([1, 1000, 790]) | 11\n",
-      "28750    | torch.Size([1, 1000, 762]) | 11\n",
-      "28751    | torch.Size([1, 1000, 785]) | 11\n",
-      "28752    | torch.Size([1, 1000, 754]) | 11\n",
-      "28753    | torch.Size([1, 1000, 754]) | 11\n",
-      "28754    | torch.Size([1, 1000, 762]) | 11\n",
-      "28755    | torch.Size([1, 1000, 754]) | 11\n",
-      "28756    | torch.Size([1, 1000, 762]) | 11\n",
-      "28757    | torch.Size([1, 1000, 754]) | 11\n",
-      "28758    | torch.Size([1, 1000, 762]) | 11\n",
-      "28759    | torch.Size([1, 1000, 754]) | 11\n",
-      "28760    | torch.Size([1, 1000, 762]) | 11\n",
-      "28761    | torch.Size([1, 1000, 754]) | 11\n",
-      "28762    | torch.Size([1, 1000, 754]) | 11\n",
-      "28763    | torch.Size([1, 1000, 762]) | 11\n",
-      "28764    | torch.Size([1, 1000, 713]) | 11\n",
-      "28765    | torch.Size([1, 1000, 774]) | 11\n",
-      "28766    | torch.Size([1, 1000, 754]) | 11\n",
-      "28767    | torch.Size([1, 1000, 754]) | 11\n",
-      "28768    | torch.Size([1, 1000, 754]) | 11\n",
-      "28769    | torch.Size([1, 1000, 762]) | 11\n",
-      "28770    | torch.Size([1, 1000, 754]) | 11\n",
-      "28771    | torch.Size([1, 1000, 773]) | 11\n",
-      "28772    | torch.Size([1, 1000, 762]) | 11\n",
-      "28773    | torch.Size([1, 1000, 805]) | 11\n",
-      "28774    | torch.Size([1, 1000, 780]) | 11\n",
-      "28775    | torch.Size([1, 1000, 786]) | 11\n",
-      "28776    | torch.Size([1, 1000, 802]) | 11\n",
-      "28777    | torch.Size([1, 1000, 762]) | 11\n",
-      "28778    | torch.Size([1, 1000, 754]) | 11\n",
-      "28779    | torch.Size([1, 1000, 762]) | 11\n",
-      "28780    | torch.Size([1, 1000, 762]) | 11\n",
-      "28781    | torch.Size([1, 1000, 642]) | 11\n",
-      "28782    | torch.Size([1, 1000, 777]) | 11\n",
-      "28783    | torch.Size([1, 1000, 762]) | 11\n",
-      "28784    | torch.Size([1, 1000, 754]) | 11\n",
-      "28785    | torch.Size([1, 1000, 777]) | 11\n",
-      "28786    | torch.Size([1, 1000, 754]) | 11\n",
-      "28787    | torch.Size([1, 1000, 636]) | 11\n",
-      "28788    | torch.Size([1, 1000, 754]) | 11\n",
-      "28789    | torch.Size([1, 1000, 754]) | 11\n",
-      "28790    | torch.Size([1, 1000, 762]) | 11\n",
-      "28791    | torch.Size([1, 1000, 754]) | 11\n",
-      "28792    | torch.Size([1, 1000, 762]) | 11\n",
-      "28793    | torch.Size([1, 1000, 762]) | 11\n",
-      "28794    | torch.Size([1, 1000, 762]) | 11\n",
-      "28795    | torch.Size([1, 1000, 777]) | 11\n",
-      "28796    | torch.Size([1, 1000, 780]) | 11\n",
-      "28797    | torch.Size([1, 1000, 850]) | 11\n",
-      "28798    | torch.Size([1, 1000, 754]) | 11\n",
-      "28799    | torch.Size([1, 1000, 754]) | 11\n",
-      "28800    | torch.Size([1, 1000, 825]) | 11\n",
-      "28801    | torch.Size([1, 1000, 754]) | 11\n",
-      "28802    | torch.Size([1, 1000, 780]) | 11\n",
-      "28803    | torch.Size([1, 1000, 799]) | 11\n",
-      "28804    | torch.Size([1, 1000, 762]) | 11\n",
-      "28805    | torch.Size([1, 1000, 777]) | 11\n",
-      "28806    | torch.Size([1, 1000, 762]) | 11\n",
-      "28807    | torch.Size([1, 1000, 762]) | 11\n",
-      "28808    | torch.Size([1, 1000, 754]) | 11\n",
-      "28809    | torch.Size([1, 1000, 802]) | 11\n",
-      "28810    | torch.Size([1, 1000, 754]) | 11\n",
-      "28811    | torch.Size([1, 1000, 787]) | 11\n",
-      "28812    | torch.Size([1, 1000, 611]) | 11\n",
-      "28813    | torch.Size([1, 1000, 754]) | 11\n",
-      "28814    | torch.Size([1, 1000, 762]) | 11\n",
-      "28815    | torch.Size([1, 1000, 781]) | 11\n",
-      "28816    | torch.Size([1, 1000, 801]) | 11\n",
-      "28817    | torch.Size([1, 1000, 845]) | 11\n",
-      "28818    | torch.Size([1, 1000, 790]) | 11\n",
-      "28819    | torch.Size([1, 1000, 774]) | 11\n",
-      "28820    | torch.Size([1, 1000, 754]) | 11\n",
-      "28821    | torch.Size([1, 1000, 762]) | 11\n",
-      "28822    | torch.Size([1, 1000, 762]) | 11\n",
-      "28823    | torch.Size([1, 1000, 754]) | 11\n",
-      "28824    | torch.Size([1, 1000, 754]) | 11\n",
-      "28825    | torch.Size([1, 1000, 754]) | 11\n",
-      "28826    | torch.Size([1, 1000, 762]) | 11\n",
-      "28827    | torch.Size([1, 1000, 754]) | 11\n",
-      "28828    | torch.Size([1, 1000, 802]) | 11\n",
-      "28829    | torch.Size([1, 1000, 754]) | 11\n",
-      "28830    | torch.Size([1, 1000, 762]) | 11\n",
-      "28831    | torch.Size([1, 1000, 808]) | 11\n",
-      "28832    | torch.Size([1, 1000, 754]) | 11\n",
-      "28833    | torch.Size([1, 1000, 762]) | 11\n",
-      "28834    | torch.Size([1, 1000, 777]) | 11\n",
-      "28835    | torch.Size([1, 1000, 614]) | 11\n",
-      "28836    | torch.Size([1, 1000, 762]) | 11\n",
-      "28837    | torch.Size([1, 1000, 754]) | 11\n",
-      "28838    | torch.Size([1, 1000, 762]) | 11\n",
-      "28839    | torch.Size([1, 1000, 762]) | 11\n",
-      "28840    | torch.Size([1, 1000, 754]) | 11\n",
-      "28841    | torch.Size([1, 1000, 777]) | 11\n",
-      "28842    | torch.Size([1, 1000, 754]) | 11\n",
-      "28843    | torch.Size([1, 1000, 762]) | 11\n",
-      "28844    | torch.Size([1, 1000, 754]) | 11\n",
-      "28845    | torch.Size([1, 1000, 785]) | 11\n",
-      "28846    | torch.Size([1, 1000, 777]) | 11\n",
-      "28847    | torch.Size([1, 1000, 762]) | 11\n",
-      "28848    | torch.Size([1, 1000, 777]) | 11\n",
-      "28849    | torch.Size([1, 1000, 777]) | 11\n",
-      "28850    | torch.Size([1, 1000, 767]) | 11\n",
-      "28851    | torch.Size([1, 1000, 762]) | 11\n",
-      "28852    | torch.Size([1, 1000, 754]) | 11\n",
-      "28853    | torch.Size([1, 1000, 777]) | 11\n",
-      "28854    | torch.Size([1, 1000, 790]) | 11\n",
-      "28855    | torch.Size([1, 1000, 754]) | 11\n",
-      "28856    | torch.Size([1, 1000, 785]) | 11\n",
-      "28857    | torch.Size([1, 1000, 754]) | 11\n",
-      "28858    | torch.Size([1, 1000, 791]) | 11\n",
-      "28859    | torch.Size([1, 1000, 762]) | 11\n",
-      "28860    | torch.Size([1, 1000, 774]) | 11\n",
-      "28861    | torch.Size([1, 1000, 782]) | 11\n",
-      "28862    | torch.Size([1, 1000, 780]) | 11\n",
-      "28863    | torch.Size([1, 1000, 785]) | 11\n",
-      "28864    | torch.Size([1, 1000, 762]) | 11\n",
-      "28865    | torch.Size([1, 1000, 754]) | 11\n",
-      "28866    | torch.Size([1, 1000, 754]) | 11\n",
-      "28867    | torch.Size([1, 1000, 754]) | 11\n",
-      "28868    | torch.Size([1, 1000, 802]) | 11\n",
-      "28869    | torch.Size([1, 1000, 754]) | 11\n",
-      "28870    | torch.Size([1, 1000, 776]) | 11\n",
-      "28871    | torch.Size([1, 1000, 754]) | 11\n",
-      "28872    | torch.Size([1, 1000, 789]) | 11\n",
-      "28873    | torch.Size([1, 1000, 636]) | 11\n",
-      "28874    | torch.Size([1, 1000, 754]) | 11\n",
-      "28875    | torch.Size([1, 1000, 615]) | 11\n",
-      "28876    | torch.Size([1, 1000, 771]) | 11\n",
-      "28877    | torch.Size([1, 1000, 754]) | 11\n",
-      "28878    | torch.Size([1, 1000, 754]) | 11\n",
-      "28879    | torch.Size([1, 1000, 754]) | 11\n",
-      "28880    | torch.Size([1, 1000, 644]) | 11\n",
-      "28881    | torch.Size([1, 1000, 754]) | 11\n",
-      "28882    | torch.Size([1, 1000, 777]) | 11\n",
-      "28883    | torch.Size([1, 1000, 784]) | 11\n",
-      "28884    | torch.Size([1, 1000, 762]) | 11\n",
-      "28885    | torch.Size([1, 1000, 754]) | 11\n",
-      "28886    | torch.Size([1, 1000, 780]) | 11\n",
-      "28887    | torch.Size([1, 1000, 781]) | 11\n",
-      "28888    | torch.Size([1, 1000, 754]) | 11\n",
-      "28889    | torch.Size([1, 1000, 777]) | 11\n",
-      "28890    | torch.Size([1, 1000, 754]) | 11\n",
-      "28891    | torch.Size([1, 1000, 762]) | 11\n",
-      "28892    | torch.Size([1, 1000, 643]) | 11\n",
-      "28893    | torch.Size([1, 1000, 773]) | 11\n",
-      "28894    | torch.Size([1, 1000, 766]) | 11\n",
-      "28895    | torch.Size([1, 1000, 762]) | 11\n",
-      "28896    | torch.Size([1, 1000, 762]) | 11\n",
-      "28897    | torch.Size([1, 1000, 754]) | 11\n",
-      "28898    | torch.Size([1, 1000, 754]) | 11\n",
-      "28899    | torch.Size([1, 1000, 754]) | 11\n",
-      "28900    | torch.Size([1, 1000, 754]) | 11\n",
-      "28901    | torch.Size([1, 1000, 754]) | 11\n",
-      "28902    | torch.Size([1, 1000, 848]) | 11\n",
-      "28903    | torch.Size([1, 1000, 754]) | 11\n",
-      "28904    | torch.Size([1, 1000, 787]) | 11\n",
-      "28905    | torch.Size([1, 1000, 773]) | 11\n",
-      "28906    | torch.Size([1, 1000, 765]) | 11\n",
-      "28907    | torch.Size([1, 1000, 773]) | 11\n",
-      "28908    | torch.Size([1, 1000, 762]) | 11\n",
-      "28909    | torch.Size([1, 1000, 792]) | 11\n",
-      "28910    | torch.Size([1, 1000, 771]) | 11\n",
-      "28911    | torch.Size([1, 1000, 754]) | 11\n",
-      "28912    | torch.Size([1, 1000, 784]) | 11\n",
-      "28913    | torch.Size([1, 1000, 754]) | 11\n",
-      "28914    | torch.Size([1, 1000, 754]) | 11\n",
-      "28915    | torch.Size([1, 1000, 771]) | 11\n",
-      "28916    | torch.Size([1, 1000, 803]) | 11\n",
-      "28917    | torch.Size([1, 1000, 762]) | 11\n",
-      "28918    | torch.Size([1, 1000, 780]) | 11\n",
-      "28919    | torch.Size([1, 1000, 762]) | 11\n",
-      "28920    | torch.Size([1, 1000, 847]) | 11\n",
-      "28921    | torch.Size([1, 1000, 779]) | 11\n",
-      "28922    | torch.Size([1, 1000, 777]) | 11\n",
-      "28923    | torch.Size([1, 1000, 754]) | 11\n",
-      "28924    | torch.Size([1, 1000, 776]) | 11\n",
-      "28925    | torch.Size([1, 1000, 754]) | 11\n",
-      "28926    | torch.Size([1, 1000, 777]) | 11\n",
-      "28927    | torch.Size([1, 1000, 754]) | 11\n",
-      "28928    | torch.Size([1, 1000, 777]) | 11\n",
-      "28929    | torch.Size([1, 1000, 762]) | 11\n",
-      "28930    | torch.Size([1, 1000, 762]) | 11\n",
-      "28931    | torch.Size([1, 1000, 754]) | 11\n",
-      "28932    | torch.Size([1, 1000, 777]) | 11\n",
-      "28933    | torch.Size([1, 1000, 777]) | 11\n",
-      "28934    | torch.Size([1, 1000, 762]) | 11\n",
-      "28935    | torch.Size([1, 1000, 611]) | 11\n",
-      "28936    | torch.Size([1, 1000, 771]) | 11\n",
-      "28937    | torch.Size([1, 1000, 802]) | 11\n",
-      "28938    | torch.Size([1, 1000, 754]) | 11\n",
-      "28939    | torch.Size([1, 1000, 762]) | 11\n",
-      "28940    | torch.Size([1, 1000, 754]) | 11\n",
-      "28941    | torch.Size([1, 1000, 762]) | 11\n",
-      "28942    | torch.Size([1, 1000, 762]) | 11\n",
-      "28943    | torch.Size([1, 1000, 777]) | 11\n",
-      "28944    | torch.Size([1, 1000, 796]) | 11\n",
-      "28945    | torch.Size([1, 1000, 756]) | 11\n",
-      "28946    | torch.Size([1, 1000, 771]) | 11\n",
-      "28947    | torch.Size([1, 1000, 762]) | 11\n",
-      "28948    | torch.Size([1, 1000, 762]) | 11\n",
-      "28949    | torch.Size([1, 1000, 762]) | 11\n",
-      "28950    | torch.Size([1, 1000, 754]) | 11\n",
-      "28951    | torch.Size([1, 1000, 777]) | 11\n",
-      "28952    | torch.Size([1, 1000, 754]) | 11\n",
-      "28953    | torch.Size([1, 1000, 762]) | 11\n",
-      "28954    | torch.Size([1, 1000, 754]) | 11\n",
-      "28955    | torch.Size([1, 1000, 787]) | 11\n",
-      "28956    | torch.Size([1, 1000, 786]) | 11\n",
-      "28957    | torch.Size([1, 1000, 780]) | 11\n",
-      "28958    | torch.Size([1, 1000, 754]) | 11\n",
-      "28959    | torch.Size([1, 1000, 768]) | 11\n",
-      "28960    | torch.Size([1, 1000, 754]) | 11\n",
-      "28961    | torch.Size([1, 1000, 754]) | 11\n",
-      "28962    | torch.Size([1, 1000, 762]) | 11\n",
-      "28963    | torch.Size([1, 1000, 765]) | 11\n",
-      "28964    | torch.Size([1, 1000, 762]) | 11\n",
-      "28965    | torch.Size([1, 1000, 754]) | 11\n",
-      "28966    | torch.Size([1, 1000, 763]) | 11\n",
-      "28967    | torch.Size([1, 1000, 754]) | 11\n",
-      "28968    | torch.Size([1, 1000, 754]) | 11\n",
-      "28969    | torch.Size([1, 1000, 849]) | 11\n",
-      "28970    | torch.Size([1, 1000, 777]) | 11\n",
-      "28971    | torch.Size([1, 1000, 762]) | 11\n",
-      "28972    | torch.Size([1, 1000, 777]) | 11\n",
-      "28973    | torch.Size([1, 1000, 795]) | 11\n",
-      "28974    | torch.Size([1, 1000, 786]) | 11\n",
-      "28975    | torch.Size([1, 1000, 762]) | 11\n",
-      "28976    | torch.Size([1, 1000, 780]) | 11\n",
-      "28977    | torch.Size([1, 1000, 810]) | 11\n",
-      "28978    | torch.Size([1, 1000, 809]) | 11\n",
-      "28979    | torch.Size([1, 1000, 762]) | 11\n",
-      "28980    | torch.Size([1, 1000, 787]) | 11\n",
-      "28981    | torch.Size([1, 1000, 614]) | 11\n",
-      "28982    | torch.Size([1, 1000, 754]) | 11\n",
-      "28983    | torch.Size([1, 1000, 777]) | 11\n",
-      "28984    | torch.Size([1, 1000, 754]) | 11\n",
-      "28985    | torch.Size([1, 1000, 765]) | 11\n",
-      "28986    | torch.Size([1, 1000, 762]) | 11\n",
-      "28987    | torch.Size([1, 1000, 754]) | 11\n",
-      "28988    | torch.Size([1, 1000, 777]) | 11\n",
-      "28989    | torch.Size([1, 1000, 754]) | 11\n",
-      "28990    | torch.Size([1, 1000, 792]) | 11\n",
-      "28991    | torch.Size([1, 1000, 752]) | 11\n",
-      "28992    | torch.Size([1, 1000, 777]) | 11\n",
-      "28993    | torch.Size([1, 1000, 779]) | 11\n",
-      "28994    | torch.Size([1, 1000, 762]) | 11\n",
-      "28995    | torch.Size([1, 1000, 769]) | 11\n",
-      "28996    | torch.Size([1, 1000, 808]) | 11\n",
-      "28997    | torch.Size([1, 1000, 754]) | 11\n",
-      "28998    | torch.Size([1, 1000, 811]) | 11\n",
-      "28999    | torch.Size([1, 1000, 823]) | 11\n",
-      "29000    | torch.Size([1, 1000, 754]) | 11\n",
-      "29001    | torch.Size([1, 1000, 754]) | 11\n",
-      "29002    | torch.Size([1, 1000, 777]) | 11\n",
-      "29003    | torch.Size([1, 1000, 803]) | 11\n",
-      "29004    | torch.Size([1, 1000, 754]) | 11\n",
-      "29005    | torch.Size([1, 1000, 775]) | 11\n",
-      "29006    | torch.Size([1, 1000, 777]) | 11\n",
-      "29007    | torch.Size([1, 1000, 754]) | 11\n",
-      "29008    | torch.Size([1, 1000, 754]) | 11\n",
-      "29009    | torch.Size([1, 1000, 754]) | 11\n",
-      "29010    | torch.Size([1, 1000, 762]) | 11\n",
-      "29011    | torch.Size([1, 1000, 803]) | 11\n",
-      "29012    | torch.Size([1, 1000, 762]) | 11\n",
-      "29013    | torch.Size([1, 1000, 754]) | 11\n",
-      "29014    | torch.Size([1, 1000, 762]) | 11\n",
-      "29015    | torch.Size([1, 1000, 777]) | 11\n",
-      "29016    | torch.Size([1, 1000, 754]) | 11\n",
-      "29017    | torch.Size([1, 1000, 777]) | 11\n",
-      "29018    | torch.Size([1, 1000, 781]) | 11\n",
-      "29019    | torch.Size([1, 1000, 754]) | 11\n",
-      "29020    | torch.Size([1, 1000, 779]) | 11\n",
-      "29021    | torch.Size([1, 1000, 777]) | 11\n",
-      "29022    | torch.Size([1, 1000, 777]) | 11\n",
-      "29023    | torch.Size([1, 1000, 754]) | 11\n",
-      "29024    | torch.Size([1, 1000, 754]) | 11\n",
-      "29025    | torch.Size([1, 1000, 754]) | 11\n",
-      "29026    | torch.Size([1, 1000, 754]) | 11\n",
-      "29027    | torch.Size([1, 1000, 764]) | 11\n",
-      "29028    | torch.Size([1, 1000, 782]) | 11\n",
-      "29029    | torch.Size([1, 1000, 775]) | 11\n",
-      "29030    | torch.Size([1, 1000, 808]) | 11\n",
-      "29031    | torch.Size([1, 1000, 754]) | 11\n",
-      "29032    | torch.Size([1, 1000, 754]) | 11\n",
-      "29033    | torch.Size([1, 1000, 754]) | 11\n",
-      "29034    | torch.Size([1, 1000, 754]) | 11\n",
-      "29035    | torch.Size([1, 1000, 777]) | 11\n",
-      "29036    | torch.Size([1, 1000, 762]) | 11\n",
-      "29037    | torch.Size([1, 1000, 796]) | 11\n",
-      "29038    | torch.Size([1, 1000, 762]) | 11\n",
-      "29039    | torch.Size([1, 1000, 777]) | 11\n",
-      "29040    | torch.Size([1, 1000, 771]) | 11\n",
-      "29041    | torch.Size([1, 1000, 754]) | 11\n",
-      "29042    | torch.Size([1, 1000, 780]) | 11\n",
-      "29043    | torch.Size([1, 1000, 754]) | 11\n",
-      "29044    | torch.Size([1, 1000, 774]) | 11\n",
-      "29045    | torch.Size([1, 1000, 754]) | 11\n",
-      "29046    | torch.Size([1, 1000, 777]) | 11\n",
-      "29047    | torch.Size([1, 1000, 761]) | 11\n",
-      "29048    | torch.Size([1, 1000, 777]) | 11\n",
-      "29049    | torch.Size([1, 1000, 811]) | 11\n",
-      "29050    | torch.Size([1, 1000, 773]) | 11\n",
-      "29051    | torch.Size([1, 1000, 754]) | 11\n",
-      "29052    | torch.Size([1, 1000, 802]) | 11\n",
-      "29053    | torch.Size([1, 1000, 754]) | 11\n",
-      "29054    | torch.Size([1, 1000, 777]) | 11\n",
-      "29055    | torch.Size([1, 1000, 754]) | 11\n",
-      "29056    | torch.Size([1, 1000, 770]) | 11\n",
-      "29057    | torch.Size([1, 1000, 754]) | 11\n",
-      "29058    | torch.Size([1, 1000, 762]) | 11\n",
-      "29059    | torch.Size([1, 1000, 795]) | 11\n",
-      "29060    | torch.Size([1, 1000, 788]) | 11\n",
-      "29061    | torch.Size([1, 1000, 754]) | 11\n",
-      "29062    | torch.Size([1, 1000, 778]) | 11\n",
-      "29063    | torch.Size([1, 1000, 754]) | 11\n",
-      "29064    | torch.Size([1, 1000, 754]) | 11\n",
-      "29065    | torch.Size([1, 1000, 754]) | 11\n",
-      "29066    | torch.Size([1, 1000, 754]) | 11\n",
-      "29067    | torch.Size([1, 1000, 778]) | 11\n",
-      "29068    | torch.Size([1, 1000, 777]) | 11\n",
-      "29069    | torch.Size([1, 1000, 762]) | 11\n",
-      "29070    | torch.Size([1, 1000, 811]) | 11\n",
-      "29071    | torch.Size([1, 1000, 776]) | 11\n",
-      "29072    | torch.Size([1, 1000, 754]) | 11\n",
-      "29073    | torch.Size([1, 1000, 754]) | 11\n",
-      "29074    | torch.Size([1, 1000, 754]) | 11\n",
-      "29075    | torch.Size([1, 1000, 762]) | 11\n",
-      "29076    | torch.Size([1, 1000, 754]) | 11\n",
-      "29077    | torch.Size([1, 1000, 788]) | 11\n",
-      "29078    | torch.Size([1, 1000, 762]) | 11\n",
-      "29079    | torch.Size([1, 1000, 742]) | 11\n",
-      "29080    | torch.Size([1, 1000, 754]) | 11\n",
-      "29081    | torch.Size([1, 1000, 784]) | 11\n",
-      "29082    | torch.Size([1, 1000, 774]) | 11\n",
-      "29083    | torch.Size([1, 1000, 762]) | 11\n",
-      "29084    | torch.Size([1, 1000, 777]) | 11\n",
-      "29085    | torch.Size([1, 1000, 762]) | 11\n",
-      "29086    | torch.Size([1, 1000, 787]) | 11\n",
-      "29087    | torch.Size([1, 1000, 762]) | 11\n",
-      "29088    | torch.Size([1, 1000, 754]) | 11\n",
-      "29089    | torch.Size([1, 1000, 722]) | 11\n",
-      "29090    | torch.Size([1, 1000, 799]) | 11\n",
-      "29091    | torch.Size([1, 1000, 780]) | 11\n",
-      "29092    | torch.Size([1, 1000, 780]) | 11\n",
-      "29093    | torch.Size([1, 1000, 754]) | 11\n",
-      "29094    | torch.Size([1, 1000, 780]) | 11\n",
-      "29095    | torch.Size([1, 1000, 783]) | 11\n",
-      "29096    | torch.Size([1, 1000, 762]) | 11\n",
-      "29097    | torch.Size([1, 1000, 762]) | 11\n",
-      "29098    | torch.Size([1, 1000, 777]) | 11\n",
-      "29099    | torch.Size([1, 1000, 776]) | 11\n",
-      "29100    | torch.Size([1, 1000, 785]) | 11\n",
-      "29101    | torch.Size([1, 1000, 762]) | 11\n",
-      "29102    | torch.Size([1, 1000, 777]) | 11\n",
-      "29103    | torch.Size([1, 1000, 771]) | 11\n",
-      "29104    | torch.Size([1, 1000, 778]) | 11\n",
-      "29105    | torch.Size([1, 1000, 754]) | 11\n",
-      "29106    | torch.Size([1, 1000, 762]) | 11\n",
-      "29107    | torch.Size([1, 1000, 754]) | 11\n",
-      "29108    | torch.Size([1, 1000, 843]) | 11\n",
-      "29109    | torch.Size([1, 1000, 772]) | 11\n",
-      "29110    | torch.Size([1, 1000, 762]) | 11\n",
-      "29111    | torch.Size([1, 1000, 802]) | 11\n",
-      "29112    | torch.Size([1, 1000, 776]) | 11\n",
-      "29113    | torch.Size([1, 1000, 754]) | 11\n",
-      "29114    | torch.Size([1, 1000, 754]) | 11\n",
-      "29115    | torch.Size([1, 1000, 771]) | 11\n",
-      "29116    | torch.Size([1, 1000, 754]) | 11\n",
-      "29117    | torch.Size([1, 1000, 762]) | 11\n",
-      "29118    | torch.Size([1, 1000, 766]) | 11\n",
-      "29119    | torch.Size([1, 1000, 767]) | 11\n",
-      "29120    | torch.Size([1, 1000, 636]) | 11\n",
-      "29121    | torch.Size([1, 1000, 776]) | 11\n",
-      "29122    | torch.Size([1, 1000, 762]) | 11\n",
-      "29123    | torch.Size([1, 1000, 809]) | 11\n",
-      "29124    | torch.Size([1, 1000, 777]) | 11\n",
-      "29125    | torch.Size([1, 1000, 754]) | 11\n",
-      "29126    | torch.Size([1, 1000, 754]) | 11\n",
-      "29127    | torch.Size([1, 1000, 776]) | 11\n",
-      "29128    | torch.Size([1, 1000, 754]) | 11\n",
-      "29129    | torch.Size([1, 1000, 777]) | 11\n",
-      "29130    | torch.Size([1, 1000, 762]) | 11\n",
-      "29131    | torch.Size([1, 1000, 777]) | 11\n",
-      "29132    | torch.Size([1, 1000, 773]) | 11\n",
-      "29133    | torch.Size([1, 1000, 754]) | 11\n",
-      "29134    | torch.Size([1, 1000, 762]) | 11\n",
-      "29135    | torch.Size([1, 1000, 754]) | 11\n",
-      "29136    | torch.Size([1, 1000, 767]) | 11\n",
-      "29137    | torch.Size([1, 1000, 754]) | 11\n",
-      "29138    | torch.Size([1, 1000, 754]) | 11\n",
-      "29139    | torch.Size([1, 1000, 772]) | 11\n",
-      "29140    | torch.Size([1, 1000, 765]) | 11\n",
-      "29141    | torch.Size([1, 1000, 754]) | 11\n",
-      "29142    | torch.Size([1, 1000, 784]) | 11\n",
-      "29143    | torch.Size([1, 1000, 762]) | 11\n",
-      "29144    | torch.Size([1, 1000, 828]) | 11\n",
-      "29145    | torch.Size([1, 1000, 762]) | 11\n",
-      "29146    | torch.Size([1, 1000, 780]) | 11\n",
-      "29147    | torch.Size([1, 1000, 762]) | 11\n",
-      "29148    | torch.Size([1, 1000, 754]) | 11\n",
-      "29149    | torch.Size([1, 1000, 762]) | 11\n",
-      "29150    | torch.Size([1, 1000, 783]) | 11\n",
-      "29151    | torch.Size([1, 1000, 754]) | 11\n",
-      "29152    | torch.Size([1, 1000, 787]) | 11\n",
-      "29153    | torch.Size([1, 1000, 762]) | 11\n",
-      "29154    | torch.Size([1, 1000, 614]) | 11\n",
-      "29155    | torch.Size([1, 1000, 796]) | 11\n",
-      "29156    | torch.Size([1, 1000, 772]) | 11\n",
-      "29157    | torch.Size([1, 1000, 762]) | 11\n",
-      "29158    | torch.Size([1, 1000, 754]) | 11\n",
-      "29159    | torch.Size([1, 1000, 777]) | 11\n",
-      "29160    | torch.Size([1, 1000, 754]) | 11\n",
-      "29161    | torch.Size([1, 1000, 767]) | 11\n",
-      "29162    | torch.Size([1, 1000, 778]) | 11\n",
-      "29163    | torch.Size([1, 1000, 769]) | 11\n",
-      "29164    | torch.Size([1, 1000, 771]) | 11\n",
-      "29165    | torch.Size([1, 1000, 754]) | 11\n",
-      "29166    | torch.Size([1, 1000, 777]) | 11\n",
-      "29167    | torch.Size([1, 1000, 762]) | 11\n",
-      "29168    | torch.Size([1, 1000, 754]) | 11\n",
-      "29169    | torch.Size([1, 1000, 762]) | 11\n",
-      "29170    | torch.Size([1, 1000, 754]) | 11\n",
-      "29171    | torch.Size([1, 1000, 754]) | 11\n",
-      "29172    | torch.Size([1, 1000, 774]) | 11\n",
-      "29173    | torch.Size([1, 1000, 777]) | 11\n",
-      "29174    | torch.Size([1, 1000, 762]) | 11\n",
-      "29175    | torch.Size([1, 1000, 762]) | 11\n",
-      "29176    | torch.Size([1, 1000, 754]) | 11\n",
-      "29177    | torch.Size([1, 1000, 762]) | 11\n",
-      "29178    | torch.Size([1, 1000, 754]) | 11\n",
-      "29179    | torch.Size([1, 1000, 793]) | 11\n",
-      "29180    | torch.Size([1, 1000, 754]) | 11\n",
-      "29181    | torch.Size([1, 1000, 724]) | 11\n",
-      "29182    | torch.Size([1, 1000, 754]) | 11\n",
-      "29183    | torch.Size([1, 1000, 762]) | 11\n",
-      "29184    | torch.Size([1, 1000, 777]) | 11\n",
-      "29185    | torch.Size([1, 1000, 777]) | 11\n",
-      "29186    | torch.Size([1, 1000, 762]) | 11\n",
-      "29187    | torch.Size([1, 1000, 754]) | 11\n",
-      "29188    | torch.Size([1, 1000, 783]) | 11\n",
-      "29189    | torch.Size([1, 1000, 780]) | 11\n",
-      "29190    | torch.Size([1, 1000, 762]) | 11\n",
-      "29191    | torch.Size([1, 1000, 754]) | 11\n",
-      "29192    | torch.Size([1, 1000, 762]) | 11\n",
-      "29193    | torch.Size([1, 1000, 762]) | 11\n",
-      "29194    | torch.Size([1, 1000, 777]) | 11\n",
-      "29195    | torch.Size([1, 1000, 784]) | 11\n",
-      "29196    | torch.Size([1, 1000, 777]) | 11\n",
-      "29197    | torch.Size([1, 1000, 762]) | 11\n",
-      "29198    | torch.Size([1, 1000, 797]) | 11\n",
-      "29199    | torch.Size([1, 1000, 777]) | 11\n",
-      "29200    | torch.Size([1, 1000, 778]) | 11\n",
-      "29201    | torch.Size([1, 1000, 754]) | 11\n",
-      "29202    | torch.Size([1, 1000, 611]) | 11\n",
-      "29203    | torch.Size([1, 1000, 754]) | 11\n",
-      "29204    | torch.Size([1, 1000, 802]) | 11\n",
-      "29205    | torch.Size([1, 1000, 754]) | 11\n",
-      "29206    | torch.Size([1, 1000, 754]) | 11\n",
-      "29207    | torch.Size([1, 1000, 800]) | 11\n",
-      "29208    | torch.Size([1, 1000, 810]) | 11\n",
-      "29209    | torch.Size([1, 1000, 754]) | 11\n",
-      "29210    | torch.Size([1, 1000, 754]) | 11\n",
-      "29211    | torch.Size([1, 1000, 762]) | 11\n",
-      "29212    | torch.Size([1, 1000, 794]) | 11\n",
-      "29213    | torch.Size([1, 1000, 616]) | 11\n",
-      "29214    | torch.Size([1, 1000, 754]) | 11\n",
-      "29215    | torch.Size([1, 1000, 743]) | 11\n",
-      "29216    | torch.Size([1, 1000, 782]) | 11\n",
-      "29217    | torch.Size([1, 1000, 762]) | 11\n",
-      "29218    | torch.Size([1, 1000, 762]) | 11\n",
-      "29219    | torch.Size([1, 1000, 754]) | 11\n",
-      "29220    | torch.Size([1, 1000, 780]) | 11\n",
-      "29221    | torch.Size([1, 1000, 761]) | 11\n",
-      "29222    | torch.Size([1, 1000, 790]) | 11\n",
-      "29223    | torch.Size([1, 1000, 798]) | 11\n",
-      "29224    | torch.Size([1, 1000, 771]) | 11\n",
-      "29225    | torch.Size([1, 1000, 766]) | 11\n",
-      "29226    | torch.Size([1, 1000, 762]) | 11\n",
-      "29227    | torch.Size([1, 1000, 777]) | 11\n",
-      "29228    | torch.Size([1, 1000, 762]) | 11\n",
-      "29229    | torch.Size([1, 1000, 614]) | 11\n",
-      "29230    | torch.Size([1, 1000, 762]) | 11\n",
-      "29231    | torch.Size([1, 1000, 754]) | 11\n",
-      "29232    | torch.Size([1, 1000, 778]) | 11\n",
-      "29233    | torch.Size([1, 1000, 773]) | 11\n",
-      "29234    | torch.Size([1, 1000, 762]) | 11\n",
-      "29235    | torch.Size([1, 1000, 762]) | 11\n",
-      "29236    | torch.Size([1, 1000, 762]) | 11\n",
-      "29237    | torch.Size([1, 1000, 795]) | 11\n",
-      "29238    | torch.Size([1, 1000, 776]) | 11\n",
-      "29239    | torch.Size([1, 1000, 777]) | 11\n",
-      "29240    | torch.Size([1, 1000, 754]) | 11\n",
-      "29241    | torch.Size([1, 1000, 754]) | 11\n",
-      "29242    | torch.Size([1, 1000, 754]) | 11\n",
-      "29243    | torch.Size([1, 1000, 754]) | 11\n",
-      "29244    | torch.Size([1, 1000, 754]) | 11\n",
-      "29245    | torch.Size([1, 1000, 762]) | 11\n",
-      "29246    | torch.Size([1, 1000, 762]) | 11\n",
-      "29247    | torch.Size([1, 1000, 612]) | 11\n",
-      "29248    | torch.Size([1, 1000, 762]) | 11\n",
-      "29249    | torch.Size([1, 1000, 754]) | 11\n",
-      "29250    | torch.Size([1, 1000, 777]) | 11\n",
-      "29251    | torch.Size([1, 1000, 754]) | 11\n",
-      "29252    | torch.Size([1, 1000, 762]) | 11\n",
-      "29253    | torch.Size([1, 1000, 754]) | 11\n",
-      "29254    | torch.Size([1, 1000, 776]) | 11\n",
-      "29255    | torch.Size([1, 1000, 777]) | 11\n",
-      "29256    | torch.Size([1, 1000, 762]) | 11\n",
-      "29257    | torch.Size([1, 1000, 762]) | 11\n",
-      "29258    | torch.Size([1, 1000, 762]) | 11\n",
-      "29259    | torch.Size([1, 1000, 762]) | 11\n",
-      "29260    | torch.Size([1, 1000, 615]) | 11\n",
-      "29261    | torch.Size([1, 1000, 777]) | 11\n",
-      "29262    | torch.Size([1, 1000, 796]) | 11\n",
-      "29263    | torch.Size([1, 1000, 754]) | 11\n",
-      "29264    | torch.Size([1, 1000, 754]) | 11\n",
-      "29265    | torch.Size([1, 1000, 762]) | 11\n",
-      "29266    | torch.Size([1, 1000, 760]) | 11\n",
-      "29267    | torch.Size([1, 1000, 754]) | 11\n",
-      "29268    | torch.Size([1, 1000, 754]) | 11\n",
-      "29269    | torch.Size([1, 1000, 762]) | 11\n",
-      "29270    | torch.Size([1, 1000, 809]) | 11\n",
-      "29271    | torch.Size([1, 1000, 762]) | 11\n",
-      "29272    | torch.Size([1, 1000, 754]) | 11\n",
-      "29273    | torch.Size([1, 1000, 762]) | 11\n",
-      "29274    | torch.Size([1, 1000, 754]) | 11\n",
-      "29275    | torch.Size([1, 1000, 777]) | 11\n",
-      "29276    | torch.Size([1, 1000, 754]) | 11\n",
-      "29277    | torch.Size([1, 1000, 771]) | 11\n",
-      "29278    | torch.Size([1, 1000, 754]) | 11\n",
-      "29279    | torch.Size([1, 1000, 754]) | 11\n",
-      "29280    | torch.Size([1, 1000, 741]) | 11\n",
-      "29281    | torch.Size([1, 1000, 760]) | 11\n",
-      "29282    | torch.Size([1, 1000, 777]) | 11\n",
-      "29283    | torch.Size([1, 1000, 762]) | 11\n",
-      "29284    | torch.Size([1, 1000, 768]) | 11\n",
-      "29285    | torch.Size([1, 1000, 762]) | 11\n",
-      "29286    | torch.Size([1, 1000, 762]) | 11\n",
-      "29287    | torch.Size([1, 1000, 777]) | 11\n",
-      "29288    | torch.Size([1, 1000, 754]) | 11\n",
-      "29289    | torch.Size([1, 1000, 772]) | 11\n",
-      "29290    | torch.Size([1, 1000, 754]) | 11\n",
-      "29291    | torch.Size([1, 1000, 754]) | 11\n",
-      "29292    | torch.Size([1, 1000, 784]) | 11\n",
-      "29293    | torch.Size([1, 1000, 777]) | 11\n",
-      "29294    | torch.Size([1, 1000, 762]) | 11\n",
-      "29295    | torch.Size([1, 1000, 754]) | 11\n",
-      "29296    | torch.Size([1, 1000, 783]) | 11\n",
-      "29297    | torch.Size([1, 1000, 754]) | 11\n",
-      "29298    | torch.Size([1, 1000, 754]) | 11\n",
-      "29299    | torch.Size([1, 1000, 754]) | 11\n",
-      "29300    | torch.Size([1, 1000, 776]) | 11\n",
-      "29301    | torch.Size([1, 1000, 762]) | 11\n",
-      "29302    | torch.Size([1, 1000, 763]) | 11\n",
-      "29303    | torch.Size([1, 1000, 784]) | 11\n",
-      "29304    | torch.Size([1, 1000, 762]) | 11\n",
-      "29305    | torch.Size([1, 1000, 754]) | 11\n",
-      "29306    | torch.Size([1, 1000, 772]) | 11\n",
-      "29307    | torch.Size([1, 1000, 776]) | 11\n",
-      "29308    | torch.Size([1, 1000, 768]) | 11\n",
-      "29309    | torch.Size([1, 1000, 762]) | 11\n",
-      "29310    | torch.Size([1, 1000, 849]) | 11\n",
-      "29311    | torch.Size([1, 1000, 799]) | 11\n",
-      "29312    | torch.Size([1, 1000, 779]) | 11\n",
-      "29313    | torch.Size([1, 1000, 754]) | 11\n",
-      "29314    | torch.Size([1, 1000, 754]) | 11\n",
-      "29315    | torch.Size([1, 1000, 789]) | 11\n",
-      "29316    | torch.Size([1, 1000, 848]) | 11\n",
-      "29317    | torch.Size([1, 1000, 773]) | 11\n",
-      "29318    | torch.Size([1, 1000, 777]) | 11\n",
-      "29319    | torch.Size([1, 1000, 741]) | 11\n",
-      "29320    | torch.Size([1, 1000, 809]) | 11\n",
-      "29321    | torch.Size([1, 1000, 754]) | 11\n",
-      "29322    | torch.Size([1, 1000, 754]) | 11\n",
-      "29323    | torch.Size([1, 1000, 775]) | 11\n",
-      "29324    | torch.Size([1, 1000, 772]) | 11\n",
-      "29325    | torch.Size([1, 1000, 777]) | 11\n",
-      "29326    | torch.Size([1, 1000, 754]) | 11\n",
-      "29327    | torch.Size([1, 1000, 754]) | 11\n",
-      "29328    | torch.Size([1, 1000, 754]) | 11\n",
-      "29329    | torch.Size([1, 1000, 754]) | 11\n",
-      "29330    | torch.Size([1, 1000, 754]) | 11\n",
-      "29331    | torch.Size([1, 1000, 762]) | 11\n",
-      "29332    | torch.Size([1, 1000, 777]) | 11\n",
-      "29333    | torch.Size([1, 1000, 762]) | 11\n",
-      "29334    | torch.Size([1, 1000, 777]) | 11\n",
-      "29335    | torch.Size([1, 1000, 813]) | 11\n",
-      "29336    | torch.Size([1, 1000, 777]) | 11\n",
-      "29337    | torch.Size([1, 1000, 754]) | 11\n",
-      "29338    | torch.Size([1, 1000, 778]) | 11\n",
-      "29339    | torch.Size([1, 1000, 777]) | 11\n",
-      "29340    | torch.Size([1, 1000, 762]) | 11\n",
-      "29341    | torch.Size([1, 1000, 777]) | 11\n",
-      "29342    | torch.Size([1, 1000, 804]) | 11\n",
-      "29343    | torch.Size([1, 1000, 762]) | 11\n",
-      "29344    | torch.Size([1, 1000, 754]) | 11\n",
-      "29345    | torch.Size([1, 1000, 659]) | 11\n",
-      "29346    | torch.Size([1, 1000, 777]) | 11\n",
-      "29347    | torch.Size([1, 1000, 762]) | 11\n",
-      "29348    | torch.Size([1, 1000, 767]) | 11\n",
-      "29349    | torch.Size([1, 1000, 762]) | 11\n",
-      "29350    | torch.Size([1, 1000, 762]) | 11\n",
-      "29351    | torch.Size([1, 1000, 754]) | 11\n",
-      "29352    | torch.Size([1, 1000, 778]) | 11\n",
-      "29353    | torch.Size([1, 1000, 781]) | 11\n",
-      "29354    | torch.Size([1, 1000, 754]) | 11\n",
-      "29355    | torch.Size([1, 1000, 754]) | 11\n",
-      "29356    | torch.Size([1, 1000, 754]) | 11\n",
-      "29357    | torch.Size([1, 1000, 762]) | 11\n",
-      "29358    | torch.Size([1, 1000, 754]) | 11\n",
-      "29359    | torch.Size([1, 1000, 802]) | 11\n",
-      "29360    | torch.Size([1, 1000, 754]) | 11\n",
-      "29361    | torch.Size([1, 1000, 776]) | 11\n",
-      "29362    | torch.Size([1, 1000, 848]) | 11\n",
-      "29363    | torch.Size([1, 1000, 754]) | 11\n",
-      "29364    | torch.Size([1, 1000, 762]) | 11\n",
-      "29365    | torch.Size([1, 1000, 840]) | 11\n",
-      "29366    | torch.Size([1, 1000, 777]) | 11\n",
-      "29367    | torch.Size([1, 1000, 762]) | 11\n",
-      "29368    | torch.Size([1, 1000, 762]) | 11\n",
-      "29369    | torch.Size([1, 1000, 754]) | 11\n",
-      "29370    | torch.Size([1, 1000, 754]) | 11\n",
-      "29371    | torch.Size([1, 1000, 777]) | 11\n",
-      "29372    | torch.Size([1, 1000, 754]) | 11\n",
-      "29373    | torch.Size([1, 1000, 777]) | 11\n",
-      "29374    | torch.Size([1, 1000, 754]) | 11\n",
-      "29375    | torch.Size([1, 1000, 762]) | 11\n",
-      "29376    | torch.Size([1, 1000, 754]) | 11\n",
-      "29377    | torch.Size([1, 1000, 781]) | 11\n",
-      "29378    | torch.Size([1, 1000, 760]) | 11\n",
-      "29379    | torch.Size([1, 1000, 762]) | 11\n",
-      "29380    | torch.Size([1, 1000, 613]) | 11\n",
-      "29381    | torch.Size([1, 1000, 777]) | 11\n",
-      "29382    | torch.Size([1, 1000, 742]) | 11\n",
-      "29383    | torch.Size([1, 1000, 777]) | 11\n",
-      "29384    | torch.Size([1, 1000, 754]) | 11\n",
-      "29385    | torch.Size([1, 1000, 777]) | 11\n",
-      "29386    | torch.Size([1, 1000, 777]) | 11\n",
-      "29387    | torch.Size([1, 1000, 775]) | 11\n",
-      "29388    | torch.Size([1, 1000, 754]) | 11\n",
-      "29389    | torch.Size([1, 1000, 762]) | 11\n",
-      "29390    | torch.Size([1, 1000, 713]) | 11\n",
-      "29391    | torch.Size([1, 1000, 754]) | 11\n",
-      "29392    | torch.Size([1, 1000, 762]) | 11\n",
-      "29393    | torch.Size([1, 1000, 785]) | 11\n",
-      "29394    | torch.Size([1, 1000, 754]) | 11\n",
-      "29395    | torch.Size([1, 1000, 768]) | 11\n",
-      "29396    | torch.Size([1, 1000, 810]) | 11\n",
-      "29397    | torch.Size([1, 1000, 773]) | 11\n",
-      "29398    | torch.Size([1, 1000, 754]) | 11\n",
-      "29399    | torch.Size([1, 1000, 770]) | 11\n",
-      "29400    | torch.Size([1, 1000, 754]) | 11\n",
-      "29401    | torch.Size([1, 1000, 754]) | 11\n",
-      "29402    | torch.Size([1, 1000, 785]) | 11\n",
-      "29403    | torch.Size([1, 1000, 811]) | 11\n",
-      "29404    | torch.Size([1, 1000, 762]) | 11\n",
-      "29405    | torch.Size([1, 1000, 811]) | 11\n",
-      "29406    | torch.Size([1, 1000, 802]) | 11\n",
-      "29407    | torch.Size([1, 1000, 777]) | 11\n",
-      "29408    | torch.Size([1, 1000, 781]) | 11\n",
-      "29409    | torch.Size([1, 1000, 613]) | 11\n",
-      "29410    | torch.Size([1, 1000, 774]) | 11\n",
-      "29411    | torch.Size([1, 1000, 777]) | 11\n",
-      "29412    | torch.Size([1, 1000, 790]) | 11\n",
-      "29413    | torch.Size([1, 1000, 802]) | 11\n",
-      "29414    | torch.Size([1, 1000, 754]) | 11\n",
-      "29415    | torch.Size([1, 1000, 772]) | 11\n",
-      "29416    | torch.Size([1, 1000, 773]) | 11\n",
-      "29417    | torch.Size([1, 1000, 754]) | 11\n",
-      "29418    | torch.Size([1, 1000, 762]) | 11\n",
-      "29419    | torch.Size([1, 1000, 809]) | 11\n",
-      "29420    | torch.Size([1, 1000, 793]) | 11\n",
-      "29421    | torch.Size([1, 1000, 754]) | 11\n",
-      "29422    | torch.Size([1, 1000, 754]) | 11\n",
-      "29423    | torch.Size([1, 1000, 754]) | 11\n",
-      "29424    | torch.Size([1, 1000, 754]) | 11\n",
-      "29425    | torch.Size([1, 1000, 777]) | 11\n",
-      "29426    | torch.Size([1, 1000, 754]) | 11\n",
-      "29427    | torch.Size([1, 1000, 754]) | 11\n",
-      "29428    | torch.Size([1, 1000, 754]) | 11\n",
-      "29429    | torch.Size([1, 1000, 780]) | 11\n",
-      "29430    | torch.Size([1, 1000, 806]) | 11\n",
-      "29431    | torch.Size([1, 1000, 754]) | 11\n",
-      "29432    | torch.Size([1, 1000, 850]) | 11\n",
-      "29433    | torch.Size([1, 1000, 754]) | 11\n",
-      "29434    | torch.Size([1, 1000, 754]) | 11\n",
-      "29435    | torch.Size([1, 1000, 784]) | 11\n",
-      "29436    | torch.Size([1, 1000, 754]) | 11\n",
-      "29437    | torch.Size([1, 1000, 777]) | 11\n",
-      "29438    | torch.Size([1, 1000, 802]) | 11\n",
-      "29439    | torch.Size([1, 1000, 754]) | 11\n",
-      "29440    | torch.Size([1, 1000, 777]) | 11\n",
-      "29441    | torch.Size([1, 1000, 774]) | 11\n",
-      "29442    | torch.Size([1, 1000, 780]) | 11\n",
-      "29443    | torch.Size([1, 1000, 754]) | 11\n",
-      "29444    | torch.Size([1, 1000, 615]) | 11\n",
-      "29445    | torch.Size([1, 1000, 772]) | 11\n",
-      "29446    | torch.Size([1, 1000, 777]) | 11\n",
-      "29447    | torch.Size([1, 1000, 766]) | 11\n",
-      "29448    | torch.Size([1, 1000, 762]) | 11\n",
-      "29449    | torch.Size([1, 1000, 782]) | 11\n",
-      "29450    | torch.Size([1, 1000, 614]) | 11\n",
-      "29451    | torch.Size([1, 1000, 786]) | 11\n",
-      "29452    | torch.Size([1, 1000, 636]) | 11\n",
-      "29453    | torch.Size([1, 1000, 762]) | 11\n",
-      "29454    | torch.Size([1, 1000, 754]) | 11\n",
-      "29455    | torch.Size([1, 1000, 754]) | 11\n",
-      "29456    | torch.Size([1, 1000, 775]) | 11\n",
-      "29457    | torch.Size([1, 1000, 801]) | 11\n",
-      "29458    | torch.Size([1, 1000, 781]) | 11\n",
-      "29459    | torch.Size([1, 1000, 754]) | 11\n",
-      "29460    | torch.Size([1, 1000, 762]) | 11\n",
-      "29461    | torch.Size([1, 1000, 766]) | 11\n",
-      "29462    | torch.Size([1, 1000, 754]) | 11\n",
-      "29463    | torch.Size([1, 1000, 781]) | 11\n",
-      "29464    | torch.Size([1, 1000, 754]) | 11\n",
-      "29465    | torch.Size([1, 1000, 754]) | 11\n",
-      "29466    | torch.Size([1, 1000, 754]) | 11\n",
-      "29467    | torch.Size([1, 1000, 762]) | 11\n",
-      "29468    | torch.Size([1, 1000, 766]) | 11\n",
-      "29469    | torch.Size([1, 1000, 613]) | 11\n",
-      "29470    | torch.Size([1, 1000, 754]) | 11\n",
-      "29471    | torch.Size([1, 1000, 762]) | 11\n",
-      "29472    | torch.Size([1, 1000, 777]) | 11\n",
-      "29473    | torch.Size([1, 1000, 754]) | 11\n",
-      "29474    | torch.Size([1, 1000, 778]) | 11\n",
-      "29475    | torch.Size([1, 1000, 777]) | 11\n",
-      "29476    | torch.Size([1, 1000, 607]) | 11\n",
-      "29477    | torch.Size([1, 1000, 798]) | 11\n",
-      "29478    | torch.Size([1, 1000, 787]) | 11\n",
-      "29479    | torch.Size([1, 1000, 754]) | 11\n",
-      "29480    | torch.Size([1, 1000, 779]) | 11\n",
-      "29481    | torch.Size([1, 1000, 787]) | 11\n",
-      "29482    | torch.Size([1, 1000, 790]) | 11\n",
-      "29483    | torch.Size([1, 1000, 776]) | 11\n",
-      "29484    | torch.Size([1, 1000, 754]) | 11\n",
-      "29485    | torch.Size([1, 1000, 784]) | 11\n",
-      "29486    | torch.Size([1, 1000, 646]) | 11\n",
-      "29487    | torch.Size([1, 1000, 777]) | 11\n",
-      "29488    | torch.Size([1, 1000, 754]) | 11\n",
-      "29489    | torch.Size([1, 1000, 762]) | 11\n",
-      "29490    | torch.Size([1, 1000, 754]) | 11\n",
-      "29491    | torch.Size([1, 1000, 650]) | 11\n",
-      "29492    | torch.Size([1, 1000, 772]) | 11\n",
-      "29493    | torch.Size([1, 1000, 777]) | 11\n",
-      "29494    | torch.Size([1, 1000, 773]) | 11\n",
-      "29495    | torch.Size([1, 1000, 754]) | 11\n",
-      "29496    | torch.Size([1, 1000, 754]) | 11\n",
-      "29497    | torch.Size([1, 1000, 762]) | 11\n",
-      "29498    | torch.Size([1, 1000, 754]) | 11\n",
-      "29499    | torch.Size([1, 1000, 785]) | 11\n",
-      "29500    | torch.Size([1, 1000, 754]) | 11\n",
-      "29501    | torch.Size([1, 1000, 812]) | 11\n",
-      "29502    | torch.Size([1, 1000, 844]) | 11\n",
-      "29503    | torch.Size([1, 1000, 762]) | 11\n",
-      "29504    | torch.Size([1, 1000, 615]) | 11\n",
-      "29505    | torch.Size([1, 1000, 754]) | 11\n",
-      "29506    | torch.Size([1, 1000, 788]) | 11\n",
-      "29507    | torch.Size([1, 1000, 754]) | 11\n",
-      "29508    | torch.Size([1, 1000, 772]) | 11\n",
-      "29509    | torch.Size([1, 1000, 777]) | 11\n",
-      "29510    | torch.Size([1, 1000, 611]) | 11\n",
-      "29511    | torch.Size([1, 1000, 773]) | 11\n",
-      "29512    | torch.Size([1, 1000, 784]) | 11\n",
-      "29513    | torch.Size([1, 1000, 771]) | 11\n",
-      "29514    | torch.Size([1, 1000, 809]) | 11\n",
-      "29515    | torch.Size([1, 1000, 754]) | 11\n",
-      "29516    | torch.Size([1, 1000, 754]) | 11\n",
-      "29517    | torch.Size([1, 1000, 848]) | 11\n",
-      "29518    | torch.Size([1, 1000, 853]) | 11\n",
-      "29519    | torch.Size([1, 1000, 843]) | 11\n",
-      "29520    | torch.Size([1, 1000, 777]) | 11\n",
-      "29521    | torch.Size([1, 1000, 762]) | 11\n",
-      "29522    | torch.Size([1, 1000, 762]) | 11\n",
-      "29523    | torch.Size([1, 1000, 625]) | 11\n",
-      "29524    | torch.Size([1, 1000, 809]) | 11\n",
-      "29525    | torch.Size([1, 1000, 774]) | 11\n",
-      "29526    | torch.Size([1, 1000, 754]) | 11\n",
-      "29527    | torch.Size([1, 1000, 762]) | 11\n",
-      "29528    | torch.Size([1, 1000, 777]) | 11\n",
-      "29529    | torch.Size([1, 1000, 762]) | 11\n",
-      "29530    | torch.Size([1, 1000, 754]) | 11\n",
-      "29531    | torch.Size([1, 1000, 782]) | 11\n",
-      "29532    | torch.Size([1, 1000, 754]) | 11\n",
-      "29533    | torch.Size([1, 1000, 762]) | 11\n",
-      "29534    | torch.Size([1, 1000, 754]) | 11\n",
-      "29535    | torch.Size([1, 1000, 641]) | 11\n",
-      "29536    | torch.Size([1, 1000, 777]) | 11\n",
-      "29537    | torch.Size([1, 1000, 754]) | 11\n",
-      "29538    | torch.Size([1, 1000, 754]) | 11\n",
-      "29539    | torch.Size([1, 1000, 739]) | 11\n",
-      "29540    | torch.Size([1, 1000, 754]) | 11\n",
-      "29541    | torch.Size([1, 1000, 620]) | 11\n",
-      "29542    | torch.Size([1, 1000, 754]) | 11\n",
-      "29543    | torch.Size([1, 1000, 754]) | 11\n",
-      "29544    | torch.Size([1, 1000, 777]) | 11\n",
-      "29545    | torch.Size([1, 1000, 762]) | 11\n",
-      "29546    | torch.Size([1, 1000, 754]) | 11\n",
-      "29547    | torch.Size([1, 1000, 614]) | 11\n",
-      "29548    | torch.Size([1, 1000, 754]) | 11\n",
-      "29549    | torch.Size([1, 1000, 769]) | 11\n",
-      "29550    | torch.Size([1, 1000, 804]) | 11\n",
-      "29551    | torch.Size([1, 1000, 809]) | 11\n",
-      "29552    | torch.Size([1, 1000, 663]) | 11\n",
-      "29553    | torch.Size([1, 1000, 781]) | 11\n",
-      "29554    | torch.Size([1, 1000, 754]) | 11\n",
-      "29555    | torch.Size([1, 1000, 754]) | 11\n",
-      "29556    | torch.Size([1, 1000, 781]) | 11\n",
-      "29557    | torch.Size([1, 1000, 754]) | 11\n",
-      "29558    | torch.Size([1, 1000, 713]) | 11\n",
-      "29559    | torch.Size([1, 1000, 762]) | 11\n",
-      "29560    | torch.Size([1, 1000, 754]) | 11\n",
-      "29561    | torch.Size([1, 1000, 616]) | 11\n",
-      "29562    | torch.Size([1, 1000, 754]) | 11\n",
-      "29563    | torch.Size([1, 1000, 754]) | 11\n",
-      "29564    | torch.Size([1, 1000, 754]) | 11\n",
-      "29565    | torch.Size([1, 1000, 777]) | 11\n",
-      "29566    | torch.Size([1, 1000, 771]) | 11\n",
-      "29567    | torch.Size([1, 1000, 777]) | 11\n",
-      "29568    | torch.Size([1, 1000, 782]) | 11\n",
-      "29569    | torch.Size([1, 1000, 777]) | 11\n",
-      "29570    | torch.Size([1, 1000, 762]) | 11\n",
-      "29571    | torch.Size([1, 1000, 754]) | 11\n",
-      "29572    | torch.Size([1, 1000, 794]) | 11\n",
-      "29573    | torch.Size([1, 1000, 780]) | 11\n",
-      "29574    | torch.Size([1, 1000, 762]) | 11\n",
-      "29575    | torch.Size([1, 1000, 762]) | 11\n",
-      "29576    | torch.Size([1, 1000, 662]) | 11\n",
-      "29577    | torch.Size([1, 1000, 762]) | 11\n",
-      "29578    | torch.Size([1, 1000, 762]) | 11\n",
-      "29579    | torch.Size([1, 1000, 777]) | 11\n",
-      "29580    | torch.Size([1, 1000, 786]) | 11\n",
-      "29581    | torch.Size([1, 1000, 772]) | 11\n",
-      "29582    | torch.Size([1, 1000, 762]) | 11\n",
-      "29583    | torch.Size([1, 1000, 825]) | 11\n",
-      "29584    | torch.Size([1, 1000, 762]) | 11\n",
-      "29585    | torch.Size([1, 1000, 754]) | 11\n",
-      "29586    | torch.Size([1, 1000, 754]) | 11\n",
-      "29587    | torch.Size([1, 1000, 777]) | 11\n",
-      "29588    | torch.Size([1, 1000, 754]) | 11\n",
-      "29589    | torch.Size([1, 1000, 784]) | 11\n",
-      "29590    | torch.Size([1, 1000, 754]) | 11\n",
-      "29591    | torch.Size([1, 1000, 802]) | 11\n",
-      "29592    | torch.Size([1, 1000, 761]) | 11\n",
-      "29593    | torch.Size([1, 1000, 843]) | 11\n",
-      "29594    | torch.Size([1, 1000, 784]) | 11\n",
-      "29595    | torch.Size([1, 1000, 754]) | 11\n",
-      "29596    | torch.Size([1, 1000, 611]) | 11\n",
-      "29597    | torch.Size([1, 1000, 817]) | 11\n",
-      "29598    | torch.Size([1, 1000, 784]) | 11\n",
-      "29599    | torch.Size([1, 1000, 762]) | 11\n",
-      "29600    | torch.Size([1, 1000, 764]) | 11\n",
-      "29601    | torch.Size([1, 1000, 762]) | 11\n",
-      "29602    | torch.Size([1, 1000, 777]) | 11\n",
-      "29603    | torch.Size([1, 1000, 782]) | 11\n",
-      "29604    | torch.Size([1, 1000, 762]) | 11\n",
-      "29605    | torch.Size([1, 1000, 802]) | 11\n",
-      "29606    | torch.Size([1, 1000, 754]) | 11\n",
-      "29607    | torch.Size([1, 1000, 754]) | 11\n",
-      "29608    | torch.Size([1, 1000, 754]) | 11\n",
-      "29609    | torch.Size([1, 1000, 793]) | 11\n",
-      "29610    | torch.Size([1, 1000, 762]) | 11\n",
-      "29611    | torch.Size([1, 1000, 762]) | 11\n",
-      "29612    | torch.Size([1, 1000, 754]) | 11\n",
-      "29613    | torch.Size([1, 1000, 754]) | 11\n",
-      "29614    | torch.Size([1, 1000, 762]) | 11\n",
-      "29615    | torch.Size([1, 1000, 762]) | 11\n",
-      "29616    | torch.Size([1, 1000, 635]) | 11\n",
-      "29617    | torch.Size([1, 1000, 754]) | 11\n",
-      "29618    | torch.Size([1, 1000, 777]) | 11\n",
-      "29619    | torch.Size([1, 1000, 754]) | 11\n",
-      "29620    | torch.Size([1, 1000, 773]) | 11\n",
-      "29621    | torch.Size([1, 1000, 754]) | 11\n",
-      "29622    | torch.Size([1, 1000, 777]) | 11\n",
-      "29623    | torch.Size([1, 1000, 754]) | 11\n",
-      "29624    | torch.Size([1, 1000, 789]) | 11\n",
-      "29625    | torch.Size([1, 1000, 713]) | 11\n",
-      "29626    | torch.Size([1, 1000, 762]) | 11\n",
-      "29627    | torch.Size([1, 1000, 777]) | 11\n",
-      "29628    | torch.Size([1, 1000, 754]) | 11\n",
-      "29629    | torch.Size([1, 1000, 754]) | 11\n",
-      "29630    | torch.Size([1, 1000, 764]) | 11\n",
-      "29631    | torch.Size([1, 1000, 781]) | 11\n",
-      "29632    | torch.Size([1, 1000, 802]) | 11\n",
-      "29633    | torch.Size([1, 1000, 754]) | 11\n",
-      "29634    | torch.Size([1, 1000, 754]) | 11\n",
-      "29635    | torch.Size([1, 1000, 754]) | 11\n",
-      "29636    | torch.Size([1, 1000, 762]) | 11\n",
-      "29637    | torch.Size([1, 1000, 754]) | 11\n",
-      "29638    | torch.Size([1, 1000, 777]) | 11\n",
-      "29639    | torch.Size([1, 1000, 762]) | 11\n",
-      "29640    | torch.Size([1, 1000, 777]) | 11\n",
-      "29641    | torch.Size([1, 1000, 777]) | 11\n",
-      "29642    | torch.Size([1, 1000, 780]) | 11\n",
-      "29643    | torch.Size([1, 1000, 777]) | 11\n",
-      "29644    | torch.Size([1, 1000, 771]) | 11\n",
-      "29645    | torch.Size([1, 1000, 754]) | 11\n",
-      "29646    | torch.Size([1, 1000, 762]) | 11\n",
-      "29647    | torch.Size([1, 1000, 754]) | 11\n",
-      "29648    | torch.Size([1, 1000, 754]) | 11\n",
-      "29649    | torch.Size([1, 1000, 660]) | 11\n",
-      "29650    | torch.Size([1, 1000, 754]) | 11\n",
-      "29651    | torch.Size([1, 1000, 791]) | 11\n",
-      "29652    | torch.Size([1, 1000, 781]) | 11\n",
-      "29653    | torch.Size([1, 1000, 754]) | 11\n",
-      "29654    | torch.Size([1, 1000, 794]) | 11\n",
-      "29655    | torch.Size([1, 1000, 754]) | 11\n",
-      "29656    | torch.Size([1, 1000, 754]) | 11\n",
-      "29657    | torch.Size([1, 1000, 754]) | 11\n",
-      "29658    | torch.Size([1, 1000, 762]) | 11\n",
-      "29659    | torch.Size([1, 1000, 803]) | 11\n",
-      "29660    | torch.Size([1, 1000, 778]) | 11\n",
-      "29661    | torch.Size([1, 1000, 807]) | 11\n",
-      "29662    | torch.Size([1, 1000, 785]) | 11\n",
-      "29663    | torch.Size([1, 1000, 754]) | 11\n",
-      "29664    | torch.Size([1, 1000, 762]) | 11\n",
-      "29665    | torch.Size([1, 1000, 780]) | 11\n",
-      "29666    | torch.Size([1, 1000, 754]) | 11\n",
-      "29667    | torch.Size([1, 1000, 762]) | 11\n",
-      "29668    | torch.Size([1, 1000, 834]) | 11\n",
-      "29669    | torch.Size([1, 1000, 778]) | 11\n",
-      "29670    | torch.Size([1, 1000, 762]) | 11\n",
-      "29671    | torch.Size([1, 1000, 778]) | 11\n",
-      "29672    | torch.Size([1, 1000, 762]) | 11\n",
-      "29673    | torch.Size([1, 1000, 762]) | 11\n",
-      "29674    | torch.Size([1, 1000, 754]) | 11\n",
-      "29675    | torch.Size([1, 1000, 799]) | 11\n",
-      "29676    | torch.Size([1, 1000, 754]) | 11\n",
-      "29677    | torch.Size([1, 1000, 754]) | 11\n",
-      "29678    | torch.Size([1, 1000, 775]) | 11\n",
-      "29679    | torch.Size([1, 1000, 785]) | 11\n",
-      "29680    | torch.Size([1, 1000, 777]) | 11\n",
-      "29681    | torch.Size([1, 1000, 768]) | 11\n",
-      "29682    | torch.Size([1, 1000, 776]) | 11\n",
-      "29683    | torch.Size([1, 1000, 754]) | 11\n",
-      "29684    | torch.Size([1, 1000, 788]) | 11\n",
-      "29685    | torch.Size([1, 1000, 788]) | 11\n",
-      "29686    | torch.Size([1, 1000, 777]) | 11\n",
-      "29687    | torch.Size([1, 1000, 768]) | 11\n",
-      "29688    | torch.Size([1, 1000, 787]) | 11\n",
-      "29689    | torch.Size([1, 1000, 802]) | 11\n",
-      "29690    | torch.Size([1, 1000, 754]) | 11\n",
-      "29691    | torch.Size([1, 1000, 762]) | 11\n",
-      "29692    | torch.Size([1, 1000, 762]) | 11\n",
-      "29693    | torch.Size([1, 1000, 764]) | 11\n",
-      "29694    | torch.Size([1, 1000, 774]) | 11\n",
-      "29695    | torch.Size([1, 1000, 762]) | 11\n",
-      "29696    | torch.Size([1, 1000, 777]) | 11\n",
-      "29697    | torch.Size([1, 1000, 754]) | 11\n",
-      "29698    | torch.Size([1, 1000, 762]) | 11\n",
-      "29699    | torch.Size([1, 1000, 771]) | 11\n",
-      "29700    | torch.Size([1, 1000, 754]) | 11\n",
-      "29701    | torch.Size([1, 1000, 780]) | 11\n",
-      "29702    | torch.Size([1, 1000, 754]) | 11\n",
-      "29703    | torch.Size([1, 1000, 762]) | 11\n",
-      "29704    | torch.Size([1, 1000, 762]) | 11\n",
-      "29705    | torch.Size([1, 1000, 771]) | 11\n",
-      "29706    | torch.Size([1, 1000, 754]) | 11\n",
-      "29707    | torch.Size([1, 1000, 773]) | 11\n",
-      "29708    | torch.Size([1, 1000, 778]) | 11\n",
-      "29709    | torch.Size([1, 1000, 771]) | 11\n",
-      "29710    | torch.Size([1, 1000, 754]) | 11\n",
-      "29711    | torch.Size([1, 1000, 808]) | 11\n",
-      "29712    | torch.Size([1, 1000, 754]) | 11\n",
-      "29713    | torch.Size([1, 1000, 754]) | 11\n",
-      "29714    | torch.Size([1, 1000, 754]) | 11\n",
-      "29715    | torch.Size([1, 1000, 802]) | 11\n",
-      "29716    | torch.Size([1, 1000, 777]) | 11\n",
-      "29717    | torch.Size([1, 1000, 762]) | 11\n",
-      "29718    | torch.Size([1, 1000, 754]) | 11\n",
-      "29719    | torch.Size([1, 1000, 762]) | 11\n",
-      "29720    | torch.Size([1, 1000, 776]) | 11\n",
-      "29721    | torch.Size([1, 1000, 754]) | 11\n",
-      "29722    | torch.Size([1, 1000, 762]) | 11\n",
-      "29723    | torch.Size([1, 1000, 762]) | 11\n",
-      "29724    | torch.Size([1, 1000, 762]) | 11\n",
-      "29725    | torch.Size([1, 1000, 754]) | 11\n",
-      "29726    | torch.Size([1, 1000, 762]) | 11\n",
-      "29727    | torch.Size([1, 1000, 754]) | 11\n",
-      "29728    | torch.Size([1, 1000, 754]) | 11\n",
-      "29729    | torch.Size([1, 1000, 762]) | 11\n",
-      "29730    | torch.Size([1, 1000, 762]) | 11\n",
-      "29731    | torch.Size([1, 1000, 762]) | 11\n",
-      "29732    | torch.Size([1, 1000, 754]) | 11\n",
-      "29733    | torch.Size([1, 1000, 762]) | 11\n",
-      "29734    | torch.Size([1, 1000, 790]) | 11\n",
-      "29735    | torch.Size([1, 1000, 754]) | 11\n",
-      "29736    | torch.Size([1, 1000, 762]) | 11\n",
-      "29737    | torch.Size([1, 1000, 754]) | 11\n",
-      "29738    | torch.Size([1, 1000, 768]) | 11\n",
-      "29739    | torch.Size([1, 1000, 779]) | 11\n",
-      "29740    | torch.Size([1, 1000, 796]) | 11\n",
-      "29741    | torch.Size([1, 1000, 754]) | 11\n",
-      "29742    | torch.Size([1, 1000, 802]) | 11\n",
-      "29743    | torch.Size([1, 1000, 777]) | 11\n",
-      "29744    | torch.Size([1, 1000, 777]) | 11\n",
-      "29745    | torch.Size([1, 1000, 754]) | 11\n",
-      "29746    | torch.Size([1, 1000, 779]) | 11\n",
-      "29747    | torch.Size([1, 1000, 779]) | 11\n",
-      "29748    | torch.Size([1, 1000, 754]) | 11\n",
-      "29749    | torch.Size([1, 1000, 762]) | 11\n",
-      "29750    | torch.Size([1, 1000, 743]) | 11\n",
-      "29751    | torch.Size([1, 1000, 775]) | 11\n",
-      "29752    | torch.Size([1, 1000, 841]) | 11\n",
-      "29753    | torch.Size([1, 1000, 777]) | 11\n",
-      "29754    | torch.Size([1, 1000, 754]) | 11\n",
-      "29755    | torch.Size([1, 1000, 754]) | 11\n",
-      "29756    | torch.Size([1, 1000, 791]) | 11\n",
-      "29757    | torch.Size([1, 1000, 808]) | 11\n",
-      "29758    | torch.Size([1, 1000, 777]) | 11\n",
-      "29759    | torch.Size([1, 1000, 754]) | 11\n",
-      "29760    | torch.Size([1, 1000, 762]) | 11\n",
-      "29761    | torch.Size([1, 1000, 774]) | 11\n",
-      "29762    | torch.Size([1, 1000, 754]) | 11\n",
-      "29763    | torch.Size([1, 1000, 762]) | 11\n",
-      "29764    | torch.Size([1, 1000, 754]) | 11\n",
-      "29765    | torch.Size([1, 1000, 634]) | 11\n",
-      "29766    | torch.Size([1, 1000, 821]) | 11\n",
-      "29767    | torch.Size([1, 1000, 754]) | 11\n",
-      "29768    | torch.Size([1, 1000, 846]) | 11\n",
-      "29769    | torch.Size([1, 1000, 762]) | 11\n",
-      "29770    | torch.Size([1, 1000, 810]) | 11\n",
-      "29771    | torch.Size([1, 1000, 754]) | 11\n",
-      "29772    | torch.Size([1, 1000, 762]) | 11\n",
-      "29773    | torch.Size([1, 1000, 777]) | 11\n",
-      "29774    | torch.Size([1, 1000, 754]) | 11\n",
-      "29775    | torch.Size([1, 1000, 762]) | 11\n",
-      "29776    | torch.Size([1, 1000, 777]) | 11\n",
-      "29777    | torch.Size([1, 1000, 754]) | 11\n",
-      "29778    | torch.Size([1, 1000, 754]) | 11\n",
-      "29779    | torch.Size([1, 1000, 754]) | 11\n",
-      "29780    | torch.Size([1, 1000, 754]) | 11\n",
-      "29781    | torch.Size([1, 1000, 762]) | 11\n",
-      "29782    | torch.Size([1, 1000, 762]) | 11\n",
-      "29783    | torch.Size([1, 1000, 802]) | 11\n",
-      "29784    | torch.Size([1, 1000, 754]) | 11\n",
-      "29785    | torch.Size([1, 1000, 823]) | 11\n",
-      "29786    | torch.Size([1, 1000, 766]) | 11\n",
-      "29787    | torch.Size([1, 1000, 754]) | 11\n",
-      "29788    | torch.Size([1, 1000, 754]) | 11\n",
-      "29789    | torch.Size([1, 1000, 754]) | 11\n",
-      "29790    | torch.Size([1, 1000, 754]) | 11\n",
-      "29791    | torch.Size([1, 1000, 782]) | 11\n",
-      "29792    | torch.Size([1, 1000, 771]) | 11\n",
-      "29793    | torch.Size([1, 1000, 775]) | 11\n",
-      "29794    | torch.Size([1, 1000, 795]) | 11\n",
-      "29795    | torch.Size([1, 1000, 754]) | 11\n",
-      "29796    | torch.Size([1, 1000, 762]) | 11\n",
-      "29797    | torch.Size([1, 1000, 754]) | 11\n",
-      "29798    | torch.Size([1, 1000, 754]) | 11\n",
-      "29799    | torch.Size([1, 1000, 779]) | 11\n",
-      "29800    | torch.Size([1, 1000, 787]) | 11\n",
-      "29801    | torch.Size([1, 1000, 777]) | 11\n",
-      "29802    | torch.Size([1, 1000, 777]) | 11\n",
-      "29803    | torch.Size([1, 1000, 754]) | 11\n",
-      "29804    | torch.Size([1, 1000, 762]) | 11\n",
-      "29805    | torch.Size([1, 1000, 762]) | 11\n",
-      "29806    | torch.Size([1, 1000, 614]) | 11\n",
-      "29807    | torch.Size([1, 1000, 769]) | 11\n",
-      "29808    | torch.Size([1, 1000, 762]) | 11\n",
-      "29809    | torch.Size([1, 1000, 775]) | 11\n",
-      "29810    | torch.Size([1, 1000, 799]) | 11\n",
-      "29811    | torch.Size([1, 1000, 782]) | 11\n",
-      "29812    | torch.Size([1, 1000, 754]) | 11\n",
-      "29813    | torch.Size([1, 1000, 754]) | 11\n",
-      "29814    | torch.Size([1, 1000, 849]) | 11\n",
-      "29815    | torch.Size([1, 1000, 777]) | 11\n",
-      "29816    | torch.Size([1, 1000, 762]) | 11\n",
-      "29817    | torch.Size([1, 1000, 777]) | 11\n",
-      "29818    | torch.Size([1, 1000, 841]) | 11\n",
-      "29819    | torch.Size([1, 1000, 802]) | 11\n",
-      "29820    | torch.Size([1, 1000, 767]) | 11\n",
-      "29821    | torch.Size([1, 1000, 786]) | 11\n",
-      "29822    | torch.Size([1, 1000, 762]) | 11\n",
-      "29823    | torch.Size([1, 1000, 788]) | 11\n",
-      "29824    | torch.Size([1, 1000, 754]) | 11\n",
-      "29825    | torch.Size([1, 1000, 754]) | 11\n",
-      "29826    | torch.Size([1, 1000, 777]) | 11\n",
-      "29827    | torch.Size([1, 1000, 754]) | 11\n",
-      "29828    | torch.Size([1, 1000, 802]) | 11\n",
-      "29829    | torch.Size([1, 1000, 754]) | 11\n",
-      "29830    | torch.Size([1, 1000, 762]) | 11\n",
-      "29831    | torch.Size([1, 1000, 777]) | 11\n",
-      "29832    | torch.Size([1, 1000, 762]) | 11\n",
-      "29833    | torch.Size([1, 1000, 754]) | 11\n",
-      "29834    | torch.Size([1, 1000, 762]) | 11\n",
-      "29835    | torch.Size([1, 1000, 762]) | 11\n",
-      "29836    | torch.Size([1, 1000, 754]) | 11\n",
-      "29837    | torch.Size([1, 1000, 666]) | 11\n",
-      "29838    | torch.Size([1, 1000, 777]) | 11\n",
-      "29839    | torch.Size([1, 1000, 777]) | 11\n",
-      "29840    | torch.Size([1, 1000, 802]) | 11\n",
-      "29841    | torch.Size([1, 1000, 762]) | 11\n",
-      "29842    | torch.Size([1, 1000, 762]) | 11\n",
-      "29843    | torch.Size([1, 1000, 754]) | 11\n",
-      "29844    | torch.Size([1, 1000, 762]) | 11\n",
-      "29845    | torch.Size([1, 1000, 774]) | 11\n",
-      "29846    | torch.Size([1, 1000, 777]) | 11\n",
-      "29847    | torch.Size([1, 1000, 762]) | 11\n",
-      "29848    | torch.Size([1, 1000, 762]) | 11\n",
-      "29849    | torch.Size([1, 1000, 766]) | 11\n",
-      "29850    | torch.Size([1, 1000, 766]) | 11\n",
-      "29851    | torch.Size([1, 1000, 614]) | 11\n",
-      "29852    | torch.Size([1, 1000, 777]) | 11\n",
-      "29853    | torch.Size([1, 1000, 777]) | 11\n",
-      "29854    | torch.Size([1, 1000, 777]) | 11\n",
-      "29855    | torch.Size([1, 1000, 611]) | 11\n",
-      "29856    | torch.Size([1, 1000, 780]) | 11\n",
-      "29857    | torch.Size([1, 1000, 775]) | 11\n",
-      "29858    | torch.Size([1, 1000, 616]) | 11\n",
-      "29859    | torch.Size([1, 1000, 754]) | 11\n",
-      "29860    | torch.Size([1, 1000, 762]) | 11\n",
-      "29861    | torch.Size([1, 1000, 789]) | 11\n",
-      "29862    | torch.Size([1, 1000, 795]) | 11\n",
-      "29863    | torch.Size([1, 1000, 754]) | 11\n",
-      "29864    | torch.Size([1, 1000, 762]) | 11\n",
-      "29865    | torch.Size([1, 1000, 754]) | 11\n",
-      "29866    | torch.Size([1, 1000, 762]) | 11\n",
-      "29867    | torch.Size([1, 1000, 762]) | 11\n",
-      "29868    | torch.Size([1, 1000, 738]) | 11\n",
-      "29869    | torch.Size([1, 1000, 725]) | 11\n",
-      "29870    | torch.Size([1, 1000, 754]) | 11\n",
-      "29871    | torch.Size([1, 1000, 784]) | 11\n",
-      "29872    | torch.Size([1, 1000, 754]) | 11\n",
-      "29873    | torch.Size([1, 1000, 762]) | 11\n",
-      "29874    | torch.Size([1, 1000, 762]) | 11\n",
-      "29875    | torch.Size([1, 1000, 762]) | 11\n",
-      "29876    | torch.Size([1, 1000, 754]) | 11\n",
-      "29877    | torch.Size([1, 1000, 788]) | 11\n",
-      "29878    | torch.Size([1, 1000, 754]) | 11\n",
-      "29879    | torch.Size([1, 1000, 754]) | 11\n",
-      "29880    | torch.Size([1, 1000, 840]) | 11\n",
-      "29881    | torch.Size([1, 1000, 754]) | 11\n",
-      "29882    | torch.Size([1, 1000, 807]) | 11\n",
-      "29883    | torch.Size([1, 1000, 754]) | 11\n",
-      "29884    | torch.Size([1, 1000, 754]) | 11\n",
-      "29885    | torch.Size([1, 1000, 652]) | 11\n",
-      "29886    | torch.Size([1, 1000, 628]) | 11\n",
-      "29887    | torch.Size([1, 1000, 814]) | 11\n",
-      "29888    | torch.Size([1, 1000, 754]) | 11\n",
-      "29889    | torch.Size([1, 1000, 754]) | 11\n",
-      "29890    | torch.Size([1, 1000, 776]) | 11\n",
-      "29891    | torch.Size([1, 1000, 762]) | 11\n",
-      "29892    | torch.Size([1, 1000, 754]) | 11\n",
-      "29893    | torch.Size([1, 1000, 779]) | 11\n",
-      "29894    | torch.Size([1, 1000, 754]) | 11\n",
-      "29895    | torch.Size([1, 1000, 776]) | 11\n",
-      "29896    | torch.Size([1, 1000, 780]) | 11\n",
-      "29897    | torch.Size([1, 1000, 754]) | 11\n",
-      "29898    | torch.Size([1, 1000, 777]) | 11\n",
-      "29899    | torch.Size([1, 1000, 754]) | 11\n",
-      "29900    | torch.Size([1, 1000, 777]) | 11\n",
-      "29901    | torch.Size([1, 1000, 792]) | 11\n",
-      "29902    | torch.Size([1, 1000, 762]) | 11\n",
-      "29903    | torch.Size([1, 1000, 754]) | 11\n",
-      "29904    | torch.Size([1, 1000, 780]) | 11\n",
-      "29905    | torch.Size([1, 1000, 776]) | 11\n",
-      "29906    | torch.Size([1, 1000, 770]) | 11\n",
-      "29907    | torch.Size([1, 1000, 774]) | 11\n",
-      "29908    | torch.Size([1, 1000, 809]) | 11\n",
-      "29909    | torch.Size([1, 1000, 867]) | 11\n",
-      "29910    | torch.Size([1, 1000, 754]) | 11\n",
-      "29911    | torch.Size([1, 1000, 754]) | 11\n",
-      "29912    | torch.Size([1, 1000, 807]) | 11\n",
-      "29913    | torch.Size([1, 1000, 774]) | 11\n",
-      "29914    | torch.Size([1, 1000, 767]) | 11\n",
-      "29915    | torch.Size([1, 1000, 754]) | 11\n",
-      "29916    | torch.Size([1, 1000, 769]) | 11\n",
-      "29917    | torch.Size([1, 1000, 754]) | 11\n",
-      "29918    | torch.Size([1, 1000, 741]) | 11\n",
-      "29919    | torch.Size([1, 1000, 780]) | 11\n",
-      "29920    | torch.Size([1, 1000, 777]) | 11\n",
-      "29921    | torch.Size([1, 1000, 754]) | 12\n",
-      "29922    | torch.Size([1, 1000, 754]) | 12\n",
-      "29923    | torch.Size([1, 1000, 754]) | 12\n",
-      "29924    | torch.Size([1, 1000, 754]) | 12\n",
-      "29925    | torch.Size([1, 1000, 754]) | 12\n",
-      "29926    | torch.Size([1, 1000, 752]) | 12\n",
-      "29927    | torch.Size([1, 1000, 754]) | 12\n",
-      "29928    | torch.Size([1, 1000, 754]) | 12\n",
-      "29929    | torch.Size([1, 1000, 754]) | 12\n",
-      "29930    | torch.Size([1, 1000, 754]) | 12\n",
-      "29931    | torch.Size([1, 1000, 754]) | 12\n",
-      "29932    | torch.Size([1, 1000, 754]) | 12\n",
-      "29933    | torch.Size([1, 1000, 754]) | 12\n",
-      "29934    | torch.Size([1, 1000, 754]) | 12\n",
-      "29935    | torch.Size([1, 1000, 786]) | 12\n",
-      "29936    | torch.Size([1, 1000, 772]) | 12\n",
-      "29937    | torch.Size([1, 1000, 754]) | 12\n",
-      "29938    | torch.Size([1, 1000, 754]) | 12\n",
-      "29939    | torch.Size([1, 1000, 754]) | 12\n",
-      "29940    | torch.Size([1, 1000, 754]) | 12\n",
-      "29941    | torch.Size([1, 1000, 754]) | 12\n",
-      "29942    | torch.Size([1, 1000, 754]) | 12\n",
-      "29943    | torch.Size([1, 1000, 754]) | 12\n",
-      "29944    | torch.Size([1, 1000, 754]) | 12\n",
-      "29945    | torch.Size([1, 1000, 754]) | 12\n",
-      "29946    | torch.Size([1, 1000, 754]) | 12\n",
-      "29947    | torch.Size([1, 1000, 754]) | 12\n",
-      "29948    | torch.Size([1, 1000, 754]) | 12\n",
-      "29949    | torch.Size([1, 1000, 754]) | 12\n",
-      "29950    | torch.Size([1, 1000, 754]) | 12\n",
-      "29951    | torch.Size([1, 1000, 754]) | 12\n",
-      "29952    | torch.Size([1, 1000, 795]) | 12\n",
-      "29953    | torch.Size([1, 1000, 754]) | 12\n",
-      "29954    | torch.Size([1, 1000, 754]) | 12\n",
-      "29955    | torch.Size([1, 1000, 771]) | 12\n",
-      "29956    | torch.Size([1, 1000, 754]) | 12\n",
-      "29957    | torch.Size([1, 1000, 754]) | 12\n",
-      "29958    | torch.Size([1, 1000, 754]) | 12\n",
-      "29959    | torch.Size([1, 1000, 754]) | 12\n",
-      "29960    | torch.Size([1, 1000, 754]) | 12\n",
-      "29961    | torch.Size([1, 1000, 754]) | 12\n",
-      "29962    | torch.Size([1, 1000, 754]) | 12\n",
-      "29963    | torch.Size([1, 1000, 754]) | 12\n",
-      "29964    | torch.Size([1, 1000, 754]) | 12\n",
-      "29965    | torch.Size([1, 1000, 754]) | 12\n",
-      "29966    | torch.Size([1, 1000, 754]) | 12\n",
-      "29967    | torch.Size([1, 1000, 754]) | 12\n",
-      "29968    | torch.Size([1, 1000, 754]) | 12\n",
-      "29969    | torch.Size([1, 1000, 774]) | 12\n",
-      "29970    | torch.Size([1, 1000, 754]) | 12\n",
-      "29971    | torch.Size([1, 1000, 754]) | 12\n",
-      "29972    | torch.Size([1, 1000, 754]) | 12\n",
-      "29973    | torch.Size([1, 1000, 754]) | 12\n",
-      "29974    | torch.Size([1, 1000, 754]) | 12\n",
-      "29975    | torch.Size([1, 1000, 754]) | 12\n",
-      "29976    | torch.Size([1, 1000, 754]) | 12\n",
-      "29977    | torch.Size([1, 1000, 754]) | 12\n",
-      "29978    | torch.Size([1, 1000, 754]) | 12\n",
-      "29979    | torch.Size([1, 1000, 754]) | 12\n",
-      "29980    | torch.Size([1, 1000, 754]) | 12\n",
-      "29981    | torch.Size([1, 1000, 754]) | 12\n",
-      "29982    | torch.Size([1, 1000, 754]) | 12\n",
-      "29983    | torch.Size([1, 1000, 754]) | 12\n",
-      "29984    | torch.Size([1, 1000, 754]) | 12\n",
-      "29985    | torch.Size([1, 1000, 754]) | 12\n",
-      "29986    | torch.Size([1, 1000, 754]) | 12\n",
-      "29987    | torch.Size([1, 1000, 762]) | 12\n",
-      "29988    | torch.Size([1, 1000, 754]) | 12\n",
-      "29989    | torch.Size([1, 1000, 754]) | 12\n",
-      "29990    | torch.Size([1, 1000, 754]) | 12\n",
-      "29991    | torch.Size([1, 1000, 754]) | 12\n",
-      "29992    | torch.Size([1, 1000, 754]) | 12\n",
-      "29993    | torch.Size([1, 1000, 771]) | 12\n",
-      "29994    | torch.Size([1, 1000, 754]) | 12\n",
-      "29995    | torch.Size([1, 1000, 754]) | 12\n",
-      "29996    | torch.Size([1, 1000, 802]) | 12\n",
-      "29997    | torch.Size([1, 1000, 754]) | 12\n",
-      "29998    | torch.Size([1, 1000, 771]) | 12\n",
-      "29999    | torch.Size([1, 1000, 771]) | 12\n",
-      "30000    | torch.Size([1, 1000, 757]) | 12\n",
-      "30001    | torch.Size([1, 1000, 754]) | 12\n",
-      "30002    | torch.Size([1, 1000, 754]) | 12\n",
-      "30003    | torch.Size([1, 1000, 754]) | 12\n",
-      "30004    | torch.Size([1, 1000, 754]) | 12\n",
-      "30005    | torch.Size([1, 1000, 754]) | 12\n",
-      "30006    | torch.Size([1, 1000, 754]) | 12\n",
-      "30007    | torch.Size([1, 1000, 754]) | 12\n",
-      "30008    | torch.Size([1, 1000, 754]) | 12\n",
-      "30009    | torch.Size([1, 1000, 754]) | 12\n",
-      "30010    | torch.Size([1, 1000, 754]) | 12\n",
-      "30011    | torch.Size([1, 1000, 754]) | 12\n",
-      "30012    | torch.Size([1, 1000, 754]) | 12\n",
-      "30013    | torch.Size([1, 1000, 754]) | 12\n",
-      "30014    | torch.Size([1, 1000, 754]) | 12\n",
-      "30015    | torch.Size([1, 1000, 754]) | 12\n",
-      "30016    | torch.Size([1, 1000, 754]) | 12\n",
-      "30017    | torch.Size([1, 1000, 754]) | 12\n",
-      "30018    | torch.Size([1, 1000, 754]) | 12\n",
-      "30019    | torch.Size([1, 1000, 754]) | 12\n",
-      "30020    | torch.Size([1, 1000, 754]) | 12\n",
-      "30021    | torch.Size([1, 1000, 754]) | 12\n",
-      "30022    | torch.Size([1, 1000, 754]) | 12\n",
-      "30023    | torch.Size([1, 1000, 754]) | 12\n",
-      "30024    | torch.Size([1, 1000, 754]) | 12\n",
-      "30025    | torch.Size([1, 1000, 776]) | 12\n",
-      "30026    | torch.Size([1, 1000, 754]) | 12\n",
-      "30027    | torch.Size([1, 1000, 766]) | 12\n",
-      "30028    | torch.Size([1, 1000, 754]) | 12\n",
-      "30029    | torch.Size([1, 1000, 754]) | 12\n",
-      "30030    | torch.Size([1, 1000, 767]) | 12\n",
-      "30031    | torch.Size([1, 1000, 771]) | 12\n",
-      "30032    | torch.Size([1, 1000, 754]) | 12\n",
-      "30033    | torch.Size([1, 1000, 754]) | 12\n",
-      "30034    | torch.Size([1, 1000, 754]) | 12\n",
-      "30035    | torch.Size([1, 1000, 777]) | 12\n",
-      "30036    | torch.Size([1, 1000, 772]) | 12\n",
-      "30037    | torch.Size([1, 1000, 754]) | 12\n",
-      "30038    | torch.Size([1, 1000, 777]) | 12\n",
-      "30039    | torch.Size([1, 1000, 754]) | 12\n",
-      "30040    | torch.Size([1, 1000, 754]) | 12\n",
-      "30041    | torch.Size([1, 1000, 754]) | 12\n",
-      "30042    | torch.Size([1, 1000, 754]) | 12\n",
-      "30043    | torch.Size([1, 1000, 754]) | 12\n",
-      "30044    | torch.Size([1, 1000, 754]) | 12\n",
-      "30045    | torch.Size([1, 1000, 754]) | 12\n",
-      "30046    | torch.Size([1, 1000, 754]) | 12\n",
-      "30047    | torch.Size([1, 1000, 763]) | 12\n",
-      "30048    | torch.Size([1, 1000, 769]) | 12\n",
-      "30049    | torch.Size([1, 1000, 777]) | 12\n",
-      "30050    | torch.Size([1, 1000, 754]) | 12\n",
-      "30051    | torch.Size([1, 1000, 762]) | 12\n",
-      "30052    | torch.Size([1, 1000, 754]) | 12\n",
-      "30053    | torch.Size([1, 1000, 754]) | 12\n",
-      "30054    | torch.Size([1, 1000, 754]) | 12\n",
-      "30055    | torch.Size([1, 1000, 800]) | 12\n",
-      "30056    | torch.Size([1, 1000, 772]) | 12\n",
-      "30057    | torch.Size([1, 1000, 754]) | 12\n",
-      "30058    | torch.Size([1, 1000, 764]) | 12\n",
-      "30059    | torch.Size([1, 1000, 754]) | 12\n",
-      "30060    | torch.Size([1, 1000, 754]) | 12\n",
-      "30061    | torch.Size([1, 1000, 778]) | 12\n",
-      "30062    | torch.Size([1, 1000, 777]) | 12\n",
-      "30063    | torch.Size([1, 1000, 772]) | 12\n",
-      "30064    | torch.Size([1, 1000, 754]) | 12\n",
-      "30065    | torch.Size([1, 1000, 754]) | 12\n",
-      "30066    | torch.Size([1, 1000, 754]) | 12\n",
-      "30067    | torch.Size([1, 1000, 754]) | 12\n",
-      "30068    | torch.Size([1, 1000, 754]) | 12\n",
-      "30069    | torch.Size([1, 1000, 754]) | 12\n",
-      "30070    | torch.Size([1, 1000, 754]) | 12\n",
-      "30071    | torch.Size([1, 1000, 754]) | 12\n",
-      "30072    | torch.Size([1, 1000, 777]) | 12\n",
-      "30073    | torch.Size([1, 1000, 754]) | 12\n",
-      "30074    | torch.Size([1, 1000, 754]) | 12\n",
-      "30075    | torch.Size([1, 1000, 754]) | 12\n",
-      "30076    | torch.Size([1, 1000, 754]) | 12\n",
-      "30077    | torch.Size([1, 1000, 754]) | 12\n",
-      "30078    | torch.Size([1, 1000, 754]) | 12\n",
-      "30079    | torch.Size([1, 1000, 799]) | 12\n",
-      "30080    | torch.Size([1, 1000, 754]) | 12\n",
-      "30081    | torch.Size([1, 1000, 754]) | 12\n",
-      "30082    | torch.Size([1, 1000, 754]) | 12\n",
-      "30083    | torch.Size([1, 1000, 754]) | 12\n",
-      "30084    | torch.Size([1, 1000, 754]) | 12\n",
-      "30085    | torch.Size([1, 1000, 754]) | 12\n",
-      "30086    | torch.Size([1, 1000, 754]) | 12\n",
-      "30087    | torch.Size([1, 1000, 754]) | 12\n",
-      "30088    | torch.Size([1, 1000, 754]) | 12\n",
-      "30089    | torch.Size([1, 1000, 754]) | 12\n",
-      "30090    | torch.Size([1, 1000, 754]) | 12\n",
-      "30091    | torch.Size([1, 1000, 754]) | 12\n",
-      "30092    | torch.Size([1, 1000, 754]) | 12\n",
-      "30093    | torch.Size([1, 1000, 754]) | 12\n",
-      "30094    | torch.Size([1, 1000, 754]) | 12\n",
-      "30095    | torch.Size([1, 1000, 754]) | 12\n",
-      "30096    | torch.Size([1, 1000, 754]) | 12\n",
-      "30097    | torch.Size([1, 1000, 771]) | 12\n",
-      "30098    | torch.Size([1, 1000, 754]) | 12\n",
-      "30099    | torch.Size([1, 1000, 754]) | 12\n",
-      "30100    | torch.Size([1, 1000, 754]) | 12\n",
-      "30101    | torch.Size([1, 1000, 754]) | 12\n",
-      "30102    | torch.Size([1, 1000, 754]) | 12\n",
-      "30103    | torch.Size([1, 1000, 777]) | 12\n",
-      "30104    | torch.Size([1, 1000, 754]) | 12\n",
-      "30105    | torch.Size([1, 1000, 754]) | 12\n",
-      "30106    | torch.Size([1, 1000, 754]) | 12\n",
-      "30107    | torch.Size([1, 1000, 754]) | 12\n",
-      "30108    | torch.Size([1, 1000, 754]) | 12\n",
-      "30109    | torch.Size([1, 1000, 754]) | 12\n",
-      "30110    | torch.Size([1, 1000, 754]) | 12\n",
-      "30111    | torch.Size([1, 1000, 754]) | 12\n",
-      "30112    | torch.Size([1, 1000, 754]) | 12\n",
-      "30113    | torch.Size([1, 1000, 762]) | 12\n",
-      "30114    | torch.Size([1, 1000, 754]) | 12\n",
-      "30115    | torch.Size([1, 1000, 754]) | 12\n",
-      "30116    | torch.Size([1, 1000, 754]) | 12\n",
-      "30117    | torch.Size([1, 1000, 754]) | 12\n",
-      "30118    | torch.Size([1, 1000, 754]) | 12\n",
-      "30119    | torch.Size([1, 1000, 754]) | 12\n",
-      "30120    | torch.Size([1, 1000, 772]) | 12\n",
-      "30121    | torch.Size([1, 1000, 754]) | 12\n",
-      "30122    | torch.Size([1, 1000, 754]) | 12\n",
-      "30123    | torch.Size([1, 1000, 754]) | 12\n",
-      "30124    | torch.Size([1, 1000, 754]) | 12\n",
-      "30125    | torch.Size([1, 1000, 781]) | 12\n",
-      "30126    | torch.Size([1, 1000, 754]) | 12\n",
-      "30127    | torch.Size([1, 1000, 809]) | 12\n",
-      "30128    | torch.Size([1, 1000, 772]) | 12\n",
-      "30129    | torch.Size([1, 1000, 754]) | 12\n",
-      "30130    | torch.Size([1, 1000, 754]) | 12\n",
-      "30131    | torch.Size([1, 1000, 754]) | 12\n",
-      "30132    | torch.Size([1, 1000, 754]) | 12\n",
-      "30133    | torch.Size([1, 1000, 754]) | 12\n",
-      "30134    | torch.Size([1, 1000, 754]) | 12\n",
-      "30135    | torch.Size([1, 1000, 771]) | 12\n",
-      "30136    | torch.Size([1, 1000, 799]) | 12\n",
-      "30137    | torch.Size([1, 1000, 754]) | 12\n",
-      "30138    | torch.Size([1, 1000, 754]) | 12\n",
-      "30139    | torch.Size([1, 1000, 772]) | 12\n",
-      "30140    | torch.Size([1, 1000, 754]) | 12\n",
-      "30141    | torch.Size([1, 1000, 754]) | 12\n",
-      "30142    | torch.Size([1, 1000, 754]) | 12\n",
-      "30143    | torch.Size([1, 1000, 754]) | 12\n",
-      "30144    | torch.Size([1, 1000, 754]) | 12\n",
-      "30145    | torch.Size([1, 1000, 754]) | 12\n",
-      "30146    | torch.Size([1, 1000, 767]) | 12\n",
-      "30147    | torch.Size([1, 1000, 754]) | 12\n",
-      "30148    | torch.Size([1, 1000, 754]) | 12\n",
-      "30149    | torch.Size([1, 1000, 754]) | 12\n",
-      "30150    | torch.Size([1, 1000, 754]) | 12\n",
-      "30151    | torch.Size([1, 1000, 754]) | 12\n",
-      "30152    | torch.Size([1, 1000, 754]) | 12\n",
-      "30153    | torch.Size([1, 1000, 754]) | 12\n",
-      "30154    | torch.Size([1, 1000, 779]) | 12\n",
-      "30155    | torch.Size([1, 1000, 754]) | 12\n",
-      "30156    | torch.Size([1, 1000, 799]) | 12\n",
-      "30157    | torch.Size([1, 1000, 769]) | 12\n",
-      "30158    | torch.Size([1, 1000, 754]) | 12\n",
-      "30159    | torch.Size([1, 1000, 754]) | 12\n",
-      "30160    | torch.Size([1, 1000, 754]) | 12\n",
-      "30161    | torch.Size([1, 1000, 754]) | 12\n",
-      "30162    | torch.Size([1, 1000, 783]) | 12\n",
-      "30163    | torch.Size([1, 1000, 754]) | 12\n",
-      "30164    | torch.Size([1, 1000, 754]) | 12\n",
-      "30165    | torch.Size([1, 1000, 754]) | 12\n",
-      "30166    | torch.Size([1, 1000, 799]) | 12\n",
-      "30167    | torch.Size([1, 1000, 754]) | 12\n",
-      "30168    | torch.Size([1, 1000, 754]) | 12\n",
-      "30169    | torch.Size([1, 1000, 754]) | 12\n",
-      "30170    | torch.Size([1, 1000, 754]) | 12\n",
-      "30171    | torch.Size([1, 1000, 754]) | 12\n",
-      "30172    | torch.Size([1, 1000, 771]) | 12\n",
-      "30173    | torch.Size([1, 1000, 754]) | 12\n",
-      "30174    | torch.Size([1, 1000, 754]) | 12\n",
-      "30175    | torch.Size([1, 1000, 754]) | 12\n",
-      "30176    | torch.Size([1, 1000, 754]) | 12\n",
-      "30177    | torch.Size([1, 1000, 754]) | 12\n",
-      "30178    | torch.Size([1, 1000, 754]) | 12\n",
-      "30179    | torch.Size([1, 1000, 754]) | 12\n",
-      "30180    | torch.Size([1, 1000, 754]) | 12\n",
-      "30181    | torch.Size([1, 1000, 754]) | 12\n",
-      "30182    | torch.Size([1, 1000, 754]) | 12\n",
-      "30183    | torch.Size([1, 1000, 754]) | 12\n",
-      "30184    | torch.Size([1, 1000, 777]) | 12\n",
-      "30185    | torch.Size([1, 1000, 754]) | 12\n",
-      "30186    | torch.Size([1, 1000, 754]) | 12\n",
-      "30187    | torch.Size([1, 1000, 754]) | 12\n",
-      "30188    | torch.Size([1, 1000, 754]) | 12\n",
-      "30189    | torch.Size([1, 1000, 762]) | 12\n",
-      "30190    | torch.Size([1, 1000, 754]) | 12\n",
-      "30191    | torch.Size([1, 1000, 754]) | 12\n",
-      "30192    | torch.Size([1, 1000, 754]) | 12\n",
-      "30193    | torch.Size([1, 1000, 754]) | 12\n",
-      "30194    | torch.Size([1, 1000, 754]) | 12\n",
-      "30195    | torch.Size([1, 1000, 754]) | 12\n",
-      "30196    | torch.Size([1, 1000, 754]) | 12\n",
-      "30197    | torch.Size([1, 1000, 754]) | 12\n",
-      "30198    | torch.Size([1, 1000, 754]) | 12\n",
-      "30199    | torch.Size([1, 1000, 754]) | 12\n",
-      "30200    | torch.Size([1, 1000, 763]) | 12\n",
-      "30201    | torch.Size([1, 1000, 772]) | 12\n",
-      "30202    | torch.Size([1, 1000, 754]) | 12\n",
-      "30203    | torch.Size([1, 1000, 754]) | 12\n",
-      "30204    | torch.Size([1, 1000, 754]) | 12\n",
-      "30205    | torch.Size([1, 1000, 777]) | 12\n",
-      "30206    | torch.Size([1, 1000, 754]) | 12\n",
-      "30207    | torch.Size([1, 1000, 754]) | 12\n",
-      "30208    | torch.Size([1, 1000, 754]) | 12\n",
-      "30209    | torch.Size([1, 1000, 754]) | 12\n",
-      "30210    | torch.Size([1, 1000, 754]) | 12\n",
-      "30211    | torch.Size([1, 1000, 754]) | 12\n",
-      "30212    | torch.Size([1, 1000, 754]) | 12\n",
-      "30213    | torch.Size([1, 1000, 754]) | 12\n",
-      "30214    | torch.Size([1, 1000, 795]) | 12\n",
-      "30215    | torch.Size([1, 1000, 754]) | 12\n",
-      "30216    | torch.Size([1, 1000, 772]) | 12\n",
-      "30217    | torch.Size([1, 1000, 754]) | 12\n",
-      "30218    | torch.Size([1, 1000, 754]) | 12\n",
-      "30219    | torch.Size([1, 1000, 754]) | 12\n",
-      "30220    | torch.Size([1, 1000, 777]) | 12\n",
-      "30221    | torch.Size([1, 1000, 754]) | 12\n",
-      "30222    | torch.Size([1, 1000, 754]) | 12\n",
-      "30223    | torch.Size([1, 1000, 754]) | 12\n",
-      "30224    | torch.Size([1, 1000, 754]) | 12\n",
-      "30225    | torch.Size([1, 1000, 754]) | 12\n",
-      "30226    | torch.Size([1, 1000, 754]) | 12\n",
-      "30227    | torch.Size([1, 1000, 754]) | 12\n",
-      "30228    | torch.Size([1, 1000, 776]) | 12\n",
-      "30229    | torch.Size([1, 1000, 754]) | 12\n",
-      "30230    | torch.Size([1, 1000, 754]) | 12\n",
-      "30231    | torch.Size([1, 1000, 788]) | 12\n",
-      "30232    | torch.Size([1, 1000, 772]) | 12\n",
-      "30233    | torch.Size([1, 1000, 754]) | 12\n",
-      "30234    | torch.Size([1, 1000, 754]) | 12\n",
-      "30235    | torch.Size([1, 1000, 841]) | 12\n",
-      "30236    | torch.Size([1, 1000, 754]) | 12\n",
-      "30237    | torch.Size([1, 1000, 754]) | 12\n",
-      "30238    | torch.Size([1, 1000, 754]) | 12\n",
-      "30239    | torch.Size([1, 1000, 754]) | 12\n",
-      "30240    | torch.Size([1, 1000, 754]) | 12\n",
-      "30241    | torch.Size([1, 1000, 754]) | 12\n",
-      "30242    | torch.Size([1, 1000, 754]) | 12\n",
-      "30243    | torch.Size([1, 1000, 816]) | 12\n",
-      "30244    | torch.Size([1, 1000, 754]) | 12\n",
-      "30245    | torch.Size([1, 1000, 754]) | 12\n",
-      "30246    | torch.Size([1, 1000, 754]) | 12\n",
-      "30247    | torch.Size([1, 1000, 754]) | 12\n",
-      "30248    | torch.Size([1, 1000, 754]) | 12\n",
-      "30249    | torch.Size([1, 1000, 754]) | 12\n",
-      "30250    | torch.Size([1, 1000, 778]) | 12\n",
-      "30251    | torch.Size([1, 1000, 754]) | 12\n",
-      "30252    | torch.Size([1, 1000, 754]) | 12\n",
-      "30253    | torch.Size([1, 1000, 754]) | 12\n",
-      "30254    | torch.Size([1, 1000, 754]) | 12\n",
-      "30255    | torch.Size([1, 1000, 754]) | 12\n",
-      "30256    | torch.Size([1, 1000, 754]) | 12\n",
-      "30257    | torch.Size([1, 1000, 754]) | 12\n",
-      "30258    | torch.Size([1, 1000, 754]) | 12\n",
-      "30259    | torch.Size([1, 1000, 754]) | 12\n",
-      "30260    | torch.Size([1, 1000, 771]) | 12\n",
-      "30261    | torch.Size([1, 1000, 754]) | 12\n",
-      "30262    | torch.Size([1, 1000, 771]) | 12\n",
-      "30263    | torch.Size([1, 1000, 772]) | 12\n",
-      "30264    | torch.Size([1, 1000, 754]) | 12\n",
-      "30265    | torch.Size([1, 1000, 754]) | 12\n",
-      "30266    | torch.Size([1, 1000, 772]) | 12\n",
-      "30267    | torch.Size([1, 1000, 754]) | 12\n",
-      "30268    | torch.Size([1, 1000, 765]) | 12\n",
-      "30269    | torch.Size([1, 1000, 754]) | 12\n",
-      "30270    | torch.Size([1, 1000, 754]) | 12\n",
-      "30271    | torch.Size([1, 1000, 761]) | 12\n",
-      "30272    | torch.Size([1, 1000, 772]) | 12\n",
-      "30273    | torch.Size([1, 1000, 754]) | 12\n",
-      "30274    | torch.Size([1, 1000, 754]) | 12\n",
-      "30275    | torch.Size([1, 1000, 754]) | 12\n",
-      "30276    | torch.Size([1, 1000, 754]) | 12\n",
-      "30277    | torch.Size([1, 1000, 754]) | 12\n",
-      "30278    | torch.Size([1, 1000, 754]) | 12\n",
-      "30279    | torch.Size([1, 1000, 754]) | 12\n",
-      "30280    | torch.Size([1, 1000, 754]) | 12\n",
-      "30281    | torch.Size([1, 1000, 754]) | 12\n",
-      "30282    | torch.Size([1, 1000, 754]) | 12\n",
-      "30283    | torch.Size([1, 1000, 754]) | 12\n",
-      "30284    | torch.Size([1, 1000, 754]) | 12\n",
-      "30285    | torch.Size([1, 1000, 766]) | 12\n",
-      "30286    | torch.Size([1, 1000, 754]) | 12\n",
-      "30287    | torch.Size([1, 1000, 754]) | 12\n",
-      "30288    | torch.Size([1, 1000, 754]) | 12\n",
-      "30289    | torch.Size([1, 1000, 754]) | 12\n",
-      "30290    | torch.Size([1, 1000, 754]) | 12\n",
-      "30291    | torch.Size([1, 1000, 754]) | 12\n",
-      "30292    | torch.Size([1, 1000, 784]) | 12\n",
-      "30293    | torch.Size([1, 1000, 754]) | 12\n",
-      "30294    | torch.Size([1, 1000, 754]) | 12\n",
-      "30295    | torch.Size([1, 1000, 754]) | 12\n",
-      "30296    | torch.Size([1, 1000, 760]) | 12\n",
-      "30297    | torch.Size([1, 1000, 754]) | 12\n",
-      "30298    | torch.Size([1, 1000, 754]) | 12\n",
-      "30299    | torch.Size([1, 1000, 772]) | 12\n",
-      "30300    | torch.Size([1, 1000, 754]) | 12\n",
-      "30301    | torch.Size([1, 1000, 754]) | 12\n",
-      "30302    | torch.Size([1, 1000, 754]) | 12\n",
-      "30303    | torch.Size([1, 1000, 754]) | 12\n",
-      "30304    | torch.Size([1, 1000, 754]) | 12\n",
-      "30305    | torch.Size([1, 1000, 754]) | 12\n",
-      "30306    | torch.Size([1, 1000, 754]) | 12\n",
-      "30307    | torch.Size([1, 1000, 754]) | 12\n",
-      "30308    | torch.Size([1, 1000, 754]) | 12\n",
-      "30309    | torch.Size([1, 1000, 754]) | 12\n",
-      "30310    | torch.Size([1, 1000, 754]) | 12\n",
-      "30311    | torch.Size([1, 1000, 754]) | 12\n",
-      "30312    | torch.Size([1, 1000, 754]) | 12\n",
-      "30313    | torch.Size([1, 1000, 754]) | 12\n",
-      "30314    | torch.Size([1, 1000, 754]) | 12\n",
-      "30315    | torch.Size([1, 1000, 754]) | 12\n",
-      "30316    | torch.Size([1, 1000, 754]) | 12\n",
-      "30317    | torch.Size([1, 1000, 754]) | 12\n",
-      "30318    | torch.Size([1, 1000, 754]) | 12\n",
-      "30319    | torch.Size([1, 1000, 754]) | 12\n",
-      "30320    | torch.Size([1, 1000, 754]) | 12\n",
-      "30321    | torch.Size([1, 1000, 754]) | 12\n",
-      "30322    | torch.Size([1, 1000, 754]) | 12\n",
-      "30323    | torch.Size([1, 1000, 754]) | 12\n",
-      "30324    | torch.Size([1, 1000, 754]) | 12\n",
-      "30325    | torch.Size([1, 1000, 754]) | 12\n",
-      "30326    | torch.Size([1, 1000, 754]) | 12\n",
-      "30327    | torch.Size([1, 1000, 787]) | 12\n",
-      "30328    | torch.Size([1, 1000, 754]) | 12\n",
-      "30329    | torch.Size([1, 1000, 754]) | 12\n",
-      "30330    | torch.Size([1, 1000, 754]) | 12\n",
-      "30331    | torch.Size([1, 1000, 754]) | 12\n",
-      "30332    | torch.Size([1, 1000, 754]) | 12\n",
-      "30333    | torch.Size([1, 1000, 754]) | 12\n",
-      "30334    | torch.Size([1, 1000, 754]) | 12\n",
-      "30335    | torch.Size([1, 1000, 754]) | 12\n",
-      "30336    | torch.Size([1, 1000, 754]) | 12\n",
-      "30337    | torch.Size([1, 1000, 754]) | 12\n",
-      "30338    | torch.Size([1, 1000, 754]) | 12\n",
-      "30339    | torch.Size([1, 1000, 797]) | 12\n",
-      "30340    | torch.Size([1, 1000, 754]) | 12\n",
-      "30341    | torch.Size([1, 1000, 762]) | 12\n",
-      "30342    | torch.Size([1, 1000, 754]) | 12\n",
-      "30343    | torch.Size([1, 1000, 754]) | 12\n",
-      "30344    | torch.Size([1, 1000, 754]) | 12\n",
-      "30345    | torch.Size([1, 1000, 754]) | 12\n",
-      "30346    | torch.Size([1, 1000, 754]) | 12\n",
-      "30347    | torch.Size([1, 1000, 754]) | 12\n",
-      "30348    | torch.Size([1, 1000, 754]) | 12\n",
-      "30349    | torch.Size([1, 1000, 754]) | 12\n",
-      "30350    | torch.Size([1, 1000, 754]) | 12\n",
-      "30351    | torch.Size([1, 1000, 754]) | 12\n",
-      "30352    | torch.Size([1, 1000, 763]) | 12\n",
-      "30353    | torch.Size([1, 1000, 754]) | 12\n",
-      "30354    | torch.Size([1, 1000, 754]) | 12\n",
-      "30355    | torch.Size([1, 1000, 754]) | 12\n",
-      "30356    | torch.Size([1, 1000, 754]) | 12\n",
-      "30357    | torch.Size([1, 1000, 754]) | 12\n",
-      "30358    | torch.Size([1, 1000, 754]) | 12\n",
-      "30359    | torch.Size([1, 1000, 754]) | 12\n",
-      "30360    | torch.Size([1, 1000, 754]) | 12\n",
-      "30361    | torch.Size([1, 1000, 754]) | 12\n",
-      "30362    | torch.Size([1, 1000, 754]) | 12\n",
-      "30363    | torch.Size([1, 1000, 754]) | 12\n",
-      "30364    | torch.Size([1, 1000, 754]) | 12\n",
-      "30365    | torch.Size([1, 1000, 754]) | 12\n",
-      "30366    | torch.Size([1, 1000, 754]) | 12\n",
-      "30367    | torch.Size([1, 1000, 754]) | 12\n",
-      "30368    | torch.Size([1, 1000, 754]) | 12\n",
-      "30369    | torch.Size([1, 1000, 771]) | 12\n",
-      "30370    | torch.Size([1, 1000, 754]) | 12\n",
-      "30371    | torch.Size([1, 1000, 754]) | 12\n",
-      "30372    | torch.Size([1, 1000, 800]) | 12\n",
-      "30373    | torch.Size([1, 1000, 754]) | 12\n",
-      "30374    | torch.Size([1, 1000, 754]) | 12\n",
-      "30375    | torch.Size([1, 1000, 754]) | 12\n",
-      "30376    | torch.Size([1, 1000, 754]) | 12\n",
-      "30377    | torch.Size([1, 1000, 754]) | 12\n",
-      "30378    | torch.Size([1, 1000, 754]) | 12\n",
-      "30379    | torch.Size([1, 1000, 754]) | 12\n",
-      "30380    | torch.Size([1, 1000, 754]) | 12\n",
-      "30381    | torch.Size([1, 1000, 754]) | 12\n",
-      "30382    | torch.Size([1, 1000, 754]) | 12\n",
-      "30383    | torch.Size([1, 1000, 759]) | 12\n",
-      "30384    | torch.Size([1, 1000, 754]) | 12\n",
-      "30385    | torch.Size([1, 1000, 754]) | 12\n",
-      "30386    | torch.Size([1, 1000, 754]) | 12\n",
-      "30387    | torch.Size([1, 1000, 754]) | 12\n",
-      "30388    | torch.Size([1, 1000, 754]) | 12\n",
-      "30389    | torch.Size([1, 1000, 754]) | 12\n",
-      "30390    | torch.Size([1, 1000, 754]) | 12\n",
-      "30391    | torch.Size([1, 1000, 754]) | 12\n",
-      "30392    | torch.Size([1, 1000, 754]) | 12\n",
-      "30393    | torch.Size([1, 1000, 754]) | 12\n",
-      "30394    | torch.Size([1, 1000, 754]) | 12\n",
-      "30395    | torch.Size([1, 1000, 754]) | 12\n",
-      "30396    | torch.Size([1, 1000, 754]) | 12\n",
-      "30397    | torch.Size([1, 1000, 754]) | 12\n",
-      "30398    | torch.Size([1, 1000, 754]) | 12\n",
-      "30399    | torch.Size([1, 1000, 754]) | 12\n",
-      "30400    | torch.Size([1, 1000, 754]) | 12\n",
-      "30401    | torch.Size([1, 1000, 754]) | 12\n",
-      "30402    | torch.Size([1, 1000, 754]) | 12\n",
-      "30403    | torch.Size([1, 1000, 754]) | 12\n",
-      "30404    | torch.Size([1, 1000, 769]) | 12\n",
-      "30405    | torch.Size([1, 1000, 754]) | 12\n",
-      "30406    | torch.Size([1, 1000, 771]) | 12\n",
-      "30407    | torch.Size([1, 1000, 754]) | 12\n",
-      "30408    | torch.Size([1, 1000, 754]) | 12\n",
-      "30409    | torch.Size([1, 1000, 754]) | 12\n",
-      "30410    | torch.Size([1, 1000, 754]) | 12\n",
-      "30411    | torch.Size([1, 1000, 772]) | 12\n",
-      "30412    | torch.Size([1, 1000, 754]) | 12\n",
-      "30413    | torch.Size([1, 1000, 754]) | 12\n",
-      "30414    | torch.Size([1, 1000, 754]) | 12\n",
-      "30415    | torch.Size([1, 1000, 754]) | 12\n",
-      "30416    | torch.Size([1, 1000, 785]) | 12\n",
-      "30417    | torch.Size([1, 1000, 754]) | 12\n",
-      "30418    | torch.Size([1, 1000, 754]) | 12\n",
-      "30419    | torch.Size([1, 1000, 754]) | 12\n",
-      "30420    | torch.Size([1, 1000, 754]) | 12\n",
-      "30421    | torch.Size([1, 1000, 754]) | 12\n",
-      "30422    | torch.Size([1, 1000, 754]) | 12\n",
-      "30423    | torch.Size([1, 1000, 754]) | 12\n",
-      "30424    | torch.Size([1, 1000, 754]) | 12\n",
-      "30425    | torch.Size([1, 1000, 754]) | 12\n",
-      "30426    | torch.Size([1, 1000, 754]) | 12\n",
-      "30427    | torch.Size([1, 1000, 754]) | 12\n",
-      "30428    | torch.Size([1, 1000, 762]) | 12\n",
-      "30429    | torch.Size([1, 1000, 754]) | 12\n",
-      "30430    | torch.Size([1, 1000, 754]) | 12\n",
-      "30431    | torch.Size([1, 1000, 754]) | 12\n",
-      "30432    | torch.Size([1, 1000, 754]) | 12\n",
-      "30433    | torch.Size([1, 1000, 754]) | 12\n",
-      "30434    | torch.Size([1, 1000, 754]) | 12\n",
-      "30435    | torch.Size([1, 1000, 754]) | 12\n",
-      "30436    | torch.Size([1, 1000, 754]) | 12\n",
-      "30437    | torch.Size([1, 1000, 754]) | 12\n",
-      "30438    | torch.Size([1, 1000, 754]) | 12\n",
-      "30439    | torch.Size([1, 1000, 754]) | 12\n",
-      "30440    | torch.Size([1, 1000, 754]) | 12\n",
-      "30441    | torch.Size([1, 1000, 754]) | 12\n",
-      "30442    | torch.Size([1, 1000, 758]) | 12\n",
-      "30443    | torch.Size([1, 1000, 769]) | 12\n",
-      "30444    | torch.Size([1, 1000, 778]) | 12\n",
-      "30445    | torch.Size([1, 1000, 754]) | 12\n",
-      "30446    | torch.Size([1, 1000, 763]) | 12\n",
-      "30447    | torch.Size([1, 1000, 762]) | 12\n",
-      "30448    | torch.Size([1, 1000, 754]) | 12\n",
-      "30449    | torch.Size([1, 1000, 754]) | 12\n",
-      "30450    | torch.Size([1, 1000, 754]) | 12\n",
-      "30451    | torch.Size([1, 1000, 754]) | 12\n",
-      "30452    | torch.Size([1, 1000, 754]) | 12\n",
-      "30453    | torch.Size([1, 1000, 754]) | 12\n",
-      "30454    | torch.Size([1, 1000, 754]) | 12\n",
-      "30455    | torch.Size([1, 1000, 754]) | 12\n",
-      "30456    | torch.Size([1, 1000, 772]) | 12\n",
-      "30457    | torch.Size([1, 1000, 754]) | 12\n",
-      "30458    | torch.Size([1, 1000, 754]) | 12\n",
-      "30459    | torch.Size([1, 1000, 754]) | 12\n",
-      "30460    | torch.Size([1, 1000, 754]) | 12\n",
-      "30461    | torch.Size([1, 1000, 754]) | 12\n",
-      "30462    | torch.Size([1, 1000, 754]) | 12\n",
-      "30463    | torch.Size([1, 1000, 754]) | 12\n",
-      "30464    | torch.Size([1, 1000, 754]) | 12\n",
-      "30465    | torch.Size([1, 1000, 754]) | 12\n",
-      "30466    | torch.Size([1, 1000, 754]) | 12\n",
-      "30467    | torch.Size([1, 1000, 754]) | 12\n",
-      "30468    | torch.Size([1, 1000, 754]) | 12\n",
-      "30469    | torch.Size([1, 1000, 754]) | 12\n",
-      "30470    | torch.Size([1, 1000, 754]) | 12\n",
-      "30471    | torch.Size([1, 1000, 754]) | 12\n",
-      "30472    | torch.Size([1, 1000, 754]) | 12\n",
-      "30473    | torch.Size([1, 1000, 754]) | 12\n",
-      "30474    | torch.Size([1, 1000, 754]) | 12\n",
-      "30475    | torch.Size([1, 1000, 754]) | 12\n",
-      "30476    | torch.Size([1, 1000, 754]) | 12\n",
-      "30477    | torch.Size([1, 1000, 754]) | 12\n",
-      "30478    | torch.Size([1, 1000, 754]) | 12\n",
-      "30479    | torch.Size([1, 1000, 754]) | 12\n",
-      "30480    | torch.Size([1, 1000, 754]) | 12\n",
-      "30481    | torch.Size([1, 1000, 754]) | 12\n",
-      "30482    | torch.Size([1, 1000, 754]) | 12\n",
-      "30483    | torch.Size([1, 1000, 754]) | 12\n",
-      "30484    | torch.Size([1, 1000, 754]) | 12\n",
-      "30485    | torch.Size([1, 1000, 754]) | 12\n",
-      "30486    | torch.Size([1, 1000, 754]) | 12\n",
-      "30487    | torch.Size([1, 1000, 754]) | 12\n",
-      "30488    | torch.Size([1, 1000, 754]) | 12\n",
-      "30489    | torch.Size([1, 1000, 754]) | 12\n",
-      "30490    | torch.Size([1, 1000, 754]) | 12\n",
-      "30491    | torch.Size([1, 1000, 754]) | 12\n",
-      "30492    | torch.Size([1, 1000, 754]) | 12\n",
-      "30493    | torch.Size([1, 1000, 754]) | 12\n",
-      "30494    | torch.Size([1, 1000, 754]) | 12\n",
-      "30495    | torch.Size([1, 1000, 777]) | 12\n",
-      "30496    | torch.Size([1, 1000, 754]) | 12\n",
-      "30497    | torch.Size([1, 1000, 769]) | 12\n",
-      "30498    | torch.Size([1, 1000, 754]) | 12\n",
-      "30499    | torch.Size([1, 1000, 754]) | 12\n",
-      "30500    | torch.Size([1, 1000, 762]) | 12\n",
-      "30501    | torch.Size([1, 1000, 754]) | 12\n",
-      "30502    | torch.Size([1, 1000, 754]) | 12\n",
-      "30503    | torch.Size([1, 1000, 754]) | 12\n",
-      "30504    | torch.Size([1, 1000, 754]) | 12\n",
-      "30505    | torch.Size([1, 1000, 776]) | 12\n",
-      "30506    | torch.Size([1, 1000, 754]) | 12\n",
-      "30507    | torch.Size([1, 1000, 754]) | 12\n",
-      "30508    | torch.Size([1, 1000, 754]) | 12\n",
-      "30509    | torch.Size([1, 1000, 754]) | 12\n",
-      "30510    | torch.Size([1, 1000, 754]) | 12\n",
-      "30511    | torch.Size([1, 1000, 754]) | 12\n",
-      "30512    | torch.Size([1, 1000, 799]) | 12\n",
-      "30513    | torch.Size([1, 1000, 754]) | 12\n",
-      "30514    | torch.Size([1, 1000, 782]) | 12\n",
-      "30515    | torch.Size([1, 1000, 754]) | 12\n",
-      "30516    | torch.Size([1, 1000, 754]) | 12\n",
-      "30517    | torch.Size([1, 1000, 754]) | 12\n",
-      "30518    | torch.Size([1, 1000, 754]) | 12\n",
-      "30519    | torch.Size([1, 1000, 754]) | 12\n",
-      "30520    | torch.Size([1, 1000, 777]) | 12\n",
-      "30521    | torch.Size([1, 1000, 754]) | 12\n",
-      "30522    | torch.Size([1, 1000, 754]) | 12\n",
-      "30523    | torch.Size([1, 1000, 754]) | 12\n",
-      "30524    | torch.Size([1, 1000, 754]) | 12\n",
-      "30525    | torch.Size([1, 1000, 780]) | 12\n",
-      "30526    | torch.Size([1, 1000, 754]) | 12\n",
-      "30527    | torch.Size([1, 1000, 754]) | 12\n",
-      "30528    | torch.Size([1, 1000, 789]) | 12\n",
-      "30529    | torch.Size([1, 1000, 799]) | 12\n",
-      "30530    | torch.Size([1, 1000, 754]) | 12\n",
-      "30531    | torch.Size([1, 1000, 767]) | 12\n",
-      "30532    | torch.Size([1, 1000, 754]) | 12\n",
-      "30533    | torch.Size([1, 1000, 754]) | 12\n",
-      "30534    | torch.Size([1, 1000, 754]) | 12\n",
-      "30535    | torch.Size([1, 1000, 754]) | 12\n",
-      "30536    | torch.Size([1, 1000, 754]) | 12\n",
-      "30537    | torch.Size([1, 1000, 754]) | 12\n",
-      "30538    | torch.Size([1, 1000, 781]) | 12\n",
-      "30539    | torch.Size([1, 1000, 772]) | 12\n",
-      "30540    | torch.Size([1, 1000, 754]) | 12\n",
-      "30541    | torch.Size([1, 1000, 754]) | 12\n",
-      "30542    | torch.Size([1, 1000, 754]) | 12\n",
-      "30543    | torch.Size([1, 1000, 777]) | 12\n",
-      "30544    | torch.Size([1, 1000, 754]) | 12\n",
-      "30545    | torch.Size([1, 1000, 754]) | 12\n",
-      "30546    | torch.Size([1, 1000, 754]) | 12\n",
-      "30547    | torch.Size([1, 1000, 754]) | 12\n",
-      "30548    | torch.Size([1, 1000, 754]) | 12\n",
-      "30549    | torch.Size([1, 1000, 754]) | 12\n",
-      "30550    | torch.Size([1, 1000, 754]) | 12\n",
-      "30551    | torch.Size([1, 1000, 754]) | 12\n",
-      "30552    | torch.Size([1, 1000, 754]) | 12\n",
-      "30553    | torch.Size([1, 1000, 754]) | 12\n",
-      "30554    | torch.Size([1, 1000, 754]) | 12\n",
-      "30555    | torch.Size([1, 1000, 772]) | 12\n",
-      "30556    | torch.Size([1, 1000, 754]) | 12\n",
-      "30557    | torch.Size([1, 1000, 754]) | 12\n",
-      "30558    | torch.Size([1, 1000, 754]) | 12\n",
-      "30559    | torch.Size([1, 1000, 754]) | 12\n",
-      "30560    | torch.Size([1, 1000, 754]) | 12\n",
-      "30561    | torch.Size([1, 1000, 754]) | 12\n",
-      "30562    | torch.Size([1, 1000, 754]) | 12\n",
-      "30563    | torch.Size([1, 1000, 800]) | 12\n",
-      "30564    | torch.Size([1, 1000, 754]) | 12\n",
-      "30565    | torch.Size([1, 1000, 754]) | 12\n",
-      "30566    | torch.Size([1, 1000, 754]) | 12\n",
-      "30567    | torch.Size([1, 1000, 754]) | 12\n",
-      "30568    | torch.Size([1, 1000, 762]) | 12\n",
-      "30569    | torch.Size([1, 1000, 754]) | 12\n",
-      "30570    | torch.Size([1, 1000, 754]) | 12\n",
-      "30571    | torch.Size([1, 1000, 754]) | 12\n",
-      "30572    | torch.Size([1, 1000, 754]) | 12\n",
-      "30573    | torch.Size([1, 1000, 754]) | 12\n",
-      "30574    | torch.Size([1, 1000, 806]) | 12\n",
-      "30575    | torch.Size([1, 1000, 754]) | 12\n",
-      "30576    | torch.Size([1, 1000, 754]) | 12\n",
-      "30577    | torch.Size([1, 1000, 754]) | 12\n",
-      "30578    | torch.Size([1, 1000, 754]) | 12\n",
-      "30579    | torch.Size([1, 1000, 754]) | 12\n",
-      "30580    | torch.Size([1, 1000, 754]) | 12\n",
-      "30581    | torch.Size([1, 1000, 754]) | 12\n",
-      "30582    | torch.Size([1, 1000, 754]) | 12\n",
-      "30583    | torch.Size([1, 1000, 754]) | 12\n",
-      "30584    | torch.Size([1, 1000, 754]) | 12\n",
-      "30585    | torch.Size([1, 1000, 754]) | 12\n",
-      "30586    | torch.Size([1, 1000, 754]) | 12\n",
-      "30587    | torch.Size([1, 1000, 754]) | 12\n",
-      "30588    | torch.Size([1, 1000, 754]) | 12\n",
-      "30589    | torch.Size([1, 1000, 754]) | 12\n",
-      "30590    | torch.Size([1, 1000, 754]) | 12\n",
-      "30591    | torch.Size([1, 1000, 754]) | 12\n",
-      "30592    | torch.Size([1, 1000, 754]) | 12\n",
-      "30593    | torch.Size([1, 1000, 754]) | 12\n",
-      "30594    | torch.Size([1, 1000, 754]) | 12\n",
-      "30595    | torch.Size([1, 1000, 754]) | 12\n",
-      "30596    | torch.Size([1, 1000, 754]) | 12\n",
-      "30597    | torch.Size([1, 1000, 772]) | 12\n",
-      "30598    | torch.Size([1, 1000, 754]) | 12\n",
-      "30599    | torch.Size([1, 1000, 754]) | 12\n",
-      "30600    | torch.Size([1, 1000, 754]) | 12\n",
-      "30601    | torch.Size([1, 1000, 754]) | 12\n",
-      "30602    | torch.Size([1, 1000, 754]) | 12\n",
-      "30603    | torch.Size([1, 1000, 780]) | 12\n",
-      "30604    | torch.Size([1, 1000, 754]) | 12\n",
-      "30605    | torch.Size([1, 1000, 754]) | 12\n",
-      "30606    | torch.Size([1, 1000, 754]) | 12\n",
-      "30607    | torch.Size([1, 1000, 754]) | 12\n",
-      "30608    | torch.Size([1, 1000, 754]) | 12\n",
-      "30609    | torch.Size([1, 1000, 754]) | 12\n",
-      "30610    | torch.Size([1, 1000, 754]) | 12\n",
-      "30611    | torch.Size([1, 1000, 754]) | 12\n",
-      "30612    | torch.Size([1, 1000, 754]) | 12\n",
-      "30613    | torch.Size([1, 1000, 754]) | 12\n",
-      "30614    | torch.Size([1, 1000, 754]) | 12\n",
-      "30615    | torch.Size([1, 1000, 754]) | 12\n",
-      "30616    | torch.Size([1, 1000, 754]) | 12\n",
-      "30617    | torch.Size([1, 1000, 754]) | 12\n",
-      "30618    | torch.Size([1, 1000, 754]) | 12\n",
-      "30619    | torch.Size([1, 1000, 768]) | 12\n",
-      "30620    | torch.Size([1, 1000, 754]) | 12\n",
-      "30621    | torch.Size([1, 1000, 754]) | 12\n",
-      "30622    | torch.Size([1, 1000, 754]) | 12\n",
-      "30623    | torch.Size([1, 1000, 754]) | 12\n",
-      "30624    | torch.Size([1, 1000, 754]) | 12\n",
-      "30625    | torch.Size([1, 1000, 754]) | 12\n",
-      "30626    | torch.Size([1, 1000, 754]) | 12\n",
-      "30627    | torch.Size([1, 1000, 754]) | 12\n",
-      "30628    | torch.Size([1, 1000, 754]) | 12\n",
-      "30629    | torch.Size([1, 1000, 754]) | 12\n",
-      "30630    | torch.Size([1, 1000, 791]) | 12\n",
-      "30631    | torch.Size([1, 1000, 754]) | 12\n",
-      "30632    | torch.Size([1, 1000, 754]) | 12\n",
-      "30633    | torch.Size([1, 1000, 754]) | 12\n",
-      "30634    | torch.Size([1, 1000, 754]) | 12\n",
-      "30635    | torch.Size([1, 1000, 754]) | 12\n",
-      "30636    | torch.Size([1, 1000, 754]) | 12\n",
-      "30637    | torch.Size([1, 1000, 754]) | 12\n",
-      "30638    | torch.Size([1, 1000, 754]) | 12\n",
-      "30639    | torch.Size([1, 1000, 754]) | 12\n",
-      "30640    | torch.Size([1, 1000, 754]) | 12\n",
-      "30641    | torch.Size([1, 1000, 754]) | 12\n",
-      "30642    | torch.Size([1, 1000, 754]) | 12\n",
-      "30643    | torch.Size([1, 1000, 754]) | 12\n",
-      "30644    | torch.Size([1, 1000, 754]) | 12\n",
-      "30645    | torch.Size([1, 1000, 754]) | 12\n",
-      "30646    | torch.Size([1, 1000, 754]) | 12\n",
-      "30647    | torch.Size([1, 1000, 772]) | 12\n",
-      "30648    | torch.Size([1, 1000, 754]) | 12\n",
-      "30649    | torch.Size([1, 1000, 754]) | 12\n",
-      "30650    | torch.Size([1, 1000, 754]) | 12\n",
-      "30651    | torch.Size([1, 1000, 754]) | 12\n",
-      "30652    | torch.Size([1, 1000, 754]) | 12\n",
-      "30653    | torch.Size([1, 1000, 754]) | 12\n",
-      "30654    | torch.Size([1, 1000, 754]) | 12\n",
-      "30655    | torch.Size([1, 1000, 754]) | 12\n",
-      "30656    | torch.Size([1, 1000, 754]) | 12\n",
-      "30657    | torch.Size([1, 1000, 769]) | 12\n",
-      "30658    | torch.Size([1, 1000, 754]) | 12\n",
-      "30659    | torch.Size([1, 1000, 754]) | 12\n",
-      "30660    | torch.Size([1, 1000, 754]) | 12\n",
-      "30661    | torch.Size([1, 1000, 754]) | 12\n",
-      "30662    | torch.Size([1, 1000, 802]) | 12\n",
-      "30663    | torch.Size([1, 1000, 754]) | 12\n",
-      "30664    | torch.Size([1, 1000, 754]) | 12\n",
-      "30665    | torch.Size([1, 1000, 754]) | 12\n",
-      "30666    | torch.Size([1, 1000, 779]) | 12\n",
-      "30667    | torch.Size([1, 1000, 754]) | 12\n",
-      "30668    | torch.Size([1, 1000, 754]) | 12\n",
-      "30669    | torch.Size([1, 1000, 754]) | 12\n",
-      "30670    | torch.Size([1, 1000, 754]) | 12\n",
-      "30671    | torch.Size([1, 1000, 754]) | 12\n",
-      "30672    | torch.Size([1, 1000, 754]) | 12\n",
-      "30673    | torch.Size([1, 1000, 754]) | 12\n",
-      "30674    | torch.Size([1, 1000, 799]) | 12\n",
-      "30675    | torch.Size([1, 1000, 754]) | 12\n",
-      "30676    | torch.Size([1, 1000, 754]) | 12\n",
-      "30677    | torch.Size([1, 1000, 754]) | 12\n",
-      "30678    | torch.Size([1, 1000, 754]) | 12\n",
-      "30679    | torch.Size([1, 1000, 769]) | 12\n",
-      "30680    | torch.Size([1, 1000, 754]) | 12\n",
-      "30681    | torch.Size([1, 1000, 754]) | 12\n",
-      "30682    | torch.Size([1, 1000, 754]) | 12\n",
-      "30683    | torch.Size([1, 1000, 754]) | 12\n",
-      "30684    | torch.Size([1, 1000, 777]) | 12\n",
-      "30685    | torch.Size([1, 1000, 754]) | 12\n",
-      "30686    | torch.Size([1, 1000, 754]) | 12\n",
-      "30687    | torch.Size([1, 1000, 754]) | 12\n",
-      "30688    | torch.Size([1, 1000, 749]) | 12\n",
-      "30689    | torch.Size([1, 1000, 764]) | 12\n",
-      "30690    | torch.Size([1, 1000, 754]) | 12\n",
-      "30691    | torch.Size([1, 1000, 754]) | 12\n",
-      "30692    | torch.Size([1, 1000, 754]) | 12\n",
-      "30693    | torch.Size([1, 1000, 754]) | 12\n",
-      "30694    | torch.Size([1, 1000, 772]) | 12\n",
-      "30695    | torch.Size([1, 1000, 754]) | 12\n",
-      "30696    | torch.Size([1, 1000, 754]) | 12\n",
-      "30697    | torch.Size([1, 1000, 777]) | 12\n",
-      "30698    | torch.Size([1, 1000, 754]) | 12\n",
-      "30699    | torch.Size([1, 1000, 754]) | 12\n",
-      "30700    | torch.Size([1, 1000, 754]) | 12\n",
-      "30701    | torch.Size([1, 1000, 754]) | 12\n",
-      "30702    | torch.Size([1, 1000, 777]) | 12\n",
-      "30703    | torch.Size([1, 1000, 754]) | 12\n",
-      "30704    | torch.Size([1, 1000, 754]) | 12\n",
-      "30705    | torch.Size([1, 1000, 754]) | 12\n",
-      "30706    | torch.Size([1, 1000, 754]) | 12\n",
-      "30707    | torch.Size([1, 1000, 754]) | 12\n",
-      "30708    | torch.Size([1, 1000, 754]) | 12\n",
-      "30709    | torch.Size([1, 1000, 777]) | 12\n",
-      "30710    | torch.Size([1, 1000, 754]) | 12\n",
-      "30711    | torch.Size([1, 1000, 754]) | 12\n",
-      "30712    | torch.Size([1, 1000, 754]) | 12\n",
-      "30713    | torch.Size([1, 1000, 754]) | 12\n",
-      "30714    | torch.Size([1, 1000, 754]) | 12\n",
-      "30715    | torch.Size([1, 1000, 754]) | 12\n",
-      "30716    | torch.Size([1, 1000, 806]) | 12\n",
-      "30717    | torch.Size([1, 1000, 774]) | 12\n",
-      "30718    | torch.Size([1, 1000, 754]) | 12\n",
-      "30719    | torch.Size([1, 1000, 762]) | 12\n",
-      "30720    | torch.Size([1, 1000, 754]) | 12\n",
-      "30721    | torch.Size([1, 1000, 754]) | 12\n",
-      "30722    | torch.Size([1, 1000, 754]) | 12\n",
-      "30723    | torch.Size([1, 1000, 754]) | 12\n",
-      "30724    | torch.Size([1, 1000, 754]) | 12\n",
-      "30725    | torch.Size([1, 1000, 754]) | 12\n",
-      "30726    | torch.Size([1, 1000, 762]) | 12\n",
-      "30727    | torch.Size([1, 1000, 785]) | 12\n",
-      "30728    | torch.Size([1, 1000, 754]) | 12\n",
-      "30729    | torch.Size([1, 1000, 754]) | 12\n",
-      "30730    | torch.Size([1, 1000, 754]) | 12\n",
-      "30731    | torch.Size([1, 1000, 754]) | 12\n",
-      "30732    | torch.Size([1, 1000, 754]) | 12\n",
-      "30733    | torch.Size([1, 1000, 754]) | 12\n",
-      "30734    | torch.Size([1, 1000, 776]) | 12\n",
-      "30735    | torch.Size([1, 1000, 754]) | 12\n",
-      "30736    | torch.Size([1, 1000, 772]) | 12\n",
-      "30737    | torch.Size([1, 1000, 754]) | 12\n",
-      "30738    | torch.Size([1, 1000, 754]) | 12\n",
-      "30739    | torch.Size([1, 1000, 754]) | 12\n",
-      "30740    | torch.Size([1, 1000, 777]) | 12\n",
-      "30741    | torch.Size([1, 1000, 754]) | 12\n",
-      "30742    | torch.Size([1, 1000, 754]) | 12\n",
-      "30743    | torch.Size([1, 1000, 754]) | 12\n",
-      "30744    | torch.Size([1, 1000, 754]) | 12\n",
-      "30745    | torch.Size([1, 1000, 754]) | 12\n",
-      "30746    | torch.Size([1, 1000, 754]) | 12\n",
-      "30747    | torch.Size([1, 1000, 754]) | 12\n",
-      "30748    | torch.Size([1, 1000, 754]) | 12\n",
-      "30749    | torch.Size([1, 1000, 754]) | 12\n",
-      "30750    | torch.Size([1, 1000, 754]) | 12\n",
-      "30751    | torch.Size([1, 1000, 754]) | 12\n",
-      "30752    | torch.Size([1, 1000, 754]) | 12\n",
-      "30753    | torch.Size([1, 1000, 754]) | 12\n",
-      "30754    | torch.Size([1, 1000, 754]) | 12\n",
-      "30755    | torch.Size([1, 1000, 754]) | 12\n",
-      "30756    | torch.Size([1, 1000, 754]) | 12\n",
-      "30757    | torch.Size([1, 1000, 754]) | 12\n",
-      "30758    | torch.Size([1, 1000, 754]) | 12\n",
-      "30759    | torch.Size([1, 1000, 754]) | 12\n",
-      "30760    | torch.Size([1, 1000, 754]) | 12\n",
-      "30761    | torch.Size([1, 1000, 754]) | 12\n",
-      "30762    | torch.Size([1, 1000, 754]) | 12\n",
-      "30763    | torch.Size([1, 1000, 754]) | 12\n",
-      "30764    | torch.Size([1, 1000, 754]) | 12\n",
-      "30765    | torch.Size([1, 1000, 754]) | 12\n",
-      "30766    | torch.Size([1, 1000, 754]) | 12\n",
-      "30767    | torch.Size([1, 1000, 754]) | 12\n",
-      "30768    | torch.Size([1, 1000, 754]) | 12\n",
-      "30769    | torch.Size([1, 1000, 754]) | 12\n",
-      "30770    | torch.Size([1, 1000, 754]) | 12\n",
-      "30771    | torch.Size([1, 1000, 785]) | 12\n",
-      "30772    | torch.Size([1, 1000, 754]) | 12\n",
-      "30773    | torch.Size([1, 1000, 772]) | 12\n",
-      "30774    | torch.Size([1, 1000, 754]) | 12\n",
-      "30775    | torch.Size([1, 1000, 754]) | 12\n",
-      "30776    | torch.Size([1, 1000, 754]) | 12\n",
-      "30777    | torch.Size([1, 1000, 754]) | 12\n",
-      "30778    | torch.Size([1, 1000, 807]) | 12\n",
-      "30779    | torch.Size([1, 1000, 754]) | 12\n",
-      "30780    | torch.Size([1, 1000, 754]) | 12\n",
-      "30781    | torch.Size([1, 1000, 754]) | 12\n",
-      "30782    | torch.Size([1, 1000, 754]) | 12\n",
-      "30783    | torch.Size([1, 1000, 780]) | 12\n",
-      "30784    | torch.Size([1, 1000, 754]) | 12\n",
-      "30785    | torch.Size([1, 1000, 754]) | 12\n",
-      "30786    | torch.Size([1, 1000, 762]) | 12\n",
-      "30787    | torch.Size([1, 1000, 752]) | 12\n",
-      "30788    | torch.Size([1, 1000, 754]) | 12\n",
-      "30789    | torch.Size([1, 1000, 754]) | 12\n",
-      "30790    | torch.Size([1, 1000, 754]) | 12\n",
-      "30791    | torch.Size([1, 1000, 771]) | 12\n",
-      "30792    | torch.Size([1, 1000, 754]) | 12\n",
-      "30793    | torch.Size([1, 1000, 762]) | 12\n",
-      "30794    | torch.Size([1, 1000, 754]) | 12\n",
-      "30795    | torch.Size([1, 1000, 754]) | 12\n",
-      "30796    | torch.Size([1, 1000, 754]) | 12\n",
-      "30797    | torch.Size([1, 1000, 754]) | 12\n",
-      "30798    | torch.Size([1, 1000, 754]) | 12\n",
-      "30799    | torch.Size([1, 1000, 754]) | 12\n",
-      "30800    | torch.Size([1, 1000, 785]) | 12\n",
-      "30801    | torch.Size([1, 1000, 754]) | 12\n",
-      "30802    | torch.Size([1, 1000, 763]) | 12\n",
-      "30803    | torch.Size([1, 1000, 754]) | 12\n",
-      "30804    | torch.Size([1, 1000, 754]) | 12\n",
-      "30805    | torch.Size([1, 1000, 754]) | 12\n",
-      "30806    | torch.Size([1, 1000, 754]) | 12\n",
-      "30807    | torch.Size([1, 1000, 762]) | 12\n",
-      "30808    | torch.Size([1, 1000, 754]) | 12\n",
-      "30809    | torch.Size([1, 1000, 754]) | 12\n",
-      "30810    | torch.Size([1, 1000, 754]) | 12\n",
-      "30811    | torch.Size([1, 1000, 754]) | 12\n",
-      "30812    | torch.Size([1, 1000, 754]) | 12\n",
-      "30813    | torch.Size([1, 1000, 754]) | 12\n",
-      "30814    | torch.Size([1, 1000, 754]) | 12\n",
-      "30815    | torch.Size([1, 1000, 754]) | 12\n",
-      "30816    | torch.Size([1, 1000, 754]) | 12\n",
-      "30817    | torch.Size([1, 1000, 754]) | 12\n",
-      "30818    | torch.Size([1, 1000, 767]) | 12\n",
-      "30819    | torch.Size([1, 1000, 754]) | 12\n",
-      "30820    | torch.Size([1, 1000, 745]) | 12\n",
-      "30821    | torch.Size([1, 1000, 754]) | 12\n",
-      "30822    | torch.Size([1, 1000, 754]) | 12\n",
-      "30823    | torch.Size([1, 1000, 754]) | 12\n",
-      "30824    | torch.Size([1, 1000, 754]) | 12\n",
-      "30825    | torch.Size([1, 1000, 754]) | 12\n",
-      "30826    | torch.Size([1, 1000, 754]) | 12\n",
-      "30827    | torch.Size([1, 1000, 754]) | 12\n",
-      "30828    | torch.Size([1, 1000, 754]) | 12\n",
-      "30829    | torch.Size([1, 1000, 754]) | 12\n",
-      "30830    | torch.Size([1, 1000, 754]) | 12\n",
-      "30831    | torch.Size([1, 1000, 754]) | 12\n",
-      "30832    | torch.Size([1, 1000, 754]) | 12\n",
-      "30833    | torch.Size([1, 1000, 754]) | 12\n",
-      "30834    | torch.Size([1, 1000, 754]) | 12\n",
-      "30835    | torch.Size([1, 1000, 754]) | 12\n",
-      "30836    | torch.Size([1, 1000, 754]) | 12\n",
-      "30837    | torch.Size([1, 1000, 754]) | 12\n",
-      "30838    | torch.Size([1, 1000, 754]) | 12\n",
-      "30839    | torch.Size([1, 1000, 754]) | 12\n",
-      "30840    | torch.Size([1, 1000, 754]) | 12\n",
-      "30841    | torch.Size([1, 1000, 754]) | 12\n",
-      "30842    | torch.Size([1, 1000, 754]) | 12\n",
-      "30843    | torch.Size([1, 1000, 754]) | 12\n",
-      "30844    | torch.Size([1, 1000, 754]) | 12\n",
-      "30845    | torch.Size([1, 1000, 754]) | 12\n",
-      "30846    | torch.Size([1, 1000, 754]) | 12\n",
-      "30847    | torch.Size([1, 1000, 762]) | 12\n",
-      "30848    | torch.Size([1, 1000, 754]) | 12\n",
-      "30849    | torch.Size([1, 1000, 754]) | 12\n",
-      "30850    | torch.Size([1, 1000, 754]) | 12\n",
-      "30851    | torch.Size([1, 1000, 754]) | 12\n",
-      "30852    | torch.Size([1, 1000, 754]) | 12\n",
-      "30853    | torch.Size([1, 1000, 754]) | 12\n",
-      "30854    | torch.Size([1, 1000, 754]) | 12\n",
-      "30855    | torch.Size([1, 1000, 754]) | 12\n",
-      "30856    | torch.Size([1, 1000, 754]) | 12\n",
-      "30857    | torch.Size([1, 1000, 754]) | 12\n",
-      "30858    | torch.Size([1, 1000, 754]) | 12\n",
-      "30859    | torch.Size([1, 1000, 754]) | 12\n",
-      "30860    | torch.Size([1, 1000, 754]) | 12\n",
-      "30861    | torch.Size([1, 1000, 754]) | 12\n",
-      "30862    | torch.Size([1, 1000, 783]) | 12\n",
-      "30863    | torch.Size([1, 1000, 754]) | 12\n",
-      "30864    | torch.Size([1, 1000, 754]) | 12\n",
-      "30865    | torch.Size([1, 1000, 780]) | 12\n",
-      "30866    | torch.Size([1, 1000, 754]) | 12\n",
-      "30867    | torch.Size([1, 1000, 754]) | 12\n",
-      "30868    | torch.Size([1, 1000, 754]) | 12\n",
-      "30869    | torch.Size([1, 1000, 777]) | 12\n",
-      "30870    | torch.Size([1, 1000, 754]) | 12\n",
-      "30871    | torch.Size([1, 1000, 754]) | 12\n",
-      "30872    | torch.Size([1, 1000, 787]) | 12\n",
-      "30873    | torch.Size([1, 1000, 754]) | 12\n",
-      "30874    | torch.Size([1, 1000, 754]) | 12\n",
-      "30875    | torch.Size([1, 1000, 754]) | 12\n",
-      "30876    | torch.Size([1, 1000, 754]) | 12\n",
-      "30877    | torch.Size([1, 1000, 754]) | 12\n",
-      "30878    | torch.Size([1, 1000, 754]) | 12\n",
-      "30879    | torch.Size([1, 1000, 754]) | 12\n",
-      "30880    | torch.Size([1, 1000, 782]) | 12\n",
-      "30881    | torch.Size([1, 1000, 754]) | 12\n",
-      "30882    | torch.Size([1, 1000, 754]) | 12\n",
-      "30883    | torch.Size([1, 1000, 806]) | 12\n",
-      "30884    | torch.Size([1, 1000, 772]) | 12\n",
-      "30885    | torch.Size([1, 1000, 754]) | 12\n",
-      "30886    | torch.Size([1, 1000, 754]) | 12\n",
-      "30887    | torch.Size([1, 1000, 754]) | 12\n",
-      "30888    | torch.Size([1, 1000, 754]) | 12\n",
-      "30889    | torch.Size([1, 1000, 754]) | 12\n",
-      "30890    | torch.Size([1, 1000, 754]) | 12\n",
-      "30891    | torch.Size([1, 1000, 754]) | 12\n",
-      "30892    | torch.Size([1, 1000, 754]) | 12\n",
-      "30893    | torch.Size([1, 1000, 754]) | 12\n",
-      "30894    | torch.Size([1, 1000, 800]) | 12\n",
-      "30895    | torch.Size([1, 1000, 754]) | 12\n",
-      "30896    | torch.Size([1, 1000, 754]) | 12\n",
-      "30897    | torch.Size([1, 1000, 754]) | 12\n",
-      "30898    | torch.Size([1, 1000, 754]) | 12\n",
-      "30899    | torch.Size([1, 1000, 754]) | 12\n",
-      "30900    | torch.Size([1, 1000, 754]) | 12\n",
-      "30901    | torch.Size([1, 1000, 754]) | 12\n",
-      "30902    | torch.Size([1, 1000, 754]) | 12\n",
-      "30903    | torch.Size([1, 1000, 754]) | 12\n",
-      "30904    | torch.Size([1, 1000, 754]) | 12\n",
-      "30905    | torch.Size([1, 1000, 772]) | 12\n",
-      "30906    | torch.Size([1, 1000, 754]) | 12\n",
-      "30907    | torch.Size([1, 1000, 754]) | 12\n",
-      "30908    | torch.Size([1, 1000, 754]) | 12\n",
-      "30909    | torch.Size([1, 1000, 754]) | 12\n",
-      "30910    | torch.Size([1, 1000, 754]) | 12\n",
-      "30911    | torch.Size([1, 1000, 788]) | 12\n",
-      "30912    | torch.Size([1, 1000, 754]) | 12\n",
-      "30913    | torch.Size([1, 1000, 754]) | 12\n",
-      "30914    | torch.Size([1, 1000, 754]) | 12\n",
-      "30915    | torch.Size([1, 1000, 754]) | 12\n",
-      "30916    | torch.Size([1, 1000, 754]) | 12\n",
-      "30917    | torch.Size([1, 1000, 754]) | 12\n",
-      "30918    | torch.Size([1, 1000, 754]) | 12\n",
-      "30919    | torch.Size([1, 1000, 754]) | 12\n",
-      "30920    | torch.Size([1, 1000, 754]) | 12\n",
-      "30921    | torch.Size([1, 1000, 754]) | 12\n",
-      "30922    | torch.Size([1, 1000, 754]) | 12\n",
-      "30923    | torch.Size([1, 1000, 754]) | 12\n",
-      "30924    | torch.Size([1, 1000, 754]) | 12\n",
-      "30925    | torch.Size([1, 1000, 754]) | 12\n",
-      "30926    | torch.Size([1, 1000, 754]) | 12\n",
-      "30927    | torch.Size([1, 1000, 754]) | 12\n",
-      "30928    | torch.Size([1, 1000, 754]) | 12\n",
-      "30929    | torch.Size([1, 1000, 754]) | 12\n",
-      "30930    | torch.Size([1, 1000, 754]) | 12\n",
-      "30931    | torch.Size([1, 1000, 754]) | 12\n",
-      "30932    | torch.Size([1, 1000, 754]) | 12\n",
-      "30933    | torch.Size([1, 1000, 754]) | 12\n",
-      "30934    | torch.Size([1, 1000, 754]) | 12\n",
-      "30935    | torch.Size([1, 1000, 762]) | 12\n",
-      "30936    | torch.Size([1, 1000, 754]) | 12\n",
-      "30937    | torch.Size([1, 1000, 754]) | 12\n",
-      "30938    | torch.Size([1, 1000, 754]) | 12\n",
-      "30939    | torch.Size([1, 1000, 754]) | 12\n",
-      "30940    | torch.Size([1, 1000, 754]) | 12\n",
-      "30941    | torch.Size([1, 1000, 754]) | 12\n",
-      "30942    | torch.Size([1, 1000, 754]) | 12\n",
-      "30943    | torch.Size([1, 1000, 754]) | 12\n",
-      "30944    | torch.Size([1, 1000, 754]) | 12\n",
-      "30945    | torch.Size([1, 1000, 801]) | 12\n",
-      "30946    | torch.Size([1, 1000, 754]) | 12\n",
-      "30947    | torch.Size([1, 1000, 754]) | 12\n",
-      "30948    | torch.Size([1, 1000, 754]) | 12\n",
-      "30949    | torch.Size([1, 1000, 754]) | 12\n",
-      "30950    | torch.Size([1, 1000, 754]) | 12\n",
-      "30951    | torch.Size([1, 1000, 754]) | 12\n",
-      "30952    | torch.Size([1, 1000, 760]) | 12\n",
-      "30953    | torch.Size([1, 1000, 754]) | 12\n",
-      "30954    | torch.Size([1, 1000, 754]) | 12\n",
-      "30955    | torch.Size([1, 1000, 754]) | 12\n",
-      "30956    | torch.Size([1, 1000, 754]) | 12\n",
-      "30957    | torch.Size([1, 1000, 754]) | 12\n",
-      "30958    | torch.Size([1, 1000, 754]) | 12\n",
-      "30959    | torch.Size([1, 1000, 754]) | 12\n",
-      "30960    | torch.Size([1, 1000, 754]) | 12\n",
-      "30961    | torch.Size([1, 1000, 754]) | 12\n",
-      "30962    | torch.Size([1, 1000, 799]) | 12\n",
-      "30963    | torch.Size([1, 1000, 754]) | 12\n",
-      "30964    | torch.Size([1, 1000, 754]) | 12\n",
-      "30965    | torch.Size([1, 1000, 754]) | 12\n",
-      "30966    | torch.Size([1, 1000, 754]) | 12\n",
-      "30967    | torch.Size([1, 1000, 754]) | 12\n",
-      "30968    | torch.Size([1, 1000, 754]) | 12\n",
-      "30969    | torch.Size([1, 1000, 754]) | 12\n",
-      "30970    | torch.Size([1, 1000, 786]) | 12\n",
-      "30971    | torch.Size([1, 1000, 754]) | 12\n",
-      "30972    | torch.Size([1, 1000, 775]) | 12\n",
-      "30973    | torch.Size([1, 1000, 754]) | 12\n",
-      "30974    | torch.Size([1, 1000, 754]) | 12\n",
-      "30975    | torch.Size([1, 1000, 754]) | 12\n",
-      "30976    | torch.Size([1, 1000, 754]) | 12\n",
-      "30977    | torch.Size([1, 1000, 754]) | 12\n",
-      "30978    | torch.Size([1, 1000, 754]) | 12\n",
-      "30979    | torch.Size([1, 1000, 754]) | 12\n",
-      "30980    | torch.Size([1, 1000, 754]) | 12\n",
-      "30981    | torch.Size([1, 1000, 754]) | 12\n",
-      "30982    | torch.Size([1, 1000, 754]) | 12\n",
-      "30983    | torch.Size([1, 1000, 754]) | 12\n",
-      "30984    | torch.Size([1, 1000, 754]) | 12\n",
-      "30985    | torch.Size([1, 1000, 754]) | 12\n",
-      "30986    | torch.Size([1, 1000, 754]) | 12\n",
-      "30987    | torch.Size([1, 1000, 754]) | 12\n",
-      "30988    | torch.Size([1, 1000, 754]) | 12\n",
-      "30989    | torch.Size([1, 1000, 754]) | 12\n",
-      "30990    | torch.Size([1, 1000, 754]) | 12\n",
-      "30991    | torch.Size([1, 1000, 754]) | 12\n",
-      "30992    | torch.Size([1, 1000, 754]) | 12\n",
-      "30993    | torch.Size([1, 1000, 754]) | 12\n",
-      "30994    | torch.Size([1, 1000, 754]) | 12\n",
-      "30995    | torch.Size([1, 1000, 754]) | 12\n",
-      "30996    | torch.Size([1, 1000, 754]) | 12\n",
-      "30997    | torch.Size([1, 1000, 754]) | 12\n",
-      "30998    | torch.Size([1, 1000, 772]) | 12\n",
-      "30999    | torch.Size([1, 1000, 754]) | 12\n",
-      "31000    | torch.Size([1, 1000, 754]) | 12\n",
-      "31001    | torch.Size([1, 1000, 754]) | 12\n",
-      "31002    | torch.Size([1, 1000, 795]) | 12\n",
-      "31003    | torch.Size([1, 1000, 754]) | 12\n",
-      "31004    | torch.Size([1, 1000, 754]) | 12\n",
-      "31005    | torch.Size([1, 1000, 754]) | 12\n",
-      "31006    | torch.Size([1, 1000, 754]) | 12\n",
-      "31007    | torch.Size([1, 1000, 754]) | 12\n",
-      "31008    | torch.Size([1, 1000, 754]) | 12\n",
-      "31009    | torch.Size([1, 1000, 754]) | 12\n",
-      "31010    | torch.Size([1, 1000, 777]) | 12\n",
-      "31011    | torch.Size([1, 1000, 766]) | 12\n",
-      "31012    | torch.Size([1, 1000, 772]) | 12\n",
-      "31013    | torch.Size([1, 1000, 754]) | 12\n",
-      "31014    | torch.Size([1, 1000, 754]) | 12\n",
-      "31015    | torch.Size([1, 1000, 754]) | 12\n",
-      "31016    | torch.Size([1, 1000, 754]) | 12\n",
-      "31017    | torch.Size([1, 1000, 754]) | 12\n",
-      "31018    | torch.Size([1, 1000, 754]) | 12\n",
-      "31019    | torch.Size([1, 1000, 754]) | 12\n",
-      "31020    | torch.Size([1, 1000, 777]) | 12\n",
-      "31021    | torch.Size([1, 1000, 754]) | 12\n",
-      "31022    | torch.Size([1, 1000, 772]) | 12\n",
-      "31023    | torch.Size([1, 1000, 754]) | 12\n",
-      "31024    | torch.Size([1, 1000, 754]) | 12\n",
-      "31025    | torch.Size([1, 1000, 754]) | 12\n",
-      "31026    | torch.Size([1, 1000, 754]) | 12\n",
-      "31027    | torch.Size([1, 1000, 754]) | 12\n",
-      "31028    | torch.Size([1, 1000, 800]) | 12\n",
-      "31029    | torch.Size([1, 1000, 772]) | 12\n",
-      "31030    | torch.Size([1, 1000, 754]) | 12\n",
-      "31031    | torch.Size([1, 1000, 754]) | 12\n",
-      "31032    | torch.Size([1, 1000, 754]) | 12\n",
-      "31033    | torch.Size([1, 1000, 754]) | 12\n",
-      "31034    | torch.Size([1, 1000, 754]) | 12\n",
-      "31035    | torch.Size([1, 1000, 754]) | 12\n",
-      "31036    | torch.Size([1, 1000, 754]) | 12\n",
-      "31037    | torch.Size([1, 1000, 754]) | 12\n",
-      "31038    | torch.Size([1, 1000, 754]) | 12\n",
-      "31039    | torch.Size([1, 1000, 777]) | 12\n",
-      "31040    | torch.Size([1, 1000, 754]) | 12\n",
-      "31041    | torch.Size([1, 1000, 754]) | 12\n",
-      "31042    | torch.Size([1, 1000, 777]) | 12\n",
-      "31043    | torch.Size([1, 1000, 754]) | 12\n",
-      "31044    | torch.Size([1, 1000, 754]) | 12\n",
-      "31045    | torch.Size([1, 1000, 754]) | 12\n",
-      "31046    | torch.Size([1, 1000, 754]) | 12\n",
-      "31047    | torch.Size([1, 1000, 754]) | 12\n",
-      "31048    | torch.Size([1, 1000, 754]) | 12\n",
-      "31049    | torch.Size([1, 1000, 754]) | 12\n",
-      "31050    | torch.Size([1, 1000, 754]) | 12\n",
-      "31051    | torch.Size([1, 1000, 754]) | 12\n",
-      "31052    | torch.Size([1, 1000, 754]) | 12\n",
-      "31053    | torch.Size([1, 1000, 754]) | 12\n",
-      "31054    | torch.Size([1, 1000, 762]) | 12\n",
-      "31055    | torch.Size([1, 1000, 754]) | 12\n",
-      "31056    | torch.Size([1, 1000, 754]) | 12\n",
-      "31057    | torch.Size([1, 1000, 754]) | 12\n",
-      "31058    | torch.Size([1, 1000, 771]) | 12\n",
-      "31059    | torch.Size([1, 1000, 785]) | 12\n",
-      "31060    | torch.Size([1, 1000, 754]) | 12\n",
-      "31061    | torch.Size([1, 1000, 754]) | 12\n",
-      "31062    | torch.Size([1, 1000, 754]) | 12\n",
-      "31063    | torch.Size([1, 1000, 754]) | 12\n",
-      "31064    | torch.Size([1, 1000, 754]) | 12\n",
-      "31065    | torch.Size([1, 1000, 754]) | 12\n",
-      "31066    | torch.Size([1, 1000, 754]) | 12\n",
-      "31067    | torch.Size([1, 1000, 754]) | 12\n",
-      "31068    | torch.Size([1, 1000, 754]) | 12\n",
-      "31069    | torch.Size([1, 1000, 754]) | 12\n",
-      "31070    | torch.Size([1, 1000, 754]) | 12\n",
-      "31071    | torch.Size([1, 1000, 754]) | 12\n",
-      "31072    | torch.Size([1, 1000, 754]) | 12\n",
-      "31073    | torch.Size([1, 1000, 754]) | 12\n",
-      "31074    | torch.Size([1, 1000, 754]) | 12\n",
-      "31075    | torch.Size([1, 1000, 754]) | 12\n",
-      "31076    | torch.Size([1, 1000, 754]) | 12\n",
-      "31077    | torch.Size([1, 1000, 754]) | 12\n",
-      "31078    | torch.Size([1, 1000, 775]) | 12\n",
-      "31079    | torch.Size([1, 1000, 754]) | 12\n",
-      "31080    | torch.Size([1, 1000, 776]) | 12\n",
-      "31081    | torch.Size([1, 1000, 754]) | 12\n",
-      "31082    | torch.Size([1, 1000, 754]) | 12\n",
-      "31083    | torch.Size([1, 1000, 754]) | 12\n",
-      "31084    | torch.Size([1, 1000, 754]) | 12\n",
-      "31085    | torch.Size([1, 1000, 754]) | 12\n",
-      "31086    | torch.Size([1, 1000, 754]) | 12\n",
-      "31087    | torch.Size([1, 1000, 754]) | 12\n",
-      "31088    | torch.Size([1, 1000, 754]) | 12\n",
-      "31089    | torch.Size([1, 1000, 754]) | 12\n",
-      "31090    | torch.Size([1, 1000, 754]) | 12\n",
-      "31091    | torch.Size([1, 1000, 771]) | 12\n",
-      "31092    | torch.Size([1, 1000, 754]) | 12\n",
-      "31093    | torch.Size([1, 1000, 754]) | 12\n",
-      "31094    | torch.Size([1, 1000, 754]) | 12\n",
-      "31095    | torch.Size([1, 1000, 754]) | 12\n",
-      "31096    | torch.Size([1, 1000, 754]) | 12\n",
-      "31097    | torch.Size([1, 1000, 754]) | 12\n",
-      "31098    | torch.Size([1, 1000, 754]) | 12\n",
-      "31099    | torch.Size([1, 1000, 754]) | 12\n",
-      "31100    | torch.Size([1, 1000, 771]) | 12\n",
-      "31101    | torch.Size([1, 1000, 754]) | 12\n",
-      "31102    | torch.Size([1, 1000, 754]) | 12\n",
-      "31103    | torch.Size([1, 1000, 763]) | 12\n",
-      "31104    | torch.Size([1, 1000, 754]) | 12\n",
-      "31105    | torch.Size([1, 1000, 754]) | 12\n",
-      "31106    | torch.Size([1, 1000, 754]) | 12\n",
-      "31107    | torch.Size([1, 1000, 754]) | 12\n",
-      "31108    | torch.Size([1, 1000, 754]) | 12\n",
-      "31109    | torch.Size([1, 1000, 754]) | 12\n",
-      "31110    | torch.Size([1, 1000, 754]) | 12\n",
-      "31111    | torch.Size([1, 1000, 762]) | 12\n",
-      "31112    | torch.Size([1, 1000, 754]) | 12\n",
-      "31113    | torch.Size([1, 1000, 754]) | 12\n",
-      "31114    | torch.Size([1, 1000, 754]) | 12\n",
-      "31115    | torch.Size([1, 1000, 754]) | 12\n",
-      "31116    | torch.Size([1, 1000, 754]) | 12\n",
-      "31117    | torch.Size([1, 1000, 772]) | 12\n",
-      "31118    | torch.Size([1, 1000, 772]) | 12\n",
-      "31119    | torch.Size([1, 1000, 754]) | 12\n",
-      "31120    | torch.Size([1, 1000, 770]) | 12\n",
-      "31121    | torch.Size([1, 1000, 754]) | 12\n",
-      "31122    | torch.Size([1, 1000, 754]) | 12\n",
-      "31123    | torch.Size([1, 1000, 754]) | 12\n",
-      "31124    | torch.Size([1, 1000, 754]) | 12\n",
-      "31125    | torch.Size([1, 1000, 754]) | 12\n",
-      "31126    | torch.Size([1, 1000, 754]) | 12\n",
-      "31127    | torch.Size([1, 1000, 754]) | 12\n",
-      "31128    | torch.Size([1, 1000, 793]) | 12\n",
-      "31129    | torch.Size([1, 1000, 754]) | 12\n",
-      "31130    | torch.Size([1, 1000, 754]) | 12\n",
-      "31131    | torch.Size([1, 1000, 754]) | 12\n",
-      "31132    | torch.Size([1, 1000, 754]) | 12\n",
-      "31133    | torch.Size([1, 1000, 754]) | 12\n",
-      "31134    | torch.Size([1, 1000, 754]) | 12\n",
-      "31135    | torch.Size([1, 1000, 770]) | 12\n",
-      "31136    | torch.Size([1, 1000, 754]) | 12\n",
-      "31137    | torch.Size([1, 1000, 754]) | 12\n",
-      "31138    | torch.Size([1, 1000, 754]) | 12\n",
-      "31139    | torch.Size([1, 1000, 754]) | 12\n",
-      "31140    | torch.Size([1, 1000, 754]) | 12\n",
-      "31141    | torch.Size([1, 1000, 754]) | 12\n",
-      "31142    | torch.Size([1, 1000, 754]) | 12\n",
-      "31143    | torch.Size([1, 1000, 754]) | 12\n",
-      "31144    | torch.Size([1, 1000, 777]) | 12\n",
-      "31145    | torch.Size([1, 1000, 754]) | 12\n",
-      "31146    | torch.Size([1, 1000, 779]) | 12\n",
-      "31147    | torch.Size([1, 1000, 754]) | 12\n",
-      "31148    | torch.Size([1, 1000, 754]) | 12\n",
-      "31149    | torch.Size([1, 1000, 754]) | 12\n",
-      "31150    | torch.Size([1, 1000, 781]) | 12\n",
-      "31151    | torch.Size([1, 1000, 754]) | 12\n",
-      "31152    | torch.Size([1, 1000, 754]) | 12\n",
-      "31153    | torch.Size([1, 1000, 754]) | 12\n",
-      "31154    | torch.Size([1, 1000, 754]) | 12\n",
-      "31155    | torch.Size([1, 1000, 754]) | 12\n",
-      "31156    | torch.Size([1, 1000, 754]) | 12\n",
-      "31157    | torch.Size([1, 1000, 754]) | 12\n",
-      "31158    | torch.Size([1, 1000, 754]) | 12\n",
-      "31159    | torch.Size([1, 1000, 754]) | 12\n",
-      "31160    | torch.Size([1, 1000, 754]) | 12\n",
-      "31161    | torch.Size([1, 1000, 754]) | 12\n",
-      "31162    | torch.Size([1, 1000, 754]) | 12\n",
-      "31163    | torch.Size([1, 1000, 754]) | 12\n",
-      "31164    | torch.Size([1, 1000, 789]) | 12\n",
-      "31165    | torch.Size([1, 1000, 754]) | 12\n",
-      "31166    | torch.Size([1, 1000, 754]) | 12\n",
-      "31167    | torch.Size([1, 1000, 781]) | 12\n",
-      "31168    | torch.Size([1, 1000, 754]) | 12\n",
-      "31169    | torch.Size([1, 1000, 754]) | 12\n",
-      "31170    | torch.Size([1, 1000, 754]) | 12\n",
-      "31171    | torch.Size([1, 1000, 754]) | 12\n",
-      "31172    | torch.Size([1, 1000, 754]) | 12\n",
-      "31173    | torch.Size([1, 1000, 754]) | 12\n",
-      "31174    | torch.Size([1, 1000, 754]) | 12\n",
-      "31175    | torch.Size([1, 1000, 754]) | 12\n",
-      "31176    | torch.Size([1, 1000, 754]) | 12\n",
-      "31177    | torch.Size([1, 1000, 754]) | 12\n",
-      "31178    | torch.Size([1, 1000, 754]) | 12\n",
-      "31179    | torch.Size([1, 1000, 754]) | 12\n",
-      "31180    | torch.Size([1, 1000, 754]) | 12\n",
-      "31181    | torch.Size([1, 1000, 754]) | 12\n",
-      "31182    | torch.Size([1, 1000, 754]) | 12\n",
-      "31183    | torch.Size([1, 1000, 754]) | 12\n",
-      "31184    | torch.Size([1, 1000, 754]) | 12\n",
-      "31185    | torch.Size([1, 1000, 780]) | 12\n",
-      "31186    | torch.Size([1, 1000, 762]) | 12\n",
-      "31187    | torch.Size([1, 1000, 754]) | 12\n",
-      "31188    | torch.Size([1, 1000, 754]) | 12\n",
-      "31189    | torch.Size([1, 1000, 762]) | 12\n",
-      "31190    | torch.Size([1, 1000, 801]) | 12\n",
-      "31191    | torch.Size([1, 1000, 754]) | 12\n",
-      "31192    | torch.Size([1, 1000, 780]) | 12\n",
-      "31193    | torch.Size([1, 1000, 754]) | 12\n",
-      "31194    | torch.Size([1, 1000, 763]) | 12\n",
-      "31195    | torch.Size([1, 1000, 754]) | 12\n",
-      "31196    | torch.Size([1, 1000, 772]) | 12\n",
-      "31197    | torch.Size([1, 1000, 764]) | 12\n",
-      "31198    | torch.Size([1, 1000, 754]) | 12\n",
-      "31199    | torch.Size([1, 1000, 754]) | 12\n",
-      "31200    | torch.Size([1, 1000, 754]) | 12\n",
-      "31201    | torch.Size([1, 1000, 754]) | 12\n",
-      "31202    | torch.Size([1, 1000, 796]) | 12\n",
-      "31203    | torch.Size([1, 1000, 754]) | 12\n",
-      "31204    | torch.Size([1, 1000, 754]) | 12\n",
-      "31205    | torch.Size([1, 1000, 768]) | 12\n",
-      "31206    | torch.Size([1, 1000, 754]) | 12\n",
-      "31207    | torch.Size([1, 1000, 754]) | 12\n",
-      "31208    | torch.Size([1, 1000, 754]) | 12\n",
-      "31209    | torch.Size([1, 1000, 754]) | 12\n",
-      "31210    | torch.Size([1, 1000, 754]) | 12\n",
-      "31211    | torch.Size([1, 1000, 754]) | 12\n",
-      "31212    | torch.Size([1, 1000, 754]) | 12\n",
-      "31213    | torch.Size([1, 1000, 754]) | 12\n",
-      "31214    | torch.Size([1, 1000, 754]) | 12\n",
-      "31215    | torch.Size([1, 1000, 754]) | 12\n",
-      "31216    | torch.Size([1, 1000, 754]) | 12\n",
-      "31217    | torch.Size([1, 1000, 754]) | 12\n",
-      "31218    | torch.Size([1, 1000, 763]) | 12\n",
-      "31219    | torch.Size([1, 1000, 754]) | 12\n",
-      "31220    | torch.Size([1, 1000, 754]) | 12\n",
-      "31221    | torch.Size([1, 1000, 754]) | 12\n",
-      "31222    | torch.Size([1, 1000, 754]) | 12\n",
-      "31223    | torch.Size([1, 1000, 754]) | 12\n",
-      "31224    | torch.Size([1, 1000, 754]) | 12\n",
-      "31225    | torch.Size([1, 1000, 754]) | 12\n",
-      "31226    | torch.Size([1, 1000, 754]) | 12\n",
-      "31227    | torch.Size([1, 1000, 754]) | 12\n",
-      "31228    | torch.Size([1, 1000, 780]) | 12\n",
-      "31229    | torch.Size([1, 1000, 754]) | 12\n",
-      "31230    | torch.Size([1, 1000, 777]) | 12\n",
-      "31231    | torch.Size([1, 1000, 754]) | 12\n",
-      "31232    | torch.Size([1, 1000, 754]) | 12\n",
-      "31233    | torch.Size([1, 1000, 754]) | 12\n",
-      "31234    | torch.Size([1, 1000, 754]) | 12\n",
-      "31235    | torch.Size([1, 1000, 754]) | 12\n",
-      "31236    | torch.Size([1, 1000, 754]) | 12\n",
-      "31237    | torch.Size([1, 1000, 754]) | 12\n",
-      "31238    | torch.Size([1, 1000, 754]) | 12\n",
-      "31239    | torch.Size([1, 1000, 754]) | 12\n",
-      "31240    | torch.Size([1, 1000, 795]) | 12\n",
-      "31241    | torch.Size([1, 1000, 754]) | 12\n",
-      "31242    | torch.Size([1, 1000, 754]) | 12\n",
-      "31243    | torch.Size([1, 1000, 769]) | 12\n",
-      "31244    | torch.Size([1, 1000, 763]) | 12\n",
-      "31245    | torch.Size([1, 1000, 772]) | 12\n",
-      "31246    | torch.Size([1, 1000, 754]) | 12\n",
-      "31247    | torch.Size([1, 1000, 763]) | 12\n",
-      "31248    | torch.Size([1, 1000, 754]) | 12\n",
-      "31249    | torch.Size([1, 1000, 772]) | 12\n",
-      "31250    | torch.Size([1, 1000, 771]) | 12\n",
-      "31251    | torch.Size([1, 1000, 754]) | 12\n",
-      "31252    | torch.Size([1, 1000, 754]) | 12\n",
-      "31253    | torch.Size([1, 1000, 754]) | 12\n",
-      "31254    | torch.Size([1, 1000, 754]) | 12\n",
-      "31255    | torch.Size([1, 1000, 767]) | 12\n",
-      "31256    | torch.Size([1, 1000, 754]) | 12\n",
-      "31257    | torch.Size([1, 1000, 754]) | 12\n",
-      "31258    | torch.Size([1, 1000, 754]) | 12\n",
-      "31259    | torch.Size([1, 1000, 754]) | 12\n",
-      "31260    | torch.Size([1, 1000, 754]) | 12\n",
-      "31261    | torch.Size([1, 1000, 754]) | 12\n",
-      "31262    | torch.Size([1, 1000, 754]) | 12\n",
-      "31263    | torch.Size([1, 1000, 754]) | 12\n",
-      "31264    | torch.Size([1, 1000, 754]) | 12\n",
-      "31265    | torch.Size([1, 1000, 754]) | 12\n",
-      "31266    | torch.Size([1, 1000, 754]) | 12\n",
-      "31267    | torch.Size([1, 1000, 754]) | 12\n",
-      "31268    | torch.Size([1, 1000, 754]) | 12\n",
-      "31269    | torch.Size([1, 1000, 754]) | 12\n",
-      "31270    | torch.Size([1, 1000, 754]) | 12\n",
-      "31271    | torch.Size([1, 1000, 754]) | 12\n",
-      "31272    | torch.Size([1, 1000, 754]) | 12\n",
-      "31273    | torch.Size([1, 1000, 754]) | 12\n",
-      "31274    | torch.Size([1, 1000, 754]) | 12\n",
-      "31275    | torch.Size([1, 1000, 754]) | 12\n",
-      "31276    | torch.Size([1, 1000, 754]) | 12\n",
-      "31277    | torch.Size([1, 1000, 754]) | 12\n",
-      "31278    | torch.Size([1, 1000, 771]) | 12\n",
-      "31279    | torch.Size([1, 1000, 804]) | 12\n",
-      "31280    | torch.Size([1, 1000, 754]) | 12\n",
-      "31281    | torch.Size([1, 1000, 754]) | 12\n",
-      "31282    | torch.Size([1, 1000, 771]) | 12\n",
-      "31283    | torch.Size([1, 1000, 763]) | 12\n",
-      "31284    | torch.Size([1, 1000, 754]) | 12\n",
-      "31285    | torch.Size([1, 1000, 754]) | 12\n",
-      "31286    | torch.Size([1, 1000, 754]) | 12\n",
-      "31287    | torch.Size([1, 1000, 754]) | 12\n",
-      "31288    | torch.Size([1, 1000, 754]) | 12\n",
-      "31289    | torch.Size([1, 1000, 754]) | 12\n",
-      "31290    | torch.Size([1, 1000, 754]) | 12\n",
-      "31291    | torch.Size([1, 1000, 754]) | 12\n",
-      "31292    | torch.Size([1, 1000, 754]) | 12\n",
-      "31293    | torch.Size([1, 1000, 754]) | 12\n",
-      "31294    | torch.Size([1, 1000, 772]) | 12\n",
-      "31295    | torch.Size([1, 1000, 754]) | 12\n",
-      "31296    | torch.Size([1, 1000, 754]) | 12\n",
-      "31297    | torch.Size([1, 1000, 754]) | 12\n",
-      "31298    | torch.Size([1, 1000, 754]) | 12\n",
-      "31299    | torch.Size([1, 1000, 754]) | 12\n",
-      "31300    | torch.Size([1, 1000, 754]) | 12\n",
-      "31301    | torch.Size([1, 1000, 754]) | 12\n",
-      "31302    | torch.Size([1, 1000, 754]) | 12\n",
-      "31303    | torch.Size([1, 1000, 754]) | 12\n",
-      "31304    | torch.Size([1, 1000, 754]) | 12\n",
-      "31305    | torch.Size([1, 1000, 754]) | 12\n",
-      "31306    | torch.Size([1, 1000, 754]) | 12\n",
-      "31307    | torch.Size([1, 1000, 754]) | 12\n",
-      "31308    | torch.Size([1, 1000, 754]) | 12\n",
-      "31309    | torch.Size([1, 1000, 754]) | 12\n",
-      "31310    | torch.Size([1, 1000, 754]) | 12\n",
-      "31311    | torch.Size([1, 1000, 754]) | 12\n",
-      "31312    | torch.Size([1, 1000, 754]) | 12\n",
-      "31313    | torch.Size([1, 1000, 754]) | 12\n",
-      "31314    | torch.Size([1, 1000, 754]) | 12\n",
-      "31315    | torch.Size([1, 1000, 754]) | 12\n",
-      "31316    | torch.Size([1, 1000, 754]) | 12\n",
-      "31317    | torch.Size([1, 1000, 754]) | 12\n",
-      "31318    | torch.Size([1, 1000, 754]) | 12\n",
-      "31319    | torch.Size([1, 1000, 754]) | 12\n",
-      "31320    | torch.Size([1, 1000, 754]) | 12\n",
-      "31321    | torch.Size([1, 1000, 754]) | 12\n",
-      "31322    | torch.Size([1, 1000, 754]) | 12\n",
-      "31323    | torch.Size([1, 1000, 754]) | 12\n",
-      "31324    | torch.Size([1, 1000, 754]) | 12\n",
-      "31325    | torch.Size([1, 1000, 754]) | 12\n",
-      "31326    | torch.Size([1, 1000, 754]) | 12\n",
-      "31327    | torch.Size([1, 1000, 769]) | 12\n",
-      "31328    | torch.Size([1, 1000, 754]) | 12\n",
-      "31329    | torch.Size([1, 1000, 754]) | 12\n",
-      "31330    | torch.Size([1, 1000, 754]) | 12\n",
-      "31331    | torch.Size([1, 1000, 754]) | 12\n",
-      "31332    | torch.Size([1, 1000, 754]) | 12\n",
-      "31333    | torch.Size([1, 1000, 754]) | 12\n",
-      "31334    | torch.Size([1, 1000, 767]) | 12\n",
-      "31335    | torch.Size([1, 1000, 754]) | 12\n",
-      "31336    | torch.Size([1, 1000, 754]) | 12\n",
-      "31337    | torch.Size([1, 1000, 754]) | 12\n",
-      "31338    | torch.Size([1, 1000, 754]) | 12\n",
-      "31339    | torch.Size([1, 1000, 754]) | 12\n",
-      "31340    | torch.Size([1, 1000, 754]) | 12\n",
-      "31341    | torch.Size([1, 1000, 754]) | 12\n",
-      "31342    | torch.Size([1, 1000, 754]) | 12\n",
-      "31343    | torch.Size([1, 1000, 754]) | 12\n",
-      "31344    | torch.Size([1, 1000, 773]) | 12\n",
-      "31345    | torch.Size([1, 1000, 771]) | 12\n",
-      "31346    | torch.Size([1, 1000, 754]) | 12\n",
-      "31347    | torch.Size([1, 1000, 754]) | 12\n",
-      "31348    | torch.Size([1, 1000, 769]) | 12\n",
-      "31349    | torch.Size([1, 1000, 760]) | 12\n",
-      "31350    | torch.Size([1, 1000, 754]) | 12\n",
-      "31351    | torch.Size([1, 1000, 789]) | 12\n",
-      "31352    | torch.Size([1, 1000, 754]) | 12\n",
-      "31353    | torch.Size([1, 1000, 772]) | 12\n",
-      "31354    | torch.Size([1, 1000, 754]) | 12\n",
-      "31355    | torch.Size([1, 1000, 754]) | 12\n",
-      "31356    | torch.Size([1, 1000, 754]) | 12\n",
-      "31357    | torch.Size([1, 1000, 754]) | 12\n",
-      "31358    | torch.Size([1, 1000, 754]) | 12\n",
-      "31359    | torch.Size([1, 1000, 754]) | 12\n",
-      "31360    | torch.Size([1, 1000, 754]) | 12\n",
-      "31361    | torch.Size([1, 1000, 754]) | 12\n",
-      "31362    | torch.Size([1, 1000, 754]) | 12\n",
-      "31363    | torch.Size([1, 1000, 754]) | 12\n",
-      "31364    | torch.Size([1, 1000, 754]) | 12\n",
-      "31365    | torch.Size([1, 1000, 754]) | 12\n",
-      "31366    | torch.Size([1, 1000, 772]) | 12\n",
-      "31367    | torch.Size([1, 1000, 754]) | 12\n",
-      "31368    | torch.Size([1, 1000, 754]) | 12\n",
-      "31369    | torch.Size([1, 1000, 754]) | 12\n",
-      "31370    | torch.Size([1, 1000, 777]) | 12\n",
-      "31371    | torch.Size([1, 1000, 754]) | 12\n",
-      "31372    | torch.Size([1, 1000, 754]) | 12\n",
-      "31373    | torch.Size([1, 1000, 754]) | 12\n",
-      "31374    | torch.Size([1, 1000, 754]) | 12\n",
-      "31375    | torch.Size([1, 1000, 754]) | 12\n",
-      "31376    | torch.Size([1, 1000, 772]) | 12\n",
-      "31377    | torch.Size([1, 1000, 754]) | 12\n",
-      "31378    | torch.Size([1, 1000, 754]) | 12\n",
-      "31379    | torch.Size([1, 1000, 754]) | 12\n",
-      "31380    | torch.Size([1, 1000, 754]) | 12\n",
-      "31381    | torch.Size([1, 1000, 754]) | 12\n",
-      "31382    | torch.Size([1, 1000, 754]) | 12\n",
-      "31383    | torch.Size([1, 1000, 754]) | 12\n",
-      "31384    | torch.Size([1, 1000, 754]) | 12\n",
-      "31385    | torch.Size([1, 1000, 754]) | 12\n",
-      "31386    | torch.Size([1, 1000, 754]) | 12\n",
-      "31387    | torch.Size([1, 1000, 754]) | 12\n",
-      "31388    | torch.Size([1, 1000, 754]) | 12\n",
-      "31389    | torch.Size([1, 1000, 754]) | 12\n",
-      "31390    | torch.Size([1, 1000, 754]) | 12\n",
-      "31391    | torch.Size([1, 1000, 754]) | 12\n",
-      "31392    | torch.Size([1, 1000, 754]) | 12\n",
-      "31393    | torch.Size([1, 1000, 754]) | 12\n",
-      "31394    | torch.Size([1, 1000, 754]) | 12\n",
-      "31395    | torch.Size([1, 1000, 767]) | 12\n",
-      "31396    | torch.Size([1, 1000, 768]) | 12\n",
-      "31397    | torch.Size([1, 1000, 754]) | 12\n",
-      "31398    | torch.Size([1, 1000, 754]) | 12\n",
-      "31399    | torch.Size([1, 1000, 754]) | 12\n",
-      "31400    | torch.Size([1, 1000, 807]) | 12\n",
-      "31401    | torch.Size([1, 1000, 754]) | 12\n",
-      "31402    | torch.Size([1, 1000, 754]) | 12\n",
-      "31403    | torch.Size([1, 1000, 754]) | 12\n",
-      "31404    | torch.Size([1, 1000, 754]) | 12\n",
-      "31405    | torch.Size([1, 1000, 754]) | 12\n",
-      "31406    | torch.Size([1, 1000, 762]) | 12\n",
-      "31407    | torch.Size([1, 1000, 754]) | 12\n",
-      "31408    | torch.Size([1, 1000, 754]) | 12\n",
-      "31409    | torch.Size([1, 1000, 762]) | 12\n",
-      "31410    | torch.Size([1, 1000, 754]) | 12\n",
-      "31411    | torch.Size([1, 1000, 754]) | 12\n",
-      "31412    | torch.Size([1, 1000, 754]) | 12\n",
-      "31413    | torch.Size([1, 1000, 754]) | 12\n",
-      "31414    | torch.Size([1, 1000, 754]) | 12\n",
-      "31415    | torch.Size([1, 1000, 754]) | 12\n",
-      "31416    | torch.Size([1, 1000, 754]) | 12\n",
-      "31417    | torch.Size([1, 1000, 754]) | 12\n",
-      "31418    | torch.Size([1, 1000, 754]) | 12\n",
-      "31419    | torch.Size([1, 1000, 754]) | 12\n",
-      "31420    | torch.Size([1, 1000, 754]) | 12\n",
-      "31421    | torch.Size([1, 1000, 776]) | 12\n",
-      "31422    | torch.Size([1, 1000, 754]) | 12\n",
-      "31423    | torch.Size([1, 1000, 754]) | 12\n",
-      "31424    | torch.Size([1, 1000, 754]) | 12\n",
-      "31425    | torch.Size([1, 1000, 754]) | 12\n",
-      "31426    | torch.Size([1, 1000, 777]) | 12\n",
-      "31427    | torch.Size([1, 1000, 754]) | 12\n",
-      "31428    | torch.Size([1, 1000, 754]) | 12\n",
-      "31429    | torch.Size([1, 1000, 754]) | 12\n",
-      "31430    | torch.Size([1, 1000, 784]) | 12\n",
-      "31431    | torch.Size([1, 1000, 793]) | 12\n",
-      "31432    | torch.Size([1, 1000, 777]) | 12\n",
-      "31433    | torch.Size([1, 1000, 754]) | 12\n",
-      "31434    | torch.Size([1, 1000, 754]) | 12\n",
-      "31435    | torch.Size([1, 1000, 772]) | 12\n",
-      "31436    | torch.Size([1, 1000, 754]) | 12\n",
-      "31437    | torch.Size([1, 1000, 754]) | 12\n",
-      "31438    | torch.Size([1, 1000, 754]) | 12\n",
-      "31439    | torch.Size([1, 1000, 754]) | 12\n",
-      "31440    | torch.Size([1, 1000, 761]) | 12\n",
-      "31441    | torch.Size([1, 1000, 754]) | 12\n",
-      "31442    | torch.Size([1, 1000, 754]) | 12\n",
-      "31443    | torch.Size([1, 1000, 754]) | 12\n",
-      "31444    | torch.Size([1, 1000, 754]) | 12\n",
-      "31445    | torch.Size([1, 1000, 754]) | 12\n",
-      "31446    | torch.Size([1, 1000, 754]) | 12\n",
-      "31447    | torch.Size([1, 1000, 762]) | 12\n",
-      "31448    | torch.Size([1, 1000, 754]) | 12\n",
-      "31449    | torch.Size([1, 1000, 754]) | 12\n",
-      "31450    | torch.Size([1, 1000, 754]) | 12\n",
-      "31451    | torch.Size([1, 1000, 754]) | 12\n",
-      "31452    | torch.Size([1, 1000, 754]) | 12\n",
-      "31453    | torch.Size([1, 1000, 754]) | 12\n",
-      "31454    | torch.Size([1, 1000, 777]) | 12\n",
-      "31455    | torch.Size([1, 1000, 754]) | 12\n",
-      "31456    | torch.Size([1, 1000, 776]) | 12\n",
-      "31457    | torch.Size([1, 1000, 754]) | 12\n",
-      "31458    | torch.Size([1, 1000, 754]) | 12\n",
-      "31459    | torch.Size([1, 1000, 754]) | 12\n",
-      "31460    | torch.Size([1, 1000, 754]) | 12\n",
-      "31461    | torch.Size([1, 1000, 754]) | 12\n",
-      "31462    | torch.Size([1, 1000, 776]) | 12\n",
-      "31463    | torch.Size([1, 1000, 754]) | 12\n",
-      "31464    | torch.Size([1, 1000, 772]) | 12\n",
-      "31465    | torch.Size([1, 1000, 754]) | 12\n",
-      "31466    | torch.Size([1, 1000, 754]) | 12\n",
-      "31467    | torch.Size([1, 1000, 754]) | 12\n",
-      "31468    | torch.Size([1, 1000, 754]) | 12\n",
-      "31469    | torch.Size([1, 1000, 800]) | 12\n",
-      "31470    | torch.Size([1, 1000, 754]) | 12\n",
-      "31471    | torch.Size([1, 1000, 754]) | 12\n",
-      "31472    | torch.Size([1, 1000, 754]) | 12\n",
-      "31473    | torch.Size([1, 1000, 754]) | 12\n",
-      "31474    | torch.Size([1, 1000, 754]) | 12\n",
-      "31475    | torch.Size([1, 1000, 754]) | 12\n",
-      "31476    | torch.Size([1, 1000, 754]) | 12\n",
-      "31477    | torch.Size([1, 1000, 754]) | 12\n",
-      "31478    | torch.Size([1, 1000, 754]) | 12\n",
-      "31479    | torch.Size([1, 1000, 754]) | 12\n",
-      "31480    | torch.Size([1, 1000, 754]) | 12\n",
-      "31481    | torch.Size([1, 1000, 754]) | 12\n",
-      "31482    | torch.Size([1, 1000, 754]) | 12\n",
-      "31483    | torch.Size([1, 1000, 754]) | 12\n",
-      "31484    | torch.Size([1, 1000, 754]) | 12\n",
-      "31485    | torch.Size([1, 1000, 754]) | 12\n",
-      "31486    | torch.Size([1, 1000, 754]) | 12\n",
-      "31487    | torch.Size([1, 1000, 767]) | 12\n",
-      "31488    | torch.Size([1, 1000, 754]) | 12\n",
-      "31489    | torch.Size([1, 1000, 754]) | 12\n",
-      "31490    | torch.Size([1, 1000, 754]) | 12\n",
-      "31491    | torch.Size([1, 1000, 762]) | 12\n",
-      "31492    | torch.Size([1, 1000, 800]) | 12\n",
-      "31493    | torch.Size([1, 1000, 762]) | 12\n",
-      "31494    | torch.Size([1, 1000, 766]) | 12\n",
-      "31495    | torch.Size([1, 1000, 754]) | 12\n",
-      "31496    | torch.Size([1, 1000, 754]) | 12\n",
-      "31497    | torch.Size([1, 1000, 754]) | 12\n",
-      "31498    | torch.Size([1, 1000, 754]) | 12\n",
-      "31499    | torch.Size([1, 1000, 754]) | 12\n",
-      "31500    | torch.Size([1, 1000, 703]) | 12\n",
-      "31501    | torch.Size([1, 1000, 754]) | 12\n",
-      "31502    | torch.Size([1, 1000, 754]) | 12\n",
-      "31503    | torch.Size([1, 1000, 754]) | 12\n",
-      "31504    | torch.Size([1, 1000, 754]) | 12\n",
-      "31505    | torch.Size([1, 1000, 754]) | 12\n",
-      "31506    | torch.Size([1, 1000, 754]) | 12\n",
-      "31507    | torch.Size([1, 1000, 754]) | 12\n",
-      "31508    | torch.Size([1, 1000, 754]) | 12\n",
-      "31509    | torch.Size([1, 1000, 754]) | 12\n",
-      "31510    | torch.Size([1, 1000, 754]) | 12\n",
-      "31511    | torch.Size([1, 1000, 754]) | 12\n",
-      "31512    | torch.Size([1, 1000, 754]) | 12\n",
-      "31513    | torch.Size([1, 1000, 754]) | 12\n",
-      "31514    | torch.Size([1, 1000, 754]) | 12\n",
-      "31515    | torch.Size([1, 1000, 754]) | 12\n",
-      "31516    | torch.Size([1, 1000, 772]) | 12\n",
-      "31517    | torch.Size([1, 1000, 754]) | 12\n",
-      "31518    | torch.Size([1, 1000, 754]) | 12\n",
-      "31519    | torch.Size([1, 1000, 754]) | 12\n",
-      "31520    | torch.Size([1, 1000, 771]) | 12\n",
-      "31521    | torch.Size([1, 1000, 754]) | 12\n",
-      "31522    | torch.Size([1, 1000, 754]) | 12\n",
-      "31523    | torch.Size([1, 1000, 777]) | 12\n",
-      "31524    | torch.Size([1, 1000, 754]) | 12\n",
-      "31525    | torch.Size([1, 1000, 754]) | 12\n",
-      "31526    | torch.Size([1, 1000, 777]) | 12\n",
-      "31527    | torch.Size([1, 1000, 754]) | 12\n",
-      "31528    | torch.Size([1, 1000, 762]) | 12\n",
-      "31529    | torch.Size([1, 1000, 754]) | 12\n",
-      "31530    | torch.Size([1, 1000, 754]) | 12\n",
-      "31531    | torch.Size([1, 1000, 754]) | 12\n",
-      "31532    | torch.Size([1, 1000, 754]) | 12\n",
-      "31533    | torch.Size([1, 1000, 754]) | 12\n",
-      "31534    | torch.Size([1, 1000, 754]) | 12\n",
-      "31535    | torch.Size([1, 1000, 754]) | 12\n",
-      "31536    | torch.Size([1, 1000, 754]) | 12\n",
-      "31537    | torch.Size([1, 1000, 754]) | 12\n",
-      "31538    | torch.Size([1, 1000, 754]) | 12\n",
-      "31539    | torch.Size([1, 1000, 754]) | 12\n",
-      "31540    | torch.Size([1, 1000, 809]) | 12\n",
-      "31541    | torch.Size([1, 1000, 754]) | 12\n",
-      "31542    | torch.Size([1, 1000, 754]) | 12\n",
-      "31543    | torch.Size([1, 1000, 754]) | 12\n",
-      "31544    | torch.Size([1, 1000, 754]) | 12\n",
-      "31545    | torch.Size([1, 1000, 754]) | 12\n",
-      "31546    | torch.Size([1, 1000, 754]) | 12\n",
-      "31547    | torch.Size([1, 1000, 754]) | 12\n",
-      "31548    | torch.Size([1, 1000, 754]) | 12\n",
-      "31549    | torch.Size([1, 1000, 754]) | 12\n",
-      "31550    | torch.Size([1, 1000, 754]) | 12\n",
-      "31551    | torch.Size([1, 1000, 754]) | 12\n",
-      "31552    | torch.Size([1, 1000, 754]) | 12\n",
-      "31553    | torch.Size([1, 1000, 754]) | 12\n",
-      "31554    | torch.Size([1, 1000, 754]) | 12\n",
-      "31555    | torch.Size([1, 1000, 754]) | 12\n",
-      "31556    | torch.Size([1, 1000, 754]) | 12\n",
-      "31557    | torch.Size([1, 1000, 754]) | 12\n",
-      "31558    | torch.Size([1, 1000, 754]) | 12\n",
-      "31559    | torch.Size([1, 1000, 811]) | 12\n",
-      "31560    | torch.Size([1, 1000, 754]) | 12\n",
-      "31561    | torch.Size([1, 1000, 754]) | 12\n",
-      "31562    | torch.Size([1, 1000, 754]) | 12\n",
-      "31563    | torch.Size([1, 1000, 754]) | 12\n",
-      "31564    | torch.Size([1, 1000, 754]) | 12\n",
-      "31565    | torch.Size([1, 1000, 754]) | 12\n",
-      "31566    | torch.Size([1, 1000, 754]) | 12\n",
-      "31567    | torch.Size([1, 1000, 754]) | 12\n",
-      "31568    | torch.Size([1, 1000, 768]) | 12\n",
-      "31569    | torch.Size([1, 1000, 754]) | 12\n",
-      "31570    | torch.Size([1, 1000, 754]) | 12\n",
-      "31571    | torch.Size([1, 1000, 754]) | 12\n",
-      "31572    | torch.Size([1, 1000, 754]) | 12\n",
-      "31573    | torch.Size([1, 1000, 772]) | 12\n",
-      "31574    | torch.Size([1, 1000, 809]) | 12\n",
-      "31575    | torch.Size([1, 1000, 754]) | 12\n",
-      "31576    | torch.Size([1, 1000, 754]) | 12\n",
-      "31577    | torch.Size([1, 1000, 754]) | 12\n",
-      "31578    | torch.Size([1, 1000, 754]) | 12\n",
-      "31579    | torch.Size([1, 1000, 754]) | 12\n",
-      "31580    | torch.Size([1, 1000, 754]) | 12\n",
-      "31581    | torch.Size([1, 1000, 772]) | 12\n",
-      "31582    | torch.Size([1, 1000, 754]) | 12\n",
-      "31583    | torch.Size([1, 1000, 754]) | 12\n",
-      "31584    | torch.Size([1, 1000, 754]) | 12\n",
-      "31585    | torch.Size([1, 1000, 754]) | 12\n",
-      "31586    | torch.Size([1, 1000, 754]) | 12\n",
-      "31587    | torch.Size([1, 1000, 754]) | 12\n",
-      "31588    | torch.Size([1, 1000, 754]) | 12\n",
-      "31589    | torch.Size([1, 1000, 754]) | 12\n",
-      "31590    | torch.Size([1, 1000, 766]) | 12\n",
-      "31591    | torch.Size([1, 1000, 754]) | 12\n",
-      "31592    | torch.Size([1, 1000, 754]) | 12\n",
-      "31593    | torch.Size([1, 1000, 754]) | 12\n",
-      "31594    | torch.Size([1, 1000, 754]) | 12\n",
-      "31595    | torch.Size([1, 1000, 754]) | 12\n",
-      "31596    | torch.Size([1, 1000, 754]) | 12\n",
-      "31597    | torch.Size([1, 1000, 754]) | 12\n",
-      "31598    | torch.Size([1, 1000, 771]) | 12\n",
-      "31599    | torch.Size([1, 1000, 754]) | 12\n",
-      "31600    | torch.Size([1, 1000, 754]) | 12\n",
-      "31601    | torch.Size([1, 1000, 754]) | 12\n",
-      "31602    | torch.Size([1, 1000, 762]) | 12\n",
-      "31603    | torch.Size([1, 1000, 754]) | 12\n",
-      "31604    | torch.Size([1, 1000, 777]) | 12\n",
-      "31605    | torch.Size([1, 1000, 754]) | 12\n",
-      "31606    | torch.Size([1, 1000, 754]) | 12\n",
-      "31607    | torch.Size([1, 1000, 754]) | 12\n",
-      "31608    | torch.Size([1, 1000, 780]) | 12\n",
-      "31609    | torch.Size([1, 1000, 754]) | 12\n",
-      "31610    | torch.Size([1, 1000, 754]) | 12\n",
-      "31611    | torch.Size([1, 1000, 754]) | 12\n",
-      "31612    | torch.Size([1, 1000, 754]) | 12\n",
-      "31613    | torch.Size([1, 1000, 759]) | 12\n",
-      "31614    | torch.Size([1, 1000, 754]) | 12\n",
-      "31615    | torch.Size([1, 1000, 754]) | 12\n",
-      "31616    | torch.Size([1, 1000, 754]) | 12\n",
-      "31617    | torch.Size([1, 1000, 754]) | 12\n",
-      "31618    | torch.Size([1, 1000, 754]) | 12\n",
-      "31619    | torch.Size([1, 1000, 754]) | 12\n",
-      "31620    | torch.Size([1, 1000, 754]) | 12\n",
-      "31621    | torch.Size([1, 1000, 754]) | 12\n",
-      "31622    | torch.Size([1, 1000, 764]) | 12\n",
-      "31623    | torch.Size([1, 1000, 754]) | 12\n",
-      "31624    | torch.Size([1, 1000, 754]) | 12\n",
-      "31625    | torch.Size([1, 1000, 781]) | 12\n",
-      "31626    | torch.Size([1, 1000, 754]) | 12\n",
-      "31627    | torch.Size([1, 1000, 754]) | 12\n",
-      "31628    | torch.Size([1, 1000, 754]) | 12\n",
-      "31629    | torch.Size([1, 1000, 777]) | 12\n",
-      "31630    | torch.Size([1, 1000, 771]) | 12\n",
-      "31631    | torch.Size([1, 1000, 769]) | 12\n",
-      "31632    | torch.Size([1, 1000, 771]) | 12\n",
-      "31633    | torch.Size([1, 1000, 754]) | 12\n",
-      "31634    | torch.Size([1, 1000, 754]) | 12\n",
-      "31635    | torch.Size([1, 1000, 754]) | 12\n",
-      "31636    | torch.Size([1, 1000, 754]) | 12\n",
-      "31637    | torch.Size([1, 1000, 754]) | 12\n",
-      "31638    | torch.Size([1, 1000, 754]) | 12\n",
-      "31639    | torch.Size([1, 1000, 754]) | 12\n",
-      "31640    | torch.Size([1, 1000, 754]) | 12\n",
-      "31641    | torch.Size([1, 1000, 754]) | 12\n",
-      "31642    | torch.Size([1, 1000, 754]) | 12\n",
-      "31643    | torch.Size([1, 1000, 754]) | 12\n",
-      "31644    | torch.Size([1, 1000, 762]) | 12\n",
-      "31645    | torch.Size([1, 1000, 754]) | 12\n",
-      "31646    | torch.Size([1, 1000, 754]) | 12\n",
-      "31647    | torch.Size([1, 1000, 762]) | 12\n",
-      "31648    | torch.Size([1, 1000, 754]) | 12\n",
-      "31649    | torch.Size([1, 1000, 754]) | 12\n",
-      "31650    | torch.Size([1, 1000, 809]) | 12\n",
-      "31651    | torch.Size([1, 1000, 754]) | 12\n",
-      "31652    | torch.Size([1, 1000, 754]) | 12\n",
-      "31653    | torch.Size([1, 1000, 754]) | 12\n",
-      "31654    | torch.Size([1, 1000, 754]) | 12\n",
-      "31655    | torch.Size([1, 1000, 754]) | 12\n",
-      "31656    | torch.Size([1, 1000, 754]) | 12\n",
-      "31657    | torch.Size([1, 1000, 754]) | 12\n",
-      "31658    | torch.Size([1, 1000, 754]) | 12\n",
-      "31659    | torch.Size([1, 1000, 754]) | 12\n",
-      "31660    | torch.Size([1, 1000, 754]) | 12\n",
-      "31661    | torch.Size([1, 1000, 754]) | 12\n",
-      "31662    | torch.Size([1, 1000, 754]) | 12\n",
-      "31663    | torch.Size([1, 1000, 754]) | 12\n",
-      "31664    | torch.Size([1, 1000, 754]) | 12\n",
-      "31665    | torch.Size([1, 1000, 754]) | 12\n",
-      "31666    | torch.Size([1, 1000, 754]) | 12\n",
-      "31667    | torch.Size([1, 1000, 754]) | 12\n",
-      "31668    | torch.Size([1, 1000, 754]) | 12\n",
-      "31669    | torch.Size([1, 1000, 777]) | 12\n",
-      "31670    | torch.Size([1, 1000, 754]) | 12\n",
-      "31671    | torch.Size([1, 1000, 754]) | 12\n",
-      "31672    | torch.Size([1, 1000, 754]) | 12\n",
-      "31673    | torch.Size([1, 1000, 754]) | 12\n",
-      "31674    | torch.Size([1, 1000, 754]) | 12\n",
-      "31675    | torch.Size([1, 1000, 754]) | 12\n",
-      "31676    | torch.Size([1, 1000, 747]) | 12\n",
-      "31677    | torch.Size([1, 1000, 754]) | 12\n",
-      "31678    | torch.Size([1, 1000, 754]) | 12\n",
-      "31679    | torch.Size([1, 1000, 754]) | 12\n",
-      "31680    | torch.Size([1, 1000, 754]) | 12\n",
-      "31681    | torch.Size([1, 1000, 754]) | 12\n",
-      "31682    | torch.Size([1, 1000, 754]) | 12\n",
-      "31683    | torch.Size([1, 1000, 791]) | 12\n",
-      "31684    | torch.Size([1, 1000, 754]) | 12\n",
-      "31685    | torch.Size([1, 1000, 754]) | 12\n",
-      "31686    | torch.Size([1, 1000, 797]) | 12\n",
-      "31687    | torch.Size([1, 1000, 754]) | 12\n",
-      "31688    | torch.Size([1, 1000, 802]) | 12\n",
-      "31689    | torch.Size([1, 1000, 754]) | 12\n",
-      "31690    | torch.Size([1, 1000, 805]) | 12\n",
-      "31691    | torch.Size([1, 1000, 777]) | 12\n",
-      "31692    | torch.Size([1, 1000, 754]) | 12\n",
-      "31693    | torch.Size([1, 1000, 754]) | 12\n",
-      "31694    | torch.Size([1, 1000, 754]) | 12\n",
-      "31695    | torch.Size([1, 1000, 754]) | 12\n",
-      "31696    | torch.Size([1, 1000, 817]) | 12\n",
-      "31697    | torch.Size([1, 1000, 754]) | 12\n",
-      "31698    | torch.Size([1, 1000, 754]) | 12\n",
-      "31699    | torch.Size([1, 1000, 754]) | 12\n",
-      "31700    | torch.Size([1, 1000, 754]) | 12\n",
-      "31701    | torch.Size([1, 1000, 754]) | 12\n",
-      "31702    | torch.Size([1, 1000, 754]) | 12\n",
-      "31703    | torch.Size([1, 1000, 772]) | 12\n",
-      "31704    | torch.Size([1, 1000, 771]) | 12\n",
-      "31705    | torch.Size([1, 1000, 754]) | 12\n",
-      "31706    | torch.Size([1, 1000, 754]) | 12\n",
-      "31707    | torch.Size([1, 1000, 797]) | 12\n",
-      "31708    | torch.Size([1, 1000, 754]) | 12\n",
-      "31709    | torch.Size([1, 1000, 754]) | 12\n",
-      "31710    | torch.Size([1, 1000, 788]) | 12\n",
-      "31711    | torch.Size([1, 1000, 754]) | 12\n",
-      "31712    | torch.Size([1, 1000, 754]) | 12\n",
-      "31713    | torch.Size([1, 1000, 777]) | 12\n",
-      "31714    | torch.Size([1, 1000, 754]) | 12\n",
-      "31715    | torch.Size([1, 1000, 754]) | 12\n",
-      "31716    | torch.Size([1, 1000, 754]) | 12\n",
-      "31717    | torch.Size([1, 1000, 754]) | 12\n",
-      "31718    | torch.Size([1, 1000, 754]) | 12\n",
-      "31719    | torch.Size([1, 1000, 754]) | 12\n",
-      "31720    | torch.Size([1, 1000, 754]) | 12\n",
-      "31721    | torch.Size([1, 1000, 762]) | 12\n",
-      "31722    | torch.Size([1, 1000, 771]) | 12\n",
-      "31723    | torch.Size([1, 1000, 754]) | 12\n",
-      "31724    | torch.Size([1, 1000, 754]) | 12\n",
-      "31725    | torch.Size([1, 1000, 754]) | 12\n",
-      "31726    | torch.Size([1, 1000, 754]) | 12\n",
-      "31727    | torch.Size([1, 1000, 754]) | 12\n",
-      "31728    | torch.Size([1, 1000, 754]) | 12\n",
-      "31729    | torch.Size([1, 1000, 754]) | 12\n",
-      "31730    | torch.Size([1, 1000, 754]) | 12\n",
-      "31731    | torch.Size([1, 1000, 754]) | 12\n",
-      "31732    | torch.Size([1, 1000, 754]) | 12\n",
-      "31733    | torch.Size([1, 1000, 754]) | 12\n",
-      "31734    | torch.Size([1, 1000, 754]) | 12\n",
-      "31735    | torch.Size([1, 1000, 780]) | 12\n",
-      "31736    | torch.Size([1, 1000, 771]) | 12\n",
-      "31737    | torch.Size([1, 1000, 809]) | 12\n",
-      "31738    | torch.Size([1, 1000, 754]) | 12\n",
-      "31739    | torch.Size([1, 1000, 772]) | 12\n",
-      "31740    | torch.Size([1, 1000, 754]) | 12\n",
-      "31741    | torch.Size([1, 1000, 754]) | 12\n",
-      "31742    | torch.Size([1, 1000, 754]) | 12\n",
-      "31743    | torch.Size([1, 1000, 786]) | 12\n",
-      "31744    | torch.Size([1, 1000, 754]) | 12\n",
-      "31745    | torch.Size([1, 1000, 754]) | 12\n",
-      "31746    | torch.Size([1, 1000, 754]) | 12\n",
-      "31747    | torch.Size([1, 1000, 774]) | 12\n",
-      "31748    | torch.Size([1, 1000, 754]) | 12\n",
-      "31749    | torch.Size([1, 1000, 754]) | 12\n",
-      "31750    | torch.Size([1, 1000, 754]) | 12\n",
-      "31751    | torch.Size([1, 1000, 754]) | 12\n",
-      "31752    | torch.Size([1, 1000, 754]) | 12\n",
-      "31753    | torch.Size([1, 1000, 754]) | 12\n",
-      "31754    | torch.Size([1, 1000, 754]) | 12\n",
-      "31755    | torch.Size([1, 1000, 754]) | 12\n",
-      "31756    | torch.Size([1, 1000, 754]) | 12\n",
-      "31757    | torch.Size([1, 1000, 774]) | 12\n",
-      "31758    | torch.Size([1, 1000, 754]) | 12\n",
-      "31759    | torch.Size([1, 1000, 754]) | 12\n",
-      "31760    | torch.Size([1, 1000, 754]) | 12\n",
-      "31761    | torch.Size([1, 1000, 754]) | 12\n",
-      "31762    | torch.Size([1, 1000, 754]) | 12\n",
-      "31763    | torch.Size([1, 1000, 754]) | 12\n",
-      "31764    | torch.Size([1, 1000, 754]) | 12\n",
-      "31765    | torch.Size([1, 1000, 754]) | 12\n",
-      "31766    | torch.Size([1, 1000, 754]) | 12\n",
-      "31767    | torch.Size([1, 1000, 772]) | 12\n",
-      "31768    | torch.Size([1, 1000, 754]) | 12\n",
-      "31769    | torch.Size([1, 1000, 754]) | 12\n",
-      "31770    | torch.Size([1, 1000, 754]) | 12\n",
-      "31771    | torch.Size([1, 1000, 754]) | 12\n",
-      "31772    | torch.Size([1, 1000, 754]) | 12\n",
-      "31773    | torch.Size([1, 1000, 754]) | 12\n",
-      "31774    | torch.Size([1, 1000, 754]) | 12\n",
-      "31775    | torch.Size([1, 1000, 754]) | 12\n",
-      "31776    | torch.Size([1, 1000, 754]) | 12\n",
-      "31777    | torch.Size([1, 1000, 754]) | 12\n",
-      "31778    | torch.Size([1, 1000, 754]) | 12\n",
-      "31779    | torch.Size([1, 1000, 754]) | 12\n",
-      "31780    | torch.Size([1, 1000, 754]) | 12\n",
-      "31781    | torch.Size([1, 1000, 801]) | 12\n",
-      "31782    | torch.Size([1, 1000, 754]) | 12\n",
-      "31783    | torch.Size([1, 1000, 754]) | 12\n",
-      "31784    | torch.Size([1, 1000, 795]) | 12\n",
-      "31785    | torch.Size([1, 1000, 754]) | 12\n",
-      "31786    | torch.Size([1, 1000, 754]) | 12\n",
-      "31787    | torch.Size([1, 1000, 762]) | 12\n",
-      "31788    | torch.Size([1, 1000, 754]) | 12\n",
-      "31789    | torch.Size([1, 1000, 754]) | 12\n",
-      "31790    | torch.Size([1, 1000, 754]) | 12\n",
-      "31791    | torch.Size([1, 1000, 754]) | 12\n",
-      "31792    | torch.Size([1, 1000, 754]) | 12\n",
-      "31793    | torch.Size([1, 1000, 754]) | 12\n",
-      "31794    | torch.Size([1, 1000, 754]) | 12\n",
-      "31795    | torch.Size([1, 1000, 754]) | 12\n",
-      "31796    | torch.Size([1, 1000, 754]) | 12\n",
-      "31797    | torch.Size([1, 1000, 754]) | 12\n",
-      "31798    | torch.Size([1, 1000, 754]) | 12\n",
-      "31799    | torch.Size([1, 1000, 754]) | 12\n",
-      "31800    | torch.Size([1, 1000, 754]) | 12\n",
-      "31801    | torch.Size([1, 1000, 754]) | 12\n",
-      "31802    | torch.Size([1, 1000, 754]) | 12\n",
-      "31803    | torch.Size([1, 1000, 754]) | 12\n",
-      "31804    | torch.Size([1, 1000, 754]) | 12\n",
-      "31805    | torch.Size([1, 1000, 754]) | 12\n",
-      "31806    | torch.Size([1, 1000, 754]) | 12\n",
-      "31807    | torch.Size([1, 1000, 754]) | 12\n",
-      "31808    | torch.Size([1, 1000, 754]) | 12\n",
-      "31809    | torch.Size([1, 1000, 754]) | 12\n",
-      "31810    | torch.Size([1, 1000, 754]) | 12\n",
-      "31811    | torch.Size([1, 1000, 754]) | 12\n",
-      "31812    | torch.Size([1, 1000, 754]) | 12\n",
-      "31813    | torch.Size([1, 1000, 777]) | 12\n",
-      "31814    | torch.Size([1, 1000, 754]) | 12\n",
-      "31815    | torch.Size([1, 1000, 754]) | 12\n",
-      "31816    | torch.Size([1, 1000, 771]) | 12\n",
-      "31817    | torch.Size([1, 1000, 754]) | 12\n",
-      "31818    | torch.Size([1, 1000, 754]) | 12\n",
-      "31819    | torch.Size([1, 1000, 754]) | 12\n",
-      "31820    | torch.Size([1, 1000, 754]) | 12\n",
-      "31821    | torch.Size([1, 1000, 780]) | 12\n",
-      "31822    | torch.Size([1, 1000, 754]) | 12\n",
-      "31823    | torch.Size([1, 1000, 754]) | 12\n",
-      "31824    | torch.Size([1, 1000, 754]) | 12\n",
-      "31825    | torch.Size([1, 1000, 754]) | 12\n",
-      "31826    | torch.Size([1, 1000, 754]) | 12\n",
-      "31827    | torch.Size([1, 1000, 754]) | 12\n",
-      "31828    | torch.Size([1, 1000, 754]) | 12\n",
-      "31829    | torch.Size([1, 1000, 754]) | 12\n",
-      "31830    | torch.Size([1, 1000, 754]) | 12\n",
-      "31831    | torch.Size([1, 1000, 754]) | 12\n",
-      "31832    | torch.Size([1, 1000, 754]) | 12\n",
-      "31833    | torch.Size([1, 1000, 754]) | 12\n",
-      "31834    | torch.Size([1, 1000, 754]) | 12\n",
-      "31835    | torch.Size([1, 1000, 754]) | 12\n",
-      "31836    | torch.Size([1, 1000, 754]) | 12\n",
-      "31837    | torch.Size([1, 1000, 754]) | 12\n",
-      "31838    | torch.Size([1, 1000, 754]) | 12\n",
-      "31839    | torch.Size([1, 1000, 754]) | 12\n",
-      "31840    | torch.Size([1, 1000, 754]) | 12\n",
-      "31841    | torch.Size([1, 1000, 770]) | 12\n",
-      "31842    | torch.Size([1, 1000, 754]) | 12\n",
-      "31843    | torch.Size([1, 1000, 754]) | 12\n",
-      "31844    | torch.Size([1, 1000, 754]) | 12\n",
-      "31845    | torch.Size([1, 1000, 754]) | 12\n",
-      "31846    | torch.Size([1, 1000, 754]) | 12\n",
-      "31847    | torch.Size([1, 1000, 754]) | 12\n",
-      "31848    | torch.Size([1, 1000, 754]) | 12\n",
-      "31849    | torch.Size([1, 1000, 754]) | 12\n",
-      "31850    | torch.Size([1, 1000, 754]) | 12\n",
-      "31851    | torch.Size([1, 1000, 754]) | 12\n",
-      "31852    | torch.Size([1, 1000, 772]) | 12\n",
-      "31853    | torch.Size([1, 1000, 754]) | 12\n",
-      "31854    | torch.Size([1, 1000, 754]) | 12\n",
-      "31855    | torch.Size([1, 1000, 754]) | 12\n",
-      "31856    | torch.Size([1, 1000, 754]) | 12\n",
-      "31857    | torch.Size([1, 1000, 754]) | 12\n",
-      "31858    | torch.Size([1, 1000, 754]) | 12\n",
-      "31859    | torch.Size([1, 1000, 754]) | 12\n",
-      "31860    | torch.Size([1, 1000, 754]) | 12\n",
-      "31861    | torch.Size([1, 1000, 754]) | 12\n",
-      "31862    | torch.Size([1, 1000, 754]) | 12\n",
-      "31863    | torch.Size([1, 1000, 754]) | 12\n",
-      "31864    | torch.Size([1, 1000, 754]) | 12\n",
-      "31865    | torch.Size([1, 1000, 772]) | 12\n",
-      "31866    | torch.Size([1, 1000, 781]) | 12\n",
-      "31867    | torch.Size([1, 1000, 754]) | 12\n",
-      "31868    | torch.Size([1, 1000, 754]) | 12\n",
-      "31869    | torch.Size([1, 1000, 754]) | 12\n",
-      "31870    | torch.Size([1, 1000, 754]) | 12\n",
-      "31871    | torch.Size([1, 1000, 768]) | 12\n",
-      "31872    | torch.Size([1, 1000, 771]) | 12\n",
-      "31873    | torch.Size([1, 1000, 754]) | 12\n",
-      "31874    | torch.Size([1, 1000, 763]) | 12\n",
-      "31875    | torch.Size([1, 1000, 783]) | 12\n",
-      "31876    | torch.Size([1, 1000, 754]) | 12\n",
-      "31877    | torch.Size([1, 1000, 754]) | 12\n",
-      "31878    | torch.Size([1, 1000, 754]) | 12\n",
-      "31879    | torch.Size([1, 1000, 754]) | 12\n",
-      "31880    | torch.Size([1, 1000, 754]) | 12\n",
-      "31881    | torch.Size([1, 1000, 754]) | 12\n",
-      "31882    | torch.Size([1, 1000, 754]) | 12\n",
-      "31883    | torch.Size([1, 1000, 754]) | 12\n",
-      "31884    | torch.Size([1, 1000, 754]) | 12\n",
-      "31885    | torch.Size([1, 1000, 754]) | 12\n",
-      "31886    | torch.Size([1, 1000, 754]) | 12\n",
-      "31887    | torch.Size([1, 1000, 754]) | 12\n",
-      "31888    | torch.Size([1, 1000, 754]) | 12\n",
-      "31889    | torch.Size([1, 1000, 754]) | 12\n",
-      "31890    | torch.Size([1, 1000, 754]) | 12\n",
-      "31891    | torch.Size([1, 1000, 754]) | 12\n",
-      "31892    | torch.Size([1, 1000, 754]) | 12\n",
-      "31893    | torch.Size([1, 1000, 754]) | 12\n",
-      "31894    | torch.Size([1, 1000, 772]) | 12\n",
-      "31895    | torch.Size([1, 1000, 754]) | 12\n",
-      "31896    | torch.Size([1, 1000, 754]) | 12\n",
-      "31897    | torch.Size([1, 1000, 762]) | 12\n",
-      "31898    | torch.Size([1, 1000, 754]) | 12\n",
-      "31899    | torch.Size([1, 1000, 754]) | 12\n",
-      "31900    | torch.Size([1, 1000, 754]) | 12\n",
-      "31901    | torch.Size([1, 1000, 754]) | 12\n",
-      "31902    | torch.Size([1, 1000, 754]) | 12\n",
-      "31903    | torch.Size([1, 1000, 754]) | 12\n",
-      "31904    | torch.Size([1, 1000, 754]) | 12\n",
-      "31905    | torch.Size([1, 1000, 754]) | 12\n",
-      "31906    | torch.Size([1, 1000, 754]) | 12\n",
-      "31907    | torch.Size([1, 1000, 754]) | 12\n",
-      "31908    | torch.Size([1, 1000, 768]) | 12\n",
-      "31909    | torch.Size([1, 1000, 774]) | 12\n",
-      "31910    | torch.Size([1, 1000, 754]) | 12\n",
-      "31911    | torch.Size([1, 1000, 754]) | 12\n",
-      "31912    | torch.Size([1, 1000, 788]) | 12\n",
-      "31913    | torch.Size([1, 1000, 754]) | 12\n",
-      "31914    | torch.Size([1, 1000, 754]) | 12\n",
-      "31915    | torch.Size([1, 1000, 754]) | 12\n",
-      "31916    | torch.Size([1, 1000, 754]) | 12\n",
-      "31917    | torch.Size([1, 1000, 754]) | 12\n",
-      "31918    | torch.Size([1, 1000, 754]) | 12\n",
-      "31919    | torch.Size([1, 1000, 754]) | 12\n",
-      "31920    | torch.Size([1, 1000, 754]) | 12\n",
-      "31921    | torch.Size([1, 1000, 754]) | 12\n",
-      "31922    | torch.Size([1, 1000, 754]) | 12\n",
-      "31923    | torch.Size([1, 1000, 754]) | 12\n",
-      "31924    | torch.Size([1, 1000, 754]) | 12\n",
-      "31925    | torch.Size([1, 1000, 754]) | 12\n",
-      "31926    | torch.Size([1, 1000, 800]) | 12\n",
-      "31927    | torch.Size([1, 1000, 754]) | 12\n",
-      "31928    | torch.Size([1, 1000, 754]) | 12\n",
-      "31929    | torch.Size([1, 1000, 754]) | 12\n",
-      "31930    | torch.Size([1, 1000, 754]) | 12\n",
-      "31931    | torch.Size([1, 1000, 754]) | 12\n",
-      "31932    | torch.Size([1, 1000, 754]) | 12\n",
-      "31933    | torch.Size([1, 1000, 754]) | 12\n",
-      "31934    | torch.Size([1, 1000, 754]) | 12\n",
-      "31935    | torch.Size([1, 1000, 754]) | 12\n",
-      "31936    | torch.Size([1, 1000, 754]) | 12\n",
-      "31937    | torch.Size([1, 1000, 754]) | 12\n",
-      "31938    | torch.Size([1, 1000, 754]) | 12\n",
-      "31939    | torch.Size([1, 1000, 754]) | 12\n",
-      "31940    | torch.Size([1, 1000, 754]) | 12\n",
-      "31941    | torch.Size([1, 1000, 754]) | 12\n",
-      "31942    | torch.Size([1, 1000, 754]) | 12\n",
-      "31943    | torch.Size([1, 1000, 754]) | 12\n",
-      "31944    | torch.Size([1, 1000, 779]) | 12\n",
-      "31945    | torch.Size([1, 1000, 754]) | 12\n",
-      "31946    | torch.Size([1, 1000, 754]) | 12\n",
-      "31947    | torch.Size([1, 1000, 754]) | 12\n",
-      "31948    | torch.Size([1, 1000, 754]) | 12\n",
-      "31949    | torch.Size([1, 1000, 754]) | 12\n",
-      "31950    | torch.Size([1, 1000, 772]) | 12\n",
-      "31951    | torch.Size([1, 1000, 754]) | 12\n",
-      "31952    | torch.Size([1, 1000, 754]) | 12\n",
-      "31953    | torch.Size([1, 1000, 772]) | 12\n",
-      "31954    | torch.Size([1, 1000, 754]) | 12\n",
-      "31955    | torch.Size([1, 1000, 754]) | 12\n",
-      "31956    | torch.Size([1, 1000, 754]) | 12\n",
-      "31957    | torch.Size([1, 1000, 754]) | 12\n",
-      "31958    | torch.Size([1, 1000, 754]) | 12\n",
-      "31959    | torch.Size([1, 1000, 754]) | 12\n",
-      "31960    | torch.Size([1, 1000, 754]) | 12\n",
-      "31961    | torch.Size([1, 1000, 754]) | 12\n",
-      "31962    | torch.Size([1, 1000, 754]) | 12\n",
-      "31963    | torch.Size([1, 1000, 754]) | 12\n",
-      "31964    | torch.Size([1, 1000, 754]) | 12\n",
-      "31965    | torch.Size([1, 1000, 754]) | 12\n",
-      "31966    | torch.Size([1, 1000, 754]) | 12\n",
-      "31967    | torch.Size([1, 1000, 754]) | 12\n",
-      "31968    | torch.Size([1, 1000, 754]) | 12\n",
-      "31969    | torch.Size([1, 1000, 754]) | 12\n",
-      "31970    | torch.Size([1, 1000, 754]) | 12\n",
-      "31971    | torch.Size([1, 1000, 777]) | 12\n",
-      "31972    | torch.Size([1, 1000, 754]) | 12\n",
-      "31973    | torch.Size([1, 1000, 754]) | 12\n",
-      "31974    | torch.Size([1, 1000, 754]) | 12\n",
-      "31975    | torch.Size([1, 1000, 754]) | 12\n",
-      "31976    | torch.Size([1, 1000, 754]) | 12\n",
-      "31977    | torch.Size([1, 1000, 794]) | 12\n",
-      "31978    | torch.Size([1, 1000, 754]) | 12\n",
-      "31979    | torch.Size([1, 1000, 754]) | 12\n",
-      "31980    | torch.Size([1, 1000, 799]) | 12\n",
-      "31981    | torch.Size([1, 1000, 754]) | 12\n",
-      "31982    | torch.Size([1, 1000, 754]) | 12\n",
-      "31983    | torch.Size([1, 1000, 754]) | 12\n",
-      "31984    | torch.Size([1, 1000, 754]) | 12\n",
-      "31985    | torch.Size([1, 1000, 754]) | 12\n",
-      "31986    | torch.Size([1, 1000, 754]) | 12\n",
-      "31987    | torch.Size([1, 1000, 754]) | 12\n",
-      "31988    | torch.Size([1, 1000, 754]) | 12\n",
-      "31989    | torch.Size([1, 1000, 754]) | 12\n",
-      "31990    | torch.Size([1, 1000, 754]) | 12\n",
-      "31991    | torch.Size([1, 1000, 767]) | 12\n",
-      "31992    | torch.Size([1, 1000, 754]) | 12\n",
-      "31993    | torch.Size([1, 1000, 754]) | 12\n",
-      "31994    | torch.Size([1, 1000, 754]) | 12\n",
-      "31995    | torch.Size([1, 1000, 754]) | 12\n",
-      "31996    | torch.Size([1, 1000, 754]) | 12\n",
-      "31997    | torch.Size([1, 1000, 762]) | 12\n",
-      "31998    | torch.Size([1, 1000, 754]) | 12\n",
-      "31999    | torch.Size([1, 1000, 754]) | 12\n",
-      "32000    | torch.Size([1, 1000, 754]) | 12\n",
-      "32001    | torch.Size([1, 1000, 754]) | 12\n",
-      "32002    | torch.Size([1, 1000, 754]) | 12\n",
-      "32003    | torch.Size([1, 1000, 812]) | 12\n",
-      "32004    | torch.Size([1, 1000, 754]) | 12\n",
-      "32005    | torch.Size([1, 1000, 772]) | 12\n",
-      "32006    | torch.Size([1, 1000, 754]) | 12\n",
-      "32007    | torch.Size([1, 1000, 754]) | 12\n",
-      "32008    | torch.Size([1, 1000, 754]) | 12\n",
-      "32009    | torch.Size([1, 1000, 754]) | 12\n",
-      "32010    | torch.Size([1, 1000, 754]) | 12\n",
-      "32011    | torch.Size([1, 1000, 754]) | 12\n",
-      "32012    | torch.Size([1, 1000, 754]) | 12\n",
-      "32013    | torch.Size([1, 1000, 754]) | 12\n",
-      "32014    | torch.Size([1, 1000, 754]) | 12\n",
-      "32015    | torch.Size([1, 1000, 767]) | 12\n",
-      "32016    | torch.Size([1, 1000, 754]) | 12\n",
-      "32017    | torch.Size([1, 1000, 754]) | 12\n",
-      "32018    | torch.Size([1, 1000, 754]) | 12\n",
-      "32019    | torch.Size([1, 1000, 754]) | 12\n",
-      "32020    | torch.Size([1, 1000, 754]) | 12\n",
-      "32021    | torch.Size([1, 1000, 754]) | 12\n",
-      "32022    | torch.Size([1, 1000, 754]) | 12\n",
-      "32023    | torch.Size([1, 1000, 754]) | 12\n",
-      "32024    | torch.Size([1, 1000, 754]) | 12\n",
-      "32025    | torch.Size([1, 1000, 754]) | 12\n",
-      "32026    | torch.Size([1, 1000, 754]) | 12\n",
-      "32027    | torch.Size([1, 1000, 754]) | 12\n",
-      "32028    | torch.Size([1, 1000, 754]) | 12\n",
-      "32029    | torch.Size([1, 1000, 754]) | 12\n",
-      "32030    | torch.Size([1, 1000, 778]) | 12\n",
-      "32031    | torch.Size([1, 1000, 754]) | 12\n",
-      "32032    | torch.Size([1, 1000, 754]) | 12\n",
-      "32033    | torch.Size([1, 1000, 754]) | 12\n",
-      "32034    | torch.Size([1, 1000, 754]) | 12\n",
-      "32035    | torch.Size([1, 1000, 754]) | 12\n",
-      "32036    | torch.Size([1, 1000, 772]) | 12\n",
-      "32037    | torch.Size([1, 1000, 754]) | 12\n",
-      "32038    | torch.Size([1, 1000, 754]) | 12\n",
-      "32039    | torch.Size([1, 1000, 754]) | 12\n",
-      "32040    | torch.Size([1, 1000, 754]) | 12\n",
-      "32041    | torch.Size([1, 1000, 754]) | 12\n",
-      "32042    | torch.Size([1, 1000, 754]) | 12\n",
-      "32043    | torch.Size([1, 1000, 754]) | 12\n",
-      "32044    | torch.Size([1, 1000, 754]) | 12\n",
-      "32045    | torch.Size([1, 1000, 754]) | 12\n",
-      "32046    | torch.Size([1, 1000, 754]) | 12\n",
-      "32047    | torch.Size([1, 1000, 754]) | 12\n",
-      "32048    | torch.Size([1, 1000, 754]) | 12\n",
-      "32049    | torch.Size([1, 1000, 754]) | 12\n",
-      "32050    | torch.Size([1, 1000, 762]) | 12\n",
-      "32051    | torch.Size([1, 1000, 754]) | 12\n",
-      "32052    | torch.Size([1, 1000, 754]) | 12\n",
-      "32053    | torch.Size([1, 1000, 754]) | 12\n",
-      "32054    | torch.Size([1, 1000, 754]) | 12\n",
-      "32055    | torch.Size([1, 1000, 754]) | 12\n",
-      "32056    | torch.Size([1, 1000, 754]) | 12\n",
-      "32057    | torch.Size([1, 1000, 754]) | 12\n",
-      "32058    | torch.Size([1, 1000, 754]) | 12\n",
-      "32059    | torch.Size([1, 1000, 754]) | 12\n",
-      "32060    | torch.Size([1, 1000, 754]) | 12\n",
-      "32061    | torch.Size([1, 1000, 754]) | 12\n",
-      "32062    | torch.Size([1, 1000, 754]) | 12\n",
-      "32063    | torch.Size([1, 1000, 772]) | 12\n",
-      "32064    | torch.Size([1, 1000, 754]) | 12\n",
-      "32065    | torch.Size([1, 1000, 754]) | 12\n",
-      "32066    | torch.Size([1, 1000, 754]) | 12\n",
-      "32067    | torch.Size([1, 1000, 754]) | 12\n",
-      "32068    | torch.Size([1, 1000, 778]) | 12\n",
-      "32069    | torch.Size([1, 1000, 754]) | 12\n",
-      "32070    | torch.Size([1, 1000, 754]) | 12\n",
-      "32071    | torch.Size([1, 1000, 771]) | 12\n",
-      "32072    | torch.Size([1, 1000, 754]) | 12\n",
-      "32073    | torch.Size([1, 1000, 767]) | 12\n",
-      "32074    | torch.Size([1, 1000, 754]) | 12\n",
-      "32075    | torch.Size([1, 1000, 754]) | 12\n",
-      "32076    | torch.Size([1, 1000, 754]) | 12\n",
-      "32077    | torch.Size([1, 1000, 754]) | 12\n",
-      "32078    | torch.Size([1, 1000, 754]) | 12\n",
-      "32079    | torch.Size([1, 1000, 754]) | 12\n",
-      "32080    | torch.Size([1, 1000, 757]) | 12\n",
-      "32081    | torch.Size([1, 1000, 770]) | 12\n",
-      "32082    | torch.Size([1, 1000, 754]) | 12\n",
-      "32083    | torch.Size([1, 1000, 754]) | 12\n",
-      "32084    | torch.Size([1, 1000, 754]) | 12\n",
-      "32085    | torch.Size([1, 1000, 754]) | 12\n",
-      "32086    | torch.Size([1, 1000, 754]) | 12\n",
-      "32087    | torch.Size([1, 1000, 754]) | 12\n",
-      "32088    | torch.Size([1, 1000, 754]) | 12\n",
-      "32089    | torch.Size([1, 1000, 754]) | 12\n",
-      "32090    | torch.Size([1, 1000, 754]) | 12\n",
-      "32091    | torch.Size([1, 1000, 754]) | 12\n",
-      "32092    | torch.Size([1, 1000, 754]) | 12\n",
-      "32093    | torch.Size([1, 1000, 772]) | 12\n",
-      "32094    | torch.Size([1, 1000, 754]) | 12\n",
-      "32095    | torch.Size([1, 1000, 754]) | 12\n",
-      "32096    | torch.Size([1, 1000, 754]) | 12\n",
-      "32097    | torch.Size([1, 1000, 754]) | 12\n",
-      "32098    | torch.Size([1, 1000, 772]) | 12\n",
-      "32099    | torch.Size([1, 1000, 754]) | 12\n",
-      "32100    | torch.Size([1, 1000, 754]) | 12\n",
-      "32101    | torch.Size([1, 1000, 754]) | 12\n",
-      "32102    | torch.Size([1, 1000, 754]) | 12\n",
-      "32103    | torch.Size([1, 1000, 754]) | 12\n",
-      "32104    | torch.Size([1, 1000, 754]) | 12\n",
-      "32105    | torch.Size([1, 1000, 754]) | 12\n",
-      "32106    | torch.Size([1, 1000, 754]) | 12\n",
-      "32107    | torch.Size([1, 1000, 754]) | 12\n",
-      "32108    | torch.Size([1, 1000, 754]) | 12\n",
-      "32109    | torch.Size([1, 1000, 754]) | 12\n",
-      "32110    | torch.Size([1, 1000, 754]) | 12\n",
-      "32111    | torch.Size([1, 1000, 767]) | 12\n",
-      "32112    | torch.Size([1, 1000, 754]) | 12\n",
-      "32113    | torch.Size([1, 1000, 754]) | 12\n",
-      "32114    | torch.Size([1, 1000, 754]) | 12\n",
-      "32115    | torch.Size([1, 1000, 754]) | 12\n",
-      "32116    | torch.Size([1, 1000, 771]) | 12\n",
-      "32117    | torch.Size([1, 1000, 772]) | 12\n",
-      "32118    | torch.Size([1, 1000, 754]) | 12\n",
-      "32119    | torch.Size([1, 1000, 754]) | 12\n",
-      "32120    | torch.Size([1, 1000, 812]) | 12\n",
-      "32121    | torch.Size([1, 1000, 754]) | 12\n",
-      "32122    | torch.Size([1, 1000, 754]) | 12\n",
-      "32123    | torch.Size([1, 1000, 754]) | 12\n",
-      "32124    | torch.Size([1, 1000, 754]) | 12\n",
-      "32125    | torch.Size([1, 1000, 754]) | 12\n",
-      "32126    | torch.Size([1, 1000, 754]) | 12\n",
-      "32127    | torch.Size([1, 1000, 754]) | 12\n",
-      "32128    | torch.Size([1, 1000, 754]) | 12\n",
-      "32129    | torch.Size([1, 1000, 772]) | 12\n",
-      "32130    | torch.Size([1, 1000, 754]) | 12\n",
-      "32131    | torch.Size([1, 1000, 754]) | 12\n",
-      "32132    | torch.Size([1, 1000, 754]) | 12\n",
-      "32133    | torch.Size([1, 1000, 754]) | 12\n",
-      "32134    | torch.Size([1, 1000, 754]) | 12\n",
-      "32135    | torch.Size([1, 1000, 754]) | 12\n",
-      "32136    | torch.Size([1, 1000, 754]) | 12\n",
-      "32137    | torch.Size([1, 1000, 754]) | 12\n",
-      "32138    | torch.Size([1, 1000, 754]) | 12\n",
-      "32139    | torch.Size([1, 1000, 754]) | 12\n",
-      "32140    | torch.Size([1, 1000, 754]) | 12\n",
-      "32141    | torch.Size([1, 1000, 754]) | 12\n",
-      "32142    | torch.Size([1, 1000, 754]) | 12\n",
-      "32143    | torch.Size([1, 1000, 754]) | 12\n",
-      "32144    | torch.Size([1, 1000, 754]) | 12\n",
-      "32145    | torch.Size([1, 1000, 754]) | 12\n",
-      "32146    | torch.Size([1, 1000, 754]) | 12\n",
-      "32147    | torch.Size([1, 1000, 754]) | 12\n",
-      "32148    | torch.Size([1, 1000, 754]) | 12\n",
-      "32149    | torch.Size([1, 1000, 754]) | 12\n",
-      "32150    | torch.Size([1, 1000, 754]) | 12\n",
-      "32151    | torch.Size([1, 1000, 771]) | 12\n",
-      "32152    | torch.Size([1, 1000, 754]) | 12\n",
-      "32153    | torch.Size([1, 1000, 754]) | 12\n",
-      "32154    | torch.Size([1, 1000, 754]) | 12\n",
-      "32155    | torch.Size([1, 1000, 754]) | 12\n",
-      "32156    | torch.Size([1, 1000, 752]) | 12\n",
-      "32157    | torch.Size([1, 1000, 754]) | 12\n",
-      "32158    | torch.Size([1, 1000, 754]) | 12\n",
-      "32159    | torch.Size([1, 1000, 754]) | 12\n",
-      "32160    | torch.Size([1, 1000, 754]) | 12\n",
-      "32161    | torch.Size([1, 1000, 754]) | 12\n",
-      "32162    | torch.Size([1, 1000, 754]) | 12\n",
-      "32163    | torch.Size([1, 1000, 754]) | 12\n",
-      "32164    | torch.Size([1, 1000, 754]) | 12\n",
-      "32165    | torch.Size([1, 1000, 754]) | 12\n",
-      "32166    | torch.Size([1, 1000, 754]) | 12\n",
-      "32167    | torch.Size([1, 1000, 754]) | 12\n",
-      "32168    | torch.Size([1, 1000, 754]) | 12\n",
-      "32169    | torch.Size([1, 1000, 754]) | 12\n",
-      "32170    | torch.Size([1, 1000, 754]) | 12\n",
-      "32171    | torch.Size([1, 1000, 754]) | 12\n",
-      "32172    | torch.Size([1, 1000, 776]) | 12\n",
-      "32173    | torch.Size([1, 1000, 754]) | 12\n",
-      "32174    | torch.Size([1, 1000, 754]) | 12\n",
-      "32175    | torch.Size([1, 1000, 784]) | 12\n",
-      "32176    | torch.Size([1, 1000, 754]) | 12\n",
-      "32177    | torch.Size([1, 1000, 754]) | 12\n",
-      "32178    | torch.Size([1, 1000, 754]) | 12\n",
-      "32179    | torch.Size([1, 1000, 762]) | 12\n",
-      "32180    | torch.Size([1, 1000, 754]) | 12\n",
-      "32181    | torch.Size([1, 1000, 772]) | 12\n",
-      "32182    | torch.Size([1, 1000, 772]) | 12\n",
-      "32183    | torch.Size([1, 1000, 754]) | 12\n",
-      "32184    | torch.Size([1, 1000, 767]) | 12\n",
-      "32185    | torch.Size([1, 1000, 754]) | 12\n",
-      "32186    | torch.Size([1, 1000, 754]) | 12\n",
-      "32187    | torch.Size([1, 1000, 754]) | 12\n",
-      "32188    | torch.Size([1, 1000, 754]) | 12\n",
-      "32189    | torch.Size([1, 1000, 754]) | 12\n",
-      "32190    | torch.Size([1, 1000, 754]) | 12\n",
-      "32191    | torch.Size([1, 1000, 754]) | 12\n",
-      "32192    | torch.Size([1, 1000, 754]) | 12\n",
-      "32193    | torch.Size([1, 1000, 754]) | 12\n",
-      "32194    | torch.Size([1, 1000, 782]) | 12\n",
-      "32195    | torch.Size([1, 1000, 754]) | 12\n",
-      "32196    | torch.Size([1, 1000, 754]) | 12\n",
-      "32197    | torch.Size([1, 1000, 775]) | 12\n",
-      "32198    | torch.Size([1, 1000, 776]) | 12\n",
-      "32199    | torch.Size([1, 1000, 754]) | 12\n",
-      "32200    | torch.Size([1, 1000, 754]) | 12\n",
-      "32201    | torch.Size([1, 1000, 754]) | 12\n",
-      "32202    | torch.Size([1, 1000, 754]) | 12\n",
-      "32203    | torch.Size([1, 1000, 754]) | 12\n",
-      "32204    | torch.Size([1, 1000, 754]) | 12\n",
-      "32205    | torch.Size([1, 1000, 754]) | 12\n",
-      "32206    | torch.Size([1, 1000, 779]) | 12\n",
-      "32207    | torch.Size([1, 1000, 754]) | 12\n",
-      "32208    | torch.Size([1, 1000, 754]) | 12\n",
-      "32209    | torch.Size([1, 1000, 762]) | 12\n",
-      "32210    | torch.Size([1, 1000, 772]) | 12\n",
-      "32211    | torch.Size([1, 1000, 754]) | 12\n",
-      "32212    | torch.Size([1, 1000, 754]) | 12\n",
-      "32213    | torch.Size([1, 1000, 754]) | 12\n",
-      "32214    | torch.Size([1, 1000, 754]) | 12\n",
-      "32215    | torch.Size([1, 1000, 754]) | 12\n",
-      "32216    | torch.Size([1, 1000, 754]) | 12\n",
-      "32217    | torch.Size([1, 1000, 754]) | 12\n",
-      "32218    | torch.Size([1, 1000, 754]) | 12\n",
-      "32219    | torch.Size([1, 1000, 754]) | 12\n",
-      "32220    | torch.Size([1, 1000, 754]) | 12\n",
-      "32221    | torch.Size([1, 1000, 754]) | 12\n",
-      "32222    | torch.Size([1, 1000, 754]) | 12\n",
-      "32223    | torch.Size([1, 1000, 754]) | 12\n",
-      "32224    | torch.Size([1, 1000, 754]) | 12\n",
-      "32225    | torch.Size([1, 1000, 754]) | 12\n",
-      "32226    | torch.Size([1, 1000, 754]) | 12\n",
-      "32227    | torch.Size([1, 1000, 754]) | 12\n",
-      "32228    | torch.Size([1, 1000, 754]) | 12\n",
-      "32229    | torch.Size([1, 1000, 754]) | 12\n",
-      "32230    | torch.Size([1, 1000, 754]) | 12\n",
-      "32231    | torch.Size([1, 1000, 754]) | 12\n",
-      "32232    | torch.Size([1, 1000, 754]) | 12\n",
-      "32233    | torch.Size([1, 1000, 754]) | 12\n",
-      "32234    | torch.Size([1, 1000, 754]) | 12\n",
-      "32235    | torch.Size([1, 1000, 754]) | 12\n",
-      "32236    | torch.Size([1, 1000, 754]) | 12\n",
-      "32237    | torch.Size([1, 1000, 754]) | 12\n",
-      "32238    | torch.Size([1, 1000, 754]) | 12\n",
-      "32239    | torch.Size([1, 1000, 754]) | 12\n",
-      "32240    | torch.Size([1, 1000, 754]) | 12\n",
-      "32241    | torch.Size([1, 1000, 754]) | 12\n",
-      "32242    | torch.Size([1, 1000, 754]) | 12\n",
-      "32243    | torch.Size([1, 1000, 754]) | 12\n",
-      "32244    | torch.Size([1, 1000, 754]) | 12\n",
-      "32245    | torch.Size([1, 1000, 754]) | 12\n",
-      "32246    | torch.Size([1, 1000, 754]) | 12\n",
-      "32247    | torch.Size([1, 1000, 754]) | 12\n",
-      "32248    | torch.Size([1, 1000, 754]) | 12\n",
-      "32249    | torch.Size([1, 1000, 754]) | 12\n",
-      "32250    | torch.Size([1, 1000, 754]) | 12\n",
-      "32251    | torch.Size([1, 1000, 754]) | 12\n",
-      "32252    | torch.Size([1, 1000, 754]) | 12\n",
-      "32253    | torch.Size([1, 1000, 754]) | 12\n",
-      "32254    | torch.Size([1, 1000, 754]) | 12\n",
-      "32255    | torch.Size([1, 1000, 762]) | 12\n",
-      "32256    | torch.Size([1, 1000, 754]) | 12\n",
-      "32257    | torch.Size([1, 1000, 771]) | 12\n",
-      "32258    | torch.Size([1, 1000, 754]) | 12\n",
-      "32259    | torch.Size([1, 1000, 754]) | 12\n",
-      "32260    | torch.Size([1, 1000, 754]) | 12\n",
-      "32261    | torch.Size([1, 1000, 754]) | 12\n",
-      "32262    | torch.Size([1, 1000, 754]) | 12\n",
-      "32263    | torch.Size([1, 1000, 754]) | 12\n",
-      "32264    | torch.Size([1, 1000, 754]) | 12\n",
-      "32265    | torch.Size([1, 1000, 754]) | 12\n",
-      "32266    | torch.Size([1, 1000, 754]) | 12\n",
-      "32267    | torch.Size([1, 1000, 754]) | 12\n",
-      "32268    | torch.Size([1, 1000, 754]) | 12\n",
-      "32269    | torch.Size([1, 1000, 779]) | 12\n",
-      "32270    | torch.Size([1, 1000, 767]) | 12\n",
-      "32271    | torch.Size([1, 1000, 804]) | 12\n",
-      "32272    | torch.Size([1, 1000, 754]) | 12\n",
-      "32273    | torch.Size([1, 1000, 754]) | 12\n",
-      "32274    | torch.Size([1, 1000, 754]) | 12\n",
-      "32275    | torch.Size([1, 1000, 754]) | 12\n",
-      "32276    | torch.Size([1, 1000, 754]) | 12\n",
-      "32277    | torch.Size([1, 1000, 754]) | 12\n",
-      "32278    | torch.Size([1, 1000, 754]) | 12\n",
-      "32279    | torch.Size([1, 1000, 754]) | 12\n",
-      "32280    | torch.Size([1, 1000, 754]) | 12\n",
-      "32281    | torch.Size([1, 1000, 754]) | 12\n",
-      "32282    | torch.Size([1, 1000, 754]) | 12\n",
-      "32283    | torch.Size([1, 1000, 754]) | 12\n",
-      "32284    | torch.Size([1, 1000, 754]) | 12\n",
-      "32285    | torch.Size([1, 1000, 754]) | 12\n",
-      "32286    | torch.Size([1, 1000, 777]) | 12\n",
-      "32287    | torch.Size([1, 1000, 754]) | 12\n",
-      "32288    | torch.Size([1, 1000, 762]) | 12\n",
-      "32289    | torch.Size([1, 1000, 754]) | 12\n",
-      "32290    | torch.Size([1, 1000, 754]) | 12\n",
-      "32291    | torch.Size([1, 1000, 754]) | 12\n",
-      "32292    | torch.Size([1, 1000, 754]) | 12\n",
-      "32293    | torch.Size([1, 1000, 754]) | 12\n",
-      "32294    | torch.Size([1, 1000, 754]) | 12\n",
-      "32295    | torch.Size([1, 1000, 754]) | 12\n",
-      "32296    | torch.Size([1, 1000, 776]) | 12\n",
-      "32297    | torch.Size([1, 1000, 754]) | 12\n",
-      "32298    | torch.Size([1, 1000, 754]) | 12\n",
-      "32299    | torch.Size([1, 1000, 754]) | 12\n",
-      "32300    | torch.Size([1, 1000, 762]) | 12\n",
-      "32301    | torch.Size([1, 1000, 754]) | 12\n",
-      "32302    | torch.Size([1, 1000, 754]) | 12\n",
-      "32303    | torch.Size([1, 1000, 754]) | 12\n",
-      "32304    | torch.Size([1, 1000, 754]) | 12\n",
-      "32305    | torch.Size([1, 1000, 754]) | 12\n",
-      "32306    | torch.Size([1, 1000, 754]) | 12\n",
-      "32307    | torch.Size([1, 1000, 754]) | 12\n",
-      "32308    | torch.Size([1, 1000, 754]) | 12\n",
-      "32309    | torch.Size([1, 1000, 754]) | 12\n",
-      "32310    | torch.Size([1, 1000, 754]) | 12\n",
-      "32311    | torch.Size([1, 1000, 754]) | 12\n",
-      "32312    | torch.Size([1, 1000, 754]) | 12\n",
-      "32313    | torch.Size([1, 1000, 754]) | 12\n",
-      "32314    | torch.Size([1, 1000, 754]) | 12\n",
-      "32315    | torch.Size([1, 1000, 754]) | 12\n",
-      "32316    | torch.Size([1, 1000, 754]) | 12\n",
-      "32317    | torch.Size([1, 1000, 754]) | 12\n",
-      "32318    | torch.Size([1, 1000, 754]) | 12\n",
-      "32319    | torch.Size([1, 1000, 754]) | 12\n",
-      "32320    | torch.Size([1, 1000, 754]) | 12\n",
-      "32321    | torch.Size([1, 1000, 754]) | 12\n",
-      "32322    | torch.Size([1, 1000, 754]) | 12\n",
-      "32323    | torch.Size([1, 1000, 754]) | 12\n",
-      "32324    | torch.Size([1, 1000, 754]) | 12\n",
-      "32325    | torch.Size([1, 1000, 754]) | 12\n",
-      "32326    | torch.Size([1, 1000, 767]) | 12\n",
-      "32327    | torch.Size([1, 1000, 754]) | 12\n",
-      "32328    | torch.Size([1, 1000, 752]) | 12\n",
-      "32329    | torch.Size([1, 1000, 754]) | 12\n",
-      "32330    | torch.Size([1, 1000, 754]) | 12\n",
-      "32331    | torch.Size([1, 1000, 754]) | 12\n",
-      "32332    | torch.Size([1, 1000, 754]) | 12\n",
-      "32333    | torch.Size([1, 1000, 754]) | 12\n",
-      "32334    | torch.Size([1, 1000, 754]) | 12\n",
-      "32335    | torch.Size([1, 1000, 754]) | 12\n",
-      "32336    | torch.Size([1, 1000, 754]) | 12\n",
-      "32337    | torch.Size([1, 1000, 772]) | 12\n",
-      "32338    | torch.Size([1, 1000, 754]) | 12\n",
-      "32339    | torch.Size([1, 1000, 754]) | 12\n",
-      "32340    | torch.Size([1, 1000, 754]) | 12\n",
-      "32341    | torch.Size([1, 1000, 754]) | 12\n",
-      "32342    | torch.Size([1, 1000, 754]) | 12\n",
-      "32343    | torch.Size([1, 1000, 754]) | 12\n",
-      "32344    | torch.Size([1, 1000, 754]) | 12\n",
-      "32345    | torch.Size([1, 1000, 754]) | 12\n",
-      "32346    | torch.Size([1, 1000, 754]) | 12\n",
-      "32347    | torch.Size([1, 1000, 762]) | 12\n",
-      "32348    | torch.Size([1, 1000, 754]) | 12\n",
-      "32349    | torch.Size([1, 1000, 754]) | 12\n",
-      "32350    | torch.Size([1, 1000, 754]) | 12\n",
-      "32351    | torch.Size([1, 1000, 754]) | 12\n",
-      "32352    | torch.Size([1, 1000, 754]) | 12\n",
-      "32353    | torch.Size([1, 1000, 771]) | 12\n",
-      "32354    | torch.Size([1, 1000, 754]) | 12\n",
-      "32355    | torch.Size([1, 1000, 754]) | 12\n",
-      "32356    | torch.Size([1, 1000, 754]) | 12\n",
-      "32357    | torch.Size([1, 1000, 754]) | 12\n",
-      "32358    | torch.Size([1, 1000, 754]) | 12\n",
-      "32359    | torch.Size([1, 1000, 754]) | 12\n",
-      "32360    | torch.Size([1, 1000, 754]) | 12\n",
-      "32361    | torch.Size([1, 1000, 754]) | 12\n",
-      "32362    | torch.Size([1, 1000, 754]) | 12\n",
-      "32363    | torch.Size([1, 1000, 754]) | 12\n",
-      "32364    | torch.Size([1, 1000, 775]) | 12\n",
-      "32365    | torch.Size([1, 1000, 754]) | 12\n",
-      "32366    | torch.Size([1, 1000, 754]) | 12\n",
-      "32367    | torch.Size([1, 1000, 754]) | 12\n",
-      "32368    | torch.Size([1, 1000, 754]) | 12\n",
-      "32369    | torch.Size([1, 1000, 780]) | 12\n",
-      "32370    | torch.Size([1, 1000, 754]) | 12\n",
-      "32371    | torch.Size([1, 1000, 754]) | 12\n",
-      "32372    | torch.Size([1, 1000, 754]) | 12\n",
-      "32373    | torch.Size([1, 1000, 754]) | 12\n",
-      "32374    | torch.Size([1, 1000, 754]) | 12\n",
-      "32375    | torch.Size([1, 1000, 754]) | 12\n",
-      "32376    | torch.Size([1, 1000, 754]) | 12\n",
-      "32377    | torch.Size([1, 1000, 754]) | 12\n",
-      "32378    | torch.Size([1, 1000, 754]) | 12\n",
-      "32379    | torch.Size([1, 1000, 772]) | 12\n",
-      "32380    | torch.Size([1, 1000, 754]) | 12\n",
-      "32381    | torch.Size([1, 1000, 754]) | 12\n",
-      "32382    | torch.Size([1, 1000, 754]) | 12\n",
-      "32383    | torch.Size([1, 1000, 784]) | 12\n",
-      "32384    | torch.Size([1, 1000, 754]) | 12\n",
-      "32385    | torch.Size([1, 1000, 754]) | 12\n",
-      "32386    | torch.Size([1, 1000, 781]) | 12\n",
-      "32387    | torch.Size([1, 1000, 754]) | 12\n",
-      "32388    | torch.Size([1, 1000, 754]) | 12\n",
-      "32389    | torch.Size([1, 1000, 754]) | 12\n",
-      "32390    | torch.Size([1, 1000, 754]) | 12\n",
-      "32391    | torch.Size([1, 1000, 754]) | 12\n",
-      "32392    | torch.Size([1, 1000, 754]) | 12\n",
-      "32393    | torch.Size([1, 1000, 754]) | 12\n",
-      "32394    | torch.Size([1, 1000, 754]) | 12\n",
-      "32395    | torch.Size([1, 1000, 786]) | 12\n",
-      "32396    | torch.Size([1, 1000, 754]) | 12\n",
-      "32397    | torch.Size([1, 1000, 754]) | 12\n",
-      "32398    | torch.Size([1, 1000, 754]) | 12\n",
-      "32399    | torch.Size([1, 1000, 754]) | 12\n",
-      "32400    | torch.Size([1, 1000, 754]) | 12\n",
-      "32401    | torch.Size([1, 1000, 754]) | 12\n",
-      "32402    | torch.Size([1, 1000, 754]) | 12\n",
-      "32403    | torch.Size([1, 1000, 793]) | 12\n",
-      "32404    | torch.Size([1, 1000, 754]) | 12\n",
-      "32405    | torch.Size([1, 1000, 754]) | 12\n",
-      "32406    | torch.Size([1, 1000, 754]) | 12\n",
-      "32407    | torch.Size([1, 1000, 754]) | 12\n",
-      "32408    | torch.Size([1, 1000, 754]) | 12\n",
-      "32409    | torch.Size([1, 1000, 807]) | 12\n",
-      "32410    | torch.Size([1, 1000, 752]) | 12\n",
-      "32411    | torch.Size([1, 1000, 754]) | 12\n",
-      "32412    | torch.Size([1, 1000, 754]) | 12\n",
-      "32413    | torch.Size([1, 1000, 754]) | 12\n",
-      "32414    | torch.Size([1, 1000, 754]) | 12\n",
-      "32415    | torch.Size([1, 1000, 754]) | 12\n",
-      "32416    | torch.Size([1, 1000, 754]) | 12\n",
-      "32417    | torch.Size([1, 1000, 754]) | 12\n",
-      "32418    | torch.Size([1, 1000, 754]) | 12\n",
-      "32419    | torch.Size([1, 1000, 754]) | 12\n",
-      "32420    | torch.Size([1, 1000, 754]) | 12\n",
-      "32421    | torch.Size([1, 1000, 754]) | 12\n",
-      "32422    | torch.Size([1, 1000, 754]) | 12\n",
-      "32423    | torch.Size([1, 1000, 754]) | 12\n",
-      "32424    | torch.Size([1, 1000, 754]) | 12\n",
-      "32425    | torch.Size([1, 1000, 754]) | 12\n",
-      "32426    | torch.Size([1, 1000, 754]) | 12\n",
-      "32427    | torch.Size([1, 1000, 754]) | 12\n",
-      "32428    | torch.Size([1, 1000, 754]) | 12\n",
-      "32429    | torch.Size([1, 1000, 749]) | 12\n",
-      "32430    | torch.Size([1, 1000, 754]) | 12\n",
-      "32431    | torch.Size([1, 1000, 754]) | 12\n",
-      "32432    | torch.Size([1, 1000, 754]) | 12\n",
-      "32433    | torch.Size([1, 1000, 754]) | 12\n",
-      "32434    | torch.Size([1, 1000, 796]) | 12\n",
-      "32435    | torch.Size([1, 1000, 754]) | 12\n",
-      "32436    | torch.Size([1, 1000, 754]) | 12\n",
-      "32437    | torch.Size([1, 1000, 754]) | 12\n",
-      "32438    | torch.Size([1, 1000, 762]) | 12\n",
-      "32439    | torch.Size([1, 1000, 754]) | 12\n",
-      "32440    | torch.Size([1, 1000, 754]) | 12\n",
-      "32441    | torch.Size([1, 1000, 754]) | 12\n",
-      "32442    | torch.Size([1, 1000, 754]) | 12\n",
-      "32443    | torch.Size([1, 1000, 754]) | 12\n",
-      "32444    | torch.Size([1, 1000, 784]) | 12\n",
-      "32445    | torch.Size([1, 1000, 772]) | 12\n",
-      "32446    | torch.Size([1, 1000, 754]) | 12\n",
-      "32447    | torch.Size([1, 1000, 754]) | 12\n",
-      "32448    | torch.Size([1, 1000, 754]) | 12\n",
-      "32449    | torch.Size([1, 1000, 754]) | 12\n",
-      "32450    | torch.Size([1, 1000, 754]) | 12\n",
-      "32451    | torch.Size([1, 1000, 754]) | 12\n",
-      "32452    | torch.Size([1, 1000, 762]) | 12\n",
-      "32453    | torch.Size([1, 1000, 754]) | 12\n",
-      "32454    | torch.Size([1, 1000, 754]) | 12\n",
-      "32455    | torch.Size([1, 1000, 754]) | 12\n",
-      "32456    | torch.Size([1, 1000, 754]) | 12\n",
-      "32457    | torch.Size([1, 1000, 754]) | 12\n",
-      "32458    | torch.Size([1, 1000, 772]) | 13\n",
-      "32459    | torch.Size([1, 1000, 772]) | 13\n",
-      "32460    | torch.Size([1, 1000, 754]) | 13\n",
-      "32461    | torch.Size([1, 1000, 799]) | 13\n",
-      "32462    | torch.Size([1, 1000, 772]) | 13\n",
-      "32463    | torch.Size([1, 1000, 777]) | 13\n",
-      "32464    | torch.Size([1, 1000, 777]) | 13\n",
-      "32465    | torch.Size([1, 1000, 754]) | 13\n",
-      "32466    | torch.Size([1, 1000, 754]) | 13\n",
-      "32467    | torch.Size([1, 1000, 754]) | 13\n",
-      "32468    | torch.Size([1, 1000, 777]) | 13\n",
-      "32469    | torch.Size([1, 1000, 754]) | 13\n",
-      "32470    | torch.Size([1, 1000, 772]) | 13\n",
-      "32471    | torch.Size([1, 1000, 777]) | 13\n",
-      "32472    | torch.Size([1, 1000, 777]) | 13\n",
-      "32473    | torch.Size([1, 1000, 754]) | 13\n",
-      "32474    | torch.Size([1, 1000, 754]) | 13\n",
-      "32475    | torch.Size([1, 1000, 773]) | 13\n",
-      "32476    | torch.Size([1, 1000, 772]) | 13\n",
-      "32477    | torch.Size([1, 1000, 754]) | 13\n",
-      "32478    | torch.Size([1, 1000, 754]) | 13\n",
-      "32479    | torch.Size([1, 1000, 754]) | 13\n",
-      "32480    | torch.Size([1, 1000, 754]) | 13\n",
-      "32481    | torch.Size([1, 1000, 754]) | 13\n",
-      "32482    | torch.Size([1, 1000, 777]) | 13\n",
-      "32483    | torch.Size([1, 1000, 747]) | 13\n",
-      "32484    | torch.Size([1, 1000, 772]) | 13\n",
-      "32485    | torch.Size([1, 1000, 754]) | 13\n",
-      "32486    | torch.Size([1, 1000, 754]) | 13\n",
-      "32487    | torch.Size([1, 1000, 777]) | 13\n",
-      "32488    | torch.Size([1, 1000, 772]) | 13\n",
-      "32489    | torch.Size([1, 1000, 754]) | 13\n",
-      "32490    | torch.Size([1, 1000, 754]) | 13\n",
-      "32491    | torch.Size([1, 1000, 772]) | 13\n",
-      "32492    | torch.Size([1, 1000, 754]) | 13\n",
-      "32493    | torch.Size([1, 1000, 754]) | 13\n",
-      "32494    | torch.Size([1, 1000, 777]) | 13\n",
-      "32495    | torch.Size([1, 1000, 754]) | 13\n",
-      "32496    | torch.Size([1, 1000, 754]) | 13\n",
-      "32497    | torch.Size([1, 1000, 754]) | 13\n",
-      "32498    | torch.Size([1, 1000, 772]) | 13\n",
-      "32499    | torch.Size([1, 1000, 754]) | 13\n",
-      "32500    | torch.Size([1, 1000, 768]) | 13\n",
-      "32501    | torch.Size([1, 1000, 781]) | 13\n",
-      "32502    | torch.Size([1, 1000, 754]) | 13\n",
-      "32503    | torch.Size([1, 1000, 772]) | 13\n",
-      "32504    | torch.Size([1, 1000, 785]) | 13\n",
-      "32505    | torch.Size([1, 1000, 777]) | 13\n",
-      "32506    | torch.Size([1, 1000, 754]) | 13\n",
-      "32507    | torch.Size([1, 1000, 759]) | 13\n",
-      "32508    | torch.Size([1, 1000, 810]) | 13\n",
-      "32509    | torch.Size([1, 1000, 754]) | 13\n",
-      "32510    | torch.Size([1, 1000, 795]) | 13\n",
-      "32511    | torch.Size([1, 1000, 754]) | 13\n",
-      "32512    | torch.Size([1, 1000, 754]) | 13\n",
-      "32513    | torch.Size([1, 1000, 754]) | 13\n",
-      "32514    | torch.Size([1, 1000, 754]) | 13\n",
-      "32515    | torch.Size([1, 1000, 754]) | 13\n",
-      "32516    | torch.Size([1, 1000, 777]) | 13\n",
-      "32517    | torch.Size([1, 1000, 754]) | 13\n",
-      "32518    | torch.Size([1, 1000, 771]) | 13\n",
-      "32519    | torch.Size([1, 1000, 772]) | 13\n",
-      "32520    | torch.Size([1, 1000, 754]) | 13\n",
-      "32521    | torch.Size([1, 1000, 754]) | 13\n",
-      "32522    | torch.Size([1, 1000, 788]) | 13\n",
-      "32523    | torch.Size([1, 1000, 759]) | 13\n",
-      "32524    | torch.Size([1, 1000, 754]) | 13\n",
-      "32525    | torch.Size([1, 1000, 772]) | 13\n",
-      "32526    | torch.Size([1, 1000, 777]) | 13\n",
-      "32527    | torch.Size([1, 1000, 797]) | 13\n",
-      "32528    | torch.Size([1, 1000, 754]) | 13\n",
-      "32529    | torch.Size([1, 1000, 754]) | 13\n",
-      "32530    | torch.Size([1, 1000, 754]) | 13\n",
-      "32531    | torch.Size([1, 1000, 754]) | 13\n",
-      "32532    | torch.Size([1, 1000, 754]) | 13\n",
-      "32533    | torch.Size([1, 1000, 754]) | 13\n",
-      "32534    | torch.Size([1, 1000, 754]) | 13\n",
-      "32535    | torch.Size([1, 1000, 754]) | 13\n",
-      "32536    | torch.Size([1, 1000, 754]) | 13\n",
-      "32537    | torch.Size([1, 1000, 754]) | 13\n",
-      "32538    | torch.Size([1, 1000, 754]) | 13\n",
-      "32539    | torch.Size([1, 1000, 772]) | 13\n",
-      "32540    | torch.Size([1, 1000, 754]) | 13\n",
-      "32541    | torch.Size([1, 1000, 754]) | 13\n",
-      "32542    | torch.Size([1, 1000, 771]) | 13\n",
-      "32543    | torch.Size([1, 1000, 777]) | 13\n",
-      "32544    | torch.Size([1, 1000, 754]) | 13\n",
-      "32545    | torch.Size([1, 1000, 844]) | 13\n",
-      "32546    | torch.Size([1, 1000, 754]) | 13\n",
-      "32547    | torch.Size([1, 1000, 754]) | 13\n",
-      "32548    | torch.Size([1, 1000, 754]) | 13\n",
-      "32549    | torch.Size([1, 1000, 754]) | 13\n",
-      "32550    | torch.Size([1, 1000, 754]) | 13\n",
-      "32551    | torch.Size([1, 1000, 754]) | 13\n",
-      "32552    | torch.Size([1, 1000, 754]) | 13\n",
-      "32553    | torch.Size([1, 1000, 754]) | 13\n",
-      "32554    | torch.Size([1, 1000, 777]) | 13\n",
-      "32555    | torch.Size([1, 1000, 754]) | 13\n",
-      "32556    | torch.Size([1, 1000, 777]) | 13\n",
-      "32557    | torch.Size([1, 1000, 754]) | 13\n",
-      "32558    | torch.Size([1, 1000, 754]) | 13\n",
-      "32559    | torch.Size([1, 1000, 772]) | 13\n",
-      "32560    | torch.Size([1, 1000, 772]) | 13\n",
-      "32561    | torch.Size([1, 1000, 754]) | 13\n",
-      "32562    | torch.Size([1, 1000, 772]) | 13\n",
-      "32563    | torch.Size([1, 1000, 852]) | 13\n",
-      "32564    | torch.Size([1, 1000, 809]) | 13\n",
-      "32565    | torch.Size([1, 1000, 754]) | 13\n",
-      "32566    | torch.Size([1, 1000, 754]) | 13\n",
-      "32567    | torch.Size([1, 1000, 754]) | 13\n",
-      "32568    | torch.Size([1, 1000, 777]) | 13\n",
-      "32569    | torch.Size([1, 1000, 754]) | 13\n",
-      "32570    | torch.Size([1, 1000, 772]) | 13\n",
-      "32571    | torch.Size([1, 1000, 754]) | 13\n",
-      "32572    | torch.Size([1, 1000, 800]) | 13\n",
-      "32573    | torch.Size([1, 1000, 777]) | 13\n",
-      "32574    | torch.Size([1, 1000, 772]) | 13\n",
-      "32575    | torch.Size([1, 1000, 754]) | 13\n",
-      "32576    | torch.Size([1, 1000, 777]) | 13\n",
-      "32577    | torch.Size([1, 1000, 773]) | 13\n",
-      "32578    | torch.Size([1, 1000, 754]) | 13\n",
-      "32579    | torch.Size([1, 1000, 754]) | 13\n",
-      "32580    | torch.Size([1, 1000, 777]) | 13\n",
-      "32581    | torch.Size([1, 1000, 754]) | 13\n",
-      "32582    | torch.Size([1, 1000, 754]) | 13\n",
-      "32583    | torch.Size([1, 1000, 754]) | 13\n",
-      "32584    | torch.Size([1, 1000, 770]) | 13\n",
-      "32585    | torch.Size([1, 1000, 754]) | 13\n",
-      "32586    | torch.Size([1, 1000, 772]) | 13\n",
-      "32587    | torch.Size([1, 1000, 754]) | 13\n",
-      "32588    | torch.Size([1, 1000, 772]) | 13\n",
-      "32589    | torch.Size([1, 1000, 777]) | 13\n",
-      "32590    | torch.Size([1, 1000, 772]) | 13\n",
-      "32591    | torch.Size([1, 1000, 772]) | 13\n",
-      "32592    | torch.Size([1, 1000, 839]) | 13\n",
-      "32593    | torch.Size([1, 1000, 772]) | 13\n",
-      "32594    | torch.Size([1, 1000, 754]) | 13\n",
-      "32595    | torch.Size([1, 1000, 754]) | 13\n",
-      "32596    | torch.Size([1, 1000, 772]) | 13\n",
-      "32597    | torch.Size([1, 1000, 754]) | 13\n",
-      "32598    | torch.Size([1, 1000, 754]) | 13\n",
-      "32599    | torch.Size([1, 1000, 772]) | 13\n",
-      "32600    | torch.Size([1, 1000, 777]) | 13\n",
-      "32601    | torch.Size([1, 1000, 754]) | 13\n",
-      "32602    | torch.Size([1, 1000, 777]) | 13\n",
-      "32603    | torch.Size([1, 1000, 772]) | 13\n",
-      "32604    | torch.Size([1, 1000, 754]) | 13\n",
-      "32605    | torch.Size([1, 1000, 772]) | 13\n",
-      "32606    | torch.Size([1, 1000, 785]) | 13\n",
-      "32607    | torch.Size([1, 1000, 754]) | 13\n",
-      "32608    | torch.Size([1, 1000, 754]) | 13\n",
-      "32609    | torch.Size([1, 1000, 777]) | 13\n",
-      "32610    | torch.Size([1, 1000, 754]) | 13\n",
-      "32611    | torch.Size([1, 1000, 754]) | 13\n",
-      "32612    | torch.Size([1, 1000, 771]) | 13\n",
-      "32613    | torch.Size([1, 1000, 754]) | 13\n",
-      "32614    | torch.Size([1, 1000, 754]) | 13\n",
-      "32615    | torch.Size([1, 1000, 754]) | 13\n",
-      "32616    | torch.Size([1, 1000, 754]) | 13\n",
-      "32617    | torch.Size([1, 1000, 754]) | 13\n",
-      "32618    | torch.Size([1, 1000, 777]) | 13\n",
-      "32619    | torch.Size([1, 1000, 772]) | 13\n",
-      "32620    | torch.Size([1, 1000, 754]) | 13\n",
-      "32621    | torch.Size([1, 1000, 772]) | 13\n",
-      "32622    | torch.Size([1, 1000, 777]) | 13\n",
-      "32623    | torch.Size([1, 1000, 754]) | 13\n",
-      "32624    | torch.Size([1, 1000, 754]) | 13\n",
-      "32625    | torch.Size([1, 1000, 777]) | 13\n",
-      "32626    | torch.Size([1, 1000, 777]) | 13\n",
-      "32627    | torch.Size([1, 1000, 754]) | 13\n",
-      "32628    | torch.Size([1, 1000, 777]) | 13\n",
-      "32629    | torch.Size([1, 1000, 747]) | 13\n",
-      "32630    | torch.Size([1, 1000, 754]) | 13\n",
-      "32631    | torch.Size([1, 1000, 754]) | 13\n",
-      "32632    | torch.Size([1, 1000, 777]) | 13\n",
-      "32633    | torch.Size([1, 1000, 754]) | 13\n",
-      "32634    | torch.Size([1, 1000, 754]) | 13\n",
-      "32635    | torch.Size([1, 1000, 767]) | 13\n",
-      "32636    | torch.Size([1, 1000, 772]) | 13\n",
-      "32637    | torch.Size([1, 1000, 754]) | 13\n",
-      "32638    | torch.Size([1, 1000, 772]) | 13\n",
-      "32639    | torch.Size([1, 1000, 772]) | 13\n",
-      "32640    | torch.Size([1, 1000, 754]) | 13\n",
-      "32641    | torch.Size([1, 1000, 802]) | 13\n",
-      "32642    | torch.Size([1, 1000, 786]) | 13\n",
-      "32643    | torch.Size([1, 1000, 772]) | 13\n",
-      "32644    | torch.Size([1, 1000, 754]) | 13\n",
-      "32645    | torch.Size([1, 1000, 754]) | 13\n",
-      "32646    | torch.Size([1, 1000, 777]) | 13\n",
-      "32647    | torch.Size([1, 1000, 772]) | 13\n",
-      "32648    | torch.Size([1, 1000, 754]) | 13\n",
-      "32649    | torch.Size([1, 1000, 754]) | 13\n",
-      "32650    | torch.Size([1, 1000, 754]) | 13\n",
-      "32651    | torch.Size([1, 1000, 754]) | 13\n",
-      "32652    | torch.Size([1, 1000, 754]) | 13\n",
-      "32653    | torch.Size([1, 1000, 771]) | 13\n",
-      "32654    | torch.Size([1, 1000, 754]) | 13\n",
-      "32655    | torch.Size([1, 1000, 777]) | 13\n",
-      "32656    | torch.Size([1, 1000, 760]) | 13\n",
-      "32657    | torch.Size([1, 1000, 754]) | 13\n",
-      "32658    | torch.Size([1, 1000, 754]) | 13\n",
-      "32659    | torch.Size([1, 1000, 754]) | 13\n",
-      "32660    | torch.Size([1, 1000, 772]) | 13\n",
-      "32661    | torch.Size([1, 1000, 754]) | 13\n",
-      "32662    | torch.Size([1, 1000, 754]) | 13\n",
-      "32663    | torch.Size([1, 1000, 754]) | 13\n",
-      "32664    | torch.Size([1, 1000, 754]) | 13\n",
-      "32665    | torch.Size([1, 1000, 754]) | 13\n",
-      "32666    | torch.Size([1, 1000, 786]) | 13\n",
-      "32667    | torch.Size([1, 1000, 754]) | 13\n",
-      "32668    | torch.Size([1, 1000, 754]) | 13\n",
-      "32669    | torch.Size([1, 1000, 754]) | 13\n",
-      "32670    | torch.Size([1, 1000, 754]) | 13\n",
-      "32671    | torch.Size([1, 1000, 795]) | 13\n",
-      "32672    | torch.Size([1, 1000, 777]) | 13\n",
-      "32673    | torch.Size([1, 1000, 754]) | 13\n",
-      "32674    | torch.Size([1, 1000, 777]) | 13\n",
-      "32675    | torch.Size([1, 1000, 776]) | 13\n",
-      "32676    | torch.Size([1, 1000, 754]) | 13\n",
-      "32677    | torch.Size([1, 1000, 772]) | 13\n",
-      "32678    | torch.Size([1, 1000, 754]) | 13\n",
-      "32679    | torch.Size([1, 1000, 772]) | 13\n",
-      "32680    | torch.Size([1, 1000, 778]) | 13\n",
-      "32681    | torch.Size([1, 1000, 754]) | 13\n",
-      "32682    | torch.Size([1, 1000, 772]) | 13\n",
-      "32683    | torch.Size([1, 1000, 754]) | 13\n",
-      "32684    | torch.Size([1, 1000, 777]) | 13\n",
-      "32685    | torch.Size([1, 1000, 754]) | 13\n",
-      "32686    | torch.Size([1, 1000, 754]) | 13\n",
-      "32687    | torch.Size([1, 1000, 754]) | 13\n",
-      "32688    | torch.Size([1, 1000, 754]) | 13\n",
-      "32689    | torch.Size([1, 1000, 777]) | 13\n",
-      "32690    | torch.Size([1, 1000, 772]) | 13\n",
-      "32691    | torch.Size([1, 1000, 777]) | 13\n",
-      "32692    | torch.Size([1, 1000, 754]) | 13\n",
-      "32693    | torch.Size([1, 1000, 777]) | 13\n",
-      "32694    | torch.Size([1, 1000, 777]) | 13\n",
-      "32695    | torch.Size([1, 1000, 754]) | 13\n",
-      "32696    | torch.Size([1, 1000, 754]) | 13\n",
-      "32697    | torch.Size([1, 1000, 789]) | 13\n",
-      "32698    | torch.Size([1, 1000, 754]) | 13\n",
-      "32699    | torch.Size([1, 1000, 754]) | 13\n",
-      "32700    | torch.Size([1, 1000, 754]) | 13\n",
-      "32701    | torch.Size([1, 1000, 754]) | 13\n",
-      "32702    | torch.Size([1, 1000, 777]) | 13\n",
-      "32703    | torch.Size([1, 1000, 754]) | 13\n",
-      "32704    | torch.Size([1, 1000, 779]) | 13\n",
-      "32705    | torch.Size([1, 1000, 771]) | 13\n",
-      "32706    | torch.Size([1, 1000, 777]) | 13\n",
-      "32707    | torch.Size([1, 1000, 799]) | 13\n",
-      "32708    | torch.Size([1, 1000, 754]) | 13\n",
-      "32709    | torch.Size([1, 1000, 796]) | 13\n",
-      "32710    | torch.Size([1, 1000, 772]) | 13\n",
-      "32711    | torch.Size([1, 1000, 754]) | 13\n",
-      "32712    | torch.Size([1, 1000, 777]) | 13\n",
-      "32713    | torch.Size([1, 1000, 754]) | 13\n",
-      "32714    | torch.Size([1, 1000, 777]) | 13\n",
-      "32715    | torch.Size([1, 1000, 754]) | 13\n",
-      "32716    | torch.Size([1, 1000, 754]) | 13\n",
-      "32717    | torch.Size([1, 1000, 754]) | 13\n",
-      "32718    | torch.Size([1, 1000, 772]) | 13\n",
-      "32719    | torch.Size([1, 1000, 772]) | 13\n",
-      "32720    | torch.Size([1, 1000, 816]) | 13\n",
-      "32721    | torch.Size([1, 1000, 754]) | 13\n",
-      "32722    | torch.Size([1, 1000, 754]) | 13\n",
-      "32723    | torch.Size([1, 1000, 772]) | 13\n",
-      "32724    | torch.Size([1, 1000, 777]) | 13\n",
-      "32725    | torch.Size([1, 1000, 754]) | 13\n",
-      "32726    | torch.Size([1, 1000, 772]) | 13\n",
-      "32727    | torch.Size([1, 1000, 772]) | 13\n",
-      "32728    | torch.Size([1, 1000, 771]) | 13\n",
-      "32729    | torch.Size([1, 1000, 733]) | 13\n",
-      "32730    | torch.Size([1, 1000, 754]) | 13\n",
-      "32731    | torch.Size([1, 1000, 772]) | 13\n",
-      "32732    | torch.Size([1, 1000, 754]) | 13\n",
-      "32733    | torch.Size([1, 1000, 754]) | 13\n",
-      "32734    | torch.Size([1, 1000, 777]) | 13\n",
-      "32735    | torch.Size([1, 1000, 754]) | 13\n",
-      "32736    | torch.Size([1, 1000, 773]) | 13\n",
-      "32737    | torch.Size([1, 1000, 774]) | 13\n",
-      "32738    | torch.Size([1, 1000, 772]) | 13\n",
-      "32739    | torch.Size([1, 1000, 754]) | 13\n",
-      "32740    | torch.Size([1, 1000, 777]) | 13\n",
-      "32741    | torch.Size([1, 1000, 772]) | 13\n",
-      "32742    | torch.Size([1, 1000, 772]) | 13\n",
-      "32743    | torch.Size([1, 1000, 772]) | 13\n",
-      "32744    | torch.Size([1, 1000, 777]) | 13\n",
-      "32745    | torch.Size([1, 1000, 772]) | 13\n",
-      "32746    | torch.Size([1, 1000, 754]) | 13\n",
-      "32747    | torch.Size([1, 1000, 725]) | 13\n",
-      "32748    | torch.Size([1, 1000, 772]) | 13\n",
-      "32749    | torch.Size([1, 1000, 754]) | 13\n",
-      "32750    | torch.Size([1, 1000, 772]) | 13\n",
-      "32751    | torch.Size([1, 1000, 754]) | 13\n",
-      "32752    | torch.Size([1, 1000, 754]) | 13\n",
-      "32753    | torch.Size([1, 1000, 777]) | 13\n",
-      "32754    | torch.Size([1, 1000, 754]) | 13\n",
-      "32755    | torch.Size([1, 1000, 754]) | 13\n",
-      "32756    | torch.Size([1, 1000, 777]) | 13\n",
-      "32757    | torch.Size([1, 1000, 772]) | 13\n",
-      "32758    | torch.Size([1, 1000, 772]) | 13\n",
-      "32759    | torch.Size([1, 1000, 754]) | 13\n",
-      "32760    | torch.Size([1, 1000, 754]) | 13\n",
-      "32761    | torch.Size([1, 1000, 772]) | 13\n",
-      "32762    | torch.Size([1, 1000, 802]) | 13\n",
-      "32763    | torch.Size([1, 1000, 754]) | 13\n",
-      "32764    | torch.Size([1, 1000, 754]) | 13\n",
-      "32765    | torch.Size([1, 1000, 754]) | 13\n",
-      "32766    | torch.Size([1, 1000, 772]) | 13\n",
-      "32767    | torch.Size([1, 1000, 772]) | 13\n",
-      "32768    | torch.Size([1, 1000, 754]) | 13\n",
-      "32769    | torch.Size([1, 1000, 754]) | 13\n",
-      "32770    | torch.Size([1, 1000, 777]) | 13\n",
-      "32771    | torch.Size([1, 1000, 703]) | 13\n",
-      "32772    | torch.Size([1, 1000, 772]) | 13\n",
-      "32773    | torch.Size([1, 1000, 785]) | 13\n",
-      "32774    | torch.Size([1, 1000, 754]) | 13\n",
-      "32775    | torch.Size([1, 1000, 772]) | 13\n",
-      "32776    | torch.Size([1, 1000, 754]) | 13\n",
-      "32777    | torch.Size([1, 1000, 773]) | 13\n",
-      "32778    | torch.Size([1, 1000, 754]) | 13\n",
-      "32779    | torch.Size([1, 1000, 754]) | 13\n",
-      "32780    | torch.Size([1, 1000, 777]) | 13\n",
-      "32781    | torch.Size([1, 1000, 754]) | 13\n",
-      "32782    | torch.Size([1, 1000, 754]) | 13\n",
-      "32783    | torch.Size([1, 1000, 754]) | 13\n",
-      "32784    | torch.Size([1, 1000, 754]) | 13\n",
-      "32785    | torch.Size([1, 1000, 754]) | 13\n",
-      "32786    | torch.Size([1, 1000, 785]) | 13\n",
-      "32787    | torch.Size([1, 1000, 754]) | 13\n",
-      "32788    | torch.Size([1, 1000, 754]) | 13\n",
-      "32789    | torch.Size([1, 1000, 754]) | 13\n",
-      "32790    | torch.Size([1, 1000, 865]) | 13\n",
-      "32791    | torch.Size([1, 1000, 754]) | 13\n",
-      "32792    | torch.Size([1, 1000, 777]) | 13\n",
-      "32793    | torch.Size([1, 1000, 754]) | 13\n",
-      "32794    | torch.Size([1, 1000, 754]) | 13\n",
-      "32795    | torch.Size([1, 1000, 754]) | 13\n",
-      "32796    | torch.Size([1, 1000, 707]) | 13\n",
-      "32797    | torch.Size([1, 1000, 754]) | 13\n",
-      "32798    | torch.Size([1, 1000, 754]) | 13\n",
-      "32799    | torch.Size([1, 1000, 771]) | 13\n",
-      "32800    | torch.Size([1, 1000, 777]) | 13\n",
-      "32801    | torch.Size([1, 1000, 777]) | 13\n",
-      "32802    | torch.Size([1, 1000, 754]) | 13\n",
-      "32803    | torch.Size([1, 1000, 777]) | 13\n",
-      "32804    | torch.Size([1, 1000, 869]) | 13\n",
-      "32805    | torch.Size([1, 1000, 772]) | 13\n",
-      "32806    | torch.Size([1, 1000, 772]) | 13\n",
-      "32807    | torch.Size([1, 1000, 754]) | 13\n",
-      "32808    | torch.Size([1, 1000, 777]) | 13\n",
-      "32809    | torch.Size([1, 1000, 772]) | 13\n",
-      "32810    | torch.Size([1, 1000, 754]) | 13\n",
-      "32811    | torch.Size([1, 1000, 754]) | 13\n",
-      "32812    | torch.Size([1, 1000, 754]) | 13\n",
-      "32813    | torch.Size([1, 1000, 774]) | 13\n",
-      "32814    | torch.Size([1, 1000, 754]) | 13\n",
-      "32815    | torch.Size([1, 1000, 777]) | 13\n",
-      "32816    | torch.Size([1, 1000, 754]) | 13\n",
-      "32817    | torch.Size([1, 1000, 771]) | 13\n",
-      "32818    | torch.Size([1, 1000, 754]) | 13\n",
-      "32819    | torch.Size([1, 1000, 754]) | 13\n",
-      "32820    | torch.Size([1, 1000, 754]) | 13\n",
-      "32821    | torch.Size([1, 1000, 754]) | 13\n",
-      "32822    | torch.Size([1, 1000, 777]) | 13\n",
-      "32823    | torch.Size([1, 1000, 771]) | 13\n",
-      "32824    | torch.Size([1, 1000, 754]) | 13\n",
-      "32825    | torch.Size([1, 1000, 723]) | 13\n",
-      "32826    | torch.Size([1, 1000, 870]) | 13\n",
-      "32827    | torch.Size([1, 1000, 754]) | 13\n",
-      "32828    | torch.Size([1, 1000, 777]) | 13\n",
-      "32829    | torch.Size([1, 1000, 777]) | 13\n",
-      "32830    | torch.Size([1, 1000, 754]) | 13\n",
-      "32831    | torch.Size([1, 1000, 772]) | 13\n",
-      "32832    | torch.Size([1, 1000, 754]) | 13\n",
-      "32833    | torch.Size([1, 1000, 754]) | 13\n",
-      "32834    | torch.Size([1, 1000, 772]) | 13\n",
-      "32835    | torch.Size([1, 1000, 754]) | 13\n",
-      "32836    | torch.Size([1, 1000, 754]) | 13\n",
-      "32837    | torch.Size([1, 1000, 777]) | 13\n",
-      "32838    | torch.Size([1, 1000, 772]) | 13\n",
-      "32839    | torch.Size([1, 1000, 754]) | 13\n",
-      "32840    | torch.Size([1, 1000, 754]) | 13\n",
-      "32841    | torch.Size([1, 1000, 754]) | 13\n",
-      "32842    | torch.Size([1, 1000, 772]) | 13\n",
-      "32843    | torch.Size([1, 1000, 754]) | 13\n",
-      "32844    | torch.Size([1, 1000, 754]) | 13\n",
-      "32845    | torch.Size([1, 1000, 754]) | 13\n",
-      "32846    | torch.Size([1, 1000, 777]) | 13\n",
-      "32847    | torch.Size([1, 1000, 770]) | 13\n",
-      "32848    | torch.Size([1, 1000, 752]) | 13\n",
-      "32849    | torch.Size([1, 1000, 754]) | 13\n",
-      "32850    | torch.Size([1, 1000, 754]) | 13\n",
-      "32851    | torch.Size([1, 1000, 772]) | 13\n",
-      "32852    | torch.Size([1, 1000, 754]) | 13\n",
-      "32853    | torch.Size([1, 1000, 754]) | 13\n",
-      "32854    | torch.Size([1, 1000, 817]) | 13\n",
-      "32855    | torch.Size([1, 1000, 754]) | 13\n",
-      "32856    | torch.Size([1, 1000, 772]) | 13\n",
-      "32857    | torch.Size([1, 1000, 777]) | 13\n",
-      "32858    | torch.Size([1, 1000, 754]) | 13\n",
-      "32859    | torch.Size([1, 1000, 754]) | 13\n",
-      "32860    | torch.Size([1, 1000, 772]) | 13\n",
-      "32861    | torch.Size([1, 1000, 777]) | 13\n",
-      "32862    | torch.Size([1, 1000, 772]) | 13\n",
-      "32863    | torch.Size([1, 1000, 754]) | 13\n",
-      "32864    | torch.Size([1, 1000, 772]) | 13\n",
-      "32865    | torch.Size([1, 1000, 754]) | 13\n",
-      "32866    | torch.Size([1, 1000, 777]) | 13\n",
-      "32867    | torch.Size([1, 1000, 754]) | 13\n",
-      "32868    | torch.Size([1, 1000, 754]) | 13\n",
-      "32869    | torch.Size([1, 1000, 777]) | 13\n",
-      "32870    | torch.Size([1, 1000, 777]) | 13\n",
-      "32871    | torch.Size([1, 1000, 778]) | 13\n",
-      "32872    | torch.Size([1, 1000, 754]) | 13\n",
-      "32873    | torch.Size([1, 1000, 777]) | 13\n",
-      "32874    | torch.Size([1, 1000, 754]) | 13\n",
-      "32875    | torch.Size([1, 1000, 754]) | 13\n",
-      "32876    | torch.Size([1, 1000, 777]) | 13\n",
-      "32877    | torch.Size([1, 1000, 772]) | 13\n",
-      "32878    | torch.Size([1, 1000, 772]) | 13\n",
-      "32879    | torch.Size([1, 1000, 777]) | 13\n",
-      "32880    | torch.Size([1, 1000, 754]) | 13\n",
-      "32881    | torch.Size([1, 1000, 777]) | 13\n",
-      "32882    | torch.Size([1, 1000, 754]) | 13\n",
-      "32883    | torch.Size([1, 1000, 754]) | 13\n",
-      "32884    | torch.Size([1, 1000, 754]) | 13\n",
-      "32885    | torch.Size([1, 1000, 754]) | 13\n",
-      "32886    | torch.Size([1, 1000, 754]) | 13\n",
-      "32887    | torch.Size([1, 1000, 754]) | 13\n",
-      "32888    | torch.Size([1, 1000, 754]) | 13\n",
-      "32889    | torch.Size([1, 1000, 771]) | 13\n",
-      "32890    | torch.Size([1, 1000, 777]) | 13\n",
-      "32891    | torch.Size([1, 1000, 754]) | 13\n",
-      "32892    | torch.Size([1, 1000, 777]) | 13\n",
-      "32893    | torch.Size([1, 1000, 754]) | 13\n",
-      "32894    | torch.Size([1, 1000, 754]) | 13\n",
-      "32895    | torch.Size([1, 1000, 777]) | 13\n",
-      "32896    | torch.Size([1, 1000, 754]) | 13\n",
-      "32897    | torch.Size([1, 1000, 754]) | 13\n",
-      "32898    | torch.Size([1, 1000, 777]) | 13\n",
-      "32899    | torch.Size([1, 1000, 754]) | 13\n",
-      "32900    | torch.Size([1, 1000, 754]) | 13\n",
-      "32901    | torch.Size([1, 1000, 795]) | 13\n",
-      "32902    | torch.Size([1, 1000, 754]) | 13\n",
-      "32903    | torch.Size([1, 1000, 730]) | 13\n",
-      "32904    | torch.Size([1, 1000, 772]) | 13\n",
-      "32905    | torch.Size([1, 1000, 754]) | 13\n",
-      "32906    | torch.Size([1, 1000, 772]) | 13\n",
-      "32907    | torch.Size([1, 1000, 754]) | 13\n",
-      "32908    | torch.Size([1, 1000, 754]) | 13\n",
-      "32909    | torch.Size([1, 1000, 754]) | 13\n",
-      "32910    | torch.Size([1, 1000, 754]) | 13\n",
-      "32911    | torch.Size([1, 1000, 754]) | 13\n",
-      "32912    | torch.Size([1, 1000, 754]) | 13\n",
-      "32913    | torch.Size([1, 1000, 754]) | 13\n",
-      "32914    | torch.Size([1, 1000, 772]) | 13\n",
-      "32915    | torch.Size([1, 1000, 777]) | 13\n",
-      "32916    | torch.Size([1, 1000, 777]) | 13\n",
-      "32917    | torch.Size([1, 1000, 754]) | 13\n",
-      "32918    | torch.Size([1, 1000, 773]) | 13\n",
-      "32919    | torch.Size([1, 1000, 772]) | 13\n",
-      "32920    | torch.Size([1, 1000, 808]) | 13\n",
-      "32921    | torch.Size([1, 1000, 754]) | 13\n",
-      "32922    | torch.Size([1, 1000, 754]) | 13\n",
-      "32923    | torch.Size([1, 1000, 754]) | 13\n",
-      "32924    | torch.Size([1, 1000, 754]) | 13\n",
-      "32925    | torch.Size([1, 1000, 777]) | 13\n",
-      "32926    | torch.Size([1, 1000, 777]) | 13\n",
-      "32927    | torch.Size([1, 1000, 754]) | 13\n",
-      "32928    | torch.Size([1, 1000, 772]) | 13\n",
-      "32929    | torch.Size([1, 1000, 777]) | 13\n",
-      "32930    | torch.Size([1, 1000, 754]) | 13\n",
-      "32931    | torch.Size([1, 1000, 777]) | 13\n",
-      "32932    | torch.Size([1, 1000, 754]) | 13\n",
-      "32933    | torch.Size([1, 1000, 772]) | 13\n",
-      "32934    | torch.Size([1, 1000, 754]) | 13\n",
-      "32935    | torch.Size([1, 1000, 754]) | 13\n",
-      "32936    | torch.Size([1, 1000, 772]) | 13\n",
-      "32937    | torch.Size([1, 1000, 777]) | 13\n",
-      "32938    | torch.Size([1, 1000, 754]) | 13\n",
-      "32939    | torch.Size([1, 1000, 762]) | 13\n",
-      "32940    | torch.Size([1, 1000, 772]) | 13\n",
-      "32941    | torch.Size([1, 1000, 772]) | 13\n",
-      "32942    | torch.Size([1, 1000, 777]) | 13\n",
-      "32943    | torch.Size([1, 1000, 754]) | 13\n",
-      "32944    | torch.Size([1, 1000, 777]) | 13\n",
-      "32945    | torch.Size([1, 1000, 772]) | 13\n",
-      "32946    | torch.Size([1, 1000, 754]) | 13\n",
-      "32947    | torch.Size([1, 1000, 763]) | 13\n",
-      "32948    | torch.Size([1, 1000, 772]) | 13\n",
-      "32949    | torch.Size([1, 1000, 754]) | 13\n",
-      "32950    | torch.Size([1, 1000, 754]) | 13\n",
-      "32951    | torch.Size([1, 1000, 754]) | 13\n",
-      "32952    | torch.Size([1, 1000, 774]) | 13\n",
-      "32953    | torch.Size([1, 1000, 771]) | 13\n",
-      "32954    | torch.Size([1, 1000, 754]) | 13\n",
-      "32955    | torch.Size([1, 1000, 754]) | 13\n",
-      "32956    | torch.Size([1, 1000, 754]) | 13\n",
-      "32957    | torch.Size([1, 1000, 772]) | 13\n",
-      "32958    | torch.Size([1, 1000, 754]) | 13\n",
-      "32959    | torch.Size([1, 1000, 754]) | 13\n",
-      "32960    | torch.Size([1, 1000, 777]) | 13\n",
-      "32961    | torch.Size([1, 1000, 754]) | 13\n",
-      "32962    | torch.Size([1, 1000, 777]) | 13\n",
-      "32963    | torch.Size([1, 1000, 754]) | 13\n",
-      "32964    | torch.Size([1, 1000, 777]) | 13\n",
-      "32965    | torch.Size([1, 1000, 754]) | 13\n",
-      "32966    | torch.Size([1, 1000, 772]) | 13\n",
-      "32967    | torch.Size([1, 1000, 754]) | 13\n",
-      "32968    | torch.Size([1, 1000, 754]) | 13\n",
-      "32969    | torch.Size([1, 1000, 719]) | 13\n",
-      "32970    | torch.Size([1, 1000, 754]) | 13\n",
-      "32971    | torch.Size([1, 1000, 777]) | 13\n",
-      "32972    | torch.Size([1, 1000, 754]) | 13\n",
-      "32973    | torch.Size([1, 1000, 793]) | 13\n",
-      "32974    | torch.Size([1, 1000, 789]) | 13\n",
-      "32975    | torch.Size([1, 1000, 777]) | 13\n",
-      "32976    | torch.Size([1, 1000, 787]) | 13\n",
-      "32977    | torch.Size([1, 1000, 754]) | 13\n",
-      "32978    | torch.Size([1, 1000, 772]) | 13\n",
-      "32979    | torch.Size([1, 1000, 777]) | 13\n",
-      "32980    | torch.Size([1, 1000, 777]) | 13\n",
-      "32981    | torch.Size([1, 1000, 777]) | 13\n",
-      "32982    | torch.Size([1, 1000, 754]) | 13\n",
-      "32983    | torch.Size([1, 1000, 777]) | 13\n",
-      "32984    | torch.Size([1, 1000, 782]) | 13\n",
-      "32985    | torch.Size([1, 1000, 777]) | 13\n",
-      "32986    | torch.Size([1, 1000, 777]) | 13\n",
-      "32987    | torch.Size([1, 1000, 777]) | 13\n",
-      "32988    | torch.Size([1, 1000, 754]) | 13\n",
-      "32989    | torch.Size([1, 1000, 754]) | 13\n",
-      "32990    | torch.Size([1, 1000, 754]) | 13\n",
-      "32991    | torch.Size([1, 1000, 777]) | 13\n",
-      "32992    | torch.Size([1, 1000, 752]) | 13\n",
-      "32993    | torch.Size([1, 1000, 754]) | 13\n",
-      "32994    | torch.Size([1, 1000, 747]) | 13\n",
-      "32995    | torch.Size([1, 1000, 772]) | 13\n",
-      "32996    | torch.Size([1, 1000, 772]) | 13\n",
-      "32997    | torch.Size([1, 1000, 754]) | 13\n",
-      "32998    | torch.Size([1, 1000, 777]) | 13\n",
-      "32999    | torch.Size([1, 1000, 754]) | 13\n",
-      "33000    | torch.Size([1, 1000, 800]) | 13\n",
-      "33001    | torch.Size([1, 1000, 721]) | 13\n",
-      "33002    | torch.Size([1, 1000, 754]) | 13\n",
-      "33003    | torch.Size([1, 1000, 754]) | 13\n",
-      "33004    | torch.Size([1, 1000, 777]) | 13\n",
-      "33005    | torch.Size([1, 1000, 777]) | 13\n",
-      "33006    | torch.Size([1, 1000, 771]) | 13\n",
-      "33007    | torch.Size([1, 1000, 754]) | 13\n",
-      "33008    | torch.Size([1, 1000, 754]) | 13\n",
-      "33009    | torch.Size([1, 1000, 754]) | 13\n",
-      "33010    | torch.Size([1, 1000, 754]) | 13\n",
-      "33011    | torch.Size([1, 1000, 772]) | 13\n",
-      "33012    | torch.Size([1, 1000, 772]) | 13\n",
-      "33013    | torch.Size([1, 1000, 754]) | 13\n",
-      "33014    | torch.Size([1, 1000, 777]) | 13\n",
-      "33015    | torch.Size([1, 1000, 754]) | 13\n",
-      "33016    | torch.Size([1, 1000, 772]) | 13\n",
-      "33017    | torch.Size([1, 1000, 772]) | 13\n",
-      "33018    | torch.Size([1, 1000, 777]) | 13\n",
-      "33019    | torch.Size([1, 1000, 754]) | 13\n",
-      "33020    | torch.Size([1, 1000, 772]) | 13\n",
-      "33021    | torch.Size([1, 1000, 847]) | 13\n",
-      "33022    | torch.Size([1, 1000, 777]) | 13\n",
-      "33023    | torch.Size([1, 1000, 771]) | 13\n",
-      "33024    | torch.Size([1, 1000, 754]) | 13\n",
-      "33025    | torch.Size([1, 1000, 754]) | 13\n",
-      "33026    | torch.Size([1, 1000, 754]) | 13\n",
-      "33027    | torch.Size([1, 1000, 754]) | 13\n",
-      "33028    | torch.Size([1, 1000, 754]) | 13\n",
-      "33029    | torch.Size([1, 1000, 754]) | 13\n",
-      "33030    | torch.Size([1, 1000, 754]) | 13\n",
-      "33031    | torch.Size([1, 1000, 754]) | 13\n",
-      "33032    | torch.Size([1, 1000, 784]) | 13\n",
-      "33033    | torch.Size([1, 1000, 777]) | 13\n",
-      "33034    | torch.Size([1, 1000, 754]) | 13\n",
-      "33035    | torch.Size([1, 1000, 754]) | 13\n",
-      "33036    | torch.Size([1, 1000, 777]) | 13\n",
-      "33037    | torch.Size([1, 1000, 849]) | 13\n",
-      "33038    | torch.Size([1, 1000, 790]) | 13\n",
-      "33039    | torch.Size([1, 1000, 754]) | 13\n",
-      "33040    | torch.Size([1, 1000, 777]) | 13\n",
-      "33041    | torch.Size([1, 1000, 777]) | 13\n",
-      "33042    | torch.Size([1, 1000, 754]) | 13\n",
-      "33043    | torch.Size([1, 1000, 754]) | 13\n",
-      "33044    | torch.Size([1, 1000, 754]) | 13\n",
-      "33045    | torch.Size([1, 1000, 772]) | 13\n",
-      "33046    | torch.Size([1, 1000, 754]) | 13\n",
-      "33047    | torch.Size([1, 1000, 777]) | 13\n",
-      "33048    | torch.Size([1, 1000, 754]) | 13\n",
-      "33049    | torch.Size([1, 1000, 754]) | 13\n",
-      "33050    | torch.Size([1, 1000, 772]) | 13\n",
-      "33051    | torch.Size([1, 1000, 772]) | 13\n",
-      "33052    | torch.Size([1, 1000, 777]) | 13\n",
-      "33053    | torch.Size([1, 1000, 754]) | 13\n",
-      "33054    | torch.Size([1, 1000, 772]) | 13\n",
-      "33055    | torch.Size([1, 1000, 754]) | 13\n",
-      "33056    | torch.Size([1, 1000, 754]) | 13\n",
-      "33057    | torch.Size([1, 1000, 772]) | 13\n",
-      "33058    | torch.Size([1, 1000, 772]) | 13\n",
-      "33059    | torch.Size([1, 1000, 754]) | 13\n",
-      "33060    | torch.Size([1, 1000, 772]) | 13\n",
-      "33061    | torch.Size([1, 1000, 754]) | 13\n",
-      "33062    | torch.Size([1, 1000, 777]) | 13\n",
-      "33063    | torch.Size([1, 1000, 754]) | 13\n",
-      "33064    | torch.Size([1, 1000, 754]) | 13\n",
-      "33065    | torch.Size([1, 1000, 778]) | 13\n",
-      "33066    | torch.Size([1, 1000, 767]) | 13\n",
-      "33067    | torch.Size([1, 1000, 754]) | 13\n",
-      "33068    | torch.Size([1, 1000, 772]) | 13\n",
-      "33069    | torch.Size([1, 1000, 754]) | 13\n",
-      "33070    | torch.Size([1, 1000, 754]) | 13\n",
-      "33071    | torch.Size([1, 1000, 754]) | 13\n",
-      "33072    | torch.Size([1, 1000, 754]) | 13\n",
-      "33073    | torch.Size([1, 1000, 777]) | 13\n",
-      "33074    | torch.Size([1, 1000, 777]) | 13\n",
-      "33075    | torch.Size([1, 1000, 754]) | 13\n",
-      "33076    | torch.Size([1, 1000, 754]) | 13\n",
-      "33077    | torch.Size([1, 1000, 772]) | 13\n",
-      "33078    | torch.Size([1, 1000, 772]) | 13\n",
-      "33079    | torch.Size([1, 1000, 754]) | 13\n",
-      "33080    | torch.Size([1, 1000, 754]) | 13\n",
-      "33081    | torch.Size([1, 1000, 754]) | 13\n",
-      "33082    | torch.Size([1, 1000, 770]) | 13\n",
-      "33083    | torch.Size([1, 1000, 754]) | 13\n",
-      "33084    | torch.Size([1, 1000, 772]) | 13\n",
-      "33085    | torch.Size([1, 1000, 754]) | 13\n",
-      "33086    | torch.Size([1, 1000, 807]) | 13\n",
-      "33087    | torch.Size([1, 1000, 777]) | 13\n",
-      "33088    | torch.Size([1, 1000, 754]) | 13\n",
-      "33089    | torch.Size([1, 1000, 754]) | 13\n",
-      "33090    | torch.Size([1, 1000, 754]) | 13\n",
-      "33091    | torch.Size([1, 1000, 754]) | 13\n",
-      "33092    | torch.Size([1, 1000, 754]) | 13\n",
-      "33093    | torch.Size([1, 1000, 754]) | 13\n",
-      "33094    | torch.Size([1, 1000, 772]) | 13\n",
-      "33095    | torch.Size([1, 1000, 777]) | 13\n",
-      "33096    | torch.Size([1, 1000, 754]) | 13\n",
-      "33097    | torch.Size([1, 1000, 754]) | 13\n",
-      "33098    | torch.Size([1, 1000, 754]) | 13\n",
-      "33099    | torch.Size([1, 1000, 772]) | 13\n",
-      "33100    | torch.Size([1, 1000, 777]) | 13\n",
-      "33101    | torch.Size([1, 1000, 754]) | 13\n",
-      "33102    | torch.Size([1, 1000, 777]) | 13\n",
-      "33103    | torch.Size([1, 1000, 754]) | 13\n",
-      "33104    | torch.Size([1, 1000, 754]) | 13\n",
-      "33105    | torch.Size([1, 1000, 754]) | 13\n",
-      "33106    | torch.Size([1, 1000, 754]) | 13\n",
-      "33107    | torch.Size([1, 1000, 777]) | 13\n",
-      "33108    | torch.Size([1, 1000, 754]) | 13\n",
-      "33109    | torch.Size([1, 1000, 772]) | 13\n",
-      "33110    | torch.Size([1, 1000, 810]) | 13\n",
-      "33111    | torch.Size([1, 1000, 777]) | 13\n",
-      "33112    | torch.Size([1, 1000, 754]) | 13\n",
-      "33113    | torch.Size([1, 1000, 783]) | 13\n",
-      "33114    | torch.Size([1, 1000, 772]) | 13\n",
-      "33115    | torch.Size([1, 1000, 754]) | 13\n",
-      "33116    | torch.Size([1, 1000, 754]) | 13\n",
-      "33117    | torch.Size([1, 1000, 754]) | 13\n",
-      "33118    | torch.Size([1, 1000, 754]) | 13\n",
-      "33119    | torch.Size([1, 1000, 754]) | 13\n",
-      "33120    | torch.Size([1, 1000, 754]) | 13\n",
-      "33121    | torch.Size([1, 1000, 820]) | 13\n",
-      "33122    | torch.Size([1, 1000, 754]) | 13\n",
-      "33123    | torch.Size([1, 1000, 754]) | 13\n",
-      "33124    | torch.Size([1, 1000, 779]) | 13\n",
-      "33125    | torch.Size([1, 1000, 754]) | 13\n",
-      "33126    | torch.Size([1, 1000, 754]) | 13\n",
-      "33127    | torch.Size([1, 1000, 772]) | 13\n",
-      "33128    | torch.Size([1, 1000, 847]) | 13\n",
-      "33129    | torch.Size([1, 1000, 772]) | 13\n",
-      "33130    | torch.Size([1, 1000, 754]) | 13\n",
-      "33131    | torch.Size([1, 1000, 772]) | 13\n",
-      "33132    | torch.Size([1, 1000, 754]) | 13\n",
-      "33133    | torch.Size([1, 1000, 787]) | 13\n",
-      "33134    | torch.Size([1, 1000, 802]) | 13\n",
-      "33135    | torch.Size([1, 1000, 772]) | 13\n",
-      "33136    | torch.Size([1, 1000, 772]) | 13\n",
-      "33137    | torch.Size([1, 1000, 754]) | 13\n",
-      "33138    | torch.Size([1, 1000, 754]) | 13\n",
-      "33139    | torch.Size([1, 1000, 754]) | 13\n",
-      "33140    | torch.Size([1, 1000, 754]) | 13\n",
-      "33141    | torch.Size([1, 1000, 777]) | 13\n",
-      "33142    | torch.Size([1, 1000, 754]) | 13\n",
-      "33143    | torch.Size([1, 1000, 771]) | 13\n",
-      "33144    | torch.Size([1, 1000, 754]) | 13\n",
-      "33145    | torch.Size([1, 1000, 772]) | 13\n",
-      "33146    | torch.Size([1, 1000, 781]) | 13\n",
-      "33147    | torch.Size([1, 1000, 754]) | 13\n",
-      "33148    | torch.Size([1, 1000, 778]) | 13\n",
-      "33149    | torch.Size([1, 1000, 769]) | 13\n",
-      "33150    | torch.Size([1, 1000, 777]) | 13\n",
-      "33151    | torch.Size([1, 1000, 754]) | 13\n",
-      "33152    | torch.Size([1, 1000, 772]) | 13\n",
-      "33153    | torch.Size([1, 1000, 777]) | 13\n",
-      "33154    | torch.Size([1, 1000, 781]) | 13\n",
-      "33155    | torch.Size([1, 1000, 754]) | 13\n",
-      "33156    | torch.Size([1, 1000, 772]) | 13\n",
-      "33157    | torch.Size([1, 1000, 777]) | 13\n",
-      "33158    | torch.Size([1, 1000, 766]) | 13\n",
-      "33159    | torch.Size([1, 1000, 754]) | 13\n",
-      "33160    | torch.Size([1, 1000, 754]) | 13\n",
-      "33161    | torch.Size([1, 1000, 777]) | 13\n",
-      "33162    | torch.Size([1, 1000, 786]) | 13\n",
-      "33163    | torch.Size([1, 1000, 754]) | 13\n",
-      "33164    | torch.Size([1, 1000, 772]) | 13\n",
-      "33165    | torch.Size([1, 1000, 754]) | 13\n",
-      "33166    | torch.Size([1, 1000, 754]) | 13\n",
-      "33167    | torch.Size([1, 1000, 816]) | 13\n",
-      "33168    | torch.Size([1, 1000, 754]) | 13\n",
-      "33169    | torch.Size([1, 1000, 777]) | 13\n",
-      "33170    | torch.Size([1, 1000, 754]) | 13\n",
-      "33171    | torch.Size([1, 1000, 777]) | 13\n",
-      "33172    | torch.Size([1, 1000, 772]) | 13\n",
-      "33173    | torch.Size([1, 1000, 754]) | 13\n",
-      "33174    | torch.Size([1, 1000, 754]) | 13\n",
-      "33175    | torch.Size([1, 1000, 777]) | 13\n",
-      "33176    | torch.Size([1, 1000, 754]) | 13\n",
-      "33177    | torch.Size([1, 1000, 754]) | 13\n",
-      "33178    | torch.Size([1, 1000, 754]) | 13\n",
-      "33179    | torch.Size([1, 1000, 754]) | 13\n",
-      "33180    | torch.Size([1, 1000, 777]) | 13\n",
-      "33181    | torch.Size([1, 1000, 772]) | 13\n",
-      "33182    | torch.Size([1, 1000, 754]) | 13\n",
-      "33183    | torch.Size([1, 1000, 754]) | 13\n",
-      "33184    | torch.Size([1, 1000, 772]) | 13\n",
-      "33185    | torch.Size([1, 1000, 754]) | 13\n",
-      "33186    | torch.Size([1, 1000, 779]) | 13\n",
-      "33187    | torch.Size([1, 1000, 754]) | 13\n",
-      "33188    | torch.Size([1, 1000, 754]) | 13\n",
-      "33189    | torch.Size([1, 1000, 754]) | 13\n",
-      "33190    | torch.Size([1, 1000, 782]) | 13\n",
-      "33191    | torch.Size([1, 1000, 754]) | 13\n",
-      "33192    | torch.Size([1, 1000, 754]) | 13\n",
-      "33193    | torch.Size([1, 1000, 754]) | 13\n",
-      "33194    | torch.Size([1, 1000, 777]) | 13\n",
-      "33195    | torch.Size([1, 1000, 754]) | 13\n",
-      "33196    | torch.Size([1, 1000, 754]) | 13\n",
-      "33197    | torch.Size([1, 1000, 754]) | 13\n",
-      "33198    | torch.Size([1, 1000, 813]) | 13\n",
-      "33199    | torch.Size([1, 1000, 777]) | 13\n",
-      "33200    | torch.Size([1, 1000, 754]) | 13\n",
-      "33201    | torch.Size([1, 1000, 754]) | 13\n",
-      "33202    | torch.Size([1, 1000, 772]) | 13\n",
-      "33203    | torch.Size([1, 1000, 772]) | 13\n",
-      "33204    | torch.Size([1, 1000, 777]) | 13\n",
-      "33205    | torch.Size([1, 1000, 761]) | 13\n",
-      "33206    | torch.Size([1, 1000, 754]) | 13\n",
-      "33207    | torch.Size([1, 1000, 754]) | 13\n",
-      "33208    | torch.Size([1, 1000, 754]) | 13\n",
-      "33209    | torch.Size([1, 1000, 754]) | 13\n",
-      "33210    | torch.Size([1, 1000, 754]) | 13\n",
-      "33211    | torch.Size([1, 1000, 777]) | 13\n",
-      "33212    | torch.Size([1, 1000, 754]) | 13\n",
-      "33213    | torch.Size([1, 1000, 754]) | 13\n",
-      "33214    | torch.Size([1, 1000, 754]) | 13\n",
-      "33215    | torch.Size([1, 1000, 781]) | 13\n",
-      "33216    | torch.Size([1, 1000, 766]) | 13\n",
-      "33217    | torch.Size([1, 1000, 777]) | 13\n",
-      "33218    | torch.Size([1, 1000, 754]) | 13\n",
-      "33219    | torch.Size([1, 1000, 754]) | 13\n",
-      "33220    | torch.Size([1, 1000, 754]) | 13\n",
-      "33221    | torch.Size([1, 1000, 754]) | 13\n",
-      "33222    | torch.Size([1, 1000, 754]) | 13\n",
-      "33223    | torch.Size([1, 1000, 777]) | 13\n",
-      "33224    | torch.Size([1, 1000, 754]) | 13\n",
-      "33225    | torch.Size([1, 1000, 754]) | 13\n",
-      "33226    | torch.Size([1, 1000, 754]) | 13\n",
-      "33227    | torch.Size([1, 1000, 754]) | 13\n",
-      "33228    | torch.Size([1, 1000, 754]) | 13\n",
-      "33229    | torch.Size([1, 1000, 754]) | 13\n",
-      "33230    | torch.Size([1, 1000, 754]) | 13\n",
-      "33231    | torch.Size([1, 1000, 771]) | 13\n",
-      "33232    | torch.Size([1, 1000, 754]) | 13\n",
-      "33233    | torch.Size([1, 1000, 754]) | 13\n",
-      "33234    | torch.Size([1, 1000, 730]) | 13\n",
-      "33235    | torch.Size([1, 1000, 754]) | 13\n",
-      "33236    | torch.Size([1, 1000, 754]) | 13\n",
-      "33237    | torch.Size([1, 1000, 772]) | 13\n",
-      "33238    | torch.Size([1, 1000, 754]) | 13\n",
-      "33239    | torch.Size([1, 1000, 754]) | 13\n",
-      "33240    | torch.Size([1, 1000, 772]) | 13\n",
-      "33241    | torch.Size([1, 1000, 754]) | 13\n",
-      "33242    | torch.Size([1, 1000, 754]) | 13\n",
-      "33243    | torch.Size([1, 1000, 754]) | 13\n",
-      "33244    | torch.Size([1, 1000, 772]) | 13\n",
-      "33245    | torch.Size([1, 1000, 771]) | 13\n",
-      "33246    | torch.Size([1, 1000, 788]) | 13\n",
-      "33247    | torch.Size([1, 1000, 754]) | 13\n",
-      "33248    | torch.Size([1, 1000, 772]) | 13\n",
-      "33249    | torch.Size([1, 1000, 754]) | 13\n",
-      "33250    | torch.Size([1, 1000, 754]) | 13\n",
-      "33251    | torch.Size([1, 1000, 754]) | 13\n",
-      "33252    | torch.Size([1, 1000, 772]) | 13\n",
-      "33253    | torch.Size([1, 1000, 754]) | 13\n",
-      "33254    | torch.Size([1, 1000, 777]) | 13\n",
-      "33255    | torch.Size([1, 1000, 754]) | 13\n",
-      "33256    | torch.Size([1, 1000, 754]) | 13\n",
-      "33257    | torch.Size([1, 1000, 754]) | 13\n",
-      "33258    | torch.Size([1, 1000, 790]) | 13\n",
-      "33259    | torch.Size([1, 1000, 777]) | 13\n",
-      "33260    | torch.Size([1, 1000, 805]) | 13\n",
-      "33261    | torch.Size([1, 1000, 777]) | 13\n",
-      "33262    | torch.Size([1, 1000, 772]) | 13\n",
-      "33263    | torch.Size([1, 1000, 772]) | 13\n",
-      "33264    | torch.Size([1, 1000, 772]) | 13\n",
-      "33265    | torch.Size([1, 1000, 754]) | 13\n",
-      "33266    | torch.Size([1, 1000, 777]) | 13\n",
-      "33267    | torch.Size([1, 1000, 777]) | 13\n",
-      "33268    | torch.Size([1, 1000, 754]) | 13\n",
-      "33269    | torch.Size([1, 1000, 754]) | 13\n",
-      "33270    | torch.Size([1, 1000, 772]) | 13\n",
-      "33271    | torch.Size([1, 1000, 777]) | 13\n",
-      "33272    | torch.Size([1, 1000, 772]) | 13\n",
-      "33273    | torch.Size([1, 1000, 754]) | 13\n",
-      "33274    | torch.Size([1, 1000, 777]) | 13\n",
-      "33275    | torch.Size([1, 1000, 754]) | 13\n",
-      "33276    | torch.Size([1, 1000, 816]) | 13\n",
-      "33277    | torch.Size([1, 1000, 772]) | 13\n",
-      "33278    | torch.Size([1, 1000, 772]) | 13\n",
-      "33279    | torch.Size([1, 1000, 778]) | 13\n",
-      "33280    | torch.Size([1, 1000, 754]) | 13\n",
-      "33281    | torch.Size([1, 1000, 772]) | 13\n",
-      "33282    | torch.Size([1, 1000, 777]) | 13\n",
-      "33283    | torch.Size([1, 1000, 747]) | 13\n",
-      "33284    | torch.Size([1, 1000, 754]) | 13\n",
-      "33285    | torch.Size([1, 1000, 754]) | 13\n",
-      "33286    | torch.Size([1, 1000, 754]) | 13\n",
-      "33287    | torch.Size([1, 1000, 777]) | 13\n",
-      "33288    | torch.Size([1, 1000, 754]) | 13\n",
-      "33289    | torch.Size([1, 1000, 810]) | 13\n",
-      "33290    | torch.Size([1, 1000, 754]) | 13\n",
-      "33291    | torch.Size([1, 1000, 754]) | 13\n",
-      "33292    | torch.Size([1, 1000, 777]) | 13\n",
-      "33293    | torch.Size([1, 1000, 754]) | 13\n",
-      "33294    | torch.Size([1, 1000, 754]) | 13\n",
-      "33295    | torch.Size([1, 1000, 773]) | 13\n",
-      "33296    | torch.Size([1, 1000, 754]) | 13\n",
-      "33297    | torch.Size([1, 1000, 754]) | 13\n",
-      "33298    | torch.Size([1, 1000, 754]) | 13\n",
-      "33299    | torch.Size([1, 1000, 777]) | 13\n",
-      "33300    | torch.Size([1, 1000, 754]) | 13\n",
-      "33301    | torch.Size([1, 1000, 772]) | 13\n",
-      "33302    | torch.Size([1, 1000, 754]) | 13\n",
-      "33303    | torch.Size([1, 1000, 754]) | 13\n",
-      "33304    | torch.Size([1, 1000, 772]) | 13\n",
-      "33305    | torch.Size([1, 1000, 777]) | 13\n",
-      "33306    | torch.Size([1, 1000, 772]) | 13\n",
-      "33307    | torch.Size([1, 1000, 772]) | 13\n",
-      "33308    | torch.Size([1, 1000, 754]) | 13\n",
-      "33309    | torch.Size([1, 1000, 754]) | 13\n",
-      "33310    | torch.Size([1, 1000, 754]) | 13\n",
-      "33311    | torch.Size([1, 1000, 754]) | 13\n",
-      "33312    | torch.Size([1, 1000, 754]) | 13\n",
-      "33313    | torch.Size([1, 1000, 754]) | 13\n",
-      "33314    | torch.Size([1, 1000, 754]) | 13\n",
-      "33315    | torch.Size([1, 1000, 772]) | 13\n",
-      "33316    | torch.Size([1, 1000, 779]) | 13\n",
-      "33317    | torch.Size([1, 1000, 754]) | 13\n",
-      "33318    | torch.Size([1, 1000, 754]) | 13\n",
-      "33319    | torch.Size([1, 1000, 754]) | 13\n",
-      "33320    | torch.Size([1, 1000, 754]) | 13\n",
-      "33321    | torch.Size([1, 1000, 771]) | 13\n",
-      "33322    | torch.Size([1, 1000, 754]) | 13\n",
-      "33323    | torch.Size([1, 1000, 754]) | 13\n",
-      "33324    | torch.Size([1, 1000, 754]) | 13\n",
-      "33325    | torch.Size([1, 1000, 754]) | 13\n",
-      "33326    | torch.Size([1, 1000, 777]) | 13\n",
-      "33327    | torch.Size([1, 1000, 777]) | 13\n",
-      "33328    | torch.Size([1, 1000, 754]) | 13\n",
-      "33329    | torch.Size([1, 1000, 754]) | 13\n",
-      "33330    | torch.Size([1, 1000, 754]) | 13\n",
-      "33331    | torch.Size([1, 1000, 772]) | 13\n",
-      "33332    | torch.Size([1, 1000, 754]) | 13\n",
-      "33333    | torch.Size([1, 1000, 754]) | 13\n",
-      "33334    | torch.Size([1, 1000, 771]) | 13\n",
-      "33335    | torch.Size([1, 1000, 777]) | 13\n",
-      "33336    | torch.Size([1, 1000, 777]) | 13\n",
-      "33337    | torch.Size([1, 1000, 754]) | 13\n",
-      "33338    | torch.Size([1, 1000, 772]) | 13\n",
-      "33339    | torch.Size([1, 1000, 754]) | 13\n",
-      "33340    | torch.Size([1, 1000, 836]) | 13\n",
-      "33341    | torch.Size([1, 1000, 754]) | 13\n",
-      "33342    | torch.Size([1, 1000, 754]) | 13\n",
-      "33343    | torch.Size([1, 1000, 763]) | 13\n",
-      "33344    | torch.Size([1, 1000, 754]) | 13\n",
-      "33345    | torch.Size([1, 1000, 777]) | 13\n",
-      "33346    | torch.Size([1, 1000, 777]) | 13\n",
-      "33347    | torch.Size([1, 1000, 754]) | 13\n",
-      "33348    | torch.Size([1, 1000, 775]) | 13\n",
-      "33349    | torch.Size([1, 1000, 754]) | 13\n",
-      "33350    | torch.Size([1, 1000, 754]) | 13\n",
-      "33351    | torch.Size([1, 1000, 754]) | 13\n",
-      "33352    | torch.Size([1, 1000, 772]) | 13\n",
-      "33353    | torch.Size([1, 1000, 772]) | 13\n",
-      "33354    | torch.Size([1, 1000, 754]) | 13\n",
-      "33355    | torch.Size([1, 1000, 770]) | 13\n",
-      "33356    | torch.Size([1, 1000, 772]) | 13\n",
-      "33357    | torch.Size([1, 1000, 777]) | 13\n",
-      "33358    | torch.Size([1, 1000, 754]) | 13\n",
-      "33359    | torch.Size([1, 1000, 754]) | 13\n",
-      "33360    | torch.Size([1, 1000, 777]) | 13\n",
-      "33361    | torch.Size([1, 1000, 772]) | 13\n",
-      "33362    | torch.Size([1, 1000, 754]) | 13\n",
-      "33363    | torch.Size([1, 1000, 777]) | 13\n",
-      "33364    | torch.Size([1, 1000, 802]) | 13\n",
-      "33365    | torch.Size([1, 1000, 754]) | 13\n",
-      "33366    | torch.Size([1, 1000, 754]) | 13\n",
-      "33367    | torch.Size([1, 1000, 777]) | 13\n",
-      "33368    | torch.Size([1, 1000, 772]) | 13\n",
-      "33369    | torch.Size([1, 1000, 772]) | 13\n",
-      "33370    | torch.Size([1, 1000, 771]) | 13\n",
-      "33371    | torch.Size([1, 1000, 771]) | 13\n",
-      "33372    | torch.Size([1, 1000, 754]) | 13\n",
-      "33373    | torch.Size([1, 1000, 782]) | 13\n",
-      "33374    | torch.Size([1, 1000, 754]) | 13\n",
-      "33375    | torch.Size([1, 1000, 754]) | 13\n",
-      "33376    | torch.Size([1, 1000, 754]) | 13\n",
-      "33377    | torch.Size([1, 1000, 754]) | 13\n",
-      "33378    | torch.Size([1, 1000, 754]) | 13\n",
-      "33379    | torch.Size([1, 1000, 754]) | 13\n",
-      "33380    | torch.Size([1, 1000, 754]) | 13\n",
-      "33381    | torch.Size([1, 1000, 754]) | 13\n",
-      "33382    | torch.Size([1, 1000, 802]) | 13\n",
-      "33383    | torch.Size([1, 1000, 772]) | 13\n",
-      "33384    | torch.Size([1, 1000, 713]) | 13\n",
-      "33385    | torch.Size([1, 1000, 732]) | 13\n",
-      "33386    | torch.Size([1, 1000, 754]) | 13\n",
-      "33387    | torch.Size([1, 1000, 754]) | 13\n",
-      "33388    | torch.Size([1, 1000, 754]) | 13\n",
-      "33389    | torch.Size([1, 1000, 754]) | 13\n",
-      "33390    | torch.Size([1, 1000, 794]) | 13\n",
-      "33391    | torch.Size([1, 1000, 777]) | 13\n",
-      "33392    | torch.Size([1, 1000, 754]) | 13\n",
-      "33393    | torch.Size([1, 1000, 713]) | 13\n",
-      "33394    | torch.Size([1, 1000, 754]) | 13\n",
-      "33395    | torch.Size([1, 1000, 754]) | 13\n",
-      "33396    | torch.Size([1, 1000, 772]) | 13\n",
-      "33397    | torch.Size([1, 1000, 754]) | 13\n",
-      "33398    | torch.Size([1, 1000, 777]) | 13\n",
-      "33399    | torch.Size([1, 1000, 777]) | 13\n",
-      "33400    | torch.Size([1, 1000, 754]) | 13\n",
-      "33401    | torch.Size([1, 1000, 754]) | 13\n",
-      "33402    | torch.Size([1, 1000, 789]) | 13\n",
-      "33403    | torch.Size([1, 1000, 754]) | 13\n",
-      "33404    | torch.Size([1, 1000, 754]) | 13\n",
-      "33405    | torch.Size([1, 1000, 713]) | 13\n",
-      "33406    | torch.Size([1, 1000, 762]) | 13\n",
-      "33407    | torch.Size([1, 1000, 754]) | 13\n",
-      "33408    | torch.Size([1, 1000, 777]) | 13\n",
-      "33409    | torch.Size([1, 1000, 754]) | 13\n",
-      "33410    | torch.Size([1, 1000, 754]) | 13\n",
-      "33411    | torch.Size([1, 1000, 777]) | 13\n",
-      "33412    | torch.Size([1, 1000, 772]) | 13\n",
-      "33413    | torch.Size([1, 1000, 754]) | 13\n",
-      "33414    | torch.Size([1, 1000, 772]) | 13\n",
-      "33415    | torch.Size([1, 1000, 754]) | 13\n",
-      "33416    | torch.Size([1, 1000, 837]) | 13\n",
-      "33417    | torch.Size([1, 1000, 777]) | 13\n",
-      "33418    | torch.Size([1, 1000, 754]) | 13\n",
-      "33419    | torch.Size([1, 1000, 777]) | 13\n",
-      "33420    | torch.Size([1, 1000, 754]) | 13\n",
-      "33421    | torch.Size([1, 1000, 772]) | 13\n",
-      "33422    | torch.Size([1, 1000, 777]) | 13\n",
-      "33423    | torch.Size([1, 1000, 772]) | 13\n",
-      "33424    | torch.Size([1, 1000, 754]) | 13\n",
-      "33425    | torch.Size([1, 1000, 754]) | 13\n",
-      "33426    | torch.Size([1, 1000, 772]) | 13\n",
-      "33427    | torch.Size([1, 1000, 713]) | 13\n",
-      "33428    | torch.Size([1, 1000, 772]) | 13\n",
-      "33429    | torch.Size([1, 1000, 771]) | 13\n",
-      "33430    | torch.Size([1, 1000, 810]) | 13\n",
-      "33431    | torch.Size([1, 1000, 754]) | 13\n",
-      "33432    | torch.Size([1, 1000, 777]) | 13\n",
-      "33433    | torch.Size([1, 1000, 754]) | 13\n",
-      "33434    | torch.Size([1, 1000, 772]) | 13\n",
-      "33435    | torch.Size([1, 1000, 754]) | 13\n",
-      "33436    | torch.Size([1, 1000, 754]) | 13\n",
-      "33437    | torch.Size([1, 1000, 754]) | 13\n",
-      "33438    | torch.Size([1, 1000, 754]) | 13\n",
-      "33439    | torch.Size([1, 1000, 778]) | 13\n",
-      "33440    | torch.Size([1, 1000, 772]) | 13\n",
-      "33441    | torch.Size([1, 1000, 754]) | 13\n",
-      "33442    | torch.Size([1, 1000, 754]) | 13\n",
-      "33443    | torch.Size([1, 1000, 772]) | 13\n",
-      "33444    | torch.Size([1, 1000, 754]) | 13\n",
-      "33445    | torch.Size([1, 1000, 777]) | 13\n",
-      "33446    | torch.Size([1, 1000, 754]) | 13\n",
-      "33447    | torch.Size([1, 1000, 772]) | 13\n",
-      "33448    | torch.Size([1, 1000, 754]) | 13\n",
-      "33449    | torch.Size([1, 1000, 794]) | 13\n",
-      "33450    | torch.Size([1, 1000, 754]) | 13\n",
-      "33451    | torch.Size([1, 1000, 754]) | 13\n",
-      "33452    | torch.Size([1, 1000, 777]) | 13\n",
-      "33453    | torch.Size([1, 1000, 777]) | 13\n",
-      "33454    | torch.Size([1, 1000, 754]) | 13\n",
-      "33455    | torch.Size([1, 1000, 777]) | 13\n",
-      "33456    | torch.Size([1, 1000, 754]) | 13\n",
-      "33457    | torch.Size([1, 1000, 754]) | 13\n",
-      "33458    | torch.Size([1, 1000, 754]) | 13\n",
-      "33459    | torch.Size([1, 1000, 754]) | 13\n",
-      "33460    | torch.Size([1, 1000, 777]) | 13\n",
-      "33461    | torch.Size([1, 1000, 754]) | 13\n",
-      "33462    | torch.Size([1, 1000, 772]) | 13\n",
-      "33463    | torch.Size([1, 1000, 771]) | 13\n",
-      "33464    | torch.Size([1, 1000, 754]) | 13\n",
-      "33465    | torch.Size([1, 1000, 754]) | 13\n",
-      "33466    | torch.Size([1, 1000, 754]) | 13\n",
-      "33467    | torch.Size([1, 1000, 754]) | 13\n",
-      "33468    | torch.Size([1, 1000, 754]) | 13\n",
-      "33469    | torch.Size([1, 1000, 754]) | 13\n",
-      "33470    | torch.Size([1, 1000, 754]) | 13\n",
-      "33471    | torch.Size([1, 1000, 754]) | 13\n",
-      "33472    | torch.Size([1, 1000, 786]) | 13\n",
-      "33473    | torch.Size([1, 1000, 754]) | 13\n",
-      "33474    | torch.Size([1, 1000, 754]) | 13\n",
-      "33475    | torch.Size([1, 1000, 777]) | 13\n",
-      "33476    | torch.Size([1, 1000, 783]) | 13\n",
-      "33477    | torch.Size([1, 1000, 754]) | 13\n",
-      "33478    | torch.Size([1, 1000, 779]) | 13\n",
-      "33479    | torch.Size([1, 1000, 754]) | 13\n",
-      "33480    | torch.Size([1, 1000, 810]) | 13\n",
-      "33481    | torch.Size([1, 1000, 799]) | 13\n",
-      "33482    | torch.Size([1, 1000, 759]) | 13\n",
-      "33483    | torch.Size([1, 1000, 869]) | 13\n",
-      "33484    | torch.Size([1, 1000, 754]) | 13\n",
-      "33485    | torch.Size([1, 1000, 773]) | 13\n",
-      "33486    | torch.Size([1, 1000, 776]) | 13\n",
-      "33487    | torch.Size([1, 1000, 784]) | 13\n",
-      "33488    | torch.Size([1, 1000, 754]) | 13\n",
-      "33489    | torch.Size([1, 1000, 777]) | 13\n",
-      "33490    | torch.Size([1, 1000, 766]) | 13\n",
-      "33491    | torch.Size([1, 1000, 777]) | 13\n",
-      "33492    | torch.Size([1, 1000, 754]) | 13\n",
-      "33493    | torch.Size([1, 1000, 781]) | 13\n",
-      "33494    | torch.Size([1, 1000, 772]) | 13\n",
-      "33495    | torch.Size([1, 1000, 772]) | 13\n",
-      "33496    | torch.Size([1, 1000, 754]) | 13\n",
-      "33497    | torch.Size([1, 1000, 754]) | 13\n",
-      "33498    | torch.Size([1, 1000, 754]) | 13\n",
-      "33499    | torch.Size([1, 1000, 777]) | 13\n",
-      "33500    | torch.Size([1, 1000, 771]) | 13\n",
-      "33501    | torch.Size([1, 1000, 718]) | 13\n",
-      "33502    | torch.Size([1, 1000, 791]) | 13\n",
-      "33503    | torch.Size([1, 1000, 754]) | 13\n",
-      "33504    | torch.Size([1, 1000, 754]) | 13\n",
-      "33505    | torch.Size([1, 1000, 772]) | 13\n",
-      "33506    | torch.Size([1, 1000, 874]) | 13\n",
-      "33507    | torch.Size([1, 1000, 754]) | 13\n",
-      "33508    | torch.Size([1, 1000, 754]) | 13\n",
-      "33509    | torch.Size([1, 1000, 771]) | 13\n",
-      "33510    | torch.Size([1, 1000, 812]) | 13\n",
-      "33511    | torch.Size([1, 1000, 791]) | 13\n",
-      "33512    | torch.Size([1, 1000, 777]) | 13\n",
-      "33513    | torch.Size([1, 1000, 754]) | 13\n",
-      "33514    | torch.Size([1, 1000, 754]) | 13\n",
-      "33515    | torch.Size([1, 1000, 754]) | 13\n",
-      "33516    | torch.Size([1, 1000, 754]) | 13\n",
-      "33517    | torch.Size([1, 1000, 772]) | 13\n",
-      "33518    | torch.Size([1, 1000, 754]) | 13\n",
-      "33519    | torch.Size([1, 1000, 754]) | 13\n",
-      "33520    | torch.Size([1, 1000, 754]) | 13\n",
-      "33521    | torch.Size([1, 1000, 754]) | 13\n",
-      "33522    | torch.Size([1, 1000, 771]) | 13\n",
-      "33523    | torch.Size([1, 1000, 787]) | 13\n",
-      "33524    | torch.Size([1, 1000, 772]) | 13\n",
-      "33525    | torch.Size([1, 1000, 754]) | 13\n",
-      "33526    | torch.Size([1, 1000, 754]) | 13\n",
-      "33527    | torch.Size([1, 1000, 754]) | 13\n",
-      "33528    | torch.Size([1, 1000, 772]) | 13\n",
-      "33529    | torch.Size([1, 1000, 754]) | 13\n",
-      "33530    | torch.Size([1, 1000, 754]) | 13\n",
-      "33531    | torch.Size([1, 1000, 754]) | 13\n",
-      "33532    | torch.Size([1, 1000, 790]) | 13\n",
-      "33533    | torch.Size([1, 1000, 777]) | 13\n",
-      "33534    | torch.Size([1, 1000, 703]) | 13\n",
-      "33535    | torch.Size([1, 1000, 787]) | 13\n",
-      "33536    | torch.Size([1, 1000, 754]) | 13\n",
-      "33537    | torch.Size([1, 1000, 801]) | 13\n",
-      "33538    | torch.Size([1, 1000, 754]) | 13\n",
-      "33539    | torch.Size([1, 1000, 752]) | 13\n",
-      "33540    | torch.Size([1, 1000, 754]) | 13\n",
-      "33541    | torch.Size([1, 1000, 754]) | 13\n",
-      "33542    | torch.Size([1, 1000, 771]) | 13\n",
-      "33543    | torch.Size([1, 1000, 772]) | 13\n",
-      "33544    | torch.Size([1, 1000, 777]) | 13\n",
-      "33545    | torch.Size([1, 1000, 777]) | 13\n",
-      "33546    | torch.Size([1, 1000, 772]) | 13\n",
-      "33547    | torch.Size([1, 1000, 754]) | 13\n",
-      "33548    | torch.Size([1, 1000, 777]) | 13\n",
-      "33549    | torch.Size([1, 1000, 782]) | 13\n",
-      "33550    | torch.Size([1, 1000, 754]) | 13\n",
-      "33551    | torch.Size([1, 1000, 772]) | 13\n",
-      "33552    | torch.Size([1, 1000, 771]) | 13\n",
-      "33553    | torch.Size([1, 1000, 719]) | 13\n",
-      "33554    | torch.Size([1, 1000, 772]) | 13\n",
-      "33555    | torch.Size([1, 1000, 754]) | 13\n",
-      "33556    | torch.Size([1, 1000, 754]) | 13\n",
-      "33557    | torch.Size([1, 1000, 754]) | 13\n",
-      "33558    | torch.Size([1, 1000, 754]) | 13\n",
-      "33559    | torch.Size([1, 1000, 772]) | 13\n",
-      "33560    | torch.Size([1, 1000, 752]) | 13\n",
-      "33561    | torch.Size([1, 1000, 754]) | 13\n",
-      "33562    | torch.Size([1, 1000, 867]) | 13\n",
-      "33563    | torch.Size([1, 1000, 772]) | 13\n",
-      "33564    | torch.Size([1, 1000, 772]) | 13\n",
-      "33565    | torch.Size([1, 1000, 777]) | 13\n",
-      "33566    | torch.Size([1, 1000, 777]) | 13\n",
-      "33567    | torch.Size([1, 1000, 777]) | 13\n",
-      "33568    | torch.Size([1, 1000, 754]) | 13\n",
-      "33569    | torch.Size([1, 1000, 754]) | 13\n",
-      "33570    | torch.Size([1, 1000, 754]) | 13\n",
-      "33571    | torch.Size([1, 1000, 777]) | 13\n",
-      "33572    | torch.Size([1, 1000, 754]) | 13\n",
-      "33573    | torch.Size([1, 1000, 754]) | 13\n",
-      "33574    | torch.Size([1, 1000, 777]) | 13\n",
-      "33575    | torch.Size([1, 1000, 754]) | 13\n",
-      "33576    | torch.Size([1, 1000, 772]) | 13\n",
-      "33577    | torch.Size([1, 1000, 777]) | 13\n",
-      "33578    | torch.Size([1, 1000, 754]) | 13\n",
-      "33579    | torch.Size([1, 1000, 754]) | 13\n",
-      "33580    | torch.Size([1, 1000, 754]) | 13\n",
-      "33581    | torch.Size([1, 1000, 784]) | 13\n",
-      "33582    | torch.Size([1, 1000, 772]) | 13\n",
-      "33583    | torch.Size([1, 1000, 754]) | 13\n",
-      "33584    | torch.Size([1, 1000, 787]) | 13\n",
-      "33585    | torch.Size([1, 1000, 754]) | 13\n",
-      "33586    | torch.Size([1, 1000, 754]) | 13\n",
-      "33587    | torch.Size([1, 1000, 777]) | 13\n",
-      "33588    | torch.Size([1, 1000, 754]) | 13\n",
-      "33589    | torch.Size([1, 1000, 754]) | 13\n",
-      "33590    | torch.Size([1, 1000, 817]) | 13\n",
-      "33591    | torch.Size([1, 1000, 754]) | 13\n",
-      "33592    | torch.Size([1, 1000, 772]) | 13\n",
-      "33593    | torch.Size([1, 1000, 754]) | 13\n",
-      "33594    | torch.Size([1, 1000, 754]) | 13\n",
-      "33595    | torch.Size([1, 1000, 754]) | 13\n",
-      "33596    | torch.Size([1, 1000, 772]) | 13\n",
-      "33597    | torch.Size([1, 1000, 777]) | 13\n",
-      "33598    | torch.Size([1, 1000, 754]) | 13\n",
-      "33599    | torch.Size([1, 1000, 772]) | 13\n",
-      "33600    | torch.Size([1, 1000, 777]) | 13\n",
-      "33601    | torch.Size([1, 1000, 754]) | 13\n",
-      "33602    | torch.Size([1, 1000, 777]) | 13\n",
-      "33603    | torch.Size([1, 1000, 754]) | 13\n",
-      "33604    | torch.Size([1, 1000, 762]) | 13\n",
-      "33605    | torch.Size([1, 1000, 754]) | 13\n",
-      "33606    | torch.Size([1, 1000, 754]) | 13\n",
-      "33607    | torch.Size([1, 1000, 772]) | 13\n",
-      "33608    | torch.Size([1, 1000, 808]) | 13\n",
-      "33609    | torch.Size([1, 1000, 772]) | 13\n",
-      "33610    | torch.Size([1, 1000, 772]) | 13\n",
-      "33611    | torch.Size([1, 1000, 754]) | 13\n",
-      "33612    | torch.Size([1, 1000, 754]) | 13\n",
-      "33613    | torch.Size([1, 1000, 772]) | 13\n",
-      "33614    | torch.Size([1, 1000, 754]) | 13\n",
-      "33615    | torch.Size([1, 1000, 754]) | 13\n",
-      "33616    | torch.Size([1, 1000, 772]) | 13\n",
-      "33617    | torch.Size([1, 1000, 772]) | 13\n",
-      "33618    | torch.Size([1, 1000, 754]) | 13\n",
-      "33619    | torch.Size([1, 1000, 777]) | 13\n",
-      "33620    | torch.Size([1, 1000, 754]) | 13\n",
-      "33621    | torch.Size([1, 1000, 772]) | 13\n",
-      "33622    | torch.Size([1, 1000, 777]) | 13\n",
-      "33623    | torch.Size([1, 1000, 754]) | 13\n",
-      "33624    | torch.Size([1, 1000, 754]) | 13\n",
-      "33625    | torch.Size([1, 1000, 754]) | 13\n",
-      "33626    | torch.Size([1, 1000, 777]) | 13\n",
-      "33627    | torch.Size([1, 1000, 754]) | 13\n",
-      "33628    | torch.Size([1, 1000, 775]) | 13\n",
-      "33629    | torch.Size([1, 1000, 754]) | 13\n",
-      "33630    | torch.Size([1, 1000, 754]) | 13\n",
-      "33631    | torch.Size([1, 1000, 754]) | 13\n",
-      "33632    | torch.Size([1, 1000, 754]) | 13\n",
-      "33633    | torch.Size([1, 1000, 754]) | 13\n",
-      "33634    | torch.Size([1, 1000, 772]) | 13\n",
-      "33635    | torch.Size([1, 1000, 754]) | 13\n",
-      "33636    | torch.Size([1, 1000, 772]) | 13\n",
-      "33637    | torch.Size([1, 1000, 754]) | 13\n",
-      "33638    | torch.Size([1, 1000, 778]) | 13\n",
-      "33639    | torch.Size([1, 1000, 754]) | 13\n",
-      "33640    | torch.Size([1, 1000, 754]) | 13\n",
-      "33641    | torch.Size([1, 1000, 754]) | 13\n",
-      "33642    | torch.Size([1, 1000, 775]) | 13\n",
-      "33643    | torch.Size([1, 1000, 782]) | 13\n",
-      "33644    | torch.Size([1, 1000, 754]) | 13\n",
-      "33645    | torch.Size([1, 1000, 815]) | 13\n",
-      "33646    | torch.Size([1, 1000, 754]) | 13\n",
-      "33647    | torch.Size([1, 1000, 772]) | 13\n",
-      "33648    | torch.Size([1, 1000, 754]) | 13\n",
-      "33649    | torch.Size([1, 1000, 777]) | 13\n",
-      "33650    | torch.Size([1, 1000, 754]) | 13\n",
-      "33651    | torch.Size([1, 1000, 765]) | 13\n",
-      "33652    | torch.Size([1, 1000, 771]) | 13\n",
-      "33653    | torch.Size([1, 1000, 754]) | 13\n",
-      "33654    | torch.Size([1, 1000, 754]) | 13\n",
-      "33655    | torch.Size([1, 1000, 754]) | 13\n",
-      "33656    | torch.Size([1, 1000, 754]) | 13\n",
-      "33657    | torch.Size([1, 1000, 754]) | 13\n",
-      "33658    | torch.Size([1, 1000, 754]) | 13\n",
-      "33659    | torch.Size([1, 1000, 777]) | 13\n",
-      "33660    | torch.Size([1, 1000, 754]) | 13\n",
-      "33661    | torch.Size([1, 1000, 771]) | 13\n",
-      "33662    | torch.Size([1, 1000, 754]) | 13\n",
-      "33663    | torch.Size([1, 1000, 754]) | 13\n",
-      "33664    | torch.Size([1, 1000, 754]) | 13\n",
-      "33665    | torch.Size([1, 1000, 772]) | 13\n",
-      "33666    | torch.Size([1, 1000, 754]) | 13\n",
-      "33667    | torch.Size([1, 1000, 772]) | 13\n",
-      "33668    | torch.Size([1, 1000, 777]) | 13\n",
-      "33669    | torch.Size([1, 1000, 785]) | 13\n",
-      "33670    | torch.Size([1, 1000, 754]) | 13\n",
-      "33671    | torch.Size([1, 1000, 754]) | 13\n",
-      "33672    | torch.Size([1, 1000, 778]) | 13\n",
-      "33673    | torch.Size([1, 1000, 754]) | 13\n",
-      "33674    | torch.Size([1, 1000, 754]) | 13\n",
-      "33675    | torch.Size([1, 1000, 772]) | 13\n",
-      "33676    | torch.Size([1, 1000, 754]) | 13\n",
-      "33677    | torch.Size([1, 1000, 772]) | 13\n",
-      "33678    | torch.Size([1, 1000, 771]) | 13\n",
-      "33679    | torch.Size([1, 1000, 773]) | 13\n",
-      "33680    | torch.Size([1, 1000, 772]) | 13\n",
-      "33681    | torch.Size([1, 1000, 713]) | 13\n",
-      "33682    | torch.Size([1, 1000, 754]) | 13\n",
-      "33683    | torch.Size([1, 1000, 777]) | 13\n",
-      "33684    | torch.Size([1, 1000, 772]) | 13\n",
-      "33685    | torch.Size([1, 1000, 813]) | 13\n",
-      "33686    | torch.Size([1, 1000, 754]) | 13\n",
-      "33687    | torch.Size([1, 1000, 754]) | 13\n",
-      "33688    | torch.Size([1, 1000, 754]) | 13\n",
-      "33689    | torch.Size([1, 1000, 754]) | 13\n",
-      "33690    | torch.Size([1, 1000, 771]) | 13\n",
-      "33691    | torch.Size([1, 1000, 810]) | 13\n",
-      "33692    | torch.Size([1, 1000, 772]) | 13\n",
-      "33693    | torch.Size([1, 1000, 772]) | 13\n",
-      "33694    | torch.Size([1, 1000, 781]) | 13\n",
-      "33695    | torch.Size([1, 1000, 754]) | 13\n",
-      "33696    | torch.Size([1, 1000, 754]) | 13\n",
-      "33697    | torch.Size([1, 1000, 789]) | 13\n",
-      "33698    | torch.Size([1, 1000, 747]) | 13\n",
-      "33699    | torch.Size([1, 1000, 651]) | 13\n",
-      "33700    | torch.Size([1, 1000, 754]) | 13\n",
-      "33701    | torch.Size([1, 1000, 754]) | 13\n",
-      "33702    | torch.Size([1, 1000, 754]) | 13\n",
-      "33703    | torch.Size([1, 1000, 754]) | 13\n",
-      "33704    | torch.Size([1, 1000, 747]) | 13\n",
-      "33705    | torch.Size([1, 1000, 786]) | 13\n",
-      "33706    | torch.Size([1, 1000, 772]) | 13\n",
-      "33707    | torch.Size([1, 1000, 754]) | 13\n",
-      "33708    | torch.Size([1, 1000, 762]) | 13\n",
-      "33709    | torch.Size([1, 1000, 754]) | 13\n",
-      "33710    | torch.Size([1, 1000, 772]) | 13\n",
-      "33711    | torch.Size([1, 1000, 754]) | 13\n",
-      "33712    | torch.Size([1, 1000, 754]) | 13\n",
-      "33713    | torch.Size([1, 1000, 754]) | 13\n",
-      "33714    | torch.Size([1, 1000, 754]) | 13\n",
-      "33715    | torch.Size([1, 1000, 754]) | 13\n",
-      "33716    | torch.Size([1, 1000, 754]) | 13\n",
-      "33717    | torch.Size([1, 1000, 754]) | 13\n",
-      "33718    | torch.Size([1, 1000, 754]) | 13\n",
-      "33719    | torch.Size([1, 1000, 754]) | 13\n",
-      "33720    | torch.Size([1, 1000, 754]) | 13\n",
-      "33721    | torch.Size([1, 1000, 754]) | 13\n",
-      "33722    | torch.Size([1, 1000, 754]) | 13\n",
-      "33723    | torch.Size([1, 1000, 777]) | 13\n",
-      "33724    | torch.Size([1, 1000, 754]) | 13\n",
-      "33725    | torch.Size([1, 1000, 754]) | 13\n",
-      "33726    | torch.Size([1, 1000, 754]) | 13\n",
-      "33727    | torch.Size([1, 1000, 754]) | 13\n",
-      "33728    | torch.Size([1, 1000, 754]) | 13\n",
-      "33729    | torch.Size([1, 1000, 772]) | 13\n",
-      "33730    | torch.Size([1, 1000, 754]) | 13\n",
-      "33731    | torch.Size([1, 1000, 777]) | 13\n",
-      "33732    | torch.Size([1, 1000, 754]) | 13\n",
-      "33733    | torch.Size([1, 1000, 754]) | 13\n",
-      "33734    | torch.Size([1, 1000, 754]) | 13\n",
-      "33735    | torch.Size([1, 1000, 777]) | 13\n",
-      "33736    | torch.Size([1, 1000, 754]) | 13\n",
-      "33737    | torch.Size([1, 1000, 777]) | 13\n",
-      "33738    | torch.Size([1, 1000, 777]) | 13\n",
-      "33739    | torch.Size([1, 1000, 772]) | 13\n",
-      "33740    | torch.Size([1, 1000, 747]) | 13\n",
-      "33741    | torch.Size([1, 1000, 777]) | 13\n",
-      "33742    | torch.Size([1, 1000, 772]) | 13\n",
-      "33743    | torch.Size([1, 1000, 754]) | 13\n",
-      "33744    | torch.Size([1, 1000, 777]) | 13\n",
-      "33745    | torch.Size([1, 1000, 754]) | 13\n",
-      "33746    | torch.Size([1, 1000, 772]) | 13\n",
-      "33747    | torch.Size([1, 1000, 754]) | 13\n",
-      "33748    | torch.Size([1, 1000, 754]) | 13\n",
-      "33749    | torch.Size([1, 1000, 772]) | 13\n",
-      "33750    | torch.Size([1, 1000, 754]) | 13\n",
-      "33751    | torch.Size([1, 1000, 754]) | 13\n",
-      "33752    | torch.Size([1, 1000, 754]) | 13\n",
-      "33753    | torch.Size([1, 1000, 772]) | 13\n",
-      "33754    | torch.Size([1, 1000, 754]) | 13\n",
-      "33755    | torch.Size([1, 1000, 785]) | 13\n",
-      "33756    | torch.Size([1, 1000, 772]) | 13\n",
-      "33757    | torch.Size([1, 1000, 757]) | 13\n",
-      "33758    | torch.Size([1, 1000, 777]) | 13\n",
-      "33759    | torch.Size([1, 1000, 754]) | 13\n",
-      "33760    | torch.Size([1, 1000, 754]) | 13\n",
-      "33761    | torch.Size([1, 1000, 754]) | 13\n",
-      "33762    | torch.Size([1, 1000, 754]) | 13\n",
-      "33763    | torch.Size([1, 1000, 754]) | 13\n",
-      "33764    | torch.Size([1, 1000, 754]) | 13\n",
-      "33765    | torch.Size([1, 1000, 754]) | 13\n",
-      "33766    | torch.Size([1, 1000, 777]) | 13\n",
-      "33767    | torch.Size([1, 1000, 754]) | 13\n",
-      "33768    | torch.Size([1, 1000, 777]) | 13\n",
-      "33769    | torch.Size([1, 1000, 754]) | 13\n",
-      "33770    | torch.Size([1, 1000, 777]) | 13\n",
-      "33771    | torch.Size([1, 1000, 737]) | 13\n",
-      "33772    | torch.Size([1, 1000, 777]) | 13\n",
-      "33773    | torch.Size([1, 1000, 777]) | 13\n",
-      "33774    | torch.Size([1, 1000, 754]) | 13\n",
-      "33775    | torch.Size([1, 1000, 772]) | 13\n",
-      "33776    | torch.Size([1, 1000, 754]) | 13\n",
-      "33777    | torch.Size([1, 1000, 754]) | 13\n",
-      "33778    | torch.Size([1, 1000, 777]) | 13\n",
-      "33779    | torch.Size([1, 1000, 754]) | 13\n",
-      "33780    | torch.Size([1, 1000, 772]) | 13\n",
-      "33781    | torch.Size([1, 1000, 754]) | 13\n",
-      "33782    | torch.Size([1, 1000, 777]) | 13\n",
-      "33783    | torch.Size([1, 1000, 754]) | 13\n",
-      "33784    | torch.Size([1, 1000, 777]) | 13\n",
-      "33785    | torch.Size([1, 1000, 754]) | 13\n",
-      "33786    | torch.Size([1, 1000, 754]) | 13\n",
-      "33787    | torch.Size([1, 1000, 754]) | 13\n",
-      "33788    | torch.Size([1, 1000, 754]) | 13\n",
-      "33789    | torch.Size([1, 1000, 777]) | 13\n",
-      "33790    | torch.Size([1, 1000, 754]) | 13\n",
-      "33791    | torch.Size([1, 1000, 777]) | 13\n",
-      "33792    | torch.Size([1, 1000, 776]) | 13\n",
-      "33793    | torch.Size([1, 1000, 777]) | 13\n",
-      "33794    | torch.Size([1, 1000, 777]) | 13\n",
-      "33795    | torch.Size([1, 1000, 802]) | 13\n",
-      "33796    | torch.Size([1, 1000, 762]) | 13\n",
-      "33797    | torch.Size([1, 1000, 777]) | 13\n",
-      "33798    | torch.Size([1, 1000, 754]) | 13\n",
-      "33799    | torch.Size([1, 1000, 777]) | 13\n",
-      "33800    | torch.Size([1, 1000, 754]) | 13\n",
-      "33801    | torch.Size([1, 1000, 754]) | 13\n",
-      "33802    | torch.Size([1, 1000, 754]) | 13\n",
-      "33803    | torch.Size([1, 1000, 754]) | 13\n",
-      "33804    | torch.Size([1, 1000, 754]) | 13\n",
-      "33805    | torch.Size([1, 1000, 777]) | 13\n",
-      "33806    | torch.Size([1, 1000, 777]) | 13\n",
-      "33807    | torch.Size([1, 1000, 754]) | 13\n",
-      "33808    | torch.Size([1, 1000, 754]) | 13\n",
-      "33809    | torch.Size([1, 1000, 772]) | 13\n",
-      "33810    | torch.Size([1, 1000, 754]) | 13\n",
-      "33811    | torch.Size([1, 1000, 772]) | 13\n",
-      "33812    | torch.Size([1, 1000, 754]) | 13\n",
-      "33813    | torch.Size([1, 1000, 754]) | 13\n",
-      "33814    | torch.Size([1, 1000, 772]) | 13\n",
-      "33815    | torch.Size([1, 1000, 777]) | 13\n",
-      "33816    | torch.Size([1, 1000, 783]) | 13\n",
-      "33817    | torch.Size([1, 1000, 754]) | 13\n",
-      "33818    | torch.Size([1, 1000, 777]) | 13\n",
-      "33819    | torch.Size([1, 1000, 797]) | 13\n",
-      "33820    | torch.Size([1, 1000, 805]) | 13\n",
-      "33821    | torch.Size([1, 1000, 772]) | 13\n",
-      "33822    | torch.Size([1, 1000, 754]) | 13\n",
-      "33823    | torch.Size([1, 1000, 754]) | 13\n",
-      "33824    | torch.Size([1, 1000, 754]) | 13\n",
-      "33825    | torch.Size([1, 1000, 747]) | 13\n",
-      "33826    | torch.Size([1, 1000, 754]) | 13\n",
-      "33827    | torch.Size([1, 1000, 777]) | 13\n",
-      "33828    | torch.Size([1, 1000, 777]) | 13\n",
-      "33829    | torch.Size([1, 1000, 754]) | 13\n",
-      "33830    | torch.Size([1, 1000, 772]) | 13\n",
-      "33831    | torch.Size([1, 1000, 772]) | 13\n",
-      "33832    | torch.Size([1, 1000, 754]) | 13\n",
-      "33833    | torch.Size([1, 1000, 772]) | 13\n",
-      "33834    | torch.Size([1, 1000, 754]) | 13\n",
-      "33835    | torch.Size([1, 1000, 778]) | 13\n",
-      "33836    | torch.Size([1, 1000, 754]) | 13\n",
-      "33837    | torch.Size([1, 1000, 772]) | 13\n",
-      "33838    | torch.Size([1, 1000, 771]) | 13\n",
-      "33839    | torch.Size([1, 1000, 772]) | 13\n",
-      "33840    | torch.Size([1, 1000, 777]) | 13\n",
-      "33841    | torch.Size([1, 1000, 772]) | 13\n",
-      "33842    | torch.Size([1, 1000, 754]) | 13\n",
-      "33843    | torch.Size([1, 1000, 754]) | 13\n",
-      "33844    | torch.Size([1, 1000, 754]) | 13\n",
-      "33845    | torch.Size([1, 1000, 754]) | 13\n",
-      "33846    | torch.Size([1, 1000, 754]) | 13\n",
-      "33847    | torch.Size([1, 1000, 783]) | 13\n",
-      "33848    | torch.Size([1, 1000, 754]) | 13\n",
-      "33849    | torch.Size([1, 1000, 777]) | 13\n",
-      "33850    | torch.Size([1, 1000, 810]) | 13\n",
-      "33851    | torch.Size([1, 1000, 754]) | 13\n",
-      "33852    | torch.Size([1, 1000, 754]) | 13\n",
-      "33853    | torch.Size([1, 1000, 754]) | 13\n",
-      "33854    | torch.Size([1, 1000, 754]) | 13\n",
-      "33855    | torch.Size([1, 1000, 754]) | 13\n",
-      "33856    | torch.Size([1, 1000, 771]) | 13\n",
-      "33857    | torch.Size([1, 1000, 754]) | 13\n",
-      "33858    | torch.Size([1, 1000, 782]) | 13\n",
-      "33859    | torch.Size([1, 1000, 754]) | 13\n",
-      "33860    | torch.Size([1, 1000, 754]) | 13\n",
-      "33861    | torch.Size([1, 1000, 754]) | 13\n",
-      "33862    | torch.Size([1, 1000, 754]) | 13\n",
-      "33863    | torch.Size([1, 1000, 777]) | 13\n",
-      "33864    | torch.Size([1, 1000, 772]) | 13\n",
-      "33865    | torch.Size([1, 1000, 754]) | 13\n",
-      "33866    | torch.Size([1, 1000, 754]) | 13\n",
-      "33867    | torch.Size([1, 1000, 772]) | 13\n",
-      "33868    | torch.Size([1, 1000, 754]) | 13\n",
-      "33869    | torch.Size([1, 1000, 754]) | 13\n",
-      "33870    | torch.Size([1, 1000, 754]) | 13\n",
-      "33871    | torch.Size([1, 1000, 754]) | 13\n",
-      "33872    | torch.Size([1, 1000, 772]) | 13\n",
-      "33873    | torch.Size([1, 1000, 770]) | 13\n",
-      "33874    | torch.Size([1, 1000, 777]) | 13\n",
-      "33875    | torch.Size([1, 1000, 772]) | 13\n",
-      "33876    | torch.Size([1, 1000, 772]) | 13\n",
-      "33877    | torch.Size([1, 1000, 772]) | 13\n",
-      "33878    | torch.Size([1, 1000, 782]) | 13\n",
-      "33879    | torch.Size([1, 1000, 772]) | 13\n",
-      "33880    | torch.Size([1, 1000, 754]) | 13\n",
-      "33881    | torch.Size([1, 1000, 754]) | 13\n",
-      "33882    | torch.Size([1, 1000, 772]) | 13\n",
-      "33883    | torch.Size([1, 1000, 777]) | 13\n",
-      "33884    | torch.Size([1, 1000, 771]) | 13\n",
-      "33885    | torch.Size([1, 1000, 754]) | 13\n",
-      "33886    | torch.Size([1, 1000, 754]) | 13\n",
-      "33887    | torch.Size([1, 1000, 770]) | 13\n",
-      "33888    | torch.Size([1, 1000, 772]) | 13\n",
-      "33889    | torch.Size([1, 1000, 754]) | 13\n",
-      "33890    | torch.Size([1, 1000, 754]) | 13\n",
-      "33891    | torch.Size([1, 1000, 754]) | 13\n",
-      "33892    | torch.Size([1, 1000, 803]) | 13\n",
-      "33893    | torch.Size([1, 1000, 772]) | 13\n",
-      "33894    | torch.Size([1, 1000, 772]) | 13\n",
-      "33895    | torch.Size([1, 1000, 789]) | 13\n",
-      "33896    | torch.Size([1, 1000, 754]) | 13\n",
-      "33897    | torch.Size([1, 1000, 777]) | 13\n",
-      "33898    | torch.Size([1, 1000, 773]) | 13\n",
-      "33899    | torch.Size([1, 1000, 777]) | 13\n",
-      "33900    | torch.Size([1, 1000, 754]) | 13\n",
-      "33901    | torch.Size([1, 1000, 754]) | 13\n",
-      "33902    | torch.Size([1, 1000, 754]) | 13\n",
-      "33903    | torch.Size([1, 1000, 777]) | 13\n",
-      "33904    | torch.Size([1, 1000, 754]) | 13\n",
-      "33905    | torch.Size([1, 1000, 754]) | 13\n",
-      "33906    | torch.Size([1, 1000, 777]) | 13\n",
-      "33907    | torch.Size([1, 1000, 754]) | 13\n",
-      "33908    | torch.Size([1, 1000, 754]) | 13\n",
-      "33909    | torch.Size([1, 1000, 754]) | 13\n",
-      "33910    | torch.Size([1, 1000, 777]) | 13\n",
-      "33911    | torch.Size([1, 1000, 812]) | 13\n",
-      "33912    | torch.Size([1, 1000, 777]) | 13\n",
-      "33913    | torch.Size([1, 1000, 777]) | 13\n",
-      "33914    | torch.Size([1, 1000, 772]) | 13\n",
-      "33915    | torch.Size([1, 1000, 772]) | 13\n",
-      "33916    | torch.Size([1, 1000, 772]) | 13\n",
-      "33917    | torch.Size([1, 1000, 772]) | 13\n",
-      "33918    | torch.Size([1, 1000, 777]) | 13\n",
-      "33919    | torch.Size([1, 1000, 754]) | 13\n",
-      "33920    | torch.Size([1, 1000, 754]) | 13\n",
-      "33921    | torch.Size([1, 1000, 703]) | 13\n",
-      "33922    | torch.Size([1, 1000, 772]) | 13\n",
-      "33923    | torch.Size([1, 1000, 754]) | 13\n",
-      "33924    | torch.Size([1, 1000, 772]) | 13\n",
-      "33925    | torch.Size([1, 1000, 754]) | 13\n",
-      "33926    | torch.Size([1, 1000, 802]) | 13\n",
-      "33927    | torch.Size([1, 1000, 754]) | 13\n",
-      "33928    | torch.Size([1, 1000, 772]) | 13\n",
-      "33929    | torch.Size([1, 1000, 754]) | 13\n",
-      "33930    | torch.Size([1, 1000, 777]) | 13\n",
-      "33931    | torch.Size([1, 1000, 754]) | 13\n",
-      "33932    | torch.Size([1, 1000, 777]) | 13\n",
-      "33933    | torch.Size([1, 1000, 754]) | 13\n",
-      "33934    | torch.Size([1, 1000, 807]) | 13\n",
-      "33935    | torch.Size([1, 1000, 754]) | 13\n",
-      "33936    | torch.Size([1, 1000, 754]) | 13\n",
-      "33937    | torch.Size([1, 1000, 777]) | 13\n",
-      "33938    | torch.Size([1, 1000, 754]) | 13\n",
-      "33939    | torch.Size([1, 1000, 772]) | 13\n",
-      "33940    | torch.Size([1, 1000, 772]) | 13\n",
-      "33941    | torch.Size([1, 1000, 772]) | 13\n",
-      "33942    | torch.Size([1, 1000, 784]) | 13\n",
-      "33943    | torch.Size([1, 1000, 754]) | 13\n",
-      "33944    | torch.Size([1, 1000, 703]) | 13\n",
-      "33945    | torch.Size([1, 1000, 754]) | 13\n",
-      "33946    | torch.Size([1, 1000, 777]) | 13\n",
-      "33947    | torch.Size([1, 1000, 754]) | 13\n",
-      "33948    | torch.Size([1, 1000, 772]) | 13\n",
-      "33949    | torch.Size([1, 1000, 783]) | 13\n",
-      "33950    | torch.Size([1, 1000, 754]) | 13\n",
-      "33951    | torch.Size([1, 1000, 771]) | 13\n",
-      "33952    | torch.Size([1, 1000, 754]) | 13\n",
-      "33953    | torch.Size([1, 1000, 754]) | 13\n",
-      "33954    | torch.Size([1, 1000, 754]) | 13\n",
-      "33955    | torch.Size([1, 1000, 754]) | 13\n",
-      "33956    | torch.Size([1, 1000, 772]) | 13\n",
-      "33957    | torch.Size([1, 1000, 773]) | 13\n",
-      "33958    | torch.Size([1, 1000, 754]) | 13\n",
-      "33959    | torch.Size([1, 1000, 772]) | 13\n",
-      "33960    | torch.Size([1, 1000, 754]) | 13\n",
-      "33961    | torch.Size([1, 1000, 754]) | 13\n",
-      "33962    | torch.Size([1, 1000, 754]) | 13\n",
-      "33963    | torch.Size([1, 1000, 792]) | 13\n",
-      "33964    | torch.Size([1, 1000, 754]) | 13\n",
-      "33965    | torch.Size([1, 1000, 777]) | 13\n",
-      "33966    | torch.Size([1, 1000, 754]) | 13\n",
-      "33967    | torch.Size([1, 1000, 771]) | 13\n",
-      "33968    | torch.Size([1, 1000, 777]) | 13\n",
-      "33969    | torch.Size([1, 1000, 754]) | 13\n",
-      "33970    | torch.Size([1, 1000, 754]) | 13\n",
-      "33971    | torch.Size([1, 1000, 754]) | 13\n",
-      "33972    | torch.Size([1, 1000, 754]) | 13\n",
-      "33973    | torch.Size([1, 1000, 754]) | 13\n",
-      "33974    | torch.Size([1, 1000, 772]) | 13\n",
-      "33975    | torch.Size([1, 1000, 754]) | 13\n",
-      "33976    | torch.Size([1, 1000, 754]) | 13\n",
-      "33977    | torch.Size([1, 1000, 754]) | 13\n",
-      "33978    | torch.Size([1, 1000, 772]) | 13\n",
-      "33979    | torch.Size([1, 1000, 754]) | 13\n",
-      "33980    | torch.Size([1, 1000, 778]) | 13\n",
-      "33981    | torch.Size([1, 1000, 754]) | 13\n",
-      "33982    | torch.Size([1, 1000, 777]) | 13\n",
-      "33983    | torch.Size([1, 1000, 786]) | 13\n",
-      "33984    | torch.Size([1, 1000, 754]) | 13\n",
-      "33985    | torch.Size([1, 1000, 777]) | 13\n",
-      "33986    | torch.Size([1, 1000, 772]) | 13\n",
-      "33987    | torch.Size([1, 1000, 754]) | 13\n",
-      "33988    | torch.Size([1, 1000, 754]) | 13\n",
-      "33989    | torch.Size([1, 1000, 754]) | 13\n",
-      "33990    | torch.Size([1, 1000, 771]) | 13\n",
-      "33991    | torch.Size([1, 1000, 772]) | 13\n",
-      "33992    | torch.Size([1, 1000, 754]) | 13\n",
-      "33993    | torch.Size([1, 1000, 754]) | 13\n",
-      "33994    | torch.Size([1, 1000, 772]) | 13\n",
-      "33995    | torch.Size([1, 1000, 754]) | 13\n",
-      "33996    | torch.Size([1, 1000, 754]) | 13\n",
-      "33997    | torch.Size([1, 1000, 754]) | 13\n",
-      "33998    | torch.Size([1, 1000, 772]) | 13\n",
-      "33999    | torch.Size([1, 1000, 777]) | 13\n",
-      "34000    | torch.Size([1, 1000, 777]) | 13\n",
-      "34001    | torch.Size([1, 1000, 772]) | 13\n",
-      "34002    | torch.Size([1, 1000, 772]) | 13\n",
-      "34003    | torch.Size([1, 1000, 754]) | 13\n",
-      "34004    | torch.Size([1, 1000, 754]) | 13\n",
-      "34005    | torch.Size([1, 1000, 754]) | 13\n",
-      "34006    | torch.Size([1, 1000, 754]) | 13\n",
-      "34007    | torch.Size([1, 1000, 772]) | 13\n",
-      "34008    | torch.Size([1, 1000, 754]) | 13\n",
-      "34009    | torch.Size([1, 1000, 847]) | 13\n",
-      "34010    | torch.Size([1, 1000, 872]) | 13\n",
-      "34011    | torch.Size([1, 1000, 754]) | 13\n",
-      "34012    | torch.Size([1, 1000, 777]) | 13\n",
-      "34013    | torch.Size([1, 1000, 754]) | 13\n",
-      "34014    | torch.Size([1, 1000, 777]) | 13\n",
-      "34015    | torch.Size([1, 1000, 754]) | 13\n",
-      "34016    | torch.Size([1, 1000, 772]) | 13\n",
-      "34017    | torch.Size([1, 1000, 771]) | 13\n",
-      "34018    | torch.Size([1, 1000, 754]) | 13\n",
-      "34019    | torch.Size([1, 1000, 754]) | 13\n",
-      "34020    | torch.Size([1, 1000, 754]) | 13\n",
-      "34021    | torch.Size([1, 1000, 762]) | 13\n",
-      "34022    | torch.Size([1, 1000, 754]) | 13\n",
-      "34023    | torch.Size([1, 1000, 754]) | 13\n",
-      "34024    | torch.Size([1, 1000, 793]) | 13\n",
-      "34025    | torch.Size([1, 1000, 777]) | 13\n",
-      "34026    | torch.Size([1, 1000, 754]) | 13\n",
-      "34027    | torch.Size([1, 1000, 754]) | 13\n",
-      "34028    | torch.Size([1, 1000, 754]) | 13\n",
-      "34029    | torch.Size([1, 1000, 754]) | 13\n",
-      "34030    | torch.Size([1, 1000, 754]) | 13\n",
-      "34031    | torch.Size([1, 1000, 772]) | 13\n",
-      "34032    | torch.Size([1, 1000, 777]) | 13\n",
-      "34033    | torch.Size([1, 1000, 777]) | 13\n",
-      "34034    | torch.Size([1, 1000, 777]) | 13\n",
-      "34035    | torch.Size([1, 1000, 754]) | 13\n",
-      "34036    | torch.Size([1, 1000, 777]) | 13\n",
-      "34037    | torch.Size([1, 1000, 777]) | 13\n",
-      "34038    | torch.Size([1, 1000, 754]) | 13\n",
-      "34039    | torch.Size([1, 1000, 772]) | 13\n",
-      "34040    | torch.Size([1, 1000, 772]) | 13\n",
-      "34041    | torch.Size([1, 1000, 777]) | 13\n",
-      "34042    | torch.Size([1, 1000, 754]) | 13\n",
-      "34043    | torch.Size([1, 1000, 782]) | 13\n",
-      "34044    | torch.Size([1, 1000, 754]) | 13\n",
-      "34045    | torch.Size([1, 1000, 754]) | 13\n",
-      "34046    | torch.Size([1, 1000, 786]) | 13\n",
-      "34047    | torch.Size([1, 1000, 728]) | 13\n",
-      "34048    | torch.Size([1, 1000, 777]) | 13\n",
-      "34049    | torch.Size([1, 1000, 754]) | 13\n",
-      "34050    | torch.Size([1, 1000, 754]) | 13\n",
-      "34051    | torch.Size([1, 1000, 754]) | 13\n",
-      "34052    | torch.Size([1, 1000, 754]) | 13\n",
-      "34053    | torch.Size([1, 1000, 772]) | 13\n",
-      "34054    | torch.Size([1, 1000, 754]) | 13\n",
-      "34055    | torch.Size([1, 1000, 754]) | 13\n",
-      "34056    | torch.Size([1, 1000, 768]) | 13\n",
-      "34057    | torch.Size([1, 1000, 777]) | 13\n",
-      "34058    | torch.Size([1, 1000, 754]) | 13\n",
-      "34059    | torch.Size([1, 1000, 754]) | 13\n",
-      "34060    | torch.Size([1, 1000, 754]) | 13\n",
-      "34061    | torch.Size([1, 1000, 795]) | 13\n",
-      "34062    | torch.Size([1, 1000, 703]) | 13\n",
-      "34063    | torch.Size([1, 1000, 754]) | 13\n",
-      "34064    | torch.Size([1, 1000, 777]) | 13\n",
-      "34065    | torch.Size([1, 1000, 844]) | 13\n",
-      "34066    | torch.Size([1, 1000, 754]) | 13\n",
-      "34067    | torch.Size([1, 1000, 754]) | 13\n",
-      "34068    | torch.Size([1, 1000, 754]) | 13\n",
-      "34069    | torch.Size([1, 1000, 772]) | 13\n",
-      "34070    | torch.Size([1, 1000, 772]) | 13\n",
-      "34071    | torch.Size([1, 1000, 754]) | 13\n",
-      "34072    | torch.Size([1, 1000, 764]) | 13\n",
-      "34073    | torch.Size([1, 1000, 754]) | 13\n",
-      "34074    | torch.Size([1, 1000, 777]) | 13\n",
-      "34075    | torch.Size([1, 1000, 754]) | 13\n",
-      "34076    | torch.Size([1, 1000, 754]) | 13\n",
-      "34077    | torch.Size([1, 1000, 777]) | 13\n",
-      "34078    | torch.Size([1, 1000, 754]) | 13\n",
-      "34079    | torch.Size([1, 1000, 754]) | 13\n",
-      "34080    | torch.Size([1, 1000, 773]) | 13\n",
-      "34081    | torch.Size([1, 1000, 772]) | 13\n",
-      "34082    | torch.Size([1, 1000, 754]) | 13\n",
-      "34083    | torch.Size([1, 1000, 754]) | 13\n",
-      "34084    | torch.Size([1, 1000, 792]) | 13\n",
-      "34085    | torch.Size([1, 1000, 754]) | 13\n",
-      "34086    | torch.Size([1, 1000, 754]) | 13\n",
-      "34087    | torch.Size([1, 1000, 754]) | 13\n",
-      "34088    | torch.Size([1, 1000, 754]) | 13\n",
-      "34089    | torch.Size([1, 1000, 754]) | 13\n",
-      "34090    | torch.Size([1, 1000, 784]) | 13\n",
-      "34091    | torch.Size([1, 1000, 777]) | 13\n",
-      "34092    | torch.Size([1, 1000, 754]) | 13\n",
-      "34093    | torch.Size([1, 1000, 777]) | 13\n",
-      "34094    | torch.Size([1, 1000, 754]) | 13\n",
-      "34095    | torch.Size([1, 1000, 790]) | 13\n",
-      "34096    | torch.Size([1, 1000, 754]) | 13\n",
-      "34097    | torch.Size([1, 1000, 754]) | 13\n",
-      "34098    | torch.Size([1, 1000, 754]) | 13\n",
-      "34099    | torch.Size([1, 1000, 754]) | 13\n",
-      "34100    | torch.Size([1, 1000, 754]) | 13\n",
-      "34101    | torch.Size([1, 1000, 754]) | 13\n",
-      "34102    | torch.Size([1, 1000, 754]) | 13\n",
-      "34103    | torch.Size([1, 1000, 754]) | 13\n",
-      "34104    | torch.Size([1, 1000, 771]) | 13\n",
-      "34105    | torch.Size([1, 1000, 754]) | 13\n",
-      "34106    | torch.Size([1, 1000, 754]) | 13\n",
-      "34107    | torch.Size([1, 1000, 754]) | 13\n",
-      "34108    | torch.Size([1, 1000, 777]) | 13\n",
-      "34109    | torch.Size([1, 1000, 754]) | 13\n",
-      "34110    | torch.Size([1, 1000, 754]) | 13\n",
-      "34111    | torch.Size([1, 1000, 772]) | 13\n",
-      "34112    | torch.Size([1, 1000, 777]) | 13\n",
-      "34113    | torch.Size([1, 1000, 772]) | 13\n",
-      "34114    | torch.Size([1, 1000, 754]) | 13\n",
-      "34115    | torch.Size([1, 1000, 754]) | 13\n",
-      "34116    | torch.Size([1, 1000, 772]) | 13\n",
-      "34117    | torch.Size([1, 1000, 754]) | 13\n",
-      "34118    | torch.Size([1, 1000, 754]) | 13\n",
-      "34119    | torch.Size([1, 1000, 772]) | 13\n",
-      "34120    | torch.Size([1, 1000, 777]) | 13\n",
-      "34121    | torch.Size([1, 1000, 754]) | 13\n",
-      "34122    | torch.Size([1, 1000, 754]) | 13\n",
-      "34123    | torch.Size([1, 1000, 772]) | 13\n",
-      "34124    | torch.Size([1, 1000, 754]) | 13\n",
-      "34125    | torch.Size([1, 1000, 777]) | 13\n",
-      "34126    | torch.Size([1, 1000, 772]) | 13\n",
-      "34127    | torch.Size([1, 1000, 772]) | 13\n",
-      "34128    | torch.Size([1, 1000, 754]) | 13\n",
-      "34129    | torch.Size([1, 1000, 772]) | 13\n",
-      "34130    | torch.Size([1, 1000, 754]) | 13\n",
-      "34131    | torch.Size([1, 1000, 754]) | 13\n",
-      "34132    | torch.Size([1, 1000, 776]) | 13\n",
-      "34133    | torch.Size([1, 1000, 754]) | 13\n",
-      "34134    | torch.Size([1, 1000, 777]) | 13\n",
-      "34135    | torch.Size([1, 1000, 809]) | 13\n",
-      "34136    | torch.Size([1, 1000, 777]) | 13\n",
-      "34137    | torch.Size([1, 1000, 772]) | 13\n",
-      "34138    | torch.Size([1, 1000, 777]) | 13\n",
-      "34139    | torch.Size([1, 1000, 754]) | 13\n",
-      "34140    | torch.Size([1, 1000, 754]) | 13\n",
-      "34141    | torch.Size([1, 1000, 777]) | 13\n",
-      "34142    | torch.Size([1, 1000, 754]) | 13\n",
-      "34143    | torch.Size([1, 1000, 777]) | 13\n",
-      "34144    | torch.Size([1, 1000, 815]) | 13\n",
-      "34145    | torch.Size([1, 1000, 777]) | 13\n",
-      "34146    | torch.Size([1, 1000, 754]) | 13\n",
-      "34147    | torch.Size([1, 1000, 754]) | 13\n",
-      "34148    | torch.Size([1, 1000, 772]) | 13\n",
-      "34149    | torch.Size([1, 1000, 754]) | 13\n",
-      "34150    | torch.Size([1, 1000, 754]) | 13\n",
-      "34151    | torch.Size([1, 1000, 777]) | 13\n",
-      "34152    | torch.Size([1, 1000, 754]) | 13\n",
-      "34153    | torch.Size([1, 1000, 754]) | 13\n",
-      "34154    | torch.Size([1, 1000, 777]) | 13\n",
-      "34155    | torch.Size([1, 1000, 754]) | 13\n",
-      "34156    | torch.Size([1, 1000, 772]) | 13\n",
-      "34157    | torch.Size([1, 1000, 754]) | 13\n",
-      "34158    | torch.Size([1, 1000, 777]) | 13\n",
-      "34159    | torch.Size([1, 1000, 777]) | 13\n",
-      "34160    | torch.Size([1, 1000, 772]) | 13\n",
-      "34161    | torch.Size([1, 1000, 754]) | 13\n",
-      "34162    | torch.Size([1, 1000, 754]) | 13\n",
-      "34163    | torch.Size([1, 1000, 754]) | 13\n",
-      "34164    | torch.Size([1, 1000, 754]) | 13\n",
-      "34165    | torch.Size([1, 1000, 772]) | 13\n",
-      "34166    | torch.Size([1, 1000, 777]) | 13\n",
-      "34167    | torch.Size([1, 1000, 754]) | 13\n",
-      "34168    | torch.Size([1, 1000, 754]) | 13\n",
-      "34169    | torch.Size([1, 1000, 771]) | 13\n",
-      "34170    | torch.Size([1, 1000, 772]) | 13\n",
-      "34171    | torch.Size([1, 1000, 772]) | 13\n",
-      "34172    | torch.Size([1, 1000, 754]) | 13\n",
-      "34173    | torch.Size([1, 1000, 754]) | 13\n",
-      "34174    | torch.Size([1, 1000, 754]) | 13\n",
-      "34175    | torch.Size([1, 1000, 772]) | 13\n",
-      "34176    | torch.Size([1, 1000, 757]) | 13\n",
-      "34177    | torch.Size([1, 1000, 754]) | 13\n",
-      "34178    | torch.Size([1, 1000, 754]) | 13\n",
-      "34179    | torch.Size([1, 1000, 772]) | 13\n",
-      "34180    | torch.Size([1, 1000, 772]) | 13\n",
-      "34181    | torch.Size([1, 1000, 754]) | 13\n",
-      "34182    | torch.Size([1, 1000, 791]) | 13\n",
-      "34183    | torch.Size([1, 1000, 754]) | 13\n",
-      "34184    | torch.Size([1, 1000, 754]) | 13\n",
-      "34185    | torch.Size([1, 1000, 754]) | 13\n",
-      "34186    | torch.Size([1, 1000, 777]) | 13\n",
-      "34187    | torch.Size([1, 1000, 754]) | 13\n",
-      "34188    | torch.Size([1, 1000, 754]) | 13\n",
-      "34189    | torch.Size([1, 1000, 772]) | 13\n",
-      "34190    | torch.Size([1, 1000, 777]) | 13\n",
-      "34191    | torch.Size([1, 1000, 754]) | 13\n",
-      "34192    | torch.Size([1, 1000, 754]) | 13\n",
-      "34193    | torch.Size([1, 1000, 777]) | 13\n",
-      "34194    | torch.Size([1, 1000, 752]) | 13\n",
-      "34195    | torch.Size([1, 1000, 754]) | 13\n",
-      "34196    | torch.Size([1, 1000, 754]) | 13\n",
-      "34197    | torch.Size([1, 1000, 754]) | 13\n",
-      "34198    | torch.Size([1, 1000, 772]) | 13\n",
-      "34199    | torch.Size([1, 1000, 754]) | 13\n",
-      "34200    | torch.Size([1, 1000, 772]) | 13\n",
-      "34201    | torch.Size([1, 1000, 724]) | 13\n",
-      "34202    | torch.Size([1, 1000, 777]) | 13\n",
-      "34203    | torch.Size([1, 1000, 777]) | 13\n",
-      "34204    | torch.Size([1, 1000, 754]) | 13\n",
-      "34205    | torch.Size([1, 1000, 754]) | 13\n",
-      "34206    | torch.Size([1, 1000, 754]) | 13\n",
-      "34207    | torch.Size([1, 1000, 754]) | 13\n",
-      "34208    | torch.Size([1, 1000, 777]) | 13\n",
-      "34209    | torch.Size([1, 1000, 754]) | 13\n",
-      "34210    | torch.Size([1, 1000, 772]) | 13\n",
-      "34211    | torch.Size([1, 1000, 754]) | 13\n",
-      "34212    | torch.Size([1, 1000, 754]) | 13\n",
-      "34213    | torch.Size([1, 1000, 754]) | 13\n",
-      "34214    | torch.Size([1, 1000, 754]) | 13\n",
-      "34215    | torch.Size([1, 1000, 754]) | 13\n",
-      "34216    | torch.Size([1, 1000, 754]) | 13\n",
-      "34217    | torch.Size([1, 1000, 783]) | 13\n",
-      "34218    | torch.Size([1, 1000, 754]) | 13\n",
-      "34219    | torch.Size([1, 1000, 754]) | 13\n",
-      "34220    | torch.Size([1, 1000, 754]) | 13\n",
-      "34221    | torch.Size([1, 1000, 777]) | 13\n",
-      "34222    | torch.Size([1, 1000, 777]) | 13\n",
-      "34223    | torch.Size([1, 1000, 772]) | 13\n",
-      "34224    | torch.Size([1, 1000, 777]) | 13\n",
-      "34225    | torch.Size([1, 1000, 849]) | 13\n",
-      "34226    | torch.Size([1, 1000, 754]) | 13\n",
-      "34227    | torch.Size([1, 1000, 809]) | 13\n",
-      "34228    | torch.Size([1, 1000, 754]) | 13\n",
-      "34229    | torch.Size([1, 1000, 790]) | 13\n",
-      "34230    | torch.Size([1, 1000, 793]) | 13\n",
-      "34231    | torch.Size([1, 1000, 754]) | 13\n",
-      "34232    | torch.Size([1, 1000, 754]) | 13\n",
-      "34233    | torch.Size([1, 1000, 771]) | 13\n",
-      "34234    | torch.Size([1, 1000, 797]) | 13\n",
-      "34235    | torch.Size([1, 1000, 772]) | 13\n",
-      "34236    | torch.Size([1, 1000, 777]) | 13\n",
-      "34237    | torch.Size([1, 1000, 754]) | 13\n",
-      "34238    | torch.Size([1, 1000, 754]) | 13\n",
-      "34239    | torch.Size([1, 1000, 754]) | 13\n",
-      "34240    | torch.Size([1, 1000, 777]) | 13\n",
-      "34241    | torch.Size([1, 1000, 772]) | 13\n",
-      "34242    | torch.Size([1, 1000, 754]) | 13\n",
-      "34243    | torch.Size([1, 1000, 754]) | 13\n",
-      "34244    | torch.Size([1, 1000, 754]) | 13\n",
-      "34245    | torch.Size([1, 1000, 771]) | 13\n",
-      "34246    | torch.Size([1, 1000, 754]) | 13\n",
-      "34247    | torch.Size([1, 1000, 772]) | 13\n",
-      "34248    | torch.Size([1, 1000, 777]) | 13\n",
-      "34249    | torch.Size([1, 1000, 777]) | 13\n",
-      "34250    | torch.Size([1, 1000, 754]) | 13\n",
-      "34251    | torch.Size([1, 1000, 777]) | 13\n",
-      "34252    | torch.Size([1, 1000, 771]) | 13\n",
-      "34253    | torch.Size([1, 1000, 754]) | 13\n",
-      "34254    | torch.Size([1, 1000, 771]) | 13\n",
-      "34255    | torch.Size([1, 1000, 754]) | 13\n",
-      "34256    | torch.Size([1, 1000, 772]) | 13\n",
-      "34257    | torch.Size([1, 1000, 754]) | 13\n",
-      "34258    | torch.Size([1, 1000, 777]) | 13\n",
-      "34259    | torch.Size([1, 1000, 772]) | 13\n",
-      "34260    | torch.Size([1, 1000, 754]) | 13\n",
-      "34261    | torch.Size([1, 1000, 798]) | 13\n",
-      "34262    | torch.Size([1, 1000, 754]) | 13\n",
-      "34263    | torch.Size([1, 1000, 772]) | 13\n",
-      "34264    | torch.Size([1, 1000, 754]) | 13\n",
-      "34265    | torch.Size([1, 1000, 754]) | 13\n",
-      "34266    | torch.Size([1, 1000, 777]) | 13\n",
-      "34267    | torch.Size([1, 1000, 754]) | 13\n",
-      "34268    | torch.Size([1, 1000, 777]) | 13\n",
-      "34269    | torch.Size([1, 1000, 777]) | 13\n",
-      "34270    | torch.Size([1, 1000, 777]) | 13\n",
-      "34271    | torch.Size([1, 1000, 754]) | 13\n",
-      "34272    | torch.Size([1, 1000, 754]) | 13\n",
-      "34273    | torch.Size([1, 1000, 783]) | 13\n",
-      "34274    | torch.Size([1, 1000, 754]) | 13\n",
-      "34275    | torch.Size([1, 1000, 771]) | 13\n",
-      "34276    | torch.Size([1, 1000, 754]) | 13\n",
-      "34277    | torch.Size([1, 1000, 740]) | 13\n",
-      "34278    | torch.Size([1, 1000, 754]) | 13\n",
-      "34279    | torch.Size([1, 1000, 780]) | 13\n",
-      "34280    | torch.Size([1, 1000, 777]) | 13\n",
-      "34281    | torch.Size([1, 1000, 754]) | 13\n",
-      "34282    | torch.Size([1, 1000, 777]) | 13\n",
-      "34283    | torch.Size([1, 1000, 754]) | 13\n",
-      "34284    | torch.Size([1, 1000, 754]) | 13\n",
-      "34285    | torch.Size([1, 1000, 754]) | 13\n",
-      "34286    | torch.Size([1, 1000, 777]) | 13\n",
-      "34287    | torch.Size([1, 1000, 754]) | 13\n",
-      "34288    | torch.Size([1, 1000, 754]) | 13\n",
-      "34289    | torch.Size([1, 1000, 779]) | 13\n",
-      "34290    | torch.Size([1, 1000, 754]) | 13\n",
-      "34291    | torch.Size([1, 1000, 728]) | 13\n",
-      "34292    | torch.Size([1, 1000, 754]) | 13\n",
-      "34293    | torch.Size([1, 1000, 754]) | 13\n",
-      "34294    | torch.Size([1, 1000, 777]) | 13\n",
-      "34295    | torch.Size([1, 1000, 772]) | 13\n",
-      "34296    | torch.Size([1, 1000, 754]) | 13\n",
-      "34297    | torch.Size([1, 1000, 754]) | 13\n",
-      "34298    | torch.Size([1, 1000, 754]) | 13\n",
-      "34299    | torch.Size([1, 1000, 772]) | 13\n",
-      "34300    | torch.Size([1, 1000, 772]) | 13\n",
-      "34301    | torch.Size([1, 1000, 754]) | 13\n",
-      "34302    | torch.Size([1, 1000, 754]) | 13\n",
-      "34303    | torch.Size([1, 1000, 754]) | 13\n",
-      "34304    | torch.Size([1, 1000, 754]) | 13\n",
-      "34305    | torch.Size([1, 1000, 754]) | 13\n",
-      "34306    | torch.Size([1, 1000, 772]) | 13\n",
-      "34307    | torch.Size([1, 1000, 772]) | 13\n",
-      "34308    | torch.Size([1, 1000, 772]) | 13\n",
-      "34309    | torch.Size([1, 1000, 789]) | 13\n",
-      "34310    | torch.Size([1, 1000, 777]) | 13\n",
-      "34311    | torch.Size([1, 1000, 777]) | 13\n",
-      "34312    | torch.Size([1, 1000, 777]) | 13\n",
-      "34313    | torch.Size([1, 1000, 754]) | 13\n",
-      "34314    | torch.Size([1, 1000, 754]) | 13\n",
-      "34315    | torch.Size([1, 1000, 773]) | 13\n",
-      "34316    | torch.Size([1, 1000, 754]) | 13\n",
-      "34317    | torch.Size([1, 1000, 777]) | 13\n",
-      "34318    | torch.Size([1, 1000, 754]) | 13\n",
-      "34319    | torch.Size([1, 1000, 754]) | 13\n",
-      "34320    | torch.Size([1, 1000, 761]) | 13\n",
-      "34321    | torch.Size([1, 1000, 766]) | 13\n",
-      "34322    | torch.Size([1, 1000, 777]) | 13\n",
-      "34323    | torch.Size([1, 1000, 754]) | 13\n",
-      "34324    | torch.Size([1, 1000, 782]) | 13\n",
-      "34325    | torch.Size([1, 1000, 772]) | 13\n",
-      "34326    | torch.Size([1, 1000, 781]) | 13\n",
-      "34327    | torch.Size([1, 1000, 777]) | 13\n",
-      "34328    | torch.Size([1, 1000, 772]) | 13\n",
-      "34329    | torch.Size([1, 1000, 772]) | 13\n",
-      "34330    | torch.Size([1, 1000, 754]) | 13\n",
-      "34331    | torch.Size([1, 1000, 754]) | 13\n",
-      "34332    | torch.Size([1, 1000, 777]) | 13\n",
-      "34333    | torch.Size([1, 1000, 772]) | 13\n",
-      "34334    | torch.Size([1, 1000, 754]) | 13\n",
-      "34335    | torch.Size([1, 1000, 754]) | 13\n",
-      "34336    | torch.Size([1, 1000, 777]) | 13\n",
-      "34337    | torch.Size([1, 1000, 777]) | 13\n",
-      "34338    | torch.Size([1, 1000, 772]) | 13\n",
-      "34339    | torch.Size([1, 1000, 754]) | 13\n",
-      "34340    | torch.Size([1, 1000, 772]) | 13\n",
-      "34341    | torch.Size([1, 1000, 754]) | 13\n",
-      "34342    | torch.Size([1, 1000, 754]) | 13\n",
-      "34343    | torch.Size([1, 1000, 772]) | 13\n",
-      "34344    | torch.Size([1, 1000, 754]) | 13\n",
-      "34345    | torch.Size([1, 1000, 754]) | 13\n",
-      "34346    | torch.Size([1, 1000, 815]) | 13\n",
-      "34347    | torch.Size([1, 1000, 777]) | 13\n",
-      "34348    | torch.Size([1, 1000, 754]) | 13\n",
-      "34349    | torch.Size([1, 1000, 772]) | 13\n",
-      "34350    | torch.Size([1, 1000, 775]) | 13\n",
-      "34351    | torch.Size([1, 1000, 777]) | 13\n",
-      "34352    | torch.Size([1, 1000, 772]) | 13\n",
-      "34353    | torch.Size([1, 1000, 754]) | 13\n",
-      "34354    | torch.Size([1, 1000, 754]) | 13\n",
-      "34355    | torch.Size([1, 1000, 788]) | 13\n",
-      "34356    | torch.Size([1, 1000, 754]) | 13\n",
-      "34357    | torch.Size([1, 1000, 754]) | 13\n",
-      "34358    | torch.Size([1, 1000, 754]) | 13\n",
-      "34359    | torch.Size([1, 1000, 793]) | 13\n",
-      "34360    | torch.Size([1, 1000, 754]) | 13\n",
-      "34361    | torch.Size([1, 1000, 754]) | 13\n",
-      "34362    | torch.Size([1, 1000, 772]) | 13\n",
-      "34363    | torch.Size([1, 1000, 754]) | 13\n",
-      "34364    | torch.Size([1, 1000, 772]) | 13\n",
-      "34365    | torch.Size([1, 1000, 772]) | 13\n",
-      "34366    | torch.Size([1, 1000, 777]) | 13\n",
-      "34367    | torch.Size([1, 1000, 754]) | 13\n",
-      "34368    | torch.Size([1, 1000, 754]) | 13\n",
-      "34369    | torch.Size([1, 1000, 754]) | 13\n",
-      "34370    | torch.Size([1, 1000, 754]) | 13\n",
-      "34371    | torch.Size([1, 1000, 777]) | 13\n",
-      "34372    | torch.Size([1, 1000, 731]) | 13\n",
-      "34373    | torch.Size([1, 1000, 754]) | 13\n",
-      "34374    | torch.Size([1, 1000, 754]) | 13\n",
-      "34375    | torch.Size([1, 1000, 754]) | 13\n",
-      "34376    | torch.Size([1, 1000, 797]) | 13\n",
-      "34377    | torch.Size([1, 1000, 754]) | 13\n",
-      "34378    | torch.Size([1, 1000, 754]) | 13\n",
-      "34379    | torch.Size([1, 1000, 754]) | 13\n",
-      "34380    | torch.Size([1, 1000, 754]) | 13\n",
-      "34381    | torch.Size([1, 1000, 772]) | 13\n",
-      "34382    | torch.Size([1, 1000, 754]) | 13\n",
-      "34383    | torch.Size([1, 1000, 754]) | 13\n",
-      "34384    | torch.Size([1, 1000, 754]) | 13\n",
-      "34385    | torch.Size([1, 1000, 772]) | 13\n",
-      "34386    | torch.Size([1, 1000, 754]) | 13\n",
-      "34387    | torch.Size([1, 1000, 754]) | 13\n",
-      "34388    | torch.Size([1, 1000, 772]) | 13\n",
-      "34389    | torch.Size([1, 1000, 772]) | 13\n",
-      "34390    | torch.Size([1, 1000, 771]) | 13\n",
-      "34391    | torch.Size([1, 1000, 754]) | 13\n",
-      "34392    | torch.Size([1, 1000, 754]) | 13\n",
-      "34393    | torch.Size([1, 1000, 797]) | 13\n",
-      "34394    | torch.Size([1, 1000, 754]) | 13\n",
-      "34395    | torch.Size([1, 1000, 754]) | 13\n",
-      "34396    | torch.Size([1, 1000, 754]) | 13\n",
-      "34397    | torch.Size([1, 1000, 772]) | 13\n",
-      "34398    | torch.Size([1, 1000, 777]) | 13\n",
-      "34399    | torch.Size([1, 1000, 754]) | 13\n",
-      "34400    | torch.Size([1, 1000, 777]) | 13\n",
-      "34401    | torch.Size([1, 1000, 777]) | 13\n",
-      "34402    | torch.Size([1, 1000, 754]) | 13\n",
-      "34403    | torch.Size([1, 1000, 771]) | 13\n",
-      "34404    | torch.Size([1, 1000, 754]) | 13\n",
-      "34405    | torch.Size([1, 1000, 754]) | 13\n",
-      "34406    | torch.Size([1, 1000, 754]) | 13\n",
-      "34407    | torch.Size([1, 1000, 754]) | 13\n",
-      "34408    | torch.Size([1, 1000, 770]) | 13\n",
-      "34409    | torch.Size([1, 1000, 777]) | 13\n",
-      "34410    | torch.Size([1, 1000, 754]) | 13\n",
-      "34411    | torch.Size([1, 1000, 778]) | 13\n",
-      "34412    | torch.Size([1, 1000, 772]) | 13\n",
-      "34413    | torch.Size([1, 1000, 754]) | 13\n",
-      "34414    | torch.Size([1, 1000, 772]) | 13\n",
-      "34415    | torch.Size([1, 1000, 771]) | 13\n",
-      "34416    | torch.Size([1, 1000, 772]) | 13\n",
-      "34417    | torch.Size([1, 1000, 772]) | 13\n",
-      "34418    | torch.Size([1, 1000, 777]) | 13\n",
-      "34419    | torch.Size([1, 1000, 777]) | 13\n",
-      "34420    | torch.Size([1, 1000, 772]) | 13\n",
-      "34421    | torch.Size([1, 1000, 777]) | 13\n",
-      "34422    | torch.Size([1, 1000, 754]) | 13\n",
-      "34423    | torch.Size([1, 1000, 772]) | 13\n",
-      "34424    | torch.Size([1, 1000, 841]) | 13\n",
-      "34425    | torch.Size([1, 1000, 754]) | 13\n",
-      "34426    | torch.Size([1, 1000, 772]) | 13\n",
-      "34427    | torch.Size([1, 1000, 777]) | 13\n",
-      "34428    | torch.Size([1, 1000, 754]) | 13\n",
-      "34429    | torch.Size([1, 1000, 754]) | 13\n",
-      "34430    | torch.Size([1, 1000, 773]) | 13\n",
-      "34431    | torch.Size([1, 1000, 777]) | 13\n",
-      "34432    | torch.Size([1, 1000, 754]) | 13\n",
-      "34433    | torch.Size([1, 1000, 759]) | 13\n",
-      "34434    | torch.Size([1, 1000, 772]) | 13\n",
-      "34435    | torch.Size([1, 1000, 754]) | 13\n",
-      "34436    | torch.Size([1, 1000, 754]) | 13\n",
-      "34437    | torch.Size([1, 1000, 754]) | 13\n",
-      "34438    | torch.Size([1, 1000, 754]) | 13\n",
-      "34439    | torch.Size([1, 1000, 843]) | 13\n",
-      "34440    | torch.Size([1, 1000, 782]) | 13\n",
-      "34441    | torch.Size([1, 1000, 779]) | 13\n",
-      "34442    | torch.Size([1, 1000, 772]) | 13\n",
-      "34443    | torch.Size([1, 1000, 814]) | 13\n",
-      "34444    | torch.Size([1, 1000, 763]) | 13\n",
-      "34445    | torch.Size([1, 1000, 754]) | 13\n",
-      "34446    | torch.Size([1, 1000, 772]) | 13\n",
-      "34447    | torch.Size([1, 1000, 772]) | 13\n",
-      "34448    | torch.Size([1, 1000, 777]) | 13\n",
-      "34449    | torch.Size([1, 1000, 777]) | 13\n",
-      "34450    | torch.Size([1, 1000, 777]) | 13\n",
-      "34451    | torch.Size([1, 1000, 754]) | 13\n",
-      "34452    | torch.Size([1, 1000, 754]) | 13\n",
-      "34453    | torch.Size([1, 1000, 754]) | 13\n",
-      "34454    | torch.Size([1, 1000, 754]) | 13\n",
-      "34455    | torch.Size([1, 1000, 754]) | 13\n",
-      "34456    | torch.Size([1, 1000, 772]) | 13\n",
-      "34457    | torch.Size([1, 1000, 799]) | 13\n",
-      "34458    | torch.Size([1, 1000, 777]) | 13\n",
-      "34459    | torch.Size([1, 1000, 754]) | 13\n",
-      "34460    | torch.Size([1, 1000, 754]) | 13\n",
-      "34461    | torch.Size([1, 1000, 754]) | 13\n",
-      "34462    | torch.Size([1, 1000, 772]) | 13\n",
-      "34463    | torch.Size([1, 1000, 754]) | 13\n",
-      "34464    | torch.Size([1, 1000, 777]) | 13\n",
-      "34465    | torch.Size([1, 1000, 772]) | 13\n",
-      "34466    | torch.Size([1, 1000, 777]) | 13\n",
-      "34467    | torch.Size([1, 1000, 777]) | 13\n",
-      "34468    | torch.Size([1, 1000, 754]) | 13\n",
-      "34469    | torch.Size([1, 1000, 754]) | 13\n",
-      "34470    | torch.Size([1, 1000, 754]) | 13\n",
-      "34471    | torch.Size([1, 1000, 754]) | 13\n",
-      "34472    | torch.Size([1, 1000, 754]) | 13\n",
-      "34473    | torch.Size([1, 1000, 754]) | 13\n",
-      "34474    | torch.Size([1, 1000, 754]) | 13\n",
-      "34475    | torch.Size([1, 1000, 772]) | 13\n",
-      "34476    | torch.Size([1, 1000, 772]) | 13\n",
-      "34477    | torch.Size([1, 1000, 768]) | 13\n",
-      "34478    | torch.Size([1, 1000, 789]) | 13\n",
-      "34479    | torch.Size([1, 1000, 754]) | 13\n",
-      "34480    | torch.Size([1, 1000, 725]) | 13\n",
-      "34481    | torch.Size([1, 1000, 754]) | 13\n",
-      "34482    | torch.Size([1, 1000, 703]) | 13\n",
-      "34483    | torch.Size([1, 1000, 754]) | 13\n",
-      "34484    | torch.Size([1, 1000, 754]) | 13\n",
-      "34485    | torch.Size([1, 1000, 754]) | 13\n",
-      "34486    | torch.Size([1, 1000, 813]) | 13\n",
-      "34487    | torch.Size([1, 1000, 754]) | 13\n",
-      "34488    | torch.Size([1, 1000, 754]) | 13\n",
-      "34489    | torch.Size([1, 1000, 754]) | 13\n",
-      "34490    | torch.Size([1, 1000, 754]) | 13\n",
-      "34491    | torch.Size([1, 1000, 754]) | 13\n",
-      "34492    | torch.Size([1, 1000, 754]) | 13\n",
-      "34493    | torch.Size([1, 1000, 777]) | 13\n",
-      "34494    | torch.Size([1, 1000, 777]) | 13\n",
-      "34495    | torch.Size([1, 1000, 809]) | 13\n",
-      "34496    | torch.Size([1, 1000, 772]) | 13\n",
-      "34497    | torch.Size([1, 1000, 766]) | 13\n",
-      "34498    | torch.Size([1, 1000, 754]) | 13\n",
-      "34499    | torch.Size([1, 1000, 806]) | 13\n",
-      "34500    | torch.Size([1, 1000, 754]) | 13\n",
-      "34501    | torch.Size([1, 1000, 754]) | 13\n",
-      "34502    | torch.Size([1, 1000, 785]) | 13\n",
-      "34503    | torch.Size([1, 1000, 804]) | 13\n",
-      "34504    | torch.Size([1, 1000, 777]) | 13\n",
-      "34505    | torch.Size([1, 1000, 772]) | 13\n",
-      "34506    | torch.Size([1, 1000, 754]) | 13\n",
-      "34507    | torch.Size([1, 1000, 766]) | 13\n",
-      "34508    | torch.Size([1, 1000, 777]) | 13\n",
-      "34509    | torch.Size([1, 1000, 772]) | 13\n",
-      "34510    | torch.Size([1, 1000, 754]) | 13\n",
-      "34511    | torch.Size([1, 1000, 771]) | 13\n",
-      "34512    | torch.Size([1, 1000, 777]) | 13\n",
-      "34513    | torch.Size([1, 1000, 754]) | 13\n",
-      "34514    | torch.Size([1, 1000, 754]) | 13\n",
-      "34515    | torch.Size([1, 1000, 754]) | 13\n",
-      "34516    | torch.Size([1, 1000, 777]) | 13\n",
-      "34517    | torch.Size([1, 1000, 766]) | 13\n",
-      "34518    | torch.Size([1, 1000, 772]) | 13\n",
-      "34519    | torch.Size([1, 1000, 772]) | 13\n",
-      "34520    | torch.Size([1, 1000, 754]) | 13\n",
-      "34521    | torch.Size([1, 1000, 777]) | 13\n",
-      "34522    | torch.Size([1, 1000, 777]) | 13\n",
-      "34523    | torch.Size([1, 1000, 771]) | 13\n",
-      "34524    | torch.Size([1, 1000, 754]) | 13\n",
-      "34525    | torch.Size([1, 1000, 754]) | 13\n",
-      "34526    | torch.Size([1, 1000, 754]) | 13\n",
-      "34527    | torch.Size([1, 1000, 754]) | 13\n",
-      "34528    | torch.Size([1, 1000, 754]) | 13\n",
-      "34529    | torch.Size([1, 1000, 754]) | 13\n",
-      "34530    | torch.Size([1, 1000, 772]) | 13\n",
-      "34531    | torch.Size([1, 1000, 772]) | 13\n",
-      "34532    | torch.Size([1, 1000, 762]) | 13\n",
-      "34533    | torch.Size([1, 1000, 777]) | 13\n",
-      "34534    | torch.Size([1, 1000, 754]) | 13\n",
-      "34535    | torch.Size([1, 1000, 754]) | 13\n",
-      "34536    | torch.Size([1, 1000, 754]) | 13\n",
-      "34537    | torch.Size([1, 1000, 754]) | 13\n",
-      "34538    | torch.Size([1, 1000, 772]) | 13\n",
-      "34539    | torch.Size([1, 1000, 754]) | 13\n",
-      "34540    | torch.Size([1, 1000, 754]) | 13\n",
-      "34541    | torch.Size([1, 1000, 772]) | 13\n",
-      "34542    | torch.Size([1, 1000, 754]) | 13\n",
-      "34543    | torch.Size([1, 1000, 772]) | 13\n",
-      "34544    | torch.Size([1, 1000, 754]) | 13\n",
-      "34545    | torch.Size([1, 1000, 754]) | 13\n",
-      "34546    | torch.Size([1, 1000, 772]) | 13\n",
-      "34547    | torch.Size([1, 1000, 802]) | 13\n",
-      "34548    | torch.Size([1, 1000, 754]) | 13\n",
-      "34549    | torch.Size([1, 1000, 754]) | 13\n",
-      "34550    | torch.Size([1, 1000, 772]) | 13\n",
-      "34551    | torch.Size([1, 1000, 754]) | 13\n",
-      "34552    | torch.Size([1, 1000, 772]) | 13\n",
-      "34553    | torch.Size([1, 1000, 754]) | 13\n",
-      "34554    | torch.Size([1, 1000, 754]) | 13\n",
-      "34555    | torch.Size([1, 1000, 772]) | 13\n",
-      "34556    | torch.Size([1, 1000, 754]) | 13\n",
-      "34557    | torch.Size([1, 1000, 754]) | 13\n",
-      "34558    | torch.Size([1, 1000, 754]) | 13\n",
-      "34559    | torch.Size([1, 1000, 754]) | 13\n",
-      "34560    | torch.Size([1, 1000, 754]) | 13\n",
-      "34561    | torch.Size([1, 1000, 772]) | 13\n",
-      "34562    | torch.Size([1, 1000, 774]) | 13\n",
-      "34563    | torch.Size([1, 1000, 772]) | 13\n",
-      "34564    | torch.Size([1, 1000, 754]) | 13\n",
-      "34565    | torch.Size([1, 1000, 784]) | 13\n",
-      "34566    | torch.Size([1, 1000, 754]) | 13\n",
-      "34567    | torch.Size([1, 1000, 754]) | 13\n",
-      "34568    | torch.Size([1, 1000, 754]) | 13\n",
-      "34569    | torch.Size([1, 1000, 754]) | 13\n",
-      "34570    | torch.Size([1, 1000, 754]) | 13\n",
-      "34571    | torch.Size([1, 1000, 754]) | 13\n",
-      "34572    | torch.Size([1, 1000, 772]) | 13\n",
-      "34573    | torch.Size([1, 1000, 777]) | 13\n",
-      "34574    | torch.Size([1, 1000, 772]) | 13\n",
-      "34575    | torch.Size([1, 1000, 754]) | 13\n",
-      "34576    | torch.Size([1, 1000, 777]) | 13\n",
-      "34577    | torch.Size([1, 1000, 754]) | 13\n",
-      "34578    | torch.Size([1, 1000, 777]) | 13\n",
-      "34579    | torch.Size([1, 1000, 754]) | 13\n",
-      "34580    | torch.Size([1, 1000, 754]) | 13\n",
-      "34581    | torch.Size([1, 1000, 754]) | 13\n",
-      "34582    | torch.Size([1, 1000, 754]) | 13\n",
-      "34583    | torch.Size([1, 1000, 772]) | 13\n",
-      "34584    | torch.Size([1, 1000, 777]) | 13\n",
-      "34585    | torch.Size([1, 1000, 783]) | 13\n",
-      "34586    | torch.Size([1, 1000, 787]) | 13\n",
-      "34587    | torch.Size([1, 1000, 777]) | 13\n",
-      "34588    | torch.Size([1, 1000, 772]) | 13\n",
-      "34589    | torch.Size([1, 1000, 754]) | 13\n",
-      "34590    | torch.Size([1, 1000, 777]) | 13\n",
-      "34591    | torch.Size([1, 1000, 772]) | 13\n",
-      "34592    | torch.Size([1, 1000, 754]) | 13\n",
-      "34593    | torch.Size([1, 1000, 754]) | 13\n",
-      "34594    | torch.Size([1, 1000, 754]) | 13\n",
-      "34595    | torch.Size([1, 1000, 754]) | 13\n",
-      "34596    | torch.Size([1, 1000, 754]) | 13\n",
-      "34597    | torch.Size([1, 1000, 809]) | 13\n",
-      "34598    | torch.Size([1, 1000, 777]) | 13\n",
-      "34599    | torch.Size([1, 1000, 754]) | 13\n",
-      "34600    | torch.Size([1, 1000, 754]) | 13\n",
-      "34601    | torch.Size([1, 1000, 754]) | 13\n",
-      "34602    | torch.Size([1, 1000, 777]) | 13\n",
-      "34603    | torch.Size([1, 1000, 754]) | 13\n",
-      "34604    | torch.Size([1, 1000, 754]) | 13\n",
-      "34605    | torch.Size([1, 1000, 754]) | 13\n",
-      "34606    | torch.Size([1, 1000, 754]) | 13\n",
-      "34607    | torch.Size([1, 1000, 772]) | 13\n",
-      "34608    | torch.Size([1, 1000, 754]) | 13\n",
-      "34609    | torch.Size([1, 1000, 777]) | 13\n",
-      "34610    | torch.Size([1, 1000, 778]) | 13\n",
-      "34611    | torch.Size([1, 1000, 772]) | 13\n",
-      "34612    | torch.Size([1, 1000, 754]) | 13\n",
-      "34613    | torch.Size([1, 1000, 759]) | 13\n",
-      "34614    | torch.Size([1, 1000, 754]) | 13\n",
-      "34615    | torch.Size([1, 1000, 754]) | 13\n",
-      "34616    | torch.Size([1, 1000, 760]) | 13\n",
-      "34617    | torch.Size([1, 1000, 754]) | 13\n",
-      "34618    | torch.Size([1, 1000, 777]) | 13\n",
-      "34619    | torch.Size([1, 1000, 754]) | 13\n",
-      "34620    | torch.Size([1, 1000, 754]) | 13\n",
-      "34621    | torch.Size([1, 1000, 754]) | 13\n",
-      "34622    | torch.Size([1, 1000, 754]) | 13\n",
-      "34623    | torch.Size([1, 1000, 772]) | 13\n",
-      "34624    | torch.Size([1, 1000, 777]) | 13\n",
-      "34625    | torch.Size([1, 1000, 772]) | 13\n",
-      "34626    | torch.Size([1, 1000, 772]) | 13\n",
-      "34627    | torch.Size([1, 1000, 754]) | 13\n",
-      "34628    | torch.Size([1, 1000, 799]) | 13\n",
-      "34629    | torch.Size([1, 1000, 754]) | 13\n",
-      "34630    | torch.Size([1, 1000, 754]) | 13\n",
-      "34631    | torch.Size([1, 1000, 777]) | 13\n",
-      "34632    | torch.Size([1, 1000, 754]) | 13\n",
-      "34633    | torch.Size([1, 1000, 772]) | 13\n",
-      "34634    | torch.Size([1, 1000, 754]) | 13\n",
-      "34635    | torch.Size([1, 1000, 754]) | 13\n",
-      "34636    | torch.Size([1, 1000, 777]) | 13\n",
-      "34637    | torch.Size([1, 1000, 754]) | 13\n",
-      "34638    | torch.Size([1, 1000, 773]) | 13\n",
-      "34639    | torch.Size([1, 1000, 754]) | 13\n",
-      "34640    | torch.Size([1, 1000, 754]) | 13\n",
-      "34641    | torch.Size([1, 1000, 772]) | 13\n",
-      "34642    | torch.Size([1, 1000, 754]) | 13\n",
-      "34643    | torch.Size([1, 1000, 754]) | 13\n",
-      "34644    | torch.Size([1, 1000, 777]) | 13\n",
-      "34645    | torch.Size([1, 1000, 772]) | 13\n",
-      "34646    | torch.Size([1, 1000, 754]) | 13\n",
-      "34647    | torch.Size([1, 1000, 778]) | 13\n",
-      "34648    | torch.Size([1, 1000, 754]) | 13\n",
-      "34649    | torch.Size([1, 1000, 754]) | 13\n",
-      "34650    | torch.Size([1, 1000, 754]) | 13\n",
-      "34651    | torch.Size([1, 1000, 772]) | 13\n",
-      "34652    | torch.Size([1, 1000, 754]) | 13\n",
-      "34653    | torch.Size([1, 1000, 783]) | 13\n",
-      "34654    | torch.Size([1, 1000, 754]) | 13\n",
-      "34655    | torch.Size([1, 1000, 754]) | 13\n",
-      "34656    | torch.Size([1, 1000, 760]) | 13\n",
-      "34657    | torch.Size([1, 1000, 754]) | 13\n",
-      "34658    | torch.Size([1, 1000, 762]) | 13\n",
-      "34659    | torch.Size([1, 1000, 809]) | 13\n",
-      "34660    | torch.Size([1, 1000, 772]) | 13\n",
-      "34661    | torch.Size([1, 1000, 754]) | 13\n",
-      "34662    | torch.Size([1, 1000, 754]) | 13\n",
-      "34663    | torch.Size([1, 1000, 754]) | 13\n",
-      "34664    | torch.Size([1, 1000, 754]) | 13\n",
-      "34665    | torch.Size([1, 1000, 731]) | 13\n",
-      "34666    | torch.Size([1, 1000, 754]) | 13\n",
-      "34667    | torch.Size([1, 1000, 772]) | 13\n",
-      "34668    | torch.Size([1, 1000, 754]) | 13\n",
-      "34669    | torch.Size([1, 1000, 802]) | 13\n",
-      "34670    | torch.Size([1, 1000, 754]) | 13\n",
-      "34671    | torch.Size([1, 1000, 772]) | 13\n",
-      "34672    | torch.Size([1, 1000, 777]) | 13\n",
-      "34673    | torch.Size([1, 1000, 754]) | 13\n",
-      "34674    | torch.Size([1, 1000, 754]) | 13\n",
-      "34675    | torch.Size([1, 1000, 754]) | 13\n",
-      "34676    | torch.Size([1, 1000, 772]) | 13\n",
-      "34677    | torch.Size([1, 1000, 754]) | 13\n",
-      "34678    | torch.Size([1, 1000, 777]) | 13\n",
-      "34679    | torch.Size([1, 1000, 754]) | 13\n",
-      "34680    | torch.Size([1, 1000, 781]) | 13\n",
-      "34681    | torch.Size([1, 1000, 777]) | 13\n",
-      "34682    | torch.Size([1, 1000, 772]) | 13\n",
-      "34683    | torch.Size([1, 1000, 777]) | 13\n",
-      "34684    | torch.Size([1, 1000, 772]) | 13\n",
-      "34685    | torch.Size([1, 1000, 754]) | 13\n",
-      "34686    | torch.Size([1, 1000, 754]) | 13\n",
-      "34687    | torch.Size([1, 1000, 777]) | 13\n",
-      "34688    | torch.Size([1, 1000, 754]) | 13\n",
-      "34689    | torch.Size([1, 1000, 777]) | 13\n",
-      "34690    | torch.Size([1, 1000, 754]) | 13\n",
-      "34691    | torch.Size([1, 1000, 777]) | 13\n",
-      "34692    | torch.Size([1, 1000, 754]) | 13\n",
-      "34693    | torch.Size([1, 1000, 777]) | 13\n",
-      "34694    | torch.Size([1, 1000, 772]) | 13\n",
-      "34695    | torch.Size([1, 1000, 754]) | 13\n",
-      "34696    | torch.Size([1, 1000, 772]) | 13\n",
-      "34697    | torch.Size([1, 1000, 772]) | 13\n",
-      "34698    | torch.Size([1, 1000, 754]) | 13\n",
-      "34699    | torch.Size([1, 1000, 777]) | 13\n",
-      "34700    | torch.Size([1, 1000, 754]) | 13\n",
-      "34701    | torch.Size([1, 1000, 772]) | 13\n",
-      "34702    | torch.Size([1, 1000, 754]) | 13\n",
-      "34703    | torch.Size([1, 1000, 772]) | 13\n",
-      "34704    | torch.Size([1, 1000, 754]) | 13\n",
-      "34705    | torch.Size([1, 1000, 772]) | 13\n",
-      "34706    | torch.Size([1, 1000, 754]) | 13\n",
-      "34707    | torch.Size([1, 1000, 754]) | 13\n",
-      "34708    | torch.Size([1, 1000, 777]) | 13\n",
-      "34709    | torch.Size([1, 1000, 754]) | 13\n",
-      "34710    | torch.Size([1, 1000, 772]) | 13\n",
-      "34711    | torch.Size([1, 1000, 777]) | 13\n",
-      "34712    | torch.Size([1, 1000, 777]) | 13\n",
-      "34713    | torch.Size([1, 1000, 777]) | 13\n",
-      "34714    | torch.Size([1, 1000, 754]) | 13\n",
-      "34715    | torch.Size([1, 1000, 792]) | 13\n",
-      "34716    | torch.Size([1, 1000, 754]) | 13\n",
-      "34717    | torch.Size([1, 1000, 772]) | 13\n",
-      "34718    | torch.Size([1, 1000, 754]) | 13\n",
-      "34719    | torch.Size([1, 1000, 754]) | 13\n",
-      "34720    | torch.Size([1, 1000, 772]) | 13\n",
-      "34721    | torch.Size([1, 1000, 754]) | 13\n",
-      "34722    | torch.Size([1, 1000, 772]) | 13\n",
-      "34723    | torch.Size([1, 1000, 799]) | 13\n",
-      "34724    | torch.Size([1, 1000, 754]) | 13\n",
-      "34725    | torch.Size([1, 1000, 777]) | 13\n",
-      "34726    | torch.Size([1, 1000, 777]) | 13\n",
-      "34727    | torch.Size([1, 1000, 775]) | 13\n",
-      "34728    | torch.Size([1, 1000, 772]) | 13\n",
-      "34729    | torch.Size([1, 1000, 754]) | 13\n",
-      "34730    | torch.Size([1, 1000, 754]) | 13\n",
-      "34731    | torch.Size([1, 1000, 772]) | 13\n",
-      "34732    | torch.Size([1, 1000, 754]) | 13\n",
-      "34733    | torch.Size([1, 1000, 754]) | 13\n",
-      "34734    | torch.Size([1, 1000, 777]) | 13\n",
-      "34735    | torch.Size([1, 1000, 772]) | 13\n",
-      "34736    | torch.Size([1, 1000, 754]) | 13\n",
-      "34737    | torch.Size([1, 1000, 754]) | 13\n",
-      "34738    | torch.Size([1, 1000, 754]) | 13\n",
-      "34739    | torch.Size([1, 1000, 772]) | 13\n",
-      "34740    | torch.Size([1, 1000, 772]) | 13\n",
-      "34741    | torch.Size([1, 1000, 777]) | 13\n",
-      "34742    | torch.Size([1, 1000, 754]) | 13\n",
-      "34743    | torch.Size([1, 1000, 754]) | 13\n",
-      "34744    | torch.Size([1, 1000, 772]) | 13\n",
-      "34745    | torch.Size([1, 1000, 754]) | 13\n",
-      "34746    | torch.Size([1, 1000, 788]) | 13\n",
-      "34747    | torch.Size([1, 1000, 772]) | 13\n",
-      "34748    | torch.Size([1, 1000, 754]) | 13\n",
-      "34749    | torch.Size([1, 1000, 754]) | 13\n",
-      "34750    | torch.Size([1, 1000, 754]) | 13\n",
-      "34751    | torch.Size([1, 1000, 777]) | 13\n",
-      "34752    | torch.Size([1, 1000, 754]) | 13\n",
-      "34753    | torch.Size([1, 1000, 842]) | 13\n",
-      "34754    | torch.Size([1, 1000, 772]) | 13\n",
-      "34755    | torch.Size([1, 1000, 772]) | 13\n",
-      "34756    | torch.Size([1, 1000, 772]) | 13\n",
-      "34757    | torch.Size([1, 1000, 754]) | 13\n",
-      "34758    | torch.Size([1, 1000, 772]) | 13\n",
-      "34759    | torch.Size([1, 1000, 754]) | 13\n",
-      "34760    | torch.Size([1, 1000, 787]) | 13\n",
-      "34761    | torch.Size([1, 1000, 772]) | 13\n",
-      "34762    | torch.Size([1, 1000, 754]) | 13\n",
-      "34763    | torch.Size([1, 1000, 772]) | 13\n",
-      "34764    | torch.Size([1, 1000, 754]) | 13\n",
-      "34765    | torch.Size([1, 1000, 754]) | 13\n",
-      "34766    | torch.Size([1, 1000, 754]) | 13\n",
-      "34767    | torch.Size([1, 1000, 772]) | 13\n",
-      "34768    | torch.Size([1, 1000, 777]) | 13\n",
-      "34769    | torch.Size([1, 1000, 754]) | 13\n",
-      "34770    | torch.Size([1, 1000, 772]) | 13\n",
-      "34771    | torch.Size([1, 1000, 813]) | 13\n",
-      "34772    | torch.Size([1, 1000, 754]) | 13\n",
-      "34773    | torch.Size([1, 1000, 772]) | 13\n",
-      "34774    | torch.Size([1, 1000, 777]) | 13\n",
-      "34775    | torch.Size([1, 1000, 754]) | 13\n",
-      "34776    | torch.Size([1, 1000, 754]) | 13\n",
-      "34777    | torch.Size([1, 1000, 754]) | 13\n",
-      "34778    | torch.Size([1, 1000, 754]) | 13\n",
-      "34779    | torch.Size([1, 1000, 772]) | 13\n",
-      "34780    | torch.Size([1, 1000, 772]) | 13\n",
-      "34781    | torch.Size([1, 1000, 754]) | 13\n",
-      "34782    | torch.Size([1, 1000, 754]) | 13\n",
-      "34783    | torch.Size([1, 1000, 772]) | 13\n",
-      "34784    | torch.Size([1, 1000, 772]) | 13\n",
-      "34785    | torch.Size([1, 1000, 754]) | 13\n",
-      "34786    | torch.Size([1, 1000, 754]) | 13\n",
-      "34787    | torch.Size([1, 1000, 772]) | 13\n",
-      "34788    | torch.Size([1, 1000, 754]) | 13\n",
-      "34789    | torch.Size([1, 1000, 754]) | 13\n",
-      "34790    | torch.Size([1, 1000, 754]) | 13\n",
-      "34791    | torch.Size([1, 1000, 754]) | 13\n",
-      "34792    | torch.Size([1, 1000, 754]) | 13\n",
-      "34793    | torch.Size([1, 1000, 754]) | 13\n",
-      "34794    | torch.Size([1, 1000, 754]) | 13\n",
-      "34795    | torch.Size([1, 1000, 754]) | 13\n",
-      "34796    | torch.Size([1, 1000, 754]) | 13\n",
-      "34797    | torch.Size([1, 1000, 773]) | 13\n",
-      "34798    | torch.Size([1, 1000, 754]) | 13\n",
-      "34799    | torch.Size([1, 1000, 754]) | 13\n",
-      "34800    | torch.Size([1, 1000, 754]) | 13\n",
-      "34801    | torch.Size([1, 1000, 754]) | 13\n",
-      "34802    | torch.Size([1, 1000, 777]) | 13\n",
-      "34803    | torch.Size([1, 1000, 772]) | 13\n",
-      "34804    | torch.Size([1, 1000, 754]) | 13\n",
-      "34805    | torch.Size([1, 1000, 777]) | 13\n",
-      "34806    | torch.Size([1, 1000, 782]) | 13\n",
-      "34807    | torch.Size([1, 1000, 772]) | 13\n",
-      "34808    | torch.Size([1, 1000, 754]) | 13\n",
-      "34809    | torch.Size([1, 1000, 772]) | 13\n",
-      "34810    | torch.Size([1, 1000, 772]) | 13\n",
-      "34811    | torch.Size([1, 1000, 772]) | 13\n",
-      "34812    | torch.Size([1, 1000, 717]) | 13\n",
-      "34813    | torch.Size([1, 1000, 754]) | 13\n",
-      "34814    | torch.Size([1, 1000, 754]) | 13\n",
-      "34815    | torch.Size([1, 1000, 772]) | 13\n",
-      "34816    | torch.Size([1, 1000, 754]) | 13\n",
-      "34817    | torch.Size([1, 1000, 754]) | 13\n",
-      "34818    | torch.Size([1, 1000, 754]) | 13\n",
-      "34819    | torch.Size([1, 1000, 713]) | 13\n",
-      "34820    | torch.Size([1, 1000, 754]) | 13\n",
-      "34821    | torch.Size([1, 1000, 780]) | 13\n",
-      "34822    | torch.Size([1, 1000, 754]) | 13\n",
-      "34823    | torch.Size([1, 1000, 781]) | 13\n",
-      "34824    | torch.Size([1, 1000, 754]) | 13\n",
-      "34825    | torch.Size([1, 1000, 754]) | 13\n",
-      "34826    | torch.Size([1, 1000, 754]) | 13\n",
-      "34827    | torch.Size([1, 1000, 754]) | 13\n",
-      "34828    | torch.Size([1, 1000, 772]) | 13\n",
-      "34829    | torch.Size([1, 1000, 782]) | 13\n",
-      "34830    | torch.Size([1, 1000, 754]) | 13\n",
-      "34831    | torch.Size([1, 1000, 754]) | 13\n",
-      "34832    | torch.Size([1, 1000, 772]) | 13\n",
-      "34833    | torch.Size([1, 1000, 762]) | 13\n",
-      "34834    | torch.Size([1, 1000, 777]) | 13\n",
-      "34835    | torch.Size([1, 1000, 754]) | 13\n",
-      "34836    | torch.Size([1, 1000, 754]) | 13\n",
-      "34837    | torch.Size([1, 1000, 754]) | 13\n",
-      "34838    | torch.Size([1, 1000, 777]) | 13\n",
-      "34839    | torch.Size([1, 1000, 754]) | 13\n",
-      "34840    | torch.Size([1, 1000, 772]) | 13\n",
-      "34841    | torch.Size([1, 1000, 772]) | 13\n",
-      "34842    | torch.Size([1, 1000, 762]) | 13\n",
-      "34843    | torch.Size([1, 1000, 812]) | 13\n",
-      "34844    | torch.Size([1, 1000, 754]) | 13\n",
-      "34845    | torch.Size([1, 1000, 754]) | 13\n",
-      "34846    | torch.Size([1, 1000, 806]) | 13\n",
-      "34847    | torch.Size([1, 1000, 777]) | 13\n",
-      "34848    | torch.Size([1, 1000, 777]) | 13\n",
-      "34849    | torch.Size([1, 1000, 772]) | 13\n",
-      "34850    | torch.Size([1, 1000, 802]) | 13\n",
-      "34851    | torch.Size([1, 1000, 754]) | 13\n",
-      "34852    | torch.Size([1, 1000, 754]) | 13\n",
-      "34853    | torch.Size([1, 1000, 777]) | 13\n",
-      "34854    | torch.Size([1, 1000, 754]) | 13\n",
-      "34855    | torch.Size([1, 1000, 771]) | 13\n",
-      "34856    | torch.Size([1, 1000, 754]) | 13\n",
-      "34857    | torch.Size([1, 1000, 754]) | 13\n",
-      "34858    | torch.Size([1, 1000, 777]) | 13\n",
-      "34859    | torch.Size([1, 1000, 754]) | 13\n",
-      "34860    | torch.Size([1, 1000, 754]) | 13\n",
-      "34861    | torch.Size([1, 1000, 781]) | 13\n",
-      "34862    | torch.Size([1, 1000, 772]) | 13\n",
-      "34863    | torch.Size([1, 1000, 754]) | 13\n",
-      "34864    | torch.Size([1, 1000, 777]) | 13\n",
-      "34865    | torch.Size([1, 1000, 754]) | 13\n",
-      "34866    | torch.Size([1, 1000, 754]) | 13\n",
-      "34867    | torch.Size([1, 1000, 815]) | 13\n",
-      "34868    | torch.Size([1, 1000, 754]) | 13\n",
-      "34869    | torch.Size([1, 1000, 772]) | 13\n",
-      "34870    | torch.Size([1, 1000, 784]) | 13\n",
-      "34871    | torch.Size([1, 1000, 754]) | 13\n",
-      "34872    | torch.Size([1, 1000, 772]) | 13\n",
-      "34873    | torch.Size([1, 1000, 754]) | 13\n",
-      "34874    | torch.Size([1, 1000, 754]) | 13\n",
-      "34875    | torch.Size([1, 1000, 754]) | 13\n",
-      "34876    | torch.Size([1, 1000, 754]) | 13\n",
-      "34877    | torch.Size([1, 1000, 754]) | 13\n",
-      "34878    | torch.Size([1, 1000, 777]) | 13\n",
-      "34879    | torch.Size([1, 1000, 772]) | 13\n",
-      "34880    | torch.Size([1, 1000, 754]) | 13\n",
-      "34881    | torch.Size([1, 1000, 779]) | 13\n",
-      "34882    | torch.Size([1, 1000, 754]) | 13\n",
-      "34883    | torch.Size([1, 1000, 732]) | 13\n",
-      "34884    | torch.Size([1, 1000, 754]) | 13\n",
-      "34885    | torch.Size([1, 1000, 754]) | 13\n",
-      "34886    | torch.Size([1, 1000, 772]) | 13\n",
-      "34887    | torch.Size([1, 1000, 754]) | 13\n",
-      "34888    | torch.Size([1, 1000, 772]) | 13\n",
-      "34889    | torch.Size([1, 1000, 777]) | 13\n",
-      "34890    | torch.Size([1, 1000, 754]) | 13\n",
-      "34891    | torch.Size([1, 1000, 772]) | 13\n",
-      "34892    | torch.Size([1, 1000, 754]) | 13\n",
-      "34893    | torch.Size([1, 1000, 772]) | 13\n",
-      "34894    | torch.Size([1, 1000, 782]) | 13\n",
-      "34895    | torch.Size([1, 1000, 754]) | 13\n",
-      "34896    | torch.Size([1, 1000, 713]) | 13\n",
-      "34897    | torch.Size([1, 1000, 754]) | 13\n",
-      "34898    | torch.Size([1, 1000, 754]) | 13\n",
-      "34899    | torch.Size([1, 1000, 754]) | 13\n",
-      "34900    | torch.Size([1, 1000, 754]) | 13\n",
-      "34901    | torch.Size([1, 1000, 754]) | 13\n",
-      "34902    | torch.Size([1, 1000, 703]) | 13\n",
-      "34903    | torch.Size([1, 1000, 772]) | 13\n",
-      "34904    | torch.Size([1, 1000, 777]) | 13\n",
-      "34905    | torch.Size([1, 1000, 777]) | 13\n",
-      "34906    | torch.Size([1, 1000, 754]) | 13\n",
-      "34907    | torch.Size([1, 1000, 777]) | 13\n",
-      "34908    | torch.Size([1, 1000, 754]) | 13\n",
-      "34909    | torch.Size([1, 1000, 802]) | 13\n",
-      "34910    | torch.Size([1, 1000, 754]) | 13\n",
-      "34911    | torch.Size([1, 1000, 777]) | 13\n",
-      "34912    | torch.Size([1, 1000, 772]) | 13\n",
-      "34913    | torch.Size([1, 1000, 777]) | 13\n",
-      "34914    | torch.Size([1, 1000, 754]) | 13\n",
-      "34915    | torch.Size([1, 1000, 777]) | 13\n",
-      "34916    | torch.Size([1, 1000, 777]) | 13\n",
-      "34917    | torch.Size([1, 1000, 754]) | 13\n",
-      "34918    | torch.Size([1, 1000, 772]) | 13\n",
-      "34919    | torch.Size([1, 1000, 754]) | 13\n",
-      "34920    | torch.Size([1, 1000, 754]) | 13\n",
-      "34921    | torch.Size([1, 1000, 754]) | 13\n",
-      "34922    | torch.Size([1, 1000, 772]) | 13\n",
-      "34923    | torch.Size([1, 1000, 754]) | 13\n",
-      "34924    | torch.Size([1, 1000, 754]) | 13\n",
-      "34925    | torch.Size([1, 1000, 772]) | 13\n",
-      "34926    | torch.Size([1, 1000, 754]) | 13\n",
-      "34927    | torch.Size([1, 1000, 777]) | 13\n",
-      "34928    | torch.Size([1, 1000, 772]) | 13\n",
-      "34929    | torch.Size([1, 1000, 772]) | 13\n",
-      "34930    | torch.Size([1, 1000, 772]) | 13\n",
-      "34931    | torch.Size([1, 1000, 771]) | 13\n",
-      "34932    | torch.Size([1, 1000, 754]) | 13\n",
-      "34933    | torch.Size([1, 1000, 785]) | 13\n",
-      "34934    | torch.Size([1, 1000, 754]) | 13\n",
-      "34935    | torch.Size([1, 1000, 754]) | 13\n",
-      "34936    | torch.Size([1, 1000, 754]) | 13\n",
-      "34937    | torch.Size([1, 1000, 782]) | 13\n",
-      "34938    | torch.Size([1, 1000, 754]) | 13\n",
-      "34939    | torch.Size([1, 1000, 754]) | 13\n",
-      "34940    | torch.Size([1, 1000, 777]) | 13\n",
-      "34941    | torch.Size([1, 1000, 772]) | 13\n",
-      "34942    | torch.Size([1, 1000, 754]) | 13\n",
-      "34943    | torch.Size([1, 1000, 754]) | 13\n",
-      "34944    | torch.Size([1, 1000, 754]) | 13\n",
-      "34945    | torch.Size([1, 1000, 777]) | 13\n",
-      "34946    | torch.Size([1, 1000, 777]) | 13\n",
-      "34947    | torch.Size([1, 1000, 772]) | 13\n",
-      "34948    | torch.Size([1, 1000, 754]) | 13\n",
-      "34949    | torch.Size([1, 1000, 754]) | 13\n",
-      "34950    | torch.Size([1, 1000, 772]) | 13\n",
-      "34951    | torch.Size([1, 1000, 772]) | 13\n",
-      "34952    | torch.Size([1, 1000, 777]) | 13\n",
-      "34953    | torch.Size([1, 1000, 754]) | 13\n",
-      "34954    | torch.Size([1, 1000, 777]) | 13\n",
-      "34955    | torch.Size([1, 1000, 772]) | 13\n",
-      "34956    | torch.Size([1, 1000, 772]) | 13\n",
-      "34957    | torch.Size([1, 1000, 772]) | 13\n",
-      "34958    | torch.Size([1, 1000, 754]) | 13\n",
-      "34959    | torch.Size([1, 1000, 777]) | 13\n",
-      "34960    | torch.Size([1, 1000, 754]) | 13\n",
-      "34961    | torch.Size([1, 1000, 754]) | 13\n",
-      "34962    | torch.Size([1, 1000, 772]) | 13\n",
-      "34963    | torch.Size([1, 1000, 754]) | 13\n",
-      "34964    | torch.Size([1, 1000, 777]) | 13\n"
-     ]
-    },
-    {
-     "ename": "UnidentifiedImageError",
-     "evalue": "cannot identify image file <_io.BytesIO object at 0x377cfaed0>",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[31m---------------------------------------------------------------------------\u001b[39m",
-      "\u001b[31mUnidentifiedImageError\u001b[39m                    Traceback (most recent call last)",
-      "\u001b[36mCell\u001b[39m\u001b[36m \u001b[39m\u001b[32mIn[46]\u001b[39m\u001b[32m, line 12\u001b[39m\n\u001b[32m      9\u001b[39m \u001b[38;5;28mprint\u001b[39m(\u001b[33m\"\u001b[39m\u001b[33m-\u001b[39m\u001b[33m\"\u001b[39m * \u001b[32m40\u001b[39m)\n\u001b[32m     11\u001b[39m \u001b[38;5;66;03m# Using tqdm so you can track progress across the 40,000+ images\u001b[39;00m\n\u001b[32m---> \u001b[39m\u001b[32m12\u001b[39m \u001b[38;5;28;43;01mfor\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43midx\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mitem\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;129;43;01min\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[38;5;28;43menumerate\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43mtest\u001b[49m\u001b[43m)\u001b[49m\u001b[43m:\u001b[49m\n\u001b[32m     13\u001b[39m \u001b[43m    \u001b[49m\u001b[38;5;28;43;01mtry\u001b[39;49;00m\u001b[43m:\u001b[49m\n\u001b[32m     14\u001b[39m \u001b[43m        \u001b[49m\u001b[38;5;66;43;03m# Accessing the image size triggers the file reading\u001b[39;49;00m\n\u001b[32m     15\u001b[39m \u001b[43m        \u001b[49m\u001b[43mshape\u001b[49m\u001b[43m \u001b[49m\u001b[43m=\u001b[49m\u001b[43m \u001b[49m\u001b[43mitem\u001b[49m\u001b[43m[\u001b[49m\u001b[33;43m\"\u001b[39;49m\u001b[33;43mimage\u001b[39;49m\u001b[33;43m\"\u001b[39;49m\u001b[43m]\u001b[49m\u001b[43m.\u001b[49m\u001b[43msize\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/arrow_dataset.py:2483\u001b[39m, in \u001b[36mDataset.__iter__\u001b[39m\u001b[34m(self)\u001b[39m\n\u001b[32m   2481\u001b[39m         \u001b[38;5;28;01mfor\u001b[39;00m i \u001b[38;5;129;01min\u001b[39;00m \u001b[38;5;28mrange\u001b[39m(pa_subtable.num_rows):\n\u001b[32m   2482\u001b[39m             pa_subtable_ex = pa_subtable.slice(i, \u001b[32m1\u001b[39m)\n\u001b[32m-> \u001b[39m\u001b[32m2483\u001b[39m             formatted_output = \u001b[43mformat_table\u001b[49m\u001b[43m(\u001b[49m\n\u001b[32m   2484\u001b[39m \u001b[43m                \u001b[49m\u001b[43mpa_subtable_ex\u001b[49m\u001b[43m,\u001b[49m\n\u001b[32m   2485\u001b[39m \u001b[43m                \u001b[49m\u001b[32;43m0\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[32m   2486\u001b[39m \u001b[43m                \u001b[49m\u001b[43mformatter\u001b[49m\u001b[43m=\u001b[49m\u001b[43mformatter\u001b[49m\u001b[43m,\u001b[49m\n\u001b[32m   2487\u001b[39m \u001b[43m                \u001b[49m\u001b[43mformat_columns\u001b[49m\u001b[43m=\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43m_format_columns\u001b[49m\u001b[43m,\u001b[49m\n\u001b[32m   2488\u001b[39m \u001b[43m                \u001b[49m\u001b[43moutput_all_columns\u001b[49m\u001b[43m=\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43m_output_all_columns\u001b[49m\u001b[43m,\u001b[49m\n\u001b[32m   2489\u001b[39m \u001b[43m            \u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m   2490\u001b[39m             \u001b[38;5;28;01myield\u001b[39;00m formatted_output\n\u001b[32m   2491\u001b[39m \u001b[38;5;28;01melse\u001b[39;00m:\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/formatting/formatting.py:666\u001b[39m, in \u001b[36mformat_table\u001b[39m\u001b[34m(table, key, formatter, format_columns, output_all_columns)\u001b[39m\n\u001b[32m    664\u001b[39m \u001b[38;5;28;01melse\u001b[39;00m:\n\u001b[32m    665\u001b[39m     pa_table_to_format = pa_table.drop(col \u001b[38;5;28;01mfor\u001b[39;00m col \u001b[38;5;129;01min\u001b[39;00m pa_table.column_names \u001b[38;5;28;01mif\u001b[39;00m col \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;129;01min\u001b[39;00m format_columns)\n\u001b[32m--> \u001b[39m\u001b[32m666\u001b[39m     formatted_output = \u001b[43mformatter\u001b[49m\u001b[43m(\u001b[49m\u001b[43mpa_table_to_format\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mquery_type\u001b[49m\u001b[43m=\u001b[49m\u001b[43mquery_type\u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m    667\u001b[39m     \u001b[38;5;28;01mif\u001b[39;00m output_all_columns:\n\u001b[32m    668\u001b[39m         \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28misinstance\u001b[39m(formatted_output, MutableMapping):\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/formatting/formatting.py:411\u001b[39m, in \u001b[36mFormatter.__call__\u001b[39m\u001b[34m(self, pa_table, query_type)\u001b[39m\n\u001b[32m    409\u001b[39m \u001b[38;5;28;01mdef\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34m__call__\u001b[39m(\u001b[38;5;28mself\u001b[39m, pa_table: pa.Table, query_type: \u001b[38;5;28mstr\u001b[39m) -> Union[RowFormat, ColumnFormat, BatchFormat]:\n\u001b[32m    410\u001b[39m     \u001b[38;5;28;01mif\u001b[39;00m query_type == \u001b[33m\"\u001b[39m\u001b[33mrow\u001b[39m\u001b[33m\"\u001b[39m:\n\u001b[32m--> \u001b[39m\u001b[32m411\u001b[39m         \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43mformat_row\u001b[49m\u001b[43m(\u001b[49m\u001b[43mpa_table\u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m    412\u001b[39m     \u001b[38;5;28;01melif\u001b[39;00m query_type == \u001b[33m\"\u001b[39m\u001b[33mcolumn\u001b[39m\u001b[33m\"\u001b[39m:\n\u001b[32m    413\u001b[39m         \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m.format_column(pa_table)\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/formatting/torch_formatter.py:253\u001b[39m, in \u001b[36mTorchFormatter.format_row\u001b[39m\u001b[34m(self, pa_table)\u001b[39m\n\u001b[32m    251\u001b[39m \u001b[38;5;28;01mdef\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34mformat_row\u001b[39m(\u001b[38;5;28mself\u001b[39m, pa_table: pa.Table) -> Mapping:\n\u001b[32m    252\u001b[39m     row = \u001b[38;5;28mself\u001b[39m.numpy_arrow_extractor().extract_row(pa_table)\n\u001b[32m--> \u001b[39m\u001b[32m253\u001b[39m     row = \u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43mpython_features_decoder\u001b[49m\u001b[43m.\u001b[49m\u001b[43mdecode_row\u001b[49m\u001b[43m(\u001b[49m\u001b[43mrow\u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m    254\u001b[39m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m.recursive_tensorize(row)\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/formatting/formatting.py:224\u001b[39m, in \u001b[36mPythonFeaturesDecoder.decode_row\u001b[39m\u001b[34m(self, row)\u001b[39m\n\u001b[32m    223\u001b[39m \u001b[38;5;28;01mdef\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34mdecode_row\u001b[39m(\u001b[38;5;28mself\u001b[39m, row: \u001b[38;5;28mdict\u001b[39m) -> \u001b[38;5;28mdict\u001b[39m:\n\u001b[32m--> \u001b[39m\u001b[32m224\u001b[39m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43mfeatures\u001b[49m\u001b[43m.\u001b[49m\u001b[43mdecode_example\u001b[49m\u001b[43m(\u001b[49m\u001b[43mrow\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m=\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m.\u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m)\u001b[49m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m.features \u001b[38;5;28;01melse\u001b[39;00m row\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/features/features.py:2111\u001b[39m, in \u001b[36mFeatures.decode_example\u001b[39m\u001b[34m(self, example, token_per_repo_id)\u001b[39m\n\u001b[32m   2096\u001b[39m \u001b[38;5;28;01mdef\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[34mdecode_example\u001b[39m(\u001b[38;5;28mself\u001b[39m, example: \u001b[38;5;28mdict\u001b[39m, token_per_repo_id: Optional[\u001b[38;5;28mdict\u001b[39m[\u001b[38;5;28mstr\u001b[39m, Union[\u001b[38;5;28mstr\u001b[39m, \u001b[38;5;28mbool\u001b[39m, \u001b[38;5;28;01mNone\u001b[39;00m]]] = \u001b[38;5;28;01mNone\u001b[39;00m):\n\u001b[32m   2097\u001b[39m \u001b[38;5;250m    \u001b[39m\u001b[33;03m\"\"\"Decode example with custom feature decoding.\u001b[39;00m\n\u001b[32m   2098\u001b[39m \n\u001b[32m   2099\u001b[39m \u001b[33;03m    Args:\u001b[39;00m\n\u001b[32m   (...)\u001b[39m\u001b[32m   2107\u001b[39m \u001b[33;03m        `dict[str, Any]`\u001b[39;00m\n\u001b[32m   2108\u001b[39m \u001b[33;03m    \"\"\"\u001b[39;00m\n\u001b[32m   2110\u001b[39m     \u001b[38;5;28;01mreturn\u001b[39;00m {\n\u001b[32m-> \u001b[39m\u001b[32m2111\u001b[39m         column_name: \u001b[43mdecode_nested_example\u001b[49m\u001b[43m(\u001b[49m\u001b[43mfeature\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mvalue\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m=\u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m   2112\u001b[39m         \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m._column_requires_decoding[column_name]\n\u001b[32m   2113\u001b[39m         \u001b[38;5;28;01melse\u001b[39;00m value\n\u001b[32m   2114\u001b[39m         \u001b[38;5;28;01mfor\u001b[39;00m column_name, (feature, value) \u001b[38;5;129;01min\u001b[39;00m zip_dict(\n\u001b[32m   2115\u001b[39m             {key: value \u001b[38;5;28;01mfor\u001b[39;00m key, value \u001b[38;5;129;01min\u001b[39;00m \u001b[38;5;28mself\u001b[39m.items() \u001b[38;5;28;01mif\u001b[39;00m key \u001b[38;5;129;01min\u001b[39;00m example}, example\n\u001b[32m   2116\u001b[39m         )\n\u001b[32m   2117\u001b[39m     }\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/features/features.py:1419\u001b[39m, in \u001b[36mdecode_nested_example\u001b[39m\u001b[34m(schema, obj, token_per_repo_id)\u001b[39m\n\u001b[32m   1416\u001b[39m \u001b[38;5;66;03m# Object with special decoding:\u001b[39;00m\n\u001b[32m   1417\u001b[39m \u001b[38;5;28;01melif\u001b[39;00m \u001b[38;5;28mhasattr\u001b[39m(schema, \u001b[33m\"\u001b[39m\u001b[33mdecode_example\u001b[39m\u001b[33m\"\u001b[39m) \u001b[38;5;129;01mand\u001b[39;00m \u001b[38;5;28mgetattr\u001b[39m(schema, \u001b[33m\"\u001b[39m\u001b[33mdecode\u001b[39m\u001b[33m\"\u001b[39m, \u001b[38;5;28;01mTrue\u001b[39;00m):\n\u001b[32m   1418\u001b[39m     \u001b[38;5;66;03m# we pass the token to read and decode files from private repositories in streaming mode\u001b[39;00m\n\u001b[32m-> \u001b[39m\u001b[32m1419\u001b[39m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mschema\u001b[49m\u001b[43m.\u001b[49m\u001b[43mdecode_example\u001b[49m\u001b[43m(\u001b[49m\u001b[43mobj\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m=\u001b[49m\u001b[43mtoken_per_repo_id\u001b[49m\u001b[43m)\u001b[49m \u001b[38;5;28;01mif\u001b[39;00m obj \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m \u001b[38;5;28;01melse\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m\n\u001b[32m   1420\u001b[39m \u001b[38;5;28;01mreturn\u001b[39;00m obj\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/datasets/features/image.py:192\u001b[39m, in \u001b[36mImage.decode_example\u001b[39m\u001b[34m(self, value, token_per_repo_id)\u001b[39m\n\u001b[32m    190\u001b[39m             image = PIL.Image.open(bytes_)\n\u001b[32m    191\u001b[39m \u001b[38;5;28;01melse\u001b[39;00m:\n\u001b[32m--> \u001b[39m\u001b[32m192\u001b[39m     image = \u001b[43mPIL\u001b[49m\u001b[43m.\u001b[49m\u001b[43mImage\u001b[49m\u001b[43m.\u001b[49m\u001b[43mopen\u001b[49m\u001b[43m(\u001b[49m\u001b[43mBytesIO\u001b[49m\u001b[43m(\u001b[49m\u001b[43mbytes_\u001b[49m\u001b[43m)\u001b[49m\u001b[43m)\u001b[49m\n\u001b[32m    193\u001b[39m image.load()  \u001b[38;5;66;03m# to avoid \"Too many open files\" errors\u001b[39;00m\n\u001b[32m    194\u001b[39m \u001b[38;5;28;01mif\u001b[39;00m image.getexif().get(PIL.Image.ExifTags.Base.Orientation) \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m:\n",
-      "\u001b[36mFile \u001b[39m\u001b[32m~/miniconda3/envs/lab_env/lib/python3.13/site-packages/PIL/Image.py:3560\u001b[39m, in \u001b[36mopen\u001b[39m\u001b[34m(fp, mode, formats)\u001b[39m\n\u001b[32m   3558\u001b[39m     warnings.warn(message)\n\u001b[32m   3559\u001b[39m msg = \u001b[33m\"\u001b[39m\u001b[33mcannot identify image file \u001b[39m\u001b[38;5;132;01m%r\u001b[39;00m\u001b[33m\"\u001b[39m % (filename \u001b[38;5;28;01mif\u001b[39;00m filename \u001b[38;5;28;01melse\u001b[39;00m fp)\n\u001b[32m-> \u001b[39m\u001b[32m3560\u001b[39m \u001b[38;5;28;01mraise\u001b[39;00m UnidentifiedImageError(msg)\n",
-      "\u001b[31mUnidentifiedImageError\u001b[39m: cannot identify image file <_io.BytesIO object at 0x377cfaed0>"
-     ]
-    }
-   ],
-   "source": [
-    "import warnings\n",
-    "from PIL import UnidentifiedImageError\n",
-    "from tqdm.auto import tqdm\n",
-    "\n",
-    "# 1. Silences the 'Corrupt EXIF data' warning cluttering your output\n",
-    "warnings.filterwarnings(\"ignore\", category=UserWarning, module=\"PIL.TiffImagePlugin\")\n",
-    "\n",
-    "print(f\"{'Index':<8} | {'Shape':<20} | {'Label'}\")\n",
-    "print(\"-\" * 40)\n",
-    "\n",
-    "# Using tqdm so you can track progress across the 40,000+ images\n",
-    "for idx, item in enumerate(test):\n",
-    "    try:\n",
-    "        # Accessing the image size triggers the file reading\n",
-    "        shape = item[\"image\"].size()\n",
-    "        label = item[\"label\"]\n",
-    "        print(f\"{idx:<8} | {str(shape):<20} | {label}\")\n",
-    "        \n",
-    "    except (UnidentifiedImageError, OSError) as e:\n",
-    "        # This catches the \"cannot identify image file\" error\n",
-    "        print(f\"{idx:<8} | !!! CORRUPT IMAGE !!! | {item.get('label', '??')}\")\n",
-    "        \n",
-    "    except Exception as e:\n",
-    "        # General safety net for any other weird errors\n",
-    "        print(f\"{idx:<8} | Error: {type(e).__name__} | Label: {item.get('label', '??')}\")\n",
-    "\n",
-    "    # Safety break so you don't print 40k lines at once\n",
-    "    # if idx >= 100: \n",
-    "    #     print(\"\\nStopping at 100 images for preview...\")\n",
-    "    #     break"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 57,
-   "id": "4cb4da2c",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Device: mps\n"
-     ]
-    }
-   ],
-   "source": [
-    "device = t.device(\"cuda\" if t.cuda.is_available() else\n",
-    "                      \"mps\" if t.backends.mps.is_available() else\n",
-    "                      \"cpu\")\n",
-    "\n",
-    "print(f\"Device: {device}\")"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "29aea2e7",
-   "metadata": {},
-   "source": [
-    "## Defining the Model"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 52,
-   "id": "4e00ed32",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import torch.nn as nn\n",
-    "\n",
-    "class BottleneckBlock(nn.Module):\n",
-    "    expansion = 4\n",
-    "\n",
-    "    def __init__(self, in_channels, mid_channels, stride=1):\n",
-    "        super(BottleneckBlock, self).__init__()\n",
-    "        \n",
-    "        out_channels = mid_channels * self.expansion\n",
-    "\n",
-    "        \n",
-    "        self.conv1 = nn.Conv2d(in_channels, mid_channels, kernel_size=1, bias=False)\n",
-    "        self.bn1 = nn.BatchNorm2d(mid_channels)\n",
-    "        \n",
-    "        self.conv2 = nn.Conv2d(mid_channels, mid_channels, kernel_size=3, stride=stride, padding=1, bias=False)\n",
-    "        self.bn2 = nn.BatchNorm2d(mid_channels)\n",
-    "        \n",
-    "        self.conv3 = nn.Conv2d(mid_channels, out_channels, kernel_size=1, bias=False)\n",
-    "        self.bn3 = nn.BatchNorm2d(out_channels)\n",
-    "        \n",
-    "        self.relu = nn.ReLU(inplace=True)\n",
-    "\n",
-    "        self.shortcut = nn.Sequential()\n",
-    "\n",
-    "        if stride != 1 or in_channels != out_channels:\n",
-    "            self.shortcut = nn.Sequential(\n",
-    "                nn.Conv2d(in_channels, out_channels, kernel_size=1, stride=stride, bias=False),\n",
-    "                nn.BatchNorm2d(out_channels)\n",
-    "            )\n",
-    "\n",
-    "    def forward(self, x):\n",
-    "        identity = x\n",
-    "\n",
-    "        out = self.conv1(x)\n",
-    "        out = self.bn1(out)\n",
-    "        out = self.relu(out)\n",
-    "\n",
-    "        out = self.conv2(out)\n",
-    "        out = self.bn2(out)\n",
-    "        out = self.relu(out)\n",
-    "\n",
-    "        out = self.conv3(out)\n",
-    "        out = self.bn3(out)\n",
-    "        \n",
-    "        identity = self.shortcut(identity)\n",
-    "\n",
-    "        out += identity\n",
-    "        out = self.relu(out)\n",
-    "\n",
-    "        return out        \n",
-    "        \n",
-    "\n",
-    "class ResNet50(nn.Module):\n",
-    "    def __init__(self, num_classes=16, channels_img=3):\n",
-    "        super(ResNet50, self).__init__()\n",
-    "        \n",
-    "        self.in_channels = 64\n",
-    "\n",
-    "        self.conv1 = nn.Conv2d(channels_img, 64, kernel_size=7, stride=2, padding=3, bias=False)\n",
-    "        self.bn1 = nn.BatchNorm2d(64)\n",
-    "        self.relu = nn.ReLU(inplace=True)\n",
-    "        self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)\n",
-    "\n",
-    "        self.stage1 = self._make_layer(mid_channels=64, num_blocks=3, stride=1)\n",
-    "        \n",
-    "        self.stage2 = self._make_layer(mid_channels=128, num_blocks=4, stride=2)\n",
-    "        \n",
-    "        self.stage3 = self._make_layer(mid_channels=256, num_blocks=6, stride=2)\n",
-    "        \n",
-    "        self.stage4 = self._make_layer(mid_channels=512, num_blocks=3, stride=2)\n",
-    "\n",
-    "        self.avgpool = nn.AdaptiveAvgPool2d((1, 1))\n",
-    "\n",
-    "        self.fc = nn.Linear(512 * 4, num_classes) # 512*4 = 2048\n",
-    "\n",
-    "    def _make_layer(self, mid_channels, num_blocks, stride):\n",
-    "        layers = []\n",
-    "        \n",
-    "        layers.append(BottleneckBlock(self.in_channels, mid_channels, stride))\n",
-    "        \n",
-    "        self.in_channels = mid_channels * 4\n",
-    "\n",
-    "        for _ in range(num_blocks - 1):\n",
-    "            layers.append(BottleneckBlock(self.in_channels, mid_channels, stride=1))\n",
-    "\n",
-    "        return nn.Sequential(*layers)\n",
-    "\n",
-    "    def forward(self, x):\n",
-    "        # Stem\n",
-    "        x = self.conv1(x)\n",
-    "        x = self.bn1(x)\n",
-    "        x = self.relu(x)\n",
-    "        x = self.maxpool(x)\n",
-    "\n",
-    "        # Stages\n",
-    "        x = self.stage1(x)\n",
-    "        x = self.stage2(x)\n",
-    "        x = self.stage3(x)\n",
-    "        x = self.stage4(x)\n",
-    "\n",
-    "        # Head\n",
-    "        x = self.avgpool(x)\n",
-    "        x = t.flatten(x, 1) # Flatten from (N, 2048, 1, 1) to (N, 2048)\n",
-    "        x = self.fc(x)\n",
-    "\n",
-    "        return x        \n",
-    "\n",
-    "        "
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 53,
-   "id": "44c64c24",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "mps\n"
-     ]
-    }
-   ],
-   "source": [
-    "device = t.device(\"cuda\" if t.cuda.is_available() else\n",
-    "                      \"mps\" if t.backends.mps.is_available() else\n",
-    "                      \"cpu\")\n",
-    "print(device)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 54,
-   "id": "a6c18e07",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "ResNet50(\n",
-      "  (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)\n",
-      "  (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "  (relu): ReLU(inplace=True)\n",
-      "  (maxpool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)\n",
-      "  (stage1): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(64, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "        (1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage2): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(256, 512, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (3): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage3): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(512, 1024, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (3): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (4): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (5): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage4): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(1024, 2048, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (avgpool): AdaptiveAvgPool2d(output_size=(1, 1))\n",
-      "  (fc): Linear(in_features=2048, out_features=16, bias=True)\n",
-      ")\n"
-     ]
-    }
-   ],
-   "source": [
-    "resnet50 = ResNet50().to(device)\n",
-    "print(resnet50)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 55,
-   "id": "db985cee",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "layer_name: conv1.weight, params_shape: torch.Size([64, 3, 7, 7])\n",
-      "layer_name: bn1.weight, params_shape: torch.Size([64])\n",
-      "layer_name: bn1.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.0.conv1.weight, params_shape: torch.Size([64, 64, 1, 1])\n",
-      "layer_name: stage1.0.bn1.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.0.bn1.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.0.conv2.weight, params_shape: torch.Size([64, 64, 3, 3])\n",
-      "layer_name: stage1.0.bn2.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.0.bn2.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.0.conv3.weight, params_shape: torch.Size([256, 64, 1, 1])\n",
-      "layer_name: stage1.0.bn3.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.0.bn3.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.0.shortcut.0.weight, params_shape: torch.Size([256, 64, 1, 1])\n",
-      "layer_name: stage1.0.shortcut.1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.0.shortcut.1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.1.conv1.weight, params_shape: torch.Size([64, 256, 1, 1])\n",
-      "layer_name: stage1.1.bn1.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.1.bn1.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.1.conv2.weight, params_shape: torch.Size([64, 64, 3, 3])\n",
-      "layer_name: stage1.1.bn2.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.1.bn2.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.1.conv3.weight, params_shape: torch.Size([256, 64, 1, 1])\n",
-      "layer_name: stage1.1.bn3.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.1.bn3.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.2.conv1.weight, params_shape: torch.Size([64, 256, 1, 1])\n",
-      "layer_name: stage1.2.bn1.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.2.bn1.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.2.conv2.weight, params_shape: torch.Size([64, 64, 3, 3])\n",
-      "layer_name: stage1.2.bn2.weight, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.2.bn2.bias, params_shape: torch.Size([64])\n",
-      "layer_name: stage1.2.conv3.weight, params_shape: torch.Size([256, 64, 1, 1])\n",
-      "layer_name: stage1.2.bn3.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage1.2.bn3.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage2.0.conv1.weight, params_shape: torch.Size([128, 256, 1, 1])\n",
-      "layer_name: stage2.0.bn1.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.0.bn1.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.0.conv2.weight, params_shape: torch.Size([128, 128, 3, 3])\n",
-      "layer_name: stage2.0.bn2.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.0.bn2.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.0.conv3.weight, params_shape: torch.Size([512, 128, 1, 1])\n",
-      "layer_name: stage2.0.bn3.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.0.bn3.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.0.shortcut.0.weight, params_shape: torch.Size([512, 256, 1, 1])\n",
-      "layer_name: stage2.0.shortcut.1.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.0.shortcut.1.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.1.conv1.weight, params_shape: torch.Size([128, 512, 1, 1])\n",
-      "layer_name: stage2.1.bn1.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.1.bn1.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.1.conv2.weight, params_shape: torch.Size([128, 128, 3, 3])\n",
-      "layer_name: stage2.1.bn2.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.1.bn2.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.1.conv3.weight, params_shape: torch.Size([512, 128, 1, 1])\n",
-      "layer_name: stage2.1.bn3.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.1.bn3.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.2.conv1.weight, params_shape: torch.Size([128, 512, 1, 1])\n",
-      "layer_name: stage2.2.bn1.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.2.bn1.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.2.conv2.weight, params_shape: torch.Size([128, 128, 3, 3])\n",
-      "layer_name: stage2.2.bn2.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.2.bn2.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.2.conv3.weight, params_shape: torch.Size([512, 128, 1, 1])\n",
-      "layer_name: stage2.2.bn3.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.2.bn3.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.3.conv1.weight, params_shape: torch.Size([128, 512, 1, 1])\n",
-      "layer_name: stage2.3.bn1.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.3.bn1.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.3.conv2.weight, params_shape: torch.Size([128, 128, 3, 3])\n",
-      "layer_name: stage2.3.bn2.weight, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.3.bn2.bias, params_shape: torch.Size([128])\n",
-      "layer_name: stage2.3.conv3.weight, params_shape: torch.Size([512, 128, 1, 1])\n",
-      "layer_name: stage2.3.bn3.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage2.3.bn3.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage3.0.conv1.weight, params_shape: torch.Size([256, 512, 1, 1])\n",
-      "layer_name: stage3.0.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.0.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.0.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.0.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.0.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.0.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.0.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.0.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.0.shortcut.0.weight, params_shape: torch.Size([1024, 512, 1, 1])\n",
-      "layer_name: stage3.0.shortcut.1.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.0.shortcut.1.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.1.conv1.weight, params_shape: torch.Size([256, 1024, 1, 1])\n",
-      "layer_name: stage3.1.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.1.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.1.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.1.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.1.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.1.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.1.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.1.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.2.conv1.weight, params_shape: torch.Size([256, 1024, 1, 1])\n",
-      "layer_name: stage3.2.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.2.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.2.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.2.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.2.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.2.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.2.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.2.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.3.conv1.weight, params_shape: torch.Size([256, 1024, 1, 1])\n",
-      "layer_name: stage3.3.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.3.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.3.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.3.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.3.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.3.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.3.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.3.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.4.conv1.weight, params_shape: torch.Size([256, 1024, 1, 1])\n",
-      "layer_name: stage3.4.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.4.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.4.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.4.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.4.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.4.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.4.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.4.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.5.conv1.weight, params_shape: torch.Size([256, 1024, 1, 1])\n",
-      "layer_name: stage3.5.bn1.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.5.bn1.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.5.conv2.weight, params_shape: torch.Size([256, 256, 3, 3])\n",
-      "layer_name: stage3.5.bn2.weight, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.5.bn2.bias, params_shape: torch.Size([256])\n",
-      "layer_name: stage3.5.conv3.weight, params_shape: torch.Size([1024, 256, 1, 1])\n",
-      "layer_name: stage3.5.bn3.weight, params_shape: torch.Size([1024])\n",
-      "layer_name: stage3.5.bn3.bias, params_shape: torch.Size([1024])\n",
-      "layer_name: stage4.0.conv1.weight, params_shape: torch.Size([512, 1024, 1, 1])\n",
-      "layer_name: stage4.0.bn1.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.0.bn1.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.0.conv2.weight, params_shape: torch.Size([512, 512, 3, 3])\n",
-      "layer_name: stage4.0.bn2.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.0.bn2.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.0.conv3.weight, params_shape: torch.Size([2048, 512, 1, 1])\n",
-      "layer_name: stage4.0.bn3.weight, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.0.bn3.bias, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.0.shortcut.0.weight, params_shape: torch.Size([2048, 1024, 1, 1])\n",
-      "layer_name: stage4.0.shortcut.1.weight, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.0.shortcut.1.bias, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.1.conv1.weight, params_shape: torch.Size([512, 2048, 1, 1])\n",
-      "layer_name: stage4.1.bn1.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.1.bn1.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.1.conv2.weight, params_shape: torch.Size([512, 512, 3, 3])\n",
-      "layer_name: stage4.1.bn2.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.1.bn2.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.1.conv3.weight, params_shape: torch.Size([2048, 512, 1, 1])\n",
-      "layer_name: stage4.1.bn3.weight, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.1.bn3.bias, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.2.conv1.weight, params_shape: torch.Size([512, 2048, 1, 1])\n",
-      "layer_name: stage4.2.bn1.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.2.bn1.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.2.conv2.weight, params_shape: torch.Size([512, 512, 3, 3])\n",
-      "layer_name: stage4.2.bn2.weight, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.2.bn2.bias, params_shape: torch.Size([512])\n",
-      "layer_name: stage4.2.conv3.weight, params_shape: torch.Size([2048, 512, 1, 1])\n",
-      "layer_name: stage4.2.bn3.weight, params_shape: torch.Size([2048])\n",
-      "layer_name: stage4.2.bn3.bias, params_shape: torch.Size([2048])\n",
-      "layer_name: fc.weight, params_shape: torch.Size([16, 2048])\n",
-      "layer_name: fc.bias, params_shape: torch.Size([16])\n"
-     ]
-    }
-   ],
-   "source": [
-    "for layer in resnet50.named_parameters():\n",
-    "    print(f\"layer_name: {layer[0]}, params_shape: {layer[1].shape}\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 56,
-   "id": "eff57243",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "ResNet50(\n",
-      "  (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)\n",
-      "  (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "  (relu): ReLU(inplace=True)\n",
-      "  (maxpool): MaxPool2d(kernel_size=3, stride=2, padding=1, dilation=1, ceil_mode=False)\n",
-      "  (stage1): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(64, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "        (1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage2): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(256, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(256, 512, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (3): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(128, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage3): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(512, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(512, 1024, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (3): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (4): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (5): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (stage4): Sequential(\n",
-      "    (0): BottleneckBlock(\n",
-      "      (conv1): Conv2d(1024, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential(\n",
-      "        (0): Conv2d(1024, 2048, kernel_size=(1, 1), stride=(2, 2), bias=False)\n",
-      "        (1): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      )\n",
-      "    )\n",
-      "    (1): BottleneckBlock(\n",
-      "      (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "    (2): BottleneckBlock(\n",
-      "      (conv1): Conv2d(2048, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)\n",
-      "      (bn2): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (conv3): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)\n",
-      "      (bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)\n",
-      "      (relu): ReLU(inplace=True)\n",
-      "      (shortcut): Sequential()\n",
-      "    )\n",
-      "  )\n",
-      "  (avgpool): AdaptiveAvgPool2d(output_size=(1, 1))\n",
-      "  (fc): Linear(in_features=2048, out_features=16, bias=True)\n",
-      ")\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(resnet50)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "ce8b5ce6",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  }
- ],
- "metadata": {
-  "kernelspec": {
-   "display_name": "lab_env",
-   "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.13.11"
-  }
- },
- "nbformat": 4,
- "nbformat_minor": 5
-}