{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "a6065b0a",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/root/anaconda3/envs/torch190-test/lib/python3.8/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
      "  from .autonotebook import tqdm as notebook_tqdm\n",
      "/tmp/ipykernel_2871875/1491415481.py:14: DeprecationWarning: `np.float` is a deprecated alias for the builtin `float`. To silence this warning, use `float` by itself. Doing this will not modify any behavior and is safe. If you specifically wanted the numpy scalar type, use `np.float64` here.\n",
      "Deprecated in NumPy 1.20; for more details and guidance: https://numpy.org/devdocs/release/1.20.0-notes.html#deprecations\n",
      "  from cython_bbox import bbox_overlaps as bbox_ious\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "import numpy as np\n",
    "import cv2\n",
    "import matplotlib.pyplot as plt\n",
    "from IPython.display import clear_output\n",
    "import os\n",
    "from PIL import Image\n",
    "import pandas as pd\n",
    "import time\n",
    "import natsort\n",
    "from tqdm import tqdm\n",
    "\n",
    "from multicam_utils import give_color\n",
    "from cython_bbox import bbox_overlaps as bbox_ious\n",
    "\n",
    "import cv2\n",
    "import time\n",
    "import torch\n",
    "import torch.nn.functional as F\n",
    "import argparse\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "from torchvision import transforms\n",
    "\n",
    "import copy\n",
    "import scipy\n",
    "import lap\n",
    "from scipy.spatial.distance import cdist\n",
    "import warnings\n",
    "\n",
    "%matplotlib inline\n",
    "warnings.filterwarnings(\"ignore\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d6819de4",
   "metadata": {},
   "source": [
    "## Load obj in recent frames"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "5e16dd14",
   "metadata": {},
   "outputs": [],
   "source": [
    "# load obj_img and calculate average feat\n",
    "\n",
    "# average feat based matching\n",
    "\n",
    "DISTANCE = 100\n",
    "\n",
    "# new id is in then get average\n",
    "\n",
    "tracking_result_path = '/data/cvprw/AIC23/tracking/SUBMISSION/S014'\n",
    "\n",
    "channel_list = []\n",
    "temp = [file for file in os.listdir(tracking_result_path)]\n",
    "for channel in temp:\n",
    "    d = os.path.join(tracking_result_path, channel)\n",
    "    if os.path.isdir(d):\n",
    "#         print(channel)\n",
    "        channel_list.append(channel)\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "1958df76",
   "metadata": {},
   "outputs": [],
   "source": [
    "channel_list=natsort.natsorted(channel_list)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "12b15a3b",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['c076', 'c077', 'c078', 'c079', 'c080', 'c081']"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "del channel_list[0]\n",
    "channel_list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "6d0d43d1",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "# Open two video capture objects\n",
    "cap1 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[0], 'video.mp4'))\n",
    "cap2 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[1], 'video.mp4'))\n",
    "cap3 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[2], 'video.mp4'))\n",
    "cap4 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[3], 'video.mp4'))\n",
    "cap5 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[4], 'video.mp4'))\n",
    "cap6 = cv2.VideoCapture(os.path.join(tracking_result_path, channel_list[5], 'video.mp4'))\n",
    "\n",
    "_, frame1 = cap1.read()\n",
    "_, frame2 = cap2.read()\n",
    "_, frame3 = cap3.read()\n",
    "_, frame4 = cap4.read()\n",
    "_, frame5 = cap5.read()\n",
    "_, frame6 = cap6.read()\n",
    "                        \n",
    "with open(os.path.join(tracking_result_path, channel_list[0], 'label.txt')) as label1:\n",
    "    gt1 = label1.readlines()\n",
    "with open(os.path.join(tracking_result_path, channel_list[1], 'label.txt')) as label2:\n",
    "    gt2 = label2.readlines()\n",
    "with open(os.path.join(tracking_result_path, channel_list[2], 'label.txt')) as label3:\n",
    "    gt3 = label3.readlines()\n",
    "with open(os.path.join(tracking_result_path, channel_list[3], 'label.txt')) as label4:\n",
    "    gt4 = label4.readlines()\n",
    "with open(os.path.join(tracking_result_path, channel_list[4], 'label.txt')) as label5:\n",
    "    gt5 = label5.readlines()\n",
    "with open(os.path.join(tracking_result_path, channel_list[5], 'label.txt')) as label6:\n",
    "    gt6 = label6.readlines()\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "b57613a6",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "18010\n"
     ]
    }
   ],
   "source": [
    "\n",
    "width = cap1.get(cv2.CAP_PROP_FRAME_WIDTH)  # float\n",
    "height = cap1.get(cv2.CAP_PROP_FRAME_HEIGHT)  # float\n",
    "fps = cap1.get(cv2.CAP_PROP_FPS)\n",
    "vid_length = int(cap1.get(cv2.CAP_PROP_FRAME_COUNT))\n",
    "\n",
    "print(vid_length)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "2f40821b-7ff8-4e7c-80de-811dda26c6e9",
   "metadata": {},
   "source": [
    "# Tools"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "53bf4e9a",
   "metadata": {},
   "outputs": [],
   "source": [
    "def draw_bbox_tracking(img, results, thres):\n",
    "    for obj in results:\n",
    "        if obj[4]>thres:\n",
    "            x = int(obj[0])\n",
    "            y = int(obj[1])\n",
    "            x2 = int(obj[2])\n",
    "            y2 = int(obj[3])\n",
    "            \n",
    "            tracking_id = obj[5]\n",
    "            \n",
    "            color = give_color(tracking_id)\n",
    "            \n",
    "            tk = 3\n",
    "            \n",
    "            content = str(tracking_id)\n",
    "            \n",
    "            font =  cv2.FONT_HERSHEY_PLAIN\n",
    "            img = cv2.rectangle(img, (x,y), (x2,y2), color, tk) # bbox\n",
    "            img = cv2.putText(img, content, (x, y-2), font, tk, color, tk, cv2.LINE_AA) # label\n",
    "    return img\n",
    "\n",
    "\n",
    "def extract_tracking_result(gt, frame_count, W, H):\n",
    "    track_bbox=[]\n",
    "    for row in gt:\n",
    "        if int(row.split(',')[0]) == frame_count:\n",
    "            track_id=int(row.split(',')[1])\n",
    "            \n",
    "            x = int(max(0, int(float(row.split(',')[2]))))\n",
    "            y = int(max(0, int(float(row.split(',')[3]))))\n",
    "            if int(float(row.split(',')[4]))+x >W:\n",
    "                obj_w=W-x-1\n",
    "            else:\n",
    "                obj_w=int(float(row.split(',')[4]))\n",
    "                \n",
    "            if int(float(row.split(',')[5]))+y >H:\n",
    "                obj_h=H-y-1\n",
    "            else:\n",
    "                obj_h=int(float(row.split(',')[5]))\n",
    "            \n",
    "            track_bbox.append([x,y,x+obj_w,y+obj_h,float(row.split(',')[6]),track_id])\n",
    "            \n",
    "        elif int(row.split(',')[0]) > frame_count:\n",
    "            break\n",
    "            \n",
    "    return track_bbox\n",
    "\n",
    "def plot_site(result_img1,result_img2,result_img3,result_img4,result_img5,map_img):\n",
    "    fig = plt.figure(figsize=(20,8))\n",
    "    rows = 2\n",
    "    cols = 3\n",
    "\n",
    "    ax1 = fig.add_subplot(rows, cols, 1)\n",
    "    ax1.imshow(cv2.cvtColor(result_img1, cv2.COLOR_BGR2RGB))\n",
    "    ax1.set_title('C025')\n",
    "    ax1.axis(\"off\")\n",
    "\n",
    "    ax2 = fig.add_subplot(rows, cols, 2)\n",
    "    ax2.imshow(cv2.cvtColor(result_img2, cv2.COLOR_BGR2RGB))\n",
    "    ax2.set_title('C026')\n",
    "    ax2.axis(\"off\")\n",
    "\n",
    "    ax3 = fig.add_subplot(rows, cols, 3)\n",
    "    ax3.imshow(cv2.cvtColor(result_img3, cv2.COLOR_BGR2RGB))\n",
    "    ax3.set_title('C027')\n",
    "    ax3.axis(\"off\")\n",
    "\n",
    "    ax4 = fig.add_subplot(rows, cols, 4)\n",
    "    ax4.imshow(cv2.cvtColor(result_img4, cv2.COLOR_BGR2RGB))\n",
    "    ax4.set_title('C028')\n",
    "    ax4.axis(\"off\")\n",
    "\n",
    "    ax5 = fig.add_subplot(rows, cols, 5)\n",
    "    ax5.imshow(cv2.cvtColor(result_img5, cv2.COLOR_BGR2RGB))\n",
    "    ax5.set_title('C029')\n",
    "    ax5.axis(\"off\")\n",
    "\n",
    "    ax6 = fig.add_subplot(rows, cols, 6)\n",
    "    ax6.imshow(cv2.cvtColor(map_img, cv2.COLOR_BGR2RGB))\n",
    "    ax6.set_title('MAP')\n",
    "    ax6.axis(\"off\")\n",
    "\n",
    "    plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "1ab18d4c",
   "metadata": {},
   "outputs": [],
   "source": [
    "def make_square(img):\n",
    "    \"\"\"\n",
    "    Given an image as a cv2 object, add padding to make it square.\n",
    "    \"\"\"\n",
    "    # Get the image dimensions\n",
    "    height, width = img.shape[:2]\n",
    "\n",
    "    # Determine the size of the square to make\n",
    "    square_size = max(width, height)\n",
    "\n",
    "    # Create a new image of the appropriate size, with a black background\n",
    "    new_img = np.zeros((square_size, square_size, 3), dtype=np.uint8)\n",
    "\n",
    "    # Determine where to paste the original image in the new image\n",
    "    x_offset = (square_size - width) // 2\n",
    "    y_offset = (square_size - height) // 2\n",
    "\n",
    "    # Paste the original image in the center of the new image\n",
    "    new_img[y_offset:y_offset+height, x_offset:x_offset+width] = img\n",
    "\n",
    "    return new_img\n",
    "\n",
    "def extract_global_position2(frame_bbox, H):\n",
    "    # format (x,y,raw_track_id)\n",
    "    c_x = (frame_bbox[0]+frame_bbox[2])/2\n",
    "    bottom_y = frame_bbox[1]+(frame_bbox[3]-frame_bbox[1])*0.95\n",
    "\n",
    "    point1 = np.array([[c_x, bottom_y, 1]])\n",
    "    point2 = np.dot(H, point1.T)\n",
    "    point2 /= point2[2]\n",
    "        \n",
    "    return int(point2[0]),int(point2[1])\n",
    "\n",
    "def add_projected_point2(projected_point_group, obj_list, channel_idx, global_obj_mapping):\n",
    "    # projected_point_group [[[],[]].[[]],...]\n",
    "    # Fromat : (<channel_track_id> <project_x> <project_y> <reid_feat> <state_bit>), <channel_idx>)\n",
    "    \n",
    "    distance = DISTANCE # 두개 그룹 이상 매칭 되면 feat으로 한 그룹에 묶어줌\n",
    "    \n",
    "    if len(projected_point_group)==0:\n",
    "        for row in obj_list:\n",
    "            row.append(channel_idx)\n",
    "            projected_point_group.append([row])\n",
    "    else:\n",
    "        for row in obj_list:\n",
    "            match_idx = []\n",
    "            for idx,group in enumerate(projected_point_group): # 클러스터링 된 수만큼 그룹수 존재 (idx)\n",
    "                _, avg_x, avg_y = np.mean(np.array(group)[:,:3], axis=0)\n",
    "                if (row[1] - avg_x)**2 + (row[2] - avg_y)**2 < distance**2: # distanec under 100pixel\n",
    "                    match_idx.append(idx)\n",
    "\n",
    "            if len(match_idx)==1:\n",
    "                row.append(channel_idx)\n",
    "                projected_point_group[match_idx[0]].append(row)\n",
    "            elif len(match_idx)>1:\n",
    "                min_distance = 0.05\n",
    "                match_id = None\n",
    "                for group_id in match_idx:\n",
    "                    for obj in projected_point_group[group_id]:\n",
    "                        if embedding_distance(obj[3],row[3])<min_distance:\n",
    "                            match_id = group_id\n",
    "                            \n",
    "                if match_id is not None:\n",
    "                    row.append(channel_idx)\n",
    "                    projected_point_group[match_id].append(row)\n",
    "                else:\n",
    "                    print(\"두개 이상 매칭 되는데 유사한건 없음\")\n",
    "                    row.append(channel_idx)\n",
    "                    projected_point_group.append([row])\n",
    "                \n",
    "            else:\n",
    "                row.append(channel_idx)\n",
    "                projected_point_group.append([row])\n",
    "\n",
    "    return projected_point_group\n",
    "\n",
    "def make_occlusion_matrix(dets):\n",
    "    occluded_val_list = []\n",
    "    if len(dets)>0:\n",
    "        ious = bbox_ious(\n",
    "            np.ascontiguousarray(dets, dtype=np.float64),\n",
    "            np.ascontiguousarray(dets, dtype=np.float64)\n",
    "        )\n",
    "\n",
    "        for iou_row in ious:\n",
    "            if sum(iou_row)>1.05:\n",
    "                occluded_val_list.append(True)\n",
    "            else:\n",
    "                occluded_val_list.append(False)\n",
    "                \n",
    "    return occluded_val_list\n",
    "\n",
    "def run_keypoint(frame_bbox, frame_img, keypoint_model, strong=True):\n",
    "    thres = 0.5\n",
    "    padding_size = 10\n",
    "    \n",
    "    # padding 40px and make square based on height\n",
    "    padding_height = frame_bbox[3]-frame_bbox[1]+padding_size\n",
    "    center_x = (frame_bbox[0]+frame_bbox[2])/2\n",
    "\n",
    "    patch_x1 = int(max(0, frame_bbox[0]-padding_size))\n",
    "    patch_y1 = int(max(0, frame_bbox[1]-padding_size))\n",
    "    patch_x2 = int(min(W - 1, frame_bbox[2]+padding_size))\n",
    "    patch_y2 = int(min(H - 1, frame_bbox[3]+padding_size))\n",
    "\n",
    "    patch = frame_img[patch_y1:patch_y2, patch_x1:patch_x2, :]\n",
    "    patch_squared = make_square(patch)\n",
    "\n",
    "#                 resized_patch = cv2.resize(patch, (256,256))\n",
    "    square_size = 256\n",
    "    resized_patch = cv2.resize(cv2.cvtColor(patch_squared, cv2.COLOR_BGR2RGB), (square_size,square_size))\n",
    "    \n",
    "    # %matplotlib inline\n",
    "    # plt.imshow(resized_patch)\n",
    "    # plt.show()\n",
    "    \n",
    "    resized_patch = torch.tensor(np.array([transforms.ToTensor()(resized_patch).numpy()]))\n",
    "\n",
    "    resized_patch = resized_patch.to(\"cuda:0\")  #convert image data to device\n",
    "    resized_patch = resized_patch.float() #convert image to float precision (cpu)\n",
    "\n",
    "    with torch.no_grad():  #get predictions\n",
    "        output_data, _ = keypoint_model(resized_patch)\n",
    "\n",
    "    output_data = non_max_suppression_kpt(output_data,   #Apply non max suppression\n",
    "                0.25,   # Conf. Threshold.\n",
    "                0.65, # IoU Threshold.\n",
    "                nc=keypoint_model.yaml['nc'], # Number of classes.\n",
    "                nkpt=keypoint_model.yaml['nkpt'], # Number of keypoints.\n",
    "                kpt_label=True)\n",
    "    output = output_to_keypoint(output_data)\n",
    "    \n",
    "    total_crop = frame_img[frame_bbox[1]:frame_bbox[3], frame_bbox[0]:frame_bbox[2], :]\n",
    "\n",
    "    obj_width = frame_bbox[2] - frame_bbox[0]\n",
    "    obj_height = frame_bbox[3] - frame_bbox[1]\n",
    "    # 전체 감지 : 2, 감지안됨 : 0\n",
    "    state_bit = 0\n",
    "    \n",
    "    max_area = 10\n",
    "    for o in output:\n",
    "        obj_area = o[4]*o[5]\n",
    "        if (obj_area>max_area) and (o[54]>thres or o[57]>thres) and (o[18]>thres or o[21]>thres or o[24]>thres or o[27]>thres):\n",
    "            max_area = obj_area\n",
    "            state_bit = 2  \n",
    "    \n",
    "    \n",
    "    # 전체 인데 반만 감지되거나, 아예 감지되지 않는 경우도 있음.\n",
    "    #감지는 안되지만 가로세로 비로 추정\n",
    "    # if not strong and obj_height>obj_width*2.5:\n",
    "    #     state_bit = 2\n",
    "\n",
    "    return state_bit, total_crop\n",
    "\n",
    "def embedding_distance(tracks, detections, metric='cosine'):\n",
    "    \"\"\"\n",
    "    :param tracks: list[STrack]\n",
    "    :param detections: list[BaseTrack]\n",
    "    :param metric:\n",
    "    :return: cost_matrix np.ndarray\n",
    "    \"\"\"\n",
    "\n",
    "    cost_matrix = np.zeros((len(tracks), len(detections)), dtype=np.float64)\n",
    "    if cost_matrix.size == 0:\n",
    "        return cost_matrix\n",
    "\n",
    "    det_features = np.asarray([tracks], dtype=np.float64)\n",
    "    track_features = np.asarray([detections], dtype=np.float64)\n",
    "\n",
    "    cost_matrix = np.maximum(0.0, cdist(track_features, det_features, metric))  # / 2.0  # Nomalized features\n",
    "    return cost_matrix\n",
    "\n",
    "def postprocess(features):\n",
    "    # Normalize feature to compute cosine distance\n",
    "    features = F.normalize(features)\n",
    "    features = features.cpu().data.numpy()\n",
    "    return features\n",
    "\n",
    "def extract_reid_feature(img, model):\n",
    "    img = cv2.resize(img, tuple((100,300)), interpolation=cv2.INTER_LINEAR)\n",
    "    \n",
    "    patches = []\n",
    "    \n",
    "    patch_img = torch.as_tensor(img.astype(\"float32\").transpose(2, 0, 1))\n",
    "    patch_img = patch_img.to(device=\"cuda:0\").half()\n",
    "    patches.append(patch_img)\n",
    "    \n",
    "    if len(patches):\n",
    "        patches = torch.stack(patches, dim=0)\n",
    "\n",
    "    features = np.zeros((0, 2048))\n",
    "    \n",
    "    # Run model\n",
    "    patches_ = torch.clone(patches)\n",
    "    pred = model(patches)\n",
    "    pred[torch.isinf(pred)] = 1.0\n",
    "\n",
    "    feat = postprocess(pred)\n",
    "\n",
    "    return feat[0]\n",
    "    \n",
    "def check_occlusion(local_obj_mapping):\n",
    "    \n",
    "    occlusion_xy = [] \n",
    "    \n",
    "    distance = DISTANCE\n",
    "    \n",
    "    # ***한채널에 대해서 분석해서 글로벌 좌표 뽑음\n",
    "    for channel_obj in local_obj_mapping:\n",
    "        for obj1 in channel_obj:\n",
    "            for obj2 in channel_obj:\n",
    "                if (obj1 != obj2) and (obj1[-1] == 2 and obj2[-1] == 2) and ((obj1[1]-obj2[1])**2+(obj1[2]-obj2[2])**2) < distance**2:\n",
    "                    occlusion_xy.append((int((obj1[1]+obj2[1])/2), int((obj1[2]+obj2[2])/2)))\n",
    "        \n",
    "    # local -> [[],[],[],[],[]]\n",
    "    for idx, channel_obj in enumerate(local_obj_mapping):\n",
    "        for jdx, obj in enumerate(channel_obj):\n",
    "            if obj[-1] == 2:\n",
    "                for item in occlusion_xy:\n",
    "                    if (item[0]-obj[1])**2 + (item[1]-obj[2])**2 < (distance/2+10)**2:\n",
    "                        local_obj_mapping[idx][jdx][-1]=3\n",
    "                    \n",
    "    return local_obj_mapping\n",
    "\n",
    "def update_feature(origin_feat_list, new_feat_list):\n",
    "    origin_feat_list = list(origin_feat_list)\n",
    "    new_feat_avg = np.mean(new_feat_list, axis=0)\n",
    "    \n",
    "    update_bit = False\n",
    "    for idx, feat in enumerate(origin_feat_list):\n",
    "        if embedding_distance(feat,new_feat_avg)<0.05:\n",
    "            update_bit = True\n",
    "            origin_feat_list[idx]= origin_feat_list[idx]*0.99 + new_feat_avg*0.01\n",
    "            \n",
    "    if not update_bit:\n",
    "        origin_feat_list.append(new_feat_avg)\n",
    "        \n",
    "    new_feat_list = np.array(origin_feat_list)\n",
    "\n",
    "    return new_feat_list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5dfadbde-f472-44a5-ab18-11754f7ff33e",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "markdown",
   "id": "b66fcc80",
   "metadata": {},
   "source": [
    "# Camera Calibration"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "d870dd94",
   "metadata": {},
   "outputs": [],
   "source": [
    "pts1 = np.array([(690, 603),(944, 573),(1073, 575),(1494, 483),(1586, 492),(1697, 705),(1822, 773)])\n",
    "pts2 = np.array([(761, 849),(620, 849),(576, 808),(57, 945),(57, 867),(576, 483),(621, 419)])\n",
    "H_c076, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "pts1 = np.array([(255,749),(383,548),(520,367),(623,338),(950,331),(1297,345),(1343,510),(1277, 405)])\n",
    "pts2 = np.array([(417,483),(576,483),(929,495),(999,549),(999,736),(932,922),(576,808),(761, 849)])\n",
    "H_c077, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "pts1 = np.array([(548, 614),(881,515),(1055, 510),(1711, 343), (1861, 359)])\n",
    "pts2 = np.array([(761, 849),(620,849),(576, 808),(57, 744), (58, 672)])\n",
    "H_c078, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "pts1 = np.array([(619,713),(651,561),(658,529),(724,385),(893,383),(1121,380),(1175,583),(1229,625)])\n",
    "pts2 = np.array([(57,945),(57,744),(58,672),(97,32),(233,32),(417,32),(246,808),(246,869)])\n",
    "H_c079, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "pts1 = np.array([(196, 523),(809, 456),(1016, 629),(1525, 810),(1672, 988)])\n",
    "pts2 = np.array([(999, 736),(932, 922),(761, 849),(620, 849),(576, 808)])\n",
    "H_c080, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "pts1 = np.array([(205,384),(294,367),(424,506),(763,541),(834,508),(929,516),(1061,400),(1758,653)])\n",
    "pts2 = np.array([(66,256),(66,182),(417,483),(576,483),(576,419),(621,419),(580,119),(929,495)])\n",
    "H_c081, _ = cv2.findHomography(pts1, pts2, cv2.RANSAC)\n",
    "\n",
    "H_list = [H_c076,H_c077,H_c078,H_c079,H_c080,H_c081]"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "ee620350",
   "metadata": {},
   "source": [
    "# Read Frame"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "1550164e",
   "metadata": {},
   "outputs": [],
   "source": [
    "from models.experimental import attempt_load\n",
    "from utils.general import non_max_suppression_kpt,strip_optimizer,xyxy2xywh\n",
    "from utils.plots import output_to_keypoint, plot_skeleton_kpts,colors,plot_one_box_kpt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "7cb6d3c3",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Fusing layers... \n"
     ]
    }
   ],
   "source": [
    "# Define MAX\n",
    "MAX_OBJ_NUM = 8\n",
    "\n",
    "poseweights = \"yolov7-w6-pose.pt\"\n",
    "device = \"cuda:0\"\n",
    "\n",
    "keypoint_model = attempt_load(poseweights, map_location=device)  #Load model\n",
    "_ = keypoint_model.eval()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "f06a5aff-5704-4a83-b712-67f3f5fccbf5",
   "metadata": {},
   "outputs": [],
   "source": [
    "from fast_reid.fastreid.config import get_cfg\n",
    "from fast_reid.fastreid.modeling.meta_arch import build_model\n",
    "from fast_reid.fastreid.utils.checkpoint import Checkpointer\n",
    "from fast_reid.fastreid.engine import DefaultTrainer, default_argument_parser, default_setup, launch\n",
    "\n",
    "def setup_cfg(config_file, opts):\n",
    "    # load config from file and command-line arguments\n",
    "    cfg = get_cfg()\n",
    "    cfg.merge_from_file(config_file)\n",
    "    cfg.merge_from_list(opts)\n",
    "    cfg.MODEL.BACKBONE.PRETRAIN = False\n",
    "\n",
    "    cfg.freeze()\n",
    "\n",
    "    return cfg\n",
    "\n",
    "config_file = '/data/cvprw/AIC23/BoT-SORT/logs/AIC23/sbs_S101/config.yaml'\n",
    "weights_path = '/data/cvprw/AIC23/BoT-SORT/logs/AIC23/sbs_S101/model_0099.pth'\n",
    "\n",
    "cfg = setup_cfg(config_file, ['MODEL.WEIGHTS', weights_path])\n",
    "\n",
    "reid_model = build_model(cfg)\n",
    "reid_model.eval()\n",
    "\n",
    "Checkpointer(reid_model).load(weights_path)\n",
    "\n",
    "reid_model = reid_model.eval().to(device='cuda:0').half()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "7a642bb6",
   "metadata": {},
   "outputs": [],
   "source": [
    "start = 0\n",
    "\n",
    "# channel_dic c025~c029\n",
    "# Format : ({channel},{},...) | channel -> (<channel_track_id> : <global_track_id>)\n",
    "channel_id_mapping = [{},{},{},{},{},{}]\n",
    "\n",
    "# global_dic\n",
    "# Format : ({id},{},...) | \n",
    "# id -> (<global_track_id> : <projected_x>, <projected_y>, [<feat_total>,...])\n",
    "global_obj_mapping = {}\n",
    "\n",
    "# loss for head only or occlusion\n",
    "# channel loss c025~c029\n",
    "# Format : channel_loss_mapping -> ([],[],...) | \n",
    "# channel -> (<frame_count> <channel_idx>  <channel_result>)\n",
    "channel_loss_final_output = [[],[],[],[],[],[]]\n",
    "\n",
    "# Save Final Result\n",
    "# Format : channel_final_output -> ([channel],[],...)\n",
    "# channel -> (〈channel_idx〉 〈global_track_id〉 〈frame_id〉 〈xmin〉 〈ymin〉 〈width〉 〈height〉 〈xworld〉 〈yworld〉)\n",
    "#  ==> <channel_result>\n",
    "channel_final_output = [[],[],[],[],[],[]]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "8c46b316-7307-47fe-9c1b-cd605644b517",
   "metadata": {},
   "outputs": [],
   "source": [
    "def add_local2gloabl(global_obj_mapping, channel_id_mapping, local_obj_mapping, new_id):\n",
    "    \n",
    "    local_channel_loss_mapping = [[],[],[],[],[],[]]\n",
    "    # projected_point_group [[[],[]].[[]],...]\n",
    "    # Fromat : (<channel_track_id>, <projected_x>, <projected_y>, <reid_feat>, <state_bit>, <channel_idx>)\n",
    "    projected_point_group = []\n",
    "    \n",
    "    # local_obj_mapping\n",
    "    # Format : ([channel],[],...) |\n",
    "    # id -> ( <channel_track_id> <project_x> <project_y> <reid_feat> <state_bit>)\n",
    "    \n",
    "    #1. Projection 기반 ID 생성\n",
    "    for idx,channel_obj in enumerate(copy.deepcopy(local_obj_mapping)):\n",
    "        temp = []\n",
    "        for obj in channel_obj:\n",
    "            if obj[-1]==2:\n",
    "                temp.append(obj)\n",
    "        projected_point_group = add_projected_point2(projected_point_group, temp, idx, global_obj_mapping)\n",
    "        \n",
    "    for group in projected_point_group:\n",
    "        _, avg_x, avg_y = np.mean(np.array(group)[:,:3], axis=0)\n",
    "        \n",
    "        # 글로벌 좌표 추가, 여기에다가 feat도 추가해야함\n",
    "        # (<global_track_id> : <projected_x>, <projected_y>, [<feat_total>,...])\n",
    "        global_obj_mapping[new_id] = [int(avg_x), int(avg_y), np.array(group)[:,3]]\n",
    "        \n",
    "        # 각 채널 매핑\n",
    "        for row in group:\n",
    "            channel_id_mapping[row[-1]][row[0]] = new_id\n",
    "        \n",
    "        new_id+=1\n",
    "    \n",
    "    for idx,channel_obj in enumerate(copy.deepcopy(local_obj_mapping)):\n",
    "        for obj in channel_obj:\n",
    "            if obj[-1]!=2:\n",
    "                local_channel_loss_mapping[idx].append(obj)\n",
    "    \n",
    "    return global_obj_mapping, channel_id_mapping, local_channel_loss_mapping, new_id\n",
    "\n",
    "\n",
    "\n",
    "def match_local2gloabl(global_obj_mapping, channel_id_mapping, local_obj_mapping):\n",
    "    # 틀리거나 feature 안맞으면 loss에 추가, 이때 type은 2로 그대로 가져가서 재 배치가 가능하게끔 함. -> reid 사용 하고 실패하면 add\n",
    "    local_channel_loss_mapping = [[],[],[],[],[],[]]\n",
    "    \n",
    "    origin_channel_id_mapping = copy.deepcopy(channel_id_mapping)\n",
    "\n",
    "    projected_point_group = []\n",
    "    # #1. Projection 기반 먼저추가\n",
    "    for idx,channel_result in enumerate(copy.deepcopy(local_obj_mapping)):\n",
    "        temp = []\n",
    "        for obj in channel_result:\n",
    "            if obj[-1]==2:\n",
    "                temp.append(obj)\n",
    "        projected_point_group = add_projected_point2(projected_point_group, temp, idx, global_obj_mapping)\n",
    "    \n",
    "    \n",
    "    # 그룹간 FEAT 출력\n",
    "    # print(\"그룹간 FEAT 출력\")\n",
    "    # for group1 in projected_point_group:\n",
    "    #     for group2 in projected_point_group:\n",
    "    #         if group1 != group2:\n",
    "    #             print(embedding_distance(np.mean(np.array(group1)[:,3], axis=0),np.mean(np.array(group2)[:,3], axis=0)))\n",
    "            \n",
    "            \n",
    "    # None 없음, 2에 대해서만\n",
    "    matched_id_list = []\n",
    "    for group in projected_point_group:\n",
    "        _, avg_x, avg_y = np.mean(np.array(group)[:,:3], axis=0)\n",
    "        \n",
    "        # 이전 글로벌 매핑과 매칭\n",
    "        # 글로벌 좌표 추가, 여기에다가 feat도 추가해야함 상,하체 모두\n",
    "        already_matched_global_id = np.array(matched_id_list)\n",
    "        \n",
    "        defined_global_id = []\n",
    "        for key in global_obj_mapping.keys():\n",
    "            if global_obj_mapping[key][0] is not None: # None 이 아닌 x,y좌표애 대해서만\n",
    "                defined_global_id.append(key)\n",
    "        defined_global_id = np.array(defined_global_id)\n",
    "        \n",
    "        target_global_id = np.setdiff1d(defined_global_id, already_matched_global_id)\n",
    "        \n",
    "        match_id = None\n",
    "        min_distance = DISTANCE**2\n",
    "        \n",
    "        for tg_id in target_global_id:\n",
    "            row = global_obj_mapping[tg_id]\n",
    "            frame_distance = (avg_x - row[0])**2 + (avg_y - row[1])**2\n",
    "            if frame_distance<min_distance:\n",
    "                match_id = tg_id\n",
    "                min_distance = frame_distance\n",
    "        \n",
    "        if match_id is not None:\n",
    "            state = 1 # 일관성 문제\n",
    "            for row in group: # 그룹안에 있는거는 모두 type2\n",
    "                if row[0] in list(channel_id_mapping[row[-1]].keys()):\n",
    "                    if channel_id_mapping[row[-1]][row[0]] != match_id:\n",
    "                        state = 0 # 문제상황\n",
    "                               \n",
    "            if state == 1: # 매핑 성공 X,Y,Feat 업데이트\n",
    "                matched_id_list.append(match_id)\n",
    "                \n",
    "                # 근사한것에만 -> 기존*0.9 + 새로운*0.1, 아니면 append\n",
    "                origin_feat_list = global_obj_mapping[match_id][2]\n",
    "                new_feat_list = update_feature(origin_feat_list, np.array(group)[:,3])\n",
    "                # group 내 하나의 리스트 : <channel_track_id> <project_x> <project_y> <reid_feat> <state_bit> <channel_id>\n",
    "                \n",
    "                global_obj_mapping[match_id] = [int(avg_x), int(avg_y), new_feat_list]\n",
    "                for row in group: # 각 채널상에 나타난 새로운 id 추가\n",
    "                    channel_id_mapping[row[-1]][row[0]] = match_id\n",
    "                \n",
    "            else:\n",
    "                print(\"채널간 매핑 정보가 다름\")\n",
    "                # # print(channel_id_mapping)\n",
    "                # for row in group: # 그룹안에 있는거는 모두 type2\n",
    "                #     if row[0] in list(channel_id_mapping[row[-1]].keys()):\n",
    "                #         if channel_id_mapping[row[-1]][row[0]] != match_id:\n",
    "                #             # print(\"channel\",row[-1],\"channel_track_id\",row[0])\n",
    "                #             # print(match_id)\n",
    "                            \n",
    "                for row in group:\n",
    "                    local_channel_loss_mapping[row[-1]].append(list(np.array(row)[:-1]))\n",
    "                    channel_id_mapping[row[-1]].pop(row[0], None) # 채널 매핑 정보 지워줌\n",
    "                # print(\"지워진 채널 매핑\")\n",
    "                # print(channel_id_mapping)\n",
    "                    \n",
    "        else: \n",
    "            # print(\"매치 아이디가 none 임\")\n",
    "            for row in group:\n",
    "                local_channel_loss_mapping[row[-1]].append(list(np.array(row)[:-1]))    \n",
    "    \n",
    "    # print(\"그룹핑 결과\")\n",
    "    # for idx, group in enumerate(projected_point_group):\n",
    "    #     print(idx)\n",
    "    #     print(np.array(group))\n",
    "    \n",
    "    # 업데이트 안된 글로벌 좌표?? -> None\n",
    "    for global_key in list(global_obj_mapping.keys()):\n",
    "        if global_key not in matched_id_list:\n",
    "            global_obj_mapping[global_key][0] = None\n",
    "            global_obj_mapping[global_key][1] = None\n",
    "\n",
    "    return global_obj_mapping, channel_id_mapping, local_channel_loss_mapping\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "84f87f35-869e-46be-9320-43f5bfd06a6b",
   "metadata": {},
   "outputs": [],
   "source": [
    "def process_online(global_obj_mapping, channel_id_mapping, local_channel_loss_mapping, new_id):\n",
    "    \n",
    "    for idx,channel_obj in enumerate(local_channel_loss_mapping):\n",
    "        for jdx,obj in enumerate(channel_obj):\n",
    "            if obj[-1] == 2:\n",
    "                defined_global_id = np.array(list(global_obj_mapping.keys())) # 모든 key\n",
    "                \n",
    "                target_global_id = defined_global_id\n",
    "                \n",
    "                # 자기 그룹내 re-id에 대해서, 글로벌 feature 들과 비교한 값중에 최솟값을 가지는 그룹이랑 글로벌 id 매칭\n",
    "                # 단순 비교가 아닌 글로벌 feature와 현재 매핑되어 있는 feature간 거리 vs 글로벌 feature와 매핑할 feature간 거리\n",
    "                min_distance = 0.05\n",
    "                match_id = None\n",
    "                \n",
    "                temp = []\n",
    "                \n",
    "                for key in target_global_id:\n",
    "                    # 해당 글로벌 id 에 해당하는 obj가 존재한다면 -> 그것과 글로벌 좌표간 매칭 보다 작아야 하고\n",
    "                    row = global_obj_mapping[key]\n",
    "                    if row[0] is not None: # 현재 프레임에서 업데이트되었으니, obj 존재\n",
    "                        temp.append((obj[1] - row[0])**2 + (obj[2] - row[1])**2)\n",
    "                        if (obj[1] - row[0])**2 + (obj[2] - row[1])**2 < 400**2:\n",
    "                            # 새로운 OBJ 거리\n",
    "                            for feat1 in row[2]:\n",
    "                                now_dist = embedding_distance(np.array(feat1),np.array(obj[3]))\n",
    "                                if now_dist<min_distance:\n",
    "                                    min_distance = now_dist\n",
    "                                    match_id = key\n",
    "\n",
    "                    else : \n",
    "                        for feat1 in global_obj_mapping[key][2]:\n",
    "                            now_dist = embedding_distance(np.array(feat1),np.array(obj[3]))\n",
    "                            if now_dist<min_distance:\n",
    "                                match_id = key\n",
    "                                min_distance = now_dist\n",
    "\n",
    "                if match_id is not None: # 성공\n",
    "                    # origin_feat_list = global_obj_mapping[match_id][2]\n",
    "                    # new_feat_list = update_feature(origin_feat_list, np.array(group)[:,3])\n",
    "                    \n",
    "                    if global_obj_mapping[match_id][0] is None:  \n",
    "                        global_obj_mapping[match_id][0] = int(obj[1])\n",
    "                        global_obj_mapping[match_id][1] = int(obj[2])\n",
    "                        channel_id_mapping[idx][row[0]] = match_id\n",
    "                    else:\n",
    "                        channel_id_mapping[idx][row[0]] = match_id\n",
    "\n",
    "                else: # 실패\n",
    "                    print()\n",
    "                    print(\"ID 증가\")\n",
    "                    print(new_id)\n",
    "                    print(temp)\n",
    "                    print()\n",
    "                    # print(\"연관된 채널에서 '방금전 프레임'에서 정의된 글로벌 ID (제외대상)\")\n",
    "                    # print(matched_id_list)\n",
    "\n",
    "                    # (<global_track_id> : <projected_x>, <projected_y>, [<feat_total>,...])\n",
    "                    global_obj_mapping[new_id] = [int(obj[1]), int(obj[2]), np.array([obj[3]])]\n",
    "                    channel_id_mapping[idx][row[0]] = new_id\n",
    "\n",
    "                    new_id+=1\n",
    "                   \n",
    "    # 채널 로스 local 로 돌리고 type2인건 다 처리 했으니 return 할필요 없지 않나?\n",
    "    return global_obj_mapping, channel_id_mapping, new_id\n",
    "\n",
    "\n",
    "def process_loss(channel_loss_final_output, channel_final_output, channel_id_mapping):\n",
    "    \n",
    "    new_channel_loss_final_output = [[],[],[],[],[],[]]\n",
    "    \n",
    "    for idx, channel_obj in enumerate(channel_loss_final_output):\n",
    "        for obj in channel_obj: # obj -> (final_format , local_obj_format)\n",
    "            channel_track_id = obj[1][0]\n",
    "            if channel_track_id in channel_id_mapping[idx].keys():\n",
    "                \n",
    "                submit_format = obj[0]\n",
    "                submit_format[1] = channel_id_mapping[idx][channel_track_id]\n",
    "                \n",
    "                channel_final_output[idx].append(submit_format)\n",
    "                \n",
    "            else:\n",
    "                new_channel_loss_final_output[idx].append(obj)\n",
    "    \n",
    "    return new_channel_loss_final_output, channel_final_output\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0bfdb81d",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 1/18010 [00:01<6:27:24,  1.29s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1115, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 369, 611, array([          0,           0,           0, ...,     0.10608,           0,           0], dtype=float16), 2], [2, 812, 706, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2], [3, 1184, 820, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 728, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [2, 353, 620, array([          0,           0,           0, ...,    0.083923,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 917, array([          0,           0,           0, ...,    0.065491,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 815, 716, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [2, 1178, 825, array([          0,           0,           0, ...,    0.072205,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 818, 727, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2], [2, 751, 373, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [3, 342, 607, array([          0,           0,           0, ...,    0.078064,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 2/18010 [00:02<6:12:01,  1.24s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1115, array([          0,           0,           0, ...,    0.067505,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 369, 611, array([          0,           0,           0, ...,     0.10754,           0,           0], dtype=float16), 2], [2, 812, 706, array([          0,           0,           0, ...,    0.071838,           0,           0], dtype=float16), 2], [3, 1184, 820, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 728, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [2, 353, 620, array([          0,           0,           0, ...,    0.086243,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 917, array([          0,           0,           0, ...,    0.064758,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 815, 716, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2], [2, 1178, 825, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 818, 726, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [2, 751, 373, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [3, 342, 607, array([          0,           0,           0, ...,    0.076782,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 3/18010 [00:03<6:05:14,  1.22s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0])], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1115, array([          0,           0,           0, ...,      0.0672,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 370, 611, array([          0,           0,           0, ...,    0.094666,           0,           0], dtype=float16), 2], [2, 812, 706, array([          0,           0,           0, ...,    0.072144,           0,           0], dtype=float16), 2], [3, 1184, 820, array([          0,           0,           0, ...,    0.071167,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 727, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [2, 352, 620, array([          0,           0,           0, ...,    0.093201,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 918, array([          0,           0,           0, ...,    0.064575,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 815, 715, array([          0,           0,           0, ...,    0.070251,           0,           0], dtype=float16), 2], [2, 1180, 825, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 818, 726, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, 751, 373, array([          0,           0,           0, ...,     0.07074,           0,           0], dtype=float16), 2], [3, 342, 607, array([          0,           0,           0, ...,    0.076721,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 4/18010 [00:05<6:18:35,  1.26s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]), ([4, None, 3, 1228, 172, 80, 199, -1, -1], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2])], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1111, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 370, 611, array([          0,           0,           0, ...,    0.090454,           0,           0], dtype=float16), 2], [2, 809, 706, array([          0,           0,           0, ...,    0.072205,           0,           0], dtype=float16), 2], [3, 1184, 820, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 787, 726, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [2, 350, 621, array([          0,           0,           0, ...,      0.1134,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 918, array([          0,           0,           0, ...,    0.064087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 817, 716, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2], [2, 1178, 824, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 818, 726, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [2, 751, 373, array([          0,           0,           0, ...,    0.071716,           0,           0], dtype=float16), 2], [3, 342, 607, array([          0,           0,           0, ...,    0.076721,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 5/18010 [00:06<6:11:03,  1.24s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]), ([4, None, 3, 1228, 172, 80, 199, -1, -1], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]), ([4, None, 4, 1228, 177, 74, 194, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0])], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1111, array([          0,           0,           0, ...,    0.067871,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 372, 610, array([          0,           0,           0, ...,    0.091125,           0,           0], dtype=float16), 2], [2, 814, 707, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 1184, 821, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 727, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [2, 351, 620, array([          0,           0,           0, ...,     0.09967,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 917, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 818, 717, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [2, 1178, 824, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 817, 725, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 2], [2, 750, 369, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 2], [3, 341, 606, array([          0,           0,           0, ...,    0.084045,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 6/18010 [00:07<6:03:46,  1.21s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1659, 486, 120, 424, -1, -1], [2, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]), ([4, None, 3, 1228, 172, 80, 199, -1, -1], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]), ([4, None, 4, 1228, 177, 74, 194, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([4, None, 5, 1222, 172, 77, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.074951,           0,           0], dtype=float16), 0])], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1115, array([          0,           0,           0, ...,    0.065857,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 372, 609, array([          0,           0,           0, ...,    0.082886,           0,           0], dtype=float16), 2], [2, 812, 707, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [3, 1184, 821, array([          0,           0,           0, ...,     0.07251,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 790, 721, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [2, 349, 620, array([          0,           0,           0, ...,    0.095825,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 917, array([          0,           0,           0, ...,    0.065247,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 813, 715, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, 1180, 825, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.074951,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 817, 724, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, 750, 368, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 2], [3, 343, 607, array([          0,           0,           0, ...,    0.080566,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 7/18010 [00:08<6:15:04,  1.25s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1659, 486, 120, 424, -1, -1], [2, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1653, 488, 128, 418, -1, -1], [2, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]), ([4, None, 3, 1228, 172, 80, 199, -1, -1], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]), ([4, None, 4, 1228, 177, 74, 194, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([4, None, 5, 1222, 172, 77, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.074951,           0,           0], dtype=float16), 0]), ([4, None, 6, 1220, 169, 77, 203, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.079773,           0,           0], dtype=float16), 0])], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1119, array([          0,           0,           0, ...,    0.065674,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 371, 608, array([          0,           0,           0, ...,    0.088257,           0,           0], dtype=float16), 2], [2, 814, 707, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [3, 1189, 821, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 725, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [2, 352, 619, array([          0,           0,           0, ...,    0.099182,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 915, array([          0,           0,           0, ...,    0.063904,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 810, 714, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [2, 1183, 825, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.079773,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 816, 723, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [2, 750, 366, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [3, 344, 605, array([          0,           0,           0, ...,    0.079834,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 8/18010 [00:09<6:11:37,  1.24s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1659, 486, 120, 424, -1, -1], [2, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1653, 488, 128, 418, -1, -1], [2, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1646, 488, 137, 421, -1, -1], [2, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3])], [], [], [], [([4, None, 2, 1229, 175, 82, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.075745,           0,           0], dtype=float16), 0]), ([4, None, 3, 1228, 172, 80, 199, -1, -1], [3, 916, 1207, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]), ([4, None, 4, 1228, 177, 74, 194, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([4, None, 5, 1222, 172, 77, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.074951,           0,           0], dtype=float16), 0]), ([4, None, 6, 1220, 169, 77, 203, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.079773,           0,           0], dtype=float16), 0])], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1118, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 370, 607, array([          0,           0,           0, ...,    0.088684,           0,           0], dtype=float16), 2], [2, 812, 707, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [3, 1189, 821, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 782, 727, array([          0,           0,           0, ...,    0.072693,           0,           0], dtype=float16), 2], [2, 350, 618, array([          0,           0,           0, ...,    0.089539,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 915, array([          0,           0,           0, ...,    0.065735,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 810, 714, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2], [2, 1183, 824, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2], [3, 879, 1140, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 814, 723, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [2, 750, 364, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 343, 604, array([          0,           0,           0, ...,    0.096741,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                       | 9/18010 [00:11<6:40:55,  1.34s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1659, 486, 120, 424, -1, -1], [2, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1653, 488, 128, 418, -1, -1], [2, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1646, 488, 137, 421, -1, -1], [2, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1639, 487, 145, 429, -1, -1], [2, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 851, 1118, array([          0,           0,           0, ...,    0.065308,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0], [4, None, None, None, 3]])\n",
      " list([[1, 368, 605, array([          0,           0,           0, ...,    0.087402,           0,           0], dtype=float16), 2], [2, 814, 708, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [3, 1194, 821, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 780, 727, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [2, 347, 618, array([          0,           0,           0, ...,    0.092163,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 914, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 806, 714, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2], [2, 1187, 825, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 872, 1120, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 811, 725, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [2, 751, 364, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2], [3, 341, 602, array([          0,           0,           0, ...,    0.096252,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 10/18010 [00:12<6:46:02,  1.35s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1662, 483, 117, 442, -1, -1], [2, None, None, None, 3]), ([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1662, 483, 118, 440, -1, -1], [2, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1661, 483, 121, 438, -1, -1], [2, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1660, 484, 123, 434, -1, -1], [2, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1660, 485, 122, 430, -1, -1], [2, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1659, 486, 120, 424, -1, -1], [2, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1653, 488, 128, 418, -1, -1], [2, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1646, 488, 137, 421, -1, -1], [2, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1639, 487, 145, 429, -1, -1], [2, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 850, 1121, array([          0,           0,           0, ...,    0.064941,           0,           0], dtype=float16), 2], [2, 740, 383, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 366, 603, array([          0,           0,           0, ...,     0.07489,           0,           0], dtype=float16), 2], [2, 811, 708, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [3, 1200, 821, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 781, 724, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2], [2, 347, 617, array([          0,           0,           0, ...,    0.093018,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 914, array([          0,           0,           0, ...,     0.06366,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 804, 715, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [2, 1190, 825, array([          0,           0,           0, ...,    0.072632,           0,           0], dtype=float16), 2], [3, 868, 1108, array([          0,           0,           0, ...,    0.066467,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 809, 727, array([          0,           0,           0, ...,    0.076477,           0,           0], dtype=float16), 2], [2, 752, 363, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [3, 339, 599, array([          0,           0,           0, ...,     0.09668,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 11/18010 [00:14<6:52:21,  1.37s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 757, 414, 54, 202, -1, -1], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 850, 1117, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2], [2, 741, 383, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 365, 602, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2], [2, 811, 709, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, 1200, 820, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 777, 726, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [2, 344, 616, array([          0,           0,           0, ...,    0.088135,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 912, array([          0,           0,           0, ...,     0.06311,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 802, 715, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 1192, 826, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [3, 866, 1102, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 806, 728, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 755, 362, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, 337, 597, array([          0,           0,           0, ...,     0.08905,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 12/18010 [00:15<6:53:27,  1.38s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 757, 414, 54, 202, -1, -1], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 761, 414, 59, 205, -1, -1], [3, 730, 809, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 850, 1116, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2], [2, 742, 382, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [3, 730, 809, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 364, 601, array([          0,           0,           0, ...,    0.077698,           0,           0], dtype=float16), 2], [2, 809, 710, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [3, 1205, 821, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 774, 727, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [2, 342, 616, array([          0,           0,           0, ...,    0.084167,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 911, array([          0,           0,           0, ...,    0.063354,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 799, 715, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [2, 1197, 827, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2], [3, 863, 1097, array([          0,           0,           0, ...,    0.066956,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 803, 728, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 756, 362, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2], [3, 333, 594, array([          0,           0,           0, ...,    0.086731,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 13/18010 [00:17<6:54:37,  1.38s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 757, 414, 54, 202, -1, -1], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 761, 414, 59, 205, -1, -1], [3, 730, 809, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]), ([0, None, 12, 764, 413, 64, 208, -1, -1], [3, 731, 804, array([          0,           0,           0, ...,    0.077087,           0,           0], dtype=float16), 2])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 850, 1116, array([          0,           0,           0, ...,    0.065796,           0,           0], dtype=float16), 2], [2, 746, 381, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 731, 804, array([          0,           0,           0, ...,    0.077087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 362, 601, array([          0,           0,           0, ...,    0.076782,           0,           0], dtype=float16), 2], [2, 808, 711, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2], [3, 1211, 821, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 770, 727, array([          0,           0,           0, ...,    0.076782,           0,           0], dtype=float16), 2], [2, 342, 614, array([          0,           0,           0, ...,    0.080811,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 909, array([          0,           0,           0, ...,    0.064087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 796, 715, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2], [2, 1199, 827, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [3, 862, 1095, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 800, 725, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [2, 758, 362, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2], [3, 330, 592, array([          0,           0,           0, ...,    0.079285,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 14/18010 [00:18<6:41:47,  1.34s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 757, 414, 54, 202, -1, -1], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 761, 414, 59, 205, -1, -1], [3, 730, 809, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]), ([0, None, 12, 764, 413, 64, 208, -1, -1], [3, 731, 804, array([          0,           0,           0, ...,    0.077087,           0,           0], dtype=float16), 2]), ([0, None, 13, 765, 413, 70, 210, -1, -1], [3, 731, 799, array([          0,           0,           0, ...,    0.076111,           0,           0], dtype=float16), 2])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 849, 1116, array([          0,           0,           0, ...,    0.065918,           0,           0], dtype=float16), 2], [2, 744, 383, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 2], [3, 731, 799, array([          0,           0,           0, ...,    0.076111,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 359, 600, array([          0,           0,           0, ...,    0.077026,           0,           0], dtype=float16), 2], [2, 806, 712, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [3, 1211, 820, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 768, 726, array([          0,           0,           0, ...,      0.0755,           0,           0], dtype=float16), 2], [2, 340, 614, array([          0,           0,           0, ...,    0.079529,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 907, array([          0,           0,           0, ...,    0.065002,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 797, 717, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [2, 1204, 828, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 797, 725, array([          0,           0,           0, ...,     0.07428,           0,           0], dtype=float16), 2], [2, 760, 359, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [3, 327, 588, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102}, {1: 101, 2: 102, 3: 103}, {2: 101, 1: 102}, {1: 104}, {1: 102, 2: 103, 751: 100, 3: 100}, {3: 101, 1: 102, 2: 105}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 15/18010 [00:19<6:43:41,  1.35s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 720, 414, 74, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 0]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 720, 415, 75, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 0]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 721, 415, 75, 197, -1, -1], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 726, 416, 71, 195, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 0]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 732, 416, 67, 196, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 0]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 738, 416, 61, 198, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 0]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 744, 415, 58, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 0]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 749, 415, 57, 199, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.073059,           0,           0], dtype=float16), 0]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 752, 414, 56, 200, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 757, 414, 54, 202, -1, -1], [3, 729, 817, array([          0,           0,           0, ...,    0.072876,           0,           0], dtype=float16), 2]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 761, 414, 59, 205, -1, -1], [3, 730, 809, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]), ([0, None, 12, 764, 413, 64, 208, -1, -1], [3, 731, 804, array([          0,           0,           0, ...,    0.077087,           0,           0], dtype=float16), 2]), ([0, None, 13, 765, 413, 70, 210, -1, -1], [3, 731, 799, array([          0,           0,           0, ...,    0.076111,           0,           0], dtype=float16), 2])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 853, 1101, array([          0,           0,           0, ...,    0.066101,           0,           0], dtype=float16), 2], [2, 741, 386, array([          0,           0,           0, ...,    0.072693,           0,           0], dtype=float16), 2], [3, 734, 792, array([          0,           0,           0, ...,    0.081299,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 355, 599, array([          0,           0,           0, ...,    0.073303,           0,           0], dtype=float16), 2], [2, 804, 712, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [3, 1211, 820, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 765, 725, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [2, 340, 614, array([          0,           0,           0, ...,    0.072144,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 904, array([          0,           0,           0, ...,    0.064087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 791, 715, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, 1207, 828, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2], [3, 864, 1096, array([          0,           0,           0, ...,    0.066833,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 794, 725, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 762, 356, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [3, 326, 588, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 16/18010 [00:21<7:06:46,  1.42s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1093, array([          0,           0,           0, ...,    0.065796,           0,           0], dtype=float16), 2], [2, 743, 387, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [3, 737, 779, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 353, 598, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [2, 801, 712, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [3, 1216, 821, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 764, 724, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [2, 339, 613, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 904, array([          0,           0,           0, ...,    0.064758,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 789, 715, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [2, 1213, 829, array([          0,           0,           0, ...,    0.075256,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 790, 725, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 764, 355, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, 325, 587, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 17/18010 [00:22<6:44:35,  1.35s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 857, 1082, array([          0,           0,           0, ...,    0.065918,           0,           0], dtype=float16), 2], [2, 745, 389, array([          0,           0,           0, ...,    0.069763,           0,           0], dtype=float16), 2], [3, 739, 772, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 353, 597, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [2, 799, 712, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [3, 1227, 822, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 763, 724, array([          0,           0,           0, ...,    0.079346,           0,           0], dtype=float16), 2], [2, 337, 612, array([          0,           0,           0, ...,    0.073975,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 906, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 786, 715, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [2, 1216, 830, array([          0,           0,           0, ...,    0.073975,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 786, 723, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 2], [2, 765, 355, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [3, 322, 585, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 18/18010 [00:23<6:48:21,  1.36s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 857, 1082, array([          0,           0,           0, ...,     0.06604,           0,           0], dtype=float16), 2], [2, 746, 390, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, 738, 769, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 351, 595, array([          0,           0,           0, ...,    0.075012,           0,           0], dtype=float16), 2], [2, 798, 713, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [3, 1233, 823, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 761, 724, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [2, 331, 612, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 903, array([          0,           0,           0, ...,    0.065063,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 783, 715, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, 1216, 829, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 784, 723, array([          0,           0,           0, ...,    0.072449,           0,           0], dtype=float16), 2], [2, 766, 356, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [3, 319, 583, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 19/18010 [00:25<6:46:50,  1.36s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 856, 1082, array([          0,           0,           0, ...,    0.066589,           0,           0], dtype=float16), 2], [2, 749, 390, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [3, 739, 762, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 351, 594, array([          0,           0,           0, ...,    0.077454,           0,           0], dtype=float16), 2], [2, 789, 712, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, 1228, 821, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 759, 725, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2], [2, 325, 611, array([          0,           0,           0, ...,     0.07428,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 897, array([          0,           0,           0, ...,    0.065491,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 778, 714, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [2, 1218, 828, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2], [3, 871, 1101, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 782, 722, array([          0,           0,           0, ...,    0.078552,           0,           0], dtype=float16), 2], [2, 768, 362, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [3, 316, 581, array([          0,           0,           0, ...,    0.073608,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 20/18010 [00:26<6:38:03,  1.33s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 856, 1082, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2], [2, 751, 391, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [3, 741, 759, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 349, 593, array([          0,           0,           0, ...,    0.090942,           0,           0], dtype=float16), 2], [2, 783, 711, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [3, 1228, 820, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 760, 724, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [2, 321, 610, array([          0,           0,           0, ...,    0.084717,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 208, 894, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 775, 713, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, 1217, 826, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 869, 1096, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 780, 723, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 769, 365, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [3, 313, 580, array([          0,           0,           0, ...,    0.076843,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 21/18010 [00:27<6:47:33,  1.36s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 856, 1082, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [2, 754, 391, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, 738, 759, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 349, 592, array([          0,           0,           0, ...,    0.095886,           0,           0], dtype=float16), 2], [2, 778, 711, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [3, 1223, 818, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 758, 724, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [2, 316, 608, array([          0,           0,           0, ...,     0.08551,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 895, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 773, 714, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [2, 1222, 826, array([          0,           0,           0, ...,    0.074585,           0,           0], dtype=float16), 2], [3, 870, 1095, array([          0,           0,           0, ...,    0.067566,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 779, 722, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, 770, 366, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, 313, 582, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 22/18010 [00:28<6:26:28,  1.29s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1085, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [2, 755, 391, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [3, 737, 759, array([          0,           0,           0, ...,     0.07251,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 349, 590, array([          0,           0,           0, ...,     0.10114,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 759, 722, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [2, 316, 607, array([          0,           0,           0, ...,      0.0755,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 894, array([          0,           0,           0, ...,    0.066345,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 772, 714, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [2, 1226, 825, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 874, 1095, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 776, 721, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [2, 772, 367, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [3, 311, 581, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 23/18010 [00:30<6:09:40,  1.23s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1081, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [2, 757, 391, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [3, 736, 758, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 350, 589, array([          0,           0,           0, ...,    0.090698,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 754, 726, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, 318, 606, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 208, 889, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 771, 714, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [2, 1229, 824, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2], [3, 871, 1086, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 773, 719, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [2, 775, 368, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2], [3, 309, 580, array([          0,           0,           0, ...,    0.075806,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 24/18010 [00:31<5:56:06,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1077, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 2], [2, 760, 392, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, 732, 757, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 347, 588, array([          0,           0,           0, ...,    0.092285,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 747, 729, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, 318, 606, array([          0,           0,           0, ...,    0.086365,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 208, 886, array([          0,           0,           0, ...,    0.065613,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 769, 714, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [2, 1231, 822, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.072571,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 770, 720, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [2, 778, 369, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2], [3, 308, 582, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 25/18010 [00:32<5:47:59,  1.16s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1077, array([          0,           0,           0, ...,    0.066406,           0,           0], dtype=float16), 2], [2, 763, 393, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2], [3, 730, 754, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 344, 586, array([          0,           0,           0, ...,    0.088867,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 741, 730, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 317, 605, array([          0,           0,           0, ...,    0.076599,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 885, array([          0,           0,           0, ...,    0.065796,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 774, 719, array([          0,           0,           0, ...,     0.07251,           0,           0], dtype=float16), 2], [2, 1231, 819, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 857, 1059, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 766, 723, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [2, 782, 369, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [3, 308, 583, array([          0,           0,           0, ...,    0.091614,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 26/18010 [00:33<6:02:42,  1.21s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 853, 1079, array([          0,           0,           0, ...,    0.067871,           0,           0], dtype=float16), 2], [2, 767, 394, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 2], [3, 727, 751, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 339, 583, array([          0,           0,           0, ...,     0.08728,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 738, 728, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.085876,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 207, 879, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 771, 720, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [2, 1232, 818, array([          0,           0,           0, ...,    0.072083,           0,           0], dtype=float16), 2], [3, 857, 1052, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 760, 722, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [2, 787, 371, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [3, 306, 583, array([          0,           0,           0, ...,    0.087769,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 27/18010 [00:34<6:03:15,  1.21s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1079, array([          0,           0,           0, ...,     0.06665,           0,           0], dtype=float16), 2], [2, 768, 397, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 2], [3, 723, 749, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 336, 581, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 736, 727, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 208, 873, array([          0,           0,           0, ...,    0.064575,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 766, 720, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 2], [2, 1233, 817, array([          0,           0,           0, ...,    0.071838,           0,           0], dtype=float16), 2], [3, 860, 1051, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 756, 722, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [2, 790, 376, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [3, 304, 581, array([          0,           0,           0, ...,    0.081055,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 28/18010 [00:35<5:52:02,  1.17s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1078, array([          0,           0,           0, ...,    0.066162,           0,           0], dtype=float16), 2], [2, 771, 400, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2], [3, 722, 745, array([          0,           0,           0, ...,    0.072693,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 334, 581, array([          0,           0,           0, ...,    0.073792,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 733, 726, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 208, 870, array([          0,           0,           0, ...,    0.065002,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 761, 721, array([          0,           0,           0, ...,    0.075684,           0,           0], dtype=float16), 2], [2, 1238, 818, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2], [3, 861, 1047, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 751, 723, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 794, 381, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [3, 299, 578, array([          0,           0,           0, ...,    0.076172,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 29/18010 [00:36<5:43:07,  1.14s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1078, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2], [2, 773, 403, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2], [3, 721, 742, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 329, 580, array([          0,           0,           0, ...,    0.072205,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 732, 725, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 208, 867, array([          0,           0,           0, ...,    0.065063,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 756, 721, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [2, 1241, 818, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2], [3, 862, 1044, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 746, 721, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [2, 797, 384, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [3, 292, 572, array([          0,           0,           0, ...,    0.084473,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 30/18010 [00:38<5:41:16,  1.14s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 852, 1075, array([          0,           0,           0, ...,    0.065857,           0,           0], dtype=float16), 2], [2, 775, 406, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2], [3, 716, 741, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 324, 580, array([          0,           0,           0, ...,    0.078308,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 729, 725, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 208, 867, array([          0,           0,           0, ...,    0.064575,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 752, 722, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [2, 1243, 818, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2], [3, 860, 1038, array([          0,           0,           0, ...,    0.073181,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 740, 720, array([          0,           0,           0, ...,    0.067749,           0,           0], dtype=float16), 2], [2, 798, 385, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [3, 290, 571, array([          0,           0,           0, ...,    0.077637,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 31/18010 [00:39<6:07:23,  1.23s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1063, array([          0,           0,           0, ...,    0.065979,           0,           0], dtype=float16), 2], [2, 775, 408, array([          0,           0,           0, ...,    0.073975,           0,           0], dtype=float16), 2], [3, 715, 740, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 322, 580, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 728, 724, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 207, 865, array([          0,           0,           0, ...,    0.063965,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 749, 722, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, 1248, 819, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, 857, 1033, array([          0,           0,           0, ...,    0.066895,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 737, 719, array([          0,           0,           0, ...,    0.068481,           0,           0], dtype=float16), 2], [2, 800, 386, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [3, 293, 574, array([          0,           0,           0, ...,    0.076599,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 32/18010 [00:40<5:55:57,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1060, array([          0,           0,           0, ...,    0.065308,           0,           0], dtype=float16), 2], [2, 776, 410, array([          0,           0,           0, ...,     0.07489,           0,           0], dtype=float16), 2], [3, 712, 741, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 320, 579, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]])\n",
      " list([[1, 726, 723, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [2, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 207, 862, array([          0,           0,           0, ...,    0.065002,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 747, 722, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [2, 1249, 819, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [3, 857, 1031, array([          0,           0,           0, ...,    0.066406,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 735, 718, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2], [2, 801, 386, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [3, 293, 575, array([          0,           0,           0, ...,    0.080078,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 33/18010 [00:41<5:56:09,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 854, 1056, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2], [2, 776, 411, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [3, 709, 743, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 319, 578, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 724, 723, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 858, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 742, 720, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [2, 1245, 817, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2], [3, 857, 1029, array([          0,           0,           0, ...,    0.065857,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 733, 716, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [2, 803, 391, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [3, 292, 574, array([          0,           0,           0, ...,     0.07373,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 34/18010 [00:42<5:50:27,  1.17s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1048, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2], [2, 777, 411, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, 706, 746, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 317, 577, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 723, 723, array([          0,           0,           0, ...,    0.068481,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 851, array([          0,           0,           0, ...,     0.06427,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 738, 718, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [2, 1243, 816, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 857, 1027, array([          0,           0,           0, ...,    0.066101,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 732, 716, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2], [2, 805, 395, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [3, 288, 571, array([          0,           0,           0, ...,    0.077576,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 35/18010 [00:44<5:46:03,  1.16s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1044, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 2], [2, 777, 412, array([          0,           0,           0, ...,    0.070251,           0,           0], dtype=float16), 2], [3, 709, 741, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 316, 576, array([          0,           0,           0, ...,    0.073303,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 723, 722, array([          0,           0,           0, ...,    0.067139,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 207, 849, array([          0,           0,           0, ...,    0.065308,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 735, 718, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2], [2, 1250, 817, array([          0,           0,           0, ...,    0.071167,           0,           0], dtype=float16), 2], [3, 857, 1024, array([          0,           0,           0, ...,     0.06488,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 732, 716, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [2, 806, 400, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2], [3, 281, 564, array([          0,           0,           0, ...,    0.077148,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 36/18010 [00:45<5:27:03,  1.09s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1040, array([          0,           0,           0, ...,    0.065674,           0,           0], dtype=float16), 2], [2, 779, 413, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2], [3, 707, 740, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 316, 574, array([          0,           0,           0, ...,    0.080505,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 722, 722, array([          0,           0,           0, ...,    0.066406,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 848, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 733, 717, array([          0,           0,           0, ...,    0.071716,           0,           0], dtype=float16), 2], [2, 1259, 820, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2], [3, 859, 1022, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 732, 719, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, 807, 403, array([          0,           0,           0, ...,     0.07373,           0,           0], dtype=float16), 2], [3, 276, 561, array([          0,           0,           0, ...,    0.096252,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 37/18010 [00:46<5:27:45,  1.09s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1036, array([          0,           0,           0, ...,    0.065918,           0,           0], dtype=float16), 2], [2, 778, 416, array([          0,           0,           0, ...,    0.071716,           0,           0], dtype=float16), 2], [3, 708, 736, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 316, 573, array([          0,           0,           0, ...,    0.096802,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 719, 721, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 206, 846, array([          0,           0,           0, ...,    0.064514,           0,           0], dtype=float16), 2], [2, 311, 559, array([          0,           0,           0, ...,    0.078613,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 733, 718, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2], [2, 1264, 821, array([          0,           0,           0, ...,    0.071167,           0,           0], dtype=float16), 2], [3, 860, 1019, array([          0,           0,           0, ...,    0.064758,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 731, 719, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2], [2, 808, 404, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, 274, 560, array([          0,           0,           0, ...,    0.099304,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 38/18010 [00:47<5:40:15,  1.14s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1036, array([          0,           0,           0, ...,    0.067261,           0,           0], dtype=float16), 2], [2, 777, 419, array([          0,           0,           0, ...,    0.072144,           0,           0], dtype=float16), 2], [3, 707, 734, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 316, 572, array([          0,           0,           0, ...,     0.11896,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 713, 724, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 204, 849, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 2], [2, 309, 562, array([          0,           0,           0, ...,    0.088135,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 732, 718, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [2, 1265, 820, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [3, 862, 1017, array([          0,           0,           0, ...,    0.064514,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 728, 719, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [2, 808, 406, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [3, 270, 557, array([          0,           0,           0, ...,    0.095154,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 39/18010 [00:48<5:39:48,  1.13s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1032, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [2, 779, 422, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 701, 736, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 311, 571, array([          0,           0,           0, ...,    0.086853,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 704, 728, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 204, 847, array([          0,           0,           0, ...,    0.063293,           0,           0], dtype=float16), 2], [2, 308, 559, array([          0,           0,           0, ...,    0.088501,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 731, 718, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [2, 1265, 820, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2], [3, 866, 1013, array([          0,           0,           0, ...,    0.064819,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 724, 718, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [2, 809, 407, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [3, 270, 558, array([          0,           0,           0, ...,    0.092957,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 40/18010 [00:49<5:32:25,  1.11s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1028, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 781, 425, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [3, 694, 741, array([          0,           0,           0, ...,    0.072021,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 305, 570, array([          0,           0,           0, ...,    0.094849,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 699, 730, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 843, array([          0,           0,           0, ...,    0.062744,           0,           0], dtype=float16), 2], [2, 305, 558, array([          0,           0,           0, ...,    0.091675,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 730, 719, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [2, 1268, 820, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 2], [3, 870, 1010, array([          0,           0,           0, ...,    0.065857,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 721, 720, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [2, 810, 407, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [3, 270, 558, array([          0,           0,           0, ...,     0.09375,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 41/18010 [00:51<6:11:44,  1.24s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 855, 1028, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [2, 785, 428, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [3, 692, 739, array([          0,           0,           0, ...,    0.081848,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 301, 569, array([          0,           0,           0, ...,    0.092407,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 696, 729, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 205, 839, array([          0,           0,           0, ...,    0.062927,           0,           0], dtype=float16), 2], [2, 304, 555, array([          0,           0,           0, ...,    0.095886,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 724, 718, array([          0,           0,           0, ...,    0.069763,           0,           0], dtype=float16), 2], [2, 1269, 820, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [3, 874, 1009, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 716, 722, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2], [2, 811, 406, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, 271, 559, array([          0,           0,           0, ...,    0.079773,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 42/18010 [00:52<5:58:00,  1.20s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 857, 1022, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [2, 787, 431, array([          0,           0,           0, ...,    0.070679,           0,           0], dtype=float16), 2], [3, 691, 735, array([          0,           0,           0, ...,      0.0755,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 298, 568, array([          0,           0,           0, ...,    0.081055,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 694, 728, array([          0,           0,           0, ...,    0.069763,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 720, 716, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [2, 1272, 820, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2], [3, 876, 1008, array([          0,           0,           0, ...,    0.064941,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 712, 722, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [2, 813, 408, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, 266, 555, array([          0,           0,           0, ...,    0.098267,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 43/18010 [00:53<5:33:35,  1.11s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 857, 1019, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [2, 790, 435, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [3, 688, 734, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 296, 567, array([          0,           0,           0, ...,    0.078613,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 691, 727, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 715, 715, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [2, 1274, 820, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2], [3, 875, 1005, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 709, 723, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2], [2, 815, 413, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2], [3, 263, 553, array([          0,           0,           0, ...,    0.094971,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 44/18010 [00:54<5:24:45,  1.08s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 858, 1012, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 792, 439, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [3, 683, 733, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 294, 566, array([          0,           0,           0, ...,    0.077393,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 688, 725, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 709, 713, array([          0,           0,           0, ...,      0.0672,           0,           0], dtype=float16), 2], [2, 1277, 821, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [3, 874, 1002, array([          0,           0,           0, ...,    0.064697,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 706, 723, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2], [2, 818, 419, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [3, 262, 551, array([          0,           0,           0, ...,    0.092407,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 45/18010 [00:55<5:11:07,  1.04s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 859, 1012, array([          0,           0,           0, ...,    0.067871,           0,           0], dtype=float16), 2], [2, 793, 443, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [3, 679, 732, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 290, 565, array([          0,           0,           0, ...,    0.075562,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 684, 725, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 707, 713, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2], [2, 1282, 822, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, 871, 998, array([          0,           0,           0, ...,    0.067139,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 703, 722, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2], [2, 821, 419, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [3, 259, 549, array([          0,           0,           0, ...,    0.081665,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 46/18010 [00:56<5:07:30,  1.03s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 860, 1007, array([          0,           0,           0, ...,     0.06781,           0,           0], dtype=float16), 2], [2, 793, 446, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2], [3, 673, 738, array([          0,           0,           0, ...,    0.070007,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 287, 564, array([          0,           0,           0, ...,    0.077271,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 681, 725, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 705, 713, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [2, 1287, 824, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [3, 871, 997, array([          0,           0,           0, ...,     0.06488,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 700, 722, array([          0,           0,           0, ...,    0.067261,           0,           0], dtype=float16), 2], [2, 822, 422, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, 260, 550, array([          0,           0,           0, ...,    0.085327,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 47/18010 [00:57<5:19:35,  1.07s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 862, 1004, array([          0,           0,           0, ...,    0.066956,           0,           0], dtype=float16), 2], [2, 792, 449, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 2], [3, 669, 740, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 564, array([          0,           0,           0, ...,    0.078918,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 679, 725, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 703, 712, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [3, 871, 995, array([          0,           0,           0, ...,    0.064514,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 698, 722, array([          0,           0,           0, ...,    0.066956,           0,           0], dtype=float16), 2], [2, 823, 425, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [3, 260, 550, array([          0,           0,           0, ...,    0.083557,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 48/18010 [00:58<5:14:43,  1.05s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 862, 1001, array([          0,           0,           0, ...,    0.067261,           0,           0], dtype=float16), 2], [2, 792, 450, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 2], [3, 667, 742, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 284, 563, array([          0,           0,           0, ...,     0.07843,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 680, 724, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 700, 711, array([          0,           0,           0, ...,    0.067871,           0,           0], dtype=float16), 2], [3, 869, 991, array([          0,           0,           0, ...,    0.063232,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 696, 721, array([          0,           0,           0, ...,    0.066589,           0,           0], dtype=float16), 2], [2, 823, 429, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [3, 260, 550, array([          0,           0,           0, ...,    0.091003,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 49/18010 [00:59<5:01:56,  1.01s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 1001, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [2, 794, 449, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2], [3, 666, 741, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 563, array([          0,           0,           0, ...,    0.078979,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3]]) list([[1, 678, 725, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 696, 709, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2], [3, 868, 988, array([          0,           0,           0, ...,    0.063477,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 695, 722, array([          0,           0,           0, ...,    0.066162,           0,           0], dtype=float16), 2], [2, 824, 434, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2], [3, 261, 551, array([          0,           0,           0, ...,    0.096252,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102}, {1: 101, 2: 102, 3: 103, 751: 102}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 50/18010 [01:00<5:27:14,  1.09s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]), ([0, None, 49, 1490, 371, 28, 144, -1, -1], [5, 178, 846, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 864, 995, array([          0,           0,           0, ...,    0.066406,           0,           0], dtype=float16), 2], [2, 795, 449, array([          0,           0,           0, ...,    0.070251,           0,           0], dtype=float16), 2], [3, 669, 736, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2], [5, 178, 846, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 284, 562, array([          0,           0,           0, ...,    0.076965,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 826, 438, array([          0,           0,           0, ...,    0.079468,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 681, 722, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 696, 710, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [3, 869, 985, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 695, 723, array([          0,           0,           0, ...,    0.067505,           0,           0], dtype=float16), 2], [2, 826, 435, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2], [3, 279, 568, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 51/18010 [01:01<5:22:13,  1.08s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0]), ([0, None, 49, 1490, 371, 28, 144, -1, -1], [5, 178, 846, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 864, 994, array([          0,           0,           0, ...,    0.067505,           0,           0], dtype=float16), 2], [2, 796, 450, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [3, 671, 732, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [5, 178, 846, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 284, 561, array([          0,           0,           0, ...,     0.10767,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 821, 439, array([          0,           0,           0, ...,    0.074219,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 680, 722, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 698, 712, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2], [3, 870, 981, array([          0,           0,           0, ...,    0.063049,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 693, 722, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [2, 827, 437, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [3, 264, 555, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 52/18010 [01:02<5:39:41,  1.13s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 997, array([          0,           0,           0, ...,    0.066956,           0,           0], dtype=float16), 2], [2, 798, 451, array([          0,           0,           0, ...,    0.070251,           0,           0], dtype=float16), 2], [3, 672, 729, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 284, 560, array([          0,           0,           0, ...,     0.11542,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 821, 439, array([          0,           0,           0, ...,     0.07312,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 676, 723, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 699, 713, array([          0,           0,           0, ...,    0.073364,           0,           0], dtype=float16), 2], [3, 867, 974, array([          0,           0,           0, ...,    0.062225,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 690, 720, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 829, 439, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [3, 254, 548, array([          0,           0,           0, ...,    0.075928,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 53/18010 [01:03<5:29:04,  1.10s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 993, array([          0,           0,           0, ...,    0.067871,           0,           0], dtype=float16), 2], [2, 800, 453, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [3, 668, 730, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [5, 172, 841, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 284, 559, array([          0,           0,           0, ...,     0.10248,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 821, 439, array([          0,           0,           0, ...,    0.073608,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 671, 724, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 697, 713, array([          0,           0,           0, ...,    0.073547,           0,           0], dtype=float16), 2], [3, 866, 968, array([          0,           0,           0, ...,    0.062622,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 686, 717, array([          0,           0,           0, ...,     0.06781,           0,           0], dtype=float16), 2], [2, 832, 439, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [3, 251, 546, array([          0,           0,           0, ...,    0.074036,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 54/18010 [01:04<5:28:54,  1.10s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 861, 996, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [2, 803, 454, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, 667, 728, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [5, 169, 839, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 284, 558, array([          0,           0,           0, ...,    0.089966,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 817, 440, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 666, 726, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 695, 713, array([          0,           0,           0, ...,    0.076477,           0,           0], dtype=float16), 2], [3, 867, 965, array([          0,           0,           0, ...,    0.064026,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 680, 713, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [2, 834, 438, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, 247, 544, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 55/18010 [01:05<5:28:41,  1.10s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 54, 1459, 189, 70, 172, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 0])], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 860, 995, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 2], [2, 806, 454, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, 663, 728, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [5, 168, 838, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 557, array([          0,           0,           0, ...,    0.080872,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 816, 443, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 663, 724, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 0]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 694, 715, array([          0,           0,           0, ...,     0.07843,           0,           0], dtype=float16), 2], [3, 871, 963, array([          0,           0,           0, ...,    0.064087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 674, 710, array([          0,           0,           0, ...,    0.072205,           0,           0], dtype=float16), 2], [2, 836, 439, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [3, 247, 545, array([          0,           0,           0, ...,    0.080078,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 56/18010 [01:07<5:36:19,  1.12s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 54, 1459, 189, 70, 172, -1, -1], [3, None, None, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 0])], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 861, 992, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 808, 457, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [3, 660, 726, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 2], [5, 163, 841, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 556, array([          0,           0,           0, ...,    0.094177,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 823, 444, array([          0,           0,           0, ...,    0.072998,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 660, 723, array([          0,           0,           0, ...,    0.072571,           0,           0], dtype=float16), 2], [3, 164, 820, array([          0,           0,           0, ...,    0.067749,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 693, 716, array([          0,           0,           0, ...,    0.073792,           0,           0], dtype=float16), 2], [3, 873, 961, array([          0,           0,           0, ...,    0.064758,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 671, 711, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, 838, 441, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [3, 246, 545, array([          0,           0,           0, ...,    0.083618,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 57/18010 [01:08<5:54:51,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 861, 992, array([          0,           0,           0, ...,    0.066956,           0,           0], dtype=float16), 2], [2, 812, 460, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2], [3, 660, 723, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 2], [5, 158, 844, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 555, array([          0,           0,           0, ...,    0.096191,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 825, 448, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 659, 721, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [3, 163, 820, array([          0,           0,           0, ...,    0.067444,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 214, 797, array([          0,           0,           0, ...,    0.065674,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 691, 716, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 2], [3, 873, 958, array([          0,           0,           0, ...,    0.064453,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 667, 709, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [2, 840, 446, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [3, 246, 546, array([          0,           0,           0, ...,    0.091003,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 58/18010 [01:09<5:53:07,  1.18s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 861, 992, array([          0,           0,           0, ...,    0.065979,           0,           0], dtype=float16), 2], [2, 815, 464, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [3, 655, 725, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [5, 162, 840, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 554, array([          0,           0,           0, ...,    0.096802,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 825, 452, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 658, 720, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 166, 818, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 213, 795, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 683, 715, array([          0,           0,           0, ...,    0.075806,           0,           0], dtype=float16), 2], [3, 872, 955, array([          0,           0,           0, ...,    0.062927,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 662, 710, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 842, 450, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 244, 544, array([          0,           0,           0, ...,    0.079956,           0,           0], dtype=float16), 2]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|                                      | 59/18010 [01:10<5:51:05,  1.17s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 862, 983, array([          0,           0,           0, ...,    0.064026,           0,           0], dtype=float16), 2], [2, 817, 468, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 2], [3, 650, 724, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [5, 166, 837, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 553, array([          0,           0,           0, ...,    0.095947,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 827, 455, array([          0,           0,           0, ...,    0.073853,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 654, 720, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2], [3, 169, 817, array([          0,           0,           0, ...,    0.066467,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 678, 715, array([          0,           0,           0, ...,    0.072144,           0,           0], dtype=float16), 2], [3, 871, 953, array([          0,           0,           0, ...,    0.062561,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 658, 710, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [2, 844, 451, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.076416,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 60/18010 [01:11<5:43:26,  1.15s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 977, array([          0,           0,           0, ...,    0.065247,           0,           0], dtype=float16), 2], [2, 818, 470, array([          0,           0,           0, ...,    0.072266,           0,           0], dtype=float16), 2], [3, 649, 721, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [5, 174, 830, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 552, array([          0,           0,           0, ...,     0.10046,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 831, 457, array([          0,           0,           0, ...,    0.072571,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 652, 720, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2], [3, 175, 815, array([          0,           0,           0, ...,      0.0672,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 675, 716, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2], [3, 870, 950, array([          0,           0,           0, ...,    0.063354,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 655, 709, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [2, 845, 453, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.077271,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 61/18010 [01:12<5:28:02,  1.10s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 864, 973, array([          0,           0,           0, ...,    0.063293,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 178, 826, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 285, 551, array([          0,           0,           0, ...,     0.08136,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 833, 458, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 650, 719, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 2], [3, 177, 813, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2]]) list([[1, None, None, None, 3], [2, None, None, None, 3]])\n",
      " list([[1, 672, 716, array([          0,           0,           0, ...,    0.073364,           0,           0], dtype=float16), 2], [3, 869, 948, array([          0,           0,           0, ...,    0.063965,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 652, 708, array([          0,           0,           0, ...,    0.067688,           0,           0], dtype=float16), 2], [2, 846, 455, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.075378,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 62/18010 [01:14<5:48:53,  1.17s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 970, array([          0,           0,           0, ...,    0.064026,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 176, 825, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 286, 550, array([          0,           0,           0, ...,    0.082703,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [4, 838, 459, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 648, 719, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 2], [3, 177, 813, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [4, 864, 945, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 668, 714, array([          0,           0,           0, ...,    0.075317,           0,           0], dtype=float16), 2], [3, 868, 945, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 650, 706, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [2, 848, 462, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.074585,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 63/18010 [01:15<5:49:15,  1.17s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 967, array([          0,           0,           0, ...,    0.064819,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 178, 820, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 286, 549, array([          0,           0,           0, ...,     0.10022,           0,           0], dtype=float16), 2], [2, 671, 705, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 2], [3, 1325, 801, array([          0,           0,           0, ...,    0.070923,           0,           0], dtype=float16), 2], [4, 837, 461, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 648, 719, array([          0,           0,           0, ...,    0.074341,           0,           0], dtype=float16), 2], [3, 181, 809, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [4, 865, 943, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 664, 711, array([          0,           0,           0, ...,      0.0755,           0,           0], dtype=float16), 2], [3, 869, 943, array([          0,           0,           0, ...,    0.064392,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 649, 705, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [2, 849, 465, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 64/18010 [01:16<5:54:50,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 863, 960, array([          0,           0,           0, ...,    0.066101,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 176, 818, array([          0,           0,           0, ...,    0.067139,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 286, 547, array([          0,           0,           0, ...,     0.10645,           0,           0], dtype=float16), 2], [2, 666, 705, array([          0,           0,           0, ...,    0.077759,           0,           0], dtype=float16), 2], [3, 1319, 800, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [4, 837, 462, array([          0,           0,           0, ...,    0.071167,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 645, 719, array([          0,           0,           0, ...,    0.073669,           0,           0], dtype=float16), 2], [3, 179, 807, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2], [4, 865, 942, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 661, 710, array([          0,           0,           0, ...,    0.075012,           0,           0], dtype=float16), 2], [3, 869, 938, array([          0,           0,           0, ...,    0.067505,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 648, 705, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [2, 851, 469, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.080017,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 65/18010 [01:17<5:54:54,  1.19s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 864, 957, array([          0,           0,           0, ...,    0.065186,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 178, 813, array([          0,           0,           0, ...,    0.067444,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 286, 545, array([          0,           0,           0, ...,      0.1051,           0,           0], dtype=float16), 2], [2, 662, 705, array([          0,           0,           0, ...,    0.074707,           0,           0], dtype=float16), 2], [3, 1325, 801, array([          0,           0,           0, ...,    0.069763,           0,           0], dtype=float16), 2], [4, 839, 463, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 646, 718, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2], [3, 179, 803, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [4, 864, 942, array([          0,           0,           0, ...,    0.069763,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 660, 710, array([          0,           0,           0, ...,     0.06781,           0,           0], dtype=float16), 2], [3, 871, 935, array([          0,           0,           0, ...,    0.060303,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 648, 706, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [2, 852, 471, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.074524,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 66/18010 [01:19<6:03:36,  1.22s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], []]\n",
      "local_obj\n",
      "[list([[1, 865, 949, array([          0,           0,           0, ...,    0.066895,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 170, 814, array([          0,           0,           0, ...,    0.065796,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,     0.10992,           0,           0], dtype=float16), 0], [2, 660, 706, array([          0,           0,           0, ...,    0.078735,           0,           0], dtype=float16), 2], [3, 1332, 800, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2], [4, 841, 464, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 2], [5, 912, 943, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 645, 716, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 2], [3, 174, 801, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [4, 865, 937, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 659, 710, array([          0,           0,           0, ...,    0.069031,           0,           0], dtype=float16), 2], [3, 874, 934, array([          0,           0,           0, ...,    0.062256,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 647, 707, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [2, 853, 472, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.071655,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 67/18010 [01:20<6:17:49,  1.26s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 865, 949, array([          0,           0,           0, ...,    0.068481,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 173, 809, array([          0,           0,           0, ...,    0.065308,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.099854,           0,           0], dtype=float16), 0], [2, 657, 706, array([          0,           0,           0, ...,    0.074036,           0,           0], dtype=float16), 2], [3, 1339, 801, array([          0,           0,           0, ...,     0.07074,           0,           0], dtype=float16), 2], [4, 839, 467, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2], [5, 915, 942, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 645, 716, array([          0,           0,           0, ...,    0.069275,           0,           0], dtype=float16), 2], [3, 177, 796, array([          0,           0,           0, ...,    0.067444,           0,           0], dtype=float16), 2], [4, 868, 929, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 658, 711, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [3, 874, 930, array([          0,           0,           0, ...,    0.064087,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 647, 708, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 855, 474, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.072693,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 68/18010 [01:21<6:22:08,  1.28s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 865, 948, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 167, 811, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,     0.12976,           0,           0], dtype=float16), 0], [2, 654, 707, array([          0,           0,           0, ...,    0.075378,           0,           0], dtype=float16), 2], [3, 1339, 800, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2], [4, 841, 470, array([          0,           0,           0, ...,    0.073242,           0,           0], dtype=float16), 2], [5, 913, 933, array([          0,           0,           0, ...,    0.068848,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 641, 717, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, 174, 793, array([          0,           0,           0, ...,    0.066162,           0,           0], dtype=float16), 2], [4, 869, 924, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 657, 711, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2], [3, 877, 925, array([          0,           0,           0, ...,    0.066406,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 644, 708, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [2, 856, 476, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.072388,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 69/18010 [01:23<6:27:17,  1.30s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 865, 948, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 165, 810, array([          0,           0,           0, ...,     0.06488,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,     0.11322,           0,           0], dtype=float16), 0], [2, 654, 707, array([          0,           0,           0, ...,    0.075195,           0,           0], dtype=float16), 2], [3, 1339, 800, array([          0,           0,           0, ...,    0.071533,           0,           0], dtype=float16), 2], [4, 841, 471, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [5, 916, 929, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 633, 721, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [3, 172, 790, array([          0,           0,           0, ...,    0.068054,           0,           0], dtype=float16), 2], [4, 869, 922, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 656, 712, array([          0,           0,           0, ...,    0.079895,           0,           0], dtype=float16), 2], [3, 884, 925, array([          0,           0,           0, ...,    0.066528,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 641, 708, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 858, 475, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 70/18010 [01:24<6:33:46,  1.32s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 866, 943, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 163, 808, array([          0,           0,           0, ...,    0.064392,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,     0.09137,           0,           0], dtype=float16), 0], [2, 654, 708, array([          0,           0,           0, ...,    0.073242,           0,           0], dtype=float16), 2], [3, 1346, 800, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [4, 838, 474, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [5, 913, 923, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 627, 723, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 2], [3, 168, 789, array([          0,           0,           0, ...,    0.068176,           0,           0], dtype=float16), 2], [4, 869, 922, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 655, 713, array([          0,           0,           0, ...,    0.081238,           0,           0], dtype=float16), 2], [3, 886, 924, array([          0,           0,           0, ...,    0.067017,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 640, 710, array([          0,           0,           0, ...,    0.069885,           0,           0], dtype=float16), 2], [2, 859, 474, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.072449,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 71/18010 [01:25<6:44:02,  1.35s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 866, 943, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 166, 804, array([          0,           0,           0, ...,    0.065125,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.093689,           0,           0], dtype=float16), 0], [2, 652, 708, array([          0,           0,           0, ...,    0.071716,           0,           0], dtype=float16), 2], [3, 1346, 798, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 2], [4, 841, 476, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2], [5, 917, 923, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 622, 724, array([          0,           0,           0, ...,    0.074097,           0,           0], dtype=float16), 2], [3, 169, 787, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [4, 869, 921, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, None, 3], [2, None, None, None, 3]]) list([[1, 654, 714, array([          0,           0,           0, ...,    0.084595,           0,           0], dtype=float16), 2], [3, 885, 921, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 636, 712, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [2, 861, 476, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.080322,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 72/18010 [01:27<7:03:30,  1.42s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 866, 943, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 170, 801, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.089111,           0,           0], dtype=float16), 0], [2, 650, 709, array([          0,           0,           0, ...,     0.07312,           0,           0], dtype=float16), 2], [3, 1353, 798, array([          0,           0,           0, ...,    0.071838,           0,           0], dtype=float16), 2], [4, 840, 479, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [5, 917, 921, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 622, 722, array([          0,           0,           0, ...,     0.07428,           0,           0], dtype=float16), 2], [3, 174, 784, array([          0,           0,           0, ...,    0.070435,           0,           0], dtype=float16), 2], [4, 871, 916, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 213, 764, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2], [2, 231, 518, array([          0,           0,           0, ...,    0.092957,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 652, 715, array([          0,           0,           0, ...,    0.078552,           0,           0], dtype=float16), 2], [3, 886, 921, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 633, 713, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 863, 481, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,     0.07019,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 73/18010 [01:28<7:14:31,  1.45s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 868, 938, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 173, 798, array([          0,           0,           0, ...,    0.065979,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.083679,           0,           0], dtype=float16), 0], [2, 649, 709, array([          0,           0,           0, ...,    0.073303,           0,           0], dtype=float16), 2], [3, 1359, 798, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 2], [4, 840, 483, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 2], [5, 917, 921, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 620, 722, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [3, 179, 782, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [4, 873, 911, array([          0,           0,           0, ...,    0.070862,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 213, 763, array([          0,           0,           0, ...,    0.065247,           0,           0], dtype=float16), 2], [2, 229, 514, array([          0,           0,           0, ...,    0.082397,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 651, 716, array([          0,           0,           0, ...,     0.07428,           0,           0], dtype=float16), 2], [3, 886, 919, array([          0,           0,           0, ...,    0.067749,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 631, 714, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 866, 482, array([          0,           0,           0, ...,    0.068481,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 74/18010 [01:30<7:25:57,  1.49s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 871, 931, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 180, 791, array([          0,           0,           0, ...,    0.065491,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.093018,           0,           0], dtype=float16), 0], [2, 647, 709, array([          0,           0,           0, ...,    0.071045,           0,           0], dtype=float16), 2], [3, 1366, 799, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 2], [4, 844, 487, array([          0,           0,           0, ...,    0.073425,           0,           0], dtype=float16), 2], [5, 914, 919, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 616, 722, array([          0,           0,           0, ...,    0.069519,           0,           0], dtype=float16), 2], [3, 182, 781, array([          0,           0,           0, ...,    0.067139,           0,           0], dtype=float16), 2], [4, 874, 907, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 212, 766, array([          0,           0,           0, ...,    0.064941,           0,           0], dtype=float16), 2], [2, 227, 514, array([          0,           0,           0, ...,     0.09314,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 650, 717, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [3, 885, 917, array([          0,           0,           0, ...,    0.066833,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 627, 712, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 2], [2, 868, 484, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 75/18010 [01:32<7:38:27,  1.53s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 873, 926, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [2, None, None, None, 3], [3, None, None, None, 3], [5, 179, 790, array([          0,           0,           0, ...,     0.07074,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.085144,           0,           0], dtype=float16), 0], [2, 648, 710, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 2], [3, 1366, 799, array([          0,           0,           0, ...,    0.072144,           0,           0], dtype=float16), 2], [4, 846, 490, array([          0,           0,           0, ...,     0.07251,           0,           0], dtype=float16), 2], [5, 914, 919, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 615, 720, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [3, 184, 779, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [4, 875, 905, array([          0,           0,           0, ...,    0.072754,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 212, 766, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 2], [2, 227, 513, array([          0,           0,           0, ...,    0.081543,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 649, 718, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [3, 884, 914, array([          0,           0,           0, ...,    0.067078,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 625, 712, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2], [2, 869, 486, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 76/18010 [01:33<7:51:40,  1.58s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]), ([2, None, 75, 1646, 692, 274, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]), ([5, None, 75, 1300, 884, 269, 196, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 874, 924, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2], [3, 608, 724, array([          0,           0,           0, ...,    0.069214,           0,           0], dtype=float16), 2], [5, 183, 787, array([          0,           0,           0, ...,    0.065796,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.075012,           0,           0], dtype=float16), 0], [2, 648, 710, array([          0,           0,           0, ...,    0.069641,           0,           0], dtype=float16), 2], [3, 1373, 800, array([          0,           0,           0, ...,    0.071594,           0,           0], dtype=float16), 2], [4, 853, 491, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 2], [5, 914, 917, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 612, 720, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [3, 186, 777, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 2], [4, 876, 903, array([          0,           0,           0, ...,    0.070618,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 213, 761, array([          0,           0,           0, ...,    0.064697,           0,           0], dtype=float16), 2], [2, 225, 517, array([          0,           0,           0, ...,     0.09259,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 648, 719, array([          0,           0,           0, ...,    0.068665,           0,           0], dtype=float16), 2], [3, 883, 911, array([          0,           0,           0, ...,    0.067688,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 623, 712, array([          0,           0,           0, ...,    0.072693,           0,           0], dtype=float16), 2], [2, 870, 488, array([          0,           0,           0, ...,    0.069092,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.071777,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 77/18010 [01:35<8:10:19,  1.64s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]), ([2, None, 75, 1646, 692, 274, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0]), ([2, None, 76, 1632, 690, 287, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]), ([5, None, 75, 1300, 884, 269, 196, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0]), ([5, None, 76, 1316, 874, 280, 205, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063599,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 875, 922, array([          0,           0,           0, ...,    0.067627,           0,           0], dtype=float16), 2], [3, 606, 725, array([          0,           0,           0, ...,    0.068542,           0,           0], dtype=float16), 2], [5, 185, 783, array([          0,           0,           0, ...,    0.065857,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.093018,           0,           0], dtype=float16), 0], [2, 648, 710, array([          0,           0,           0, ...,    0.069458,           0,           0], dtype=float16), 2], [3, 1359, 796, array([          0,           0,           0, ...,    0.072937,           0,           0], dtype=float16), 2], [4, 853, 492, array([          0,           0,           0, ...,    0.072571,           0,           0], dtype=float16), 2], [5, 914, 915, array([          0,           0,           0, ...,      0.0672,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 610, 720, array([          0,           0,           0, ...,    0.069336,           0,           0], dtype=float16), 2], [3, 187, 774, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 2], [4, 877, 900, array([          0,           0,           0, ...,    0.071838,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 215, 754, array([          0,           0,           0, ...,    0.065918,           0,           0], dtype=float16), 2], [2, 225, 517, array([          0,           0,           0, ...,     0.07605,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 647, 720, array([          0,           0,           0, ...,    0.069702,           0,           0], dtype=float16), 2], [3, 884, 909, array([          0,           0,           0, ...,    0.067139,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 623, 713, array([          0,           0,           0, ...,     0.06897,           0,           0], dtype=float16), 2], [2, 871, 492, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [3, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0], [4, None, None, array([          0,           0,           0, ...,    0.063599,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 78/18010 [01:37<8:10:41,  1.64s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]), ([2, None, 75, 1646, 692, 274, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0]), ([2, None, 76, 1632, 690, 287, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([2, None, 77, 1626, 689, 292, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]), ([5, None, 75, 1300, 884, 269, 196, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0]), ([5, None, 76, 1316, 874, 280, 205, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063599,           0,           0], dtype=float16), 0]), ([5, None, 77, 1328, 863, 296, 217, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.062805,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 877, 917, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [3, 604, 727, array([          0,           0,           0, ...,    0.068909,           0,           0], dtype=float16), 2], [5, 183, 781, array([          0,           0,           0, ...,    0.066467,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.081482,           0,           0], dtype=float16), 0], [2, 644, 709, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [3, 1359, 795, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 2], [4, 853, 493, array([          0,           0,           0, ...,    0.071899,           0,           0], dtype=float16), 2], [5, 914, 912, array([          0,           0,           0, ...,     0.06604,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 609, 721, array([          0,           0,           0, ...,    0.069153,           0,           0], dtype=float16), 2], [3, 185, 772, array([          0,           0,           0, ...,    0.069397,           0,           0], dtype=float16), 2], [4, 878, 899, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 215, 754, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 2], [2, 224, 516, array([          0,           0,           0, ...,    0.079529,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 647, 720, array([          0,           0,           0, ...,    0.070129,           0,           0], dtype=float16), 2], [3, 885, 907, array([          0,           0,           0, ...,    0.067566,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 621, 712, array([          0,           0,           0, ...,    0.068726,           0,           0], dtype=float16), 2], [2, 871, 500, array([          0,           0,           0, ...,     0.06958,           0,           0], dtype=float16), 2], [4, None, None, array([          0,           0,           0, ...,    0.062805,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 79/18010 [01:38<8:13:19,  1.65s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]), ([2, None, 75, 1646, 692, 274, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0]), ([2, None, 76, 1632, 690, 287, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([2, None, 77, 1626, 689, 292, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 0]), ([2, None, 78, 1616, 689, 290, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]), ([5, None, 75, 1300, 884, 269, 196, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0]), ([5, None, 76, 1316, 874, 280, 205, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063599,           0,           0], dtype=float16), 0]), ([5, None, 77, 1328, 863, 296, 217, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.062805,           0,           0], dtype=float16), 0]), ([5, None, 78, 1340, 854, 306, 225, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 877, 915, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [3, 605, 725, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [5, 180, 779, array([          0,           0,           0, ...,    0.067688,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.091064,           0,           0], dtype=float16), 0], [2, 636, 707, array([          0,           0,           0, ...,    0.070312,           0,           0], dtype=float16), 2], [3, 1359, 794, array([          0,           0,           0, ...,    0.074646,           0,           0], dtype=float16), 2], [4, 853, 493, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 2], [5, 915, 906, array([          0,           0,           0, ...,     0.06543,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 609, 721, array([          0,           0,           0, ...,    0.068787,           0,           0], dtype=float16), 2], [3, 185, 768, array([          0,           0,           0, ...,    0.067932,           0,           0], dtype=float16), 2], [4, 877, 899, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 215, 752, array([          0,           0,           0, ...,    0.065552,           0,           0], dtype=float16), 2], [2, 222, 520, array([          0,           0,           0, ...,    0.082031,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 646, 721, array([          0,           0,           0, ...,    0.069946,           0,           0], dtype=float16), 2], [3, 887, 905, array([          0,           0,           0, ...,    0.066895,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 620, 712, array([          0,           0,           0, ...,    0.067993,           0,           0], dtype=float16), 2], [2, 872, 503, array([          0,           0,           0, ...,     0.06842,           0,           0], dtype=float16), 2], [4, None, None, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "  0%|▏                                     | 80/18010 [01:40<8:30:37,  1.71s/it]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[([0, None, 0, 1660, 398, 55, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 1, 1660, 397, 56, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 2, 1660, 398, 56, 191, -1, -1], [4, None, None, None, 3]), ([0, None, 3, 1659, 397, 58, 192, -1, -1], [4, None, None, None, 3]), ([0, None, 4, 1658, 398, 61, 190, -1, -1], [4, None, None, None, 3]), ([0, None, 5, 1657, 399, 63, 187, -1, -1], [4, None, None, None, 3]), ([0, None, 6, 1661, 399, 62, 164, -1, -1], [4, None, None, None, 3]), ([0, None, 7, 1666, 399, 57, 155, -1, -1], [4, None, None, None, 3]), ([0, None, 8, 1668, 400, 54, 151, -1, -1], [4, None, None, None, 3]), ([0, None, 9, 1679, 400, 43, 142, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.070984,           0,           0], dtype=float16), 0]), ([0, None, 10, 1683, 400, 40, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.071411,           0,           0], dtype=float16), 0]), ([0, None, 11, 1686, 402, 38, 135, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.085999,           0,           0], dtype=float16), 0])], [], [([2, None, 71, 1731, 694, 188, 385, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076538,           0,           0], dtype=float16), 0]), ([2, None, 72, 1708, 689, 212, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.076233,           0,           0], dtype=float16), 0]), ([2, None, 73, 1686, 691, 233, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070496,           0,           0], dtype=float16), 0]), ([2, None, 74, 1664, 692, 256, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,     0.07135,           0,           0], dtype=float16), 0]), ([2, None, 75, 1646, 692, 274, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071106,           0,           0], dtype=float16), 0]), ([2, None, 76, 1632, 690, 287, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071472,           0,           0], dtype=float16), 0]), ([2, None, 77, 1626, 689, 292, 390, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071289,           0,           0], dtype=float16), 0]), ([2, None, 78, 1616, 689, 290, 389, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.071228,           0,           0], dtype=float16), 0]), ([2, None, 79, 1613, 690, 270, 388, -1, -1], [5, None, None, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 0])], [], [], [([5, None, 66, 1257, 967, 133, 112, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072815,           0,           0], dtype=float16), 0]), ([5, None, 67, 1274, 958, 129, 121, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.073914,           0,           0], dtype=float16), 0]), ([5, None, 68, 1294, 951, 130, 128, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.072327,           0,           0], dtype=float16), 0]), ([5, None, 69, 1315, 942, 129, 137, -1, -1], [4, None, None, array([          0,           0,           0, ...,     0.07196,           0,           0], dtype=float16), 0]), ([5, None, 70, 1269, 933, 196, 146, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.068604,           0,           0], dtype=float16), 0]), ([5, None, 71, 1258, 923, 225, 156, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.067383,           0,           0], dtype=float16), 0]), ([5, None, 72, 1260, 915, 239, 164, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.066284,           0,           0], dtype=float16), 0]), ([5, None, 73, 1272, 906, 247, 173, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.065369,           0,           0], dtype=float16), 0]), ([5, None, 74, 1285, 895, 262, 185, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 0]), ([5, None, 75, 1300, 884, 269, 196, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063538,           0,           0], dtype=float16), 0]), ([5, None, 76, 1316, 874, 280, 205, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063599,           0,           0], dtype=float16), 0]), ([5, None, 77, 1328, 863, 296, 217, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.062805,           0,           0], dtype=float16), 0]), ([5, None, 78, 1340, 854, 306, 225, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063843,           0,           0], dtype=float16), 0]), ([5, None, 79, 1351, 844, 313, 235, -1, -1], [4, None, None, array([          0,           0,           0, ...,    0.063049,           0,           0], dtype=float16), 0])]]\n",
      "local_obj\n",
      "[list([[1, 878, 915, array([          0,           0,           0, ...,    0.067078,           0,           0], dtype=float16), 2], [3, 603, 726, array([          0,           0,           0, ...,    0.068298,           0,           0], dtype=float16), 2], [5, 177, 777, array([          0,           0,           0, ...,    0.065735,           0,           0], dtype=float16), 2]])\n",
      " list([[1, None, None, array([          0,           0,           0, ...,    0.096313,           0,           0], dtype=float16), 0], [2, 636, 707, array([          0,           0,           0, ...,    0.069824,           0,           0], dtype=float16), 2], [3, 1366, 794, array([          0,           0,           0, ...,    0.073303,           0,           0], dtype=float16), 2], [4, 853, 494, array([          0,           0,           0, ...,    0.070801,           0,           0], dtype=float16), 2], [5, 915, 903, array([          0,           0,           0, ...,    0.063965,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 608, 721, array([          0,           0,           0, ...,    0.068359,           0,           0], dtype=float16), 2], [3, 180, 766, array([          0,           0,           0, ...,    0.064636,           0,           0], dtype=float16), 2], [4, 877, 900, array([          0,           0,           0, ...,    0.070557,           0,           0], dtype=float16), 2], [5, None, None, array([          0,           0,           0, ...,    0.070374,           0,           0], dtype=float16), 0]])\n",
      " list([[1, 215, 752, array([          0,           0,           0, ...,    0.065491,           0,           0], dtype=float16), 2], [2, 219, 522, array([          0,           0,           0, ...,    0.076599,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 645, 722, array([          0,           0,           0, ...,    0.070068,           0,           0], dtype=float16), 2], [3, 887, 900, array([          0,           0,           0, ...,    0.064514,           0,           0], dtype=float16), 2]])\n",
      " list([[1, 619, 712, array([          0,           0,           0, ...,    0.068115,           0,           0], dtype=float16), 2], [2, 873, 504, array([          0,           0,           0, ...,    0.068237,           0,           0], dtype=float16), 2], [4, None, None, array([          0,           0,           0, ...,    0.063049,           0,           0], dtype=float16), 0]])]\n",
      "[{1: 100, 2: 105, 748: 102, 749: 102, 752: 102, 3: 102, 815: 104, 5: 104, 817: 104}, {1: 101, 2: 102, 3: 103, 751: 102, 4: 105, 843: 103, 5: 100}, {2: 101, 1: 102, 751: 102, 3: 104, 4: 100}, {1: 104, 2: 101, 851: 101}, {1: 102, 2: 103, 751: 102, 3: 100}, {3: 101, 1: 102, 2: 105, 751: 102}]\n"
     ]
    }
   ],
   "source": [
    "frame_count = 0\n",
    "end = 500\n",
    "new_id = 100\n",
    "\n",
    "gts=[gt1,gt2,gt3,gt4,gt5,gt6]\n",
    "\n",
    "t = tqdm(total=vid_length)\n",
    "\n",
    "while frame1 is not None:\n",
    "    # read GT\n",
    "    H, W, _ = np.shape(frame1)\n",
    "    \n",
    "    # format (x,y,x2,y2,det_score,track_id)\n",
    "    track_bbox1 = extract_tracking_result(gt1, frame_count, W, H)\n",
    "    track_bbox2 = extract_tracking_result(gt2, frame_count, W, H)\n",
    "    track_bbox3 = extract_tracking_result(gt3, frame_count, W, H)\n",
    "    track_bbox4 = extract_tracking_result(gt4, frame_count, W, H)\n",
    "    track_bbox5 = extract_tracking_result(gt5, frame_count, W, H)\n",
    "    track_bbox6 = extract_tracking_result(gt6, frame_count, W, H)\n",
    "    \n",
    "    track_result = [track_bbox1,track_bbox2,track_bbox3,track_bbox4,track_bbox5,track_bbox6]\n",
    "    channel_frame = [frame1,frame2,frame3,frame4,frame5,frame6]\n",
    "    \n",
    "    # (1) 혼자인지 겹치는지 | 채널 하나에 대하여 각각 분석\n",
    "    # 채널 ID는 0,1,2,3,4로 통일\n",
    "    \n",
    "    # Format : ([channel_id],[],...) |\n",
    "    # channel_id ->  <channel_track_id> <project_x> <project_y> <reid_feat> <state_bit>\n",
    "    local_obj_mapping = [[],[],[],[],[],[]]\n",
    "    # local_channel_loss_mapping = [[],[],[],[],[],[]]\n",
    "    \n",
    "    \n",
    "    for idx,track_bbox in enumerate(track_result):\n",
    "        if len(track_bbox)>0:\n",
    "            occlusion_check=make_occlusion_matrix(np.array(track_bbox)[:,:4])\n",
    "            for jdx,v in enumerate(occlusion_check):\n",
    "                frame_img = channel_frame[idx]\n",
    "                H, W, _ = np.shape(frame_img)\n",
    "                # format : x1,y1,x2,y2,score,track_id\n",
    "                frame_bbox = track_bbox[jdx] \n",
    "                frame_img = np.array(channel_frame[idx])\n",
    "\n",
    "                if not v: # 겹치지 않음\n",
    "                    # 2: 전체, 1: 상체, 0:unkwon\n",
    "                    if frame_count == 0:\n",
    "                        state_bit, total_crop = run_keypoint(frame_bbox, frame_img, keypoint_model, strong=False)\n",
    "                    else:\n",
    "                        state_bit, total_crop = run_keypoint(frame_bbox, frame_img, keypoint_model)\n",
    "                    \n",
    "                    reid_feature_total = extract_reid_feature(total_crop, reid_model)\n",
    "                    \n",
    "                    if state_bit==2: # 다보이는 경우 <type:2>\n",
    "                        projected_x, projected_y = extract_global_position2(frame_bbox, H_list[idx])           \n",
    "                        # <channel_track_id> <project_x> <project_y> <reid_feat> <state_bit>\n",
    "                        local_obj_mapping[idx].append([frame_bbox[5], projected_x, projected_y, reid_feature_total, 2]) \n",
    "\n",
    "                    else: # Unkown <type:1,0>\n",
    "                        local_obj_mapping[idx].append([frame_bbox[5], None, None, reid_feature_total,0])\n",
    "\n",
    "                else : # 겹침 <type:3>\n",
    "                    local_obj_mapping[idx].append([frame_bbox[5], None, None, None, 3])\n",
    "    \n",
    "    \n",
    "    \n",
    "    # print(np.array(local_obj_mapping))\n",
    "    # local_obj_mapping 객체 중에서 근처로 접근 하는것 같으면 -> 해당 객체는 겹침으로 간주 class 3\n",
    "    local_obj_mapping = check_occlusion(local_obj_mapping)\n",
    "    \n",
    "    # Control type 2\n",
    "    if frame_count == 0:\n",
    "        global_obj_mapping, channel_id_mapping, local_channel_loss_mapping, new_id = add_local2gloabl(global_obj_mapping, channel_id_mapping, local_obj_mapping, new_id)\n",
    "    else:\n",
    "        global_obj_mapping, channel_id_mapping, local_channel_loss_mapping = match_local2gloabl(global_obj_mapping, channel_id_mapping, local_obj_mapping)\n",
    "        \n",
    "    # print(\"전체 OBJ\")\n",
    "    # print(np.array(local_obj_mapping))\n",
    "    \n",
    "    # print(\"매핑된 channel\")\n",
    "    # print(np.array(channel_id_mapping))\n",
    "    \n",
    "#     print(\"loss된 channel\")\n",
    "#     print(np.array(local_channel_loss_mapping))\n",
    "    \n",
    "    # print(np.array(global_obj_mapping))\n",
    "    # loss 를 처리해야함.\n",
    "    \n",
    "    # loss type 0: 전체 다 안보임, 3: 겹침 -> 있으면 그냥 쓰고 없으면 channel_loss_mapping에 추가. -> final 쓸때 처리 ***\n",
    "    #      type 2: 모델이 100% 정확한건 아니기 때문에 생기는 문제 ( 트래킹 문제일 수도 있고 감지 문제일 수도 있고) 모름 -> reid 사용하고 실패하면 new id 부여\n",
    "    \n",
    "    # online -> type2 번 loss 처리 (실시간) | local_channel_loss_mapping\n",
    "    global_obj_mapping, channel_id_mapping, new_id = process_online(global_obj_mapping, channel_id_mapping, local_channel_loss_mapping, new_id)\n",
    "    \n",
    "    \n",
    "    # local_channel_loss_mapping \n",
    "    # channel_loss_mapping\n",
    "    # local_obj_mapping\n",
    "    # Format : ([channel_id],[],...) |\n",
    "    # channel_id ->  <channel_track_id> <project_x> <project_y> <reid_feat> <state_bit>\n",
    "    \n",
    "\n",
    "    # local_obj 우선 쓰고\n",
    "    for i in range (len(channel_final_output)):\n",
    "        for row in track_result[i]:\n",
    "            channel_track_id = row[-1]\n",
    "            # print(\"채널 아이디\",i,\"채널 트랙 아이디\",channel_track_id)\n",
    "            \n",
    "            if channel_track_id in channel_id_mapping[i].keys(): # 매핑 성공 시 add  \n",
    "                global_track_id = channel_id_mapping[i][channel_track_id]\n",
    "                channel_final_output[i].append([i, global_track_id , frame_count , row[0] , row[1] , row[2]-row[0] , row[3]-row[1] , \n",
    "                                                global_obj_mapping[global_track_id][0], global_obj_mapping[global_track_id][1]])\n",
    "            else: # 매칭 실패한 현재 프레임 obj\n",
    "                temp = None\n",
    "                for obj in local_obj_mapping[i]:\n",
    "                    if obj[0] == channel_track_id:\n",
    "                        temp = obj     \n",
    "                channel_loss_final_output[i].append(([i, None , frame_count , row[0] , row[1] , row[2]-row[0] , row[3]-row[1] , \n",
    "                                                    -1, -1], temp)) # <- 추후에 매칭 기다리거나 or 매칭 안되면 offline 매칭 수행\n",
    "    \n",
    "    # print(local_channel_loss_mapping)\n",
    "    print(channel_loss_final_output)\n",
    "    print(\"local_obj\")\n",
    "    print(np.array(local_obj_mapping))\n",
    "    print(channel_id_mapping)\n",
    "    \n",
    "    # 저장된 loss_channel_mappiing에 대하여 새롭게 찾은게 있나           \n",
    "    channel_loss_final_output, channel_final_output = process_loss(channel_loss_final_output, channel_final_output, channel_id_mapping)\n",
    "    \n",
    "    frame_count+=1\n",
    "    t.update(1)\n",
    "    \n",
    "    # %matplotlib inline\n",
    "    # plot_site(frame1,frame2,frame3,frame4,frame5,frame6)\n",
    "    \n",
    "    if frame_count>end:\n",
    "        print(\"stop\")\n",
    "        break\n",
    "    \n",
    "    _, frame1 = cap1.read()\n",
    "    _, frame2 = cap2.read()\n",
    "    _, frame3 = cap3.read()\n",
    "    _, frame4 = cap4.read()\n",
    "    _, frame5 = cap5.read()\n",
    "    _, frame6 = cap6.read()\n",
    "\n",
    "t.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6d4dfbab-653c-4e7c-b9b0-f2a92b3010ab",
   "metadata": {},
   "outputs": [],
   "source": [
    "# track re-id 기반 매칭 수행\n",
    "# channel_loss_final_output"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4d8f20ce-b8b5-4c28-be50-33a46248a05e",
   "metadata": {},
   "outputs": [],
   "source": [
    "# 마지막 loss_obj 강제 매칭\n",
    "# channel_final_output,channel_loss_mapping, channel_id_mapping=process_offline(channel_final_output, channel_loss_mapping, channel_id_mapping, global_obj_mapping, gts, H_list, W, H)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bc5fc3f8",
   "metadata": {},
   "source": [
    "# Sort and Make TXT"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c98e8e9c-6c26-45a7-9b81-da951374a297",
   "metadata": {},
   "outputs": [],
   "source": [
    "np.array(channel_final_output[0]).shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "aacd039b-3de7-48e8-a32d-08394a84f09b",
   "metadata": {},
   "outputs": [],
   "source": [
    "# SAVE_PATH = '/data/cvprw/AIC23/tracking/SUBMISSION/S022'\n",
    "\n",
    "# # just save\n",
    "# for idx, channel_result in enumerate(channel_final_output):\n",
    "#     # Open a file for writing\n",
    "#     with open(os.path.join(SAVE_PATH,str(idx)+'_associated.txt'), 'w') as f:\n",
    "#         # Loop through the rows of the array and write them to the file\n",
    "#         for row in channel_result:\n",
    "#             # Convert the row to a string and add a newline character\n",
    "#             row_str = ','.join(str(elem) for elem in row) + '\\n'\n",
    "#             f.write(row_str)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2ac53167-2fc1-4d05-b4d3-790044ac7c5f",
   "metadata": {},
   "outputs": [],
   "source": [
    "# # save after sort\n",
    "# # just save\n",
    "# for idx, channel_result in enumerate(channel_final_output):\n",
    "#     sort_indices = np.argsort(np.array(channel_result)[:, 2])\n",
    "#     sorted_channel_result = list(np.array(channel_result)[sort_indices])\n",
    "    \n",
    "    \n",
    "    \n",
    "#     # Open a file for writing\n",
    "#     with open(os.path.join(tracking_result_path, channel_list[idx],'sorted_associated.txt'), 'w') as f:\n",
    "#         # Loop through the rows of the array and write them to the file\n",
    "#         for row in sorted_channel_result:\n",
    "#             # Convert the row to a string and add a newline character\n",
    "#             row_str = ','.join(str(elem) for elem in row) + '\\n'\n",
    "#             f.write(row_str)\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bd5dbb22-5503-48ad-ba22-873be2b83375",
   "metadata": {},
   "source": [
    "# Save Vid"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2a2d2487-35d2-459d-8d86-f05bd1f2139a",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "torch190-test",
   "language": "python",
   "name": "torch190-test"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}
