Copy_of_CS224W_Colab4.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "machine_shape": "hm",
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "codemirror_mode": {
        "name": "ipython",
        "version": 3
      },
      "file_extension": ".py",
      "mimetype": "text/x-python",
      "name": "python",
      "nbconvert_exporter": "python",
      "pygments_lexer": "ipython3",
      "version": "3.7.3"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Tomsha1407/GNN/blob/main/Copy_of_CS224W_Colab4.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "XuXWJLEm2UWS"
      },
      "source": [
        "# **CS224W - Colab 4**"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "8gzsP50bF6Gb"
      },
      "source": [
        "In Colab 2 we constructed GNN models by using PyTorch Geometric's built in GCN layer, `GCNConv`. In Colab 3 we implemented the  **GraphSAGE** ([Hamilton et al. (2017)](https://arxiv.org/abs/1706.02216)) layer. In this colab you'll use what you've learned and implement a more powerful layer: **GAT** ([Veličković et al. (2018)](https://arxiv.org/abs/1710.10903)). Then we will run our models on the CORA dataset, which is a standard citation network benchmark dataset.\n",
        "\n",
        "**Note**: Make sure to **sequentially run all the cells in each section** so that the intermediate variables / packages will carry over to the next cell\n",
        "\n",
        "Have fun and good luck on Colab 4 :)"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "MSaetj53YnT6"
      },
      "source": [
        "# Device\n",
        "We recommend using a GPU for this Colab.\n",
        "\n",
        "Please click `Runtime` and then `Change runtime type`. Then set the `hardware accelerator` to **GPU**."
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "67gOQITlCNQi"
      },
      "source": [
        "## Installation"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "J_m9l6OYCQZP",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "9b0cfe51-7552-4714-fca4-4b2e8cd632f7"
      },
      "source": [
        "# Install torch geometric\n",
        "import os\n",
        "if 'IS_GRADESCOPE_ENV' not in os.environ:\n",
        "  !pip install torch-scatter -f https://data.pyg.org/whl/torch-1.13.1+cu116.html\n",
        "  !pip install torch-sparse -f https://data.pyg.org/whl/torch-1.13.1+cu116.html\n",
        "  !pip install torch-geometric\n",
        "  !pip install -q git+https://github.com/snap-stanford/deepsnap.git"
      ],
      "execution_count": 1,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Looking in links: https://data.pyg.org/whl/torch-1.13.1+cu116.html\n",
            "Collecting torch-scatter\n",
            "  Downloading torch_scatter-2.1.2.tar.gz (108 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m108.0/108.0 kB\u001b[0m \u001b[31m2.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25h  Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "Building wheels for collected packages: torch-scatter\n",
            "  Building wheel for torch-scatter (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for torch-scatter: filename=torch_scatter-2.1.2-cp310-cp310-linux_x86_64.whl size=3658842 sha256=11c36dc502275c8aca66d941ed529672c231a4f149039f5c31fe3a285cbd78c1\n",
            "  Stored in directory: /root/.cache/pip/wheels/92/f1/2b/3b46d54b134259f58c8363568569053248040859b1a145b3ce\n",
            "Successfully built torch-scatter\n",
            "Installing collected packages: torch-scatter\n",
            "Successfully installed torch-scatter-2.1.2\n",
            "Looking in links: https://data.pyg.org/whl/torch-1.13.1+cu116.html\n",
            "Collecting torch-sparse\n",
            "  Downloading torch_sparse-0.6.18.tar.gz (209 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m210.0/210.0 kB\u001b[0m \u001b[31m16.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25h  Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "Requirement already satisfied: scipy in /usr/local/lib/python3.10/dist-packages (from torch-sparse) (1.13.1)\n",
            "Requirement already satisfied: numpy<2.3,>=1.22.4 in /usr/local/lib/python3.10/dist-packages (from scipy->torch-sparse) (1.26.4)\n",
            "Building wheels for collected packages: torch-sparse\n",
            "  Building wheel for torch-sparse (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for torch-sparse: filename=torch_sparse-0.6.18-cp310-cp310-linux_x86_64.whl size=2809236 sha256=dbf2387a10042674b579913ae455bb1c4a2652a55473e74945a9f3cab341b9e3\n",
            "  Stored in directory: /root/.cache/pip/wheels/c9/dd/0f/a6a16f9f3b0236733d257b4b4ea91b548b984a341ed3b8f38c\n",
            "Successfully built torch-sparse\n",
            "Installing collected packages: torch-sparse\n",
            "Successfully installed torch-sparse-0.6.18\n",
            "Collecting torch-geometric\n",
            "  Downloading torch_geometric-2.6.1-py3-none-any.whl.metadata (63 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m63.1/63.1 kB\u001b[0m \u001b[31m5.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (3.10.9)\n",
            "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (2024.6.1)\n",
            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (3.1.4)\n",
            "Requirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (1.26.4)\n",
            "Requirement already satisfied: psutil>=5.8.0 in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (5.9.5)\n",
            "Requirement already satisfied: pyparsing in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (3.1.4)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (2.32.3)\n",
            "Requirement already satisfied: tqdm in /usr/local/lib/python3.10/dist-packages (from torch-geometric) (4.66.5)\n",
            "Requirement already satisfied: aiohappyeyeballs>=2.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (2.4.3)\n",
            "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (1.3.1)\n",
            "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (24.2.0)\n",
            "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (1.4.1)\n",
            "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (6.1.0)\n",
            "Requirement already satisfied: yarl<2.0,>=1.12.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (1.13.1)\n",
            "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->torch-geometric) (4.0.3)\n",
            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch-geometric) (2.1.5)\n",
            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->torch-geometric) (3.3.2)\n",
            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->torch-geometric) (3.10)\n",
            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->torch-geometric) (2.2.3)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->torch-geometric) (2024.8.30)\n",
            "Requirement already satisfied: typing-extensions>=4.1.0 in /usr/local/lib/python3.10/dist-packages (from multidict<7.0,>=4.5->aiohttp->torch-geometric) (4.12.2)\n",
            "Downloading torch_geometric-2.6.1-py3-none-any.whl (1.1 MB)\n",
            "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.1/1.1 MB\u001b[0m \u001b[31m54.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hInstalling collected packages: torch-geometric\n",
            "Successfully installed torch-geometric-2.6.1\n",
            "  Preparing metadata (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Building wheel for deepsnap (setup.py) ... \u001b[?25l\u001b[?25hdone\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "PRfgbfTjCRD_",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 36
        },
        "outputId": "d3a6b636-2e5f-41ec-aa4a-78c0b15cb3c5"
      },
      "source": [
        "import torch_geometric\n",
        "torch_geometric.__version__"
      ],
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "'2.6.1'"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "string"
            }
          },
          "metadata": {},
          "execution_count": 2
        }
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "ZoXlf4MtYrbz"
      },
      "source": [
        "# 1) GNN Layers"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "TQy2RBfgYut4"
      },
      "source": [
        "## Implementing Layer Modules\n",
        "\n",
        "In Colab 2, we implemented a GCN model for node and graph classification tasks. However, for that notebook we took advantage of PyG's built in GCN module. For Colabs 3 and 4, we provide a build upon a general Graph Neural Network Stack, into which we will be able to plugin our own module implementations: GraphSAGE and GAT.\n",
        "\n",
        "We will then use our layer implemenations to complete node classification on the CORA dataset, a standard citation network benchmark. In this dataset, nodes correspond to documents and edges correspond to undirected citations. Each node or document in the graph is assigned a class label and features based on the documents binarized bag-of-words representation. Specifically, the Cora graph has 2708 nodes, 5429 edges, 7 prediction classes, and 1433 features per node."
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "H4ne6Gw-CT5G"
      },
      "source": [
        "## GNN Stack Module\n",
        "\n",
        "Below is the implementation of a general GNN stack, where we can plugin any GNN layer, such as **GraphSage**, **GAT**, etc. This module is provided for you. Your implementations of the **GraphSage** and **GAT** layers will function as components in the GNNStack Module."
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "Ys8vZAFPCWWe"
      },
      "source": [
        "import torch\n",
        "import torch_scatter\n",
        "import torch.nn as nn\n",
        "import torch.nn.functional as F\n",
        "\n",
        "import torch_geometric.nn as pyg_nn\n",
        "import torch_geometric.utils as pyg_utils\n",
        "\n",
        "from torch import Tensor\n",
        "from typing import Union, Tuple, Optional\n",
        "from torch_geometric.typing import (OptPairTensor, Adj, Size, NoneType,\n",
        "                                    OptTensor)\n",
        "\n",
        "from torch.nn import Parameter, Linear\n",
        "from torch_sparse import SparseTensor, set_diag\n",
        "from torch_geometric.nn.conv import MessagePassing\n",
        "from torch_geometric.utils import remove_self_loops, add_self_loops, softmax, degree\n",
        "\n",
        "class GNNStack(torch.nn.Module):\n",
        "    def __init__(self, input_dim, hidden_dim, output_dim, args, emb=False):\n",
        "        super(GNNStack, self).__init__()\n",
        "        conv_model = self.build_conv_model(args.model_type)\n",
        "        self.convs = nn.ModuleList()\n",
        "        self.convs.append(conv_model(input_dim, hidden_dim))\n",
        "        assert (args.num_layers >= 1), 'Number of layers is not >=1'\n",
        "        for l in range(args.num_layers-1):\n",
        "            self.convs.append(conv_model(args.heads * hidden_dim, hidden_dim))\n",
        "\n",
        "        # post-message-passing\n",
        "        self.post_mp = nn.Sequential(\n",
        "            nn.Linear(args.heads * hidden_dim, hidden_dim), nn.Dropout(args.dropout),\n",
        "            nn.Linear(hidden_dim, output_dim))\n",
        "\n",
        "        self.dropout = args.dropout\n",
        "        self.num_layers = args.num_layers\n",
        "\n",
        "        self.emb = emb\n",
        "\n",
        "    def build_conv_model(self, model_type):\n",
        "        if model_type == 'GraphSage':\n",
        "            return GraphSage\n",
        "        elif model_type == 'GAT':\n",
        "            # When applying GAT with num heads > 1, you need to modify the\n",
        "            # input and output dimension of the conv layers (self.convs),\n",
        "            # to ensure that the input dim of the next layer is num heads\n",
        "            # multiplied by the output dim of the previous layer.\n",
        "            # HINT: In case you want to play with multiheads, you need to change the for-loop that builds up self.convs to be\n",
        "            # self.convs.append(conv_model(hidden_dim * num_heads, hidden_dim)),\n",
        "            # and also the first nn.Linear(hidden_dim * num_heads, hidden_dim) in post-message-passing.\n",
        "            return GAT\n",
        "\n",
        "    def forward(self, data):\n",
        "        x, edge_index, batch = data.x, data.edge_index, data.batch\n",
        "\n",
        "        for i in range(self.num_layers):\n",
        "            x = self.convs[i](x, edge_index)\n",
        "            x = F.relu(x)\n",
        "            x = F.dropout(x, p=self.dropout,training=self.training)\n",
        "\n",
        "        x = self.post_mp(x)\n",
        "\n",
        "        if self.emb == True:\n",
        "            return x\n",
        "\n",
        "        return F.log_softmax(x, dim=1)\n",
        "\n",
        "    def loss(self, pred, label):\n",
        "        return F.nll_loss(pred, label)"
      ],
      "execution_count": 3,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "5nW_XpEwASNZ"
      },
      "source": [
        "## Creating Our Own Message Passing Layer\n",
        "\n",
        "Now let's start implementing our own message passing layers! Working through this part will help us become acutely familiar with the behind the scenes work of implementing Pytorch Message Passing Layers, allowing us to build our own GNN models. To do so, we will work with and implement 3 critcal functions needed to define a PyG Message Passing Layer: `forward`, `message`, and `aggregate`.\n",
        "\n",
        "Before diving head first into the coding details, let us quickly review the key components of the message passing process. To do so, we will focus on a single round of messsage passing with respect to a single central node $x$. Before message passing, $x$ is associated with a feature vector $x^{l-1}$, and the goal of message passing is to update this feature vector as $x^l$. To do so, we implement the following steps: 1) each neighboring node $v$ passes its current message $v^{l-1}$ across the edge $(x, v)$ - 2) for the node $x$, we aggregate all of the messages of the neighboring nodes (for example through a sum or mean) - and 3) we transform the aggregated information by for example applying linear and non-linear transformations. Altogether, the message passing process is applied such that every node $u$ in our graph updates its embedding by acting as the central node $x$ in step 1-3 described above.\n",
        "\n",
        "Now, we extending this process to that of a single message passing layer, the job of a message passing layer is to update the current feature representation or embedding of each node in a graph by propagating and transforming information within the graph. Overall, the general paradigm of a message passing layers is: 1) pre-processing -> 2) **message passing** / propagation -> 3) post-processing.\n",
        "\n",
        "The `forward` fuction that we will implement for our message passing layer captures this execution logic. Namely, the `forward` function handles the pre and post-processing of node features / embeddings, as well as initiates message passing by calling the `propagate` function.\n",
        "\n",
        "\n",
        "The `propagate` function encapsulates the message passing process! It does so by calling three important functions: 1) `message`, 2) `aggregate`, and 3) `update`. Our implementation will vary slightly from this, as we will not explicitly implement `update`, but instead place the logic for updating node embeddings after message passing and within the `forward` function. To be more specific, after information is propagated (message passing), we can further transform the node embeddings outputed by `propagate`. Therefore, the output of `forward` is exactly the node embeddings after one GNN layer.\n",
        "\n",
        "Lastly, before starting to implement our own layer, let us dig a bit deeper into each of the functions described above:\n",
        "\n",
        "1.\n",
        "\n",
        "```\n",
        "def propagate(edge_index, x=(x_i, x_j), extra=(extra_i, extra_j), size=size):\n",
        "```\n",
        "Calling `propagate` initiates the message passing process. Looking at the function parameters, we highlight a couple of key parameters.\n",
        "\n",
        "  - `edge_index` is passed to the forward function and captures the edge structure of the graph.\n",
        "  - `x=(x_i, x_j)` represents the node features that will be used in message passing. In order to explain why we pass the tuple `(x_i, x_j)`, we first look at how our edges are represented. For every edge $(i, j) \\in E$, we can differentiate $i$ as the source or central node ($x_{central}$) and j as the neighboring node ($x_{neighbor}$).\n",
        "  \n",
        "    Taking the example of message passing above, for a central node $u$ we will aggregate and transform all of the messages associated with the nodes $v$ s.t. $(u, v) \\in E$ (i.e. $v \\in N_{u}$). Thus we see, the subscripts `_i` and `_j` allow us to specifcally differenciate features associated with central nodes (i.e. nodes  recieving message information) and neighboring nodes (i.e. nodes passing messages).\n",
        "\n",
        "    This is definitely a somewhat confusing concept; however, one key thing to remember / wrap your head around is that depending on the perspective, a node $x$ acts as a central node or a neighboring node. In fact, in undirected graphs we store both edge directions (i.e. $(i, j)$ and $(j, i)$). From the central node perspective, `x_i`, x is collecting neighboring information to update its embedding. From a neighboring node perspective, `x_j`, x is passing its message information along the edge connecting it to a different central node.\n",
        "\n",
        "  - `extra=(extra_i, extra_j)` represents additional information that we can associate with each node beyond its current feature embedding. In fact, we can include as many additional parameters of the form `param=(param_i, param_j)` as we would like. Again, we highlight that indexing with `_i` and `_j` allows us to differentiate central and neighboring nodes.\n",
        "\n",
        "  The output of the `propagate` function is a matrix of node embeddings after the message passing process and has shape $[N, d]$.\n",
        "\n",
        "2.\n",
        "```\n",
        "def message(x_j, ...):\n",
        "```\n",
        "The `message` function is called by propagate and constructs the messages from\n",
        "neighboring nodes $j$ to central nodes $i$ for each edge $(i, j)$ in *edge_index*. This function can take any argument that was initially passed to `propagate`. Furthermore, we can again differentiate central nodes and neighboring nodes by appending `_i` or `_j` to the variable name, .e.g. `x_i` and `x_j`. Looking more specifically at the variables, we have:\n",
        "\n",
        "  - `x_j` represents a matrix of feature embeddings for all neighboring nodes passing their messages along their respective edge (i.e. all nodes $j$ for edges $(i, j) \\in E$). Thus, its shape is $[|E|, d]$!\n",
        "  - In implementing GAT we will see how to access additional variables passed to propagate\n",
        "\n",
        "  Critically, we see that the output of the `message` function is a matrix of neighboring node embeddings ready to be aggregated, having shape $[|E|, d]$.\n",
        "\n",
        "3.\n",
        "```\n",
        "def aggregate(self, inputs, index, dim_size = None):\n",
        "```\n",
        "Lastly, the `aggregate` function is used to aggregate the messages from neighboring nodes. Looking at the parameters we highlight:\n",
        "\n",
        "  - `inputs` represents a matrix of the messages passed from neighboring nodes (i.e. the output of the `message` function).\n",
        "  - `index` has the same shape as `inputs` and tells us the central node that corresponding to each of the rows / messages $j$ in the `inputs` matrix. Thus, `index` tells us which rows / messages to aggregate for each central node.\n",
        "\n",
        "  The output of `aggregate` is of shape $[N, d]$.\n",
        "\n",
        "\n",
        "For additional resources refer to the PyG documentation for implementing custom message passing layers: https://pytorch-geometric.readthedocs.io/en/latest/notes/create_gnn.html"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "qjcfF3RACdLD"
      },
      "source": [
        "## GAT Implementation\n",
        "\n",
        "Attention mechanisms have become the state-of-the-art in many sequence-based tasks such as machine translation and learning sentence representations. One of the major benefits of attention-based mechanisms is their ability to focus on the most relevant parts of the input to make decisions. In this problem, we will see how attention mechanisms can be used to perform node classification over graph-structured data through the usage of Graph Attention Networks (GATs) ([Veličković et al. (2018)](https://arxiv.org/abs/1710.10903)).\n",
        "\n",
        "The building block of the Graph Attention Network is the graph attention layer, which is a variant of the aggregation function. Let $N$ be the number of nodes and $F$ be the dimension of the feature vector for each node. The input to each graph attentional layer is a set of node features: $\\mathbf{h} = \\{\\overrightarrow{h_1}, \\overrightarrow{h_2}, \\dots, \\overrightarrow{h_N}$\\}, $\\overrightarrow{h_i} \\in R^F$. The output of each graph attentional layer is a new set of node features, which may have a new dimension $F'$: $\\mathbf{h'} = \\{\\overrightarrow{h_1'}, \\overrightarrow{h_2'}, \\dots, \\overrightarrow{h_N'}\\}$, with $\\overrightarrow{h_i'} \\in \\mathbb{R}^{F'}$.\n",
        "\n",
        "We will now describe how this transformation is performed for each graph attention layer. First, a shared linear transformation parametrized by the weight matrix $\\mathbf{W} \\in \\mathbb{R}^{F' \\times F}$ is applied to every node.\n",
        "\n",
        "Next, we perform self-attention on the nodes. We use a shared attention function $a$:\n",
        "\\begin{equation}\n",
        "a : \\mathbb{R}^{F'} \\times \\mathbb{R}^{F'} \\rightarrow \\mathbb{R}.\n",
        "\\end{equation}\n",
        "\n",
        "that computes the attention coefficients capturing the importance of node $j$'s features to node $i$:\n",
        "\\begin{equation}\n",
        "e_{ij} = a(\\mathbf{W_l}\\overrightarrow{h_i}, \\mathbf{W_r} \\overrightarrow{h_j})\n",
        "\\end{equation}\n",
        "\n",
        "The most general formulation of self-attention allows every node to attend to all other nodes which drops all structural information. However, to utilize graph structure in the attention mechanisms, we use **masked attention**. In masked attention, we only compute attention coefficients $e_{ij}$ for nodes $j \\in N_i$ where $N_i$ is some neighborhood of node $i$ in the graph.\n",
        "\n",
        "To easily compare coefficients across different nodes, we normalize the coefficients across $j$ using a softmax function:\n",
        "\\begin{equation}\n",
        "\\alpha_{ij} = \\text{softmax}_j(e_{ij}) = \\frac{\\exp(e_{ij})}{\\sum_{k \\in N_i} \\exp(e_{ik})}\n",
        "\\end{equation}\n",
        "\n",
        "For this problem, our attention mechanism $a$ will be a single-layer feedforward neural network parametrized by a weight vectors $\\overrightarrow{a_l} \\in \\mathbb{R}^{F'}$ and $\\overrightarrow{a_r} \\in \\mathbb{R}^{F'}$, followed by a LeakyReLU nonlinearity (with negative input slope 0.2). Let $\\cdot^T$ represent transposition and $||$ represent concatenation. The coefficients computed by our attention mechanism may be expressed as:\n",
        "\n",
        "\\begin{equation}\n",
        "\\alpha_{ij} = \\frac{\\exp\\Big(\\text{LeakyReLU}\\Big(\\overrightarrow{a_l}^T \\mathbf{W_l} \\overrightarrow{h_i} + \\overrightarrow{a_r}^T\\mathbf{W_r}\\overrightarrow{h_j}\\Big)\\Big)}{\\sum_{k\\in N_i} \\exp\\Big(\\text{LeakyReLU}\\Big(\\overrightarrow{a_l}^T \\mathbf{W_l} \\overrightarrow{h_i} + \\overrightarrow{a_r}^T\\mathbf{W_r}\\overrightarrow{h_k}\\Big)\\Big)}\n",
        "\\end{equation}\n",
        "\n",
        "For the following questions, we denote `alpha_l` = $\\alpha_l = [...,\\overrightarrow{a_l}^T \\mathbf{W_l} \\overrightarrow{h_i},...] \\in R^n$ and `alpha_r` = $\\alpha_r = [..., \\overrightarrow{a_r}^T \\mathbf{W_r} \\overrightarrow{h_j}, ...] \\in R^n$.\n",
        "\n",
        "\n",
        "At every layer of GAT, after the attention coefficients are computed for that layer, the aggregation function can be computed by a weighted sum of neighborhood messages, where weights are specified by $\\alpha_{ij}$.\n",
        "\n",
        "Now, we use the normalized attention coefficients to compute a linear combination of the features corresponding to them. These aggregated features will serve as the final output features for every node.\n",
        "\n",
        "\\begin{equation}\n",
        "h_i' = \\sum_{j \\in N_i} \\alpha_{ij} \\mathbf{W_r} \\overrightarrow{h_j}.\n",
        "\\end{equation}\n",
        "\n",
        "At this point, we have covered a lot of information! Before reading further about multi-head attention, we encourage you to go again through the excersize of thinking about what components of the attention mechanism correspond with the different functions: 1) `forward`, 2) `message`, and 3 `aggregate`.\n",
        "\n",
        "- Hint 1: Our aggregation is very similar to that of GraphSage except now we are using sum aggregation\n",
        "- Hint 2: The terms we aggregate over again represent the individual message that each neighbor node j sends. Thus, we see that $\\alpha_{ij}$ is part of the message each node sends and is thus computed during the message step. This makes sense since an attention weight is associated with each edge in the graph.\n",
        "- Hint 3: Look at the terms in the definition of $\\alpha_{ij}$. What values do we want to pre-process and pass as parameters to the `propagate` function. The parameters of `message(..., x_j, alpha_j, alpha_i, ...)` should give a good hint.  \n",
        "\n",
        "### Multi-Head Attention\n",
        "To stabilize the learning process of self-attention, we use multi-head attention. To do this we use $K$ independent attention mechanisms, or ``heads'' compute output features as in the above equations. Then, we concatenate these output feature representations:\n",
        "\n",
        "\\begin{equation}\n",
        "    \\overrightarrow{h_i}' = ||_{k=1}^K \\Big(\\sum_{j \\in N_i} \\alpha_{ij}^{(k)} \\mathbf{W_r}^{(k)} \\overrightarrow{h_j}\\Big)\n",
        "\\end{equation}\n",
        "\n",
        "where $||$ is concentation, $\\alpha_{ij}^{(k)}$ are the normalized attention coefficients computed by the $k$-th attention mechanism $(a^k)$, and $\\mathbf{W}^{(k)}$ is the corresponding input linear transformation's weight matrix. Note that for this setting, $\\mathbf{h'} \\in \\mathbb{R}^{KF'}$."
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "w4j45gTpCeXO"
      },
      "source": [
        "class GAT(MessagePassing):\n",
        "\n",
        "    def __init__(self, in_channels, out_channels, heads = 2,\n",
        "                 negative_slope = 0.2, dropout = 0., **kwargs):\n",
        "        super(GAT, self).__init__(node_dim=0, **kwargs)\n",
        "\n",
        "        self.in_channels = in_channels\n",
        "        self.out_channels = out_channels\n",
        "        self.heads = heads\n",
        "        self.negative_slope = negative_slope\n",
        "        self.dropout = dropout\n",
        "\n",
        "        self.lin_l = None\n",
        "        self.lin_r = None\n",
        "        self.att_l = None\n",
        "        self.att_r = None\n",
        "\n",
        "        ############################################################################\n",
        "        # TODO: Your code here!\n",
        "        # Define the layers needed for the message functions below.\n",
        "        # self.lin_l is the linear transformation that you apply to embeddings\n",
        "        # BEFORE message passing.\n",
        "        #\n",
        "        # Pay attention to dimensions of the linear layers, since we're using\n",
        "        # multi-head attention.\n",
        "        # Our implementation is ~1 lines, but don't worry if you deviate from this.\n",
        "        # self.leakyrelu = torch.nn.LeakyReLU(self.negative_slope)\n",
        "        self.lin_l = torch.nn.Linear(self.in_channels,self.heads*self.out_channels)\n",
        "        #self.node_dim = node_dim\n",
        "        ############################################################################\n",
        "\n",
        "        self.lin_r = self.lin_l\n",
        "\n",
        "        ############################################################################\n",
        "        # TODO: Your code here!\n",
        "        # Define the attention parameters \\overrightarrow{a_l/r}^T in the above intro.\n",
        "        # You have to deal with multi-head scenarios.\n",
        "        # Use nn.Parameter instead of nn.Linear\n",
        "        # Our implementation is ~2 lines, but don't worry if you deviate from this.\n",
        "        self.att_l = torch.nn.Parameter((torch.ones(self.heads,self.out_channels)))\n",
        "        self.att_r = torch.nn.Parameter((torch.ones(self.heads,self.out_channels)))\n",
        "        ############################################################################\n",
        "\n",
        "        self.reset_parameters()\n",
        "\n",
        "    def reset_parameters(self):\n",
        "        nn.init.xavier_uniform_(self.lin_l.weight)\n",
        "        nn.init.xavier_uniform_(self.lin_r.weight)\n",
        "        nn.init.xavier_uniform_(self.att_l)\n",
        "        nn.init.xavier_uniform_(self.att_r)\n",
        "\n",
        "    def forward(self, x, edge_index, size = None):\n",
        "\n",
        "        H, C = self.heads, self.out_channels\n",
        "\n",
        "        ############################################################################\n",
        "        # TODO: Your code here!\n",
        "        # Implement message passing, as well as any pre- and post-processing (our update rule).\n",
        "        # 1. First apply linear transformation to node embeddings, and split that\n",
        "        #    into multiple heads. We use the same representations for source and\n",
        "        #    target nodes, but apply different linear weights (W_l and W_r)\n",
        "        # 2. Calculate alpha vectors for central nodes (alpha_l) and neighbor nodes (alpha_r).\n",
        "        # 3. Call propagate function to conduct the message passing.\n",
        "        #    3.1 Remember to pass alpha = (alpha_l, alpha_r) as a parameter.\n",
        "        #    3.2 See there for more information: https://pytorch-geometric.readthedocs.io/en/latest/notes/create_gnn.html\n",
        "        # 4. Transform the output back to the shape of [N, H * C].\n",
        "        # Our implementation is ~5 lines, but don't worry if you deviate from this.\n",
        "\n",
        "        surce = self.lin_l(x).reshape((-1, H,C))\n",
        "        print(f\"shape source before: {self.lin_l(x).shape}\")\n",
        "        print(f\"shape surce: {surce.shape}\")\n",
        "        targets = self.lin_r(x).reshape((-1, H,C))\n",
        "        alpha_l = torch.einsum('nhc,hc->nh', surce,self.att_l)\n",
        "        alpha_r =  torch.einsum('nhc,hc->nh', targets,self.att_r)\n",
        "        print(f\"shape of alpha_l: {alpha_l.shape} should be 2708,H\")\n",
        "        print(f\"shape H: {H}\")\n",
        "        print(f\"shape C: {C}\")\n",
        "        out = self.propagate( x=(targets,surce), edge_index=edge_index, size=size,alpha = (alpha_l, alpha_r)).reshape((-1, H*C))\n",
        "\n",
        "        ############################################################################\n",
        "\n",
        "        return out\n",
        "\n",
        "\n",
        "    def message(self, x_j, alpha_j, alpha_i, index, ptr, size_i):\n",
        "\n",
        "        ############################################################################\n",
        "        # TODO: Your code here!\n",
        "        # Implement your message function. Putting the attention in message\n",
        "        # instead of in update is a little tricky.\n",
        "        # 1. Calculate the final attention weights using alpha_i and alpha_j,\n",
        "        #    and apply leaky Relu.\n",
        "        # 2. Calculate softmax over the neighbor nodes for all the nodes. Use\n",
        "        #    torch_geometric.utils.softmax instead of the one in Pytorch.\n",
        "        # 3. Apply dropout to attention weights (alpha).\n",
        "        # 4. Multiply embeddings and attention weights. As a sanity check, the output\n",
        "        #    should be of shape [E, H, C].\n",
        "        # 5. ptr (LongTensor, optional): If given, computes the softmax based on\n",
        "        #    sorted inputs in CSR representation. You can simply pass it to softmax.\n",
        "        # Our implementation is ~4-5 lines, but don't worry if you deviate from this.\n",
        "\n",
        "        #x_j shape [E,in_channels]\n",
        "        final_att = alpha_i+alpha_j\n",
        "        final_att = torch.nn.functional.leaky_relu(final_att, self.negative_slope)#self.leakyrelu(final_att)\n",
        "        final_att = softmax(final_att,index,ptr)\n",
        "        final_att = torch.nn.functional.dropout(final_att, p=self.dropout, training=self.training)\n",
        "        print(f\"shape of final_att: {final_att.shape}\") #n,2,32\n",
        "        print(f\"shape of x_j: {x_j.shape}\") #n,2\n",
        "        out =  torch.einsum('nhc,nh->nhc', x_j,final_att)\n",
        "        print(f\"shape FINAL: {out.shape}\")\n",
        "\n",
        "        ############################################################################\n",
        "\n",
        "        return out\n",
        "\n",
        "\n",
        "    def aggregate(self, inputs, index, dim_size = None):\n",
        "\n",
        "        ############################################################################\n",
        "        # TODO: Your code here!\n",
        "        # Implement your aggregate function here.\n",
        "        # See here as how to use torch_scatter.scatter: https://pytorch-scatter.readthedocs.io/en/latest/_modules/torch_scatter/scatter.html\n",
        "        # Pay attention to \"reduce\" parameter is different from that in GraphSage.\n",
        "        # Our implementation is ~1 lines, but don't worry if you deviate from this.\n",
        "        out = torch_scatter.scatter(inputs, index, dim = self.node_dim, dim_size=dim_size, reduce=\"sum\")\n",
        "        ############################################################################\n",
        "\n",
        "        return out"
      ],
      "execution_count": 81,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "V2dkgSuWCheU"
      },
      "source": [
        "## Building Optimizers\n",
        "\n",
        "This function has been implemented for you. **For grading purposes please use the default Adam optimizer**, but feel free to play with other types of optimizers on your own."
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "f_TIQ8NPCjBP"
      },
      "source": [
        "import torch.optim as optim\n",
        "\n",
        "def build_optimizer(args, params):\n",
        "    weight_decay = args.weight_decay\n",
        "    filter_fn = filter(lambda p : p.requires_grad, params)\n",
        "    if args.opt == 'adam':\n",
        "        optimizer = optim.Adam(filter_fn, lr=args.lr, weight_decay=weight_decay)\n",
        "    elif args.opt == 'sgd':\n",
        "        optimizer = optim.SGD(filter_fn, lr=args.lr, momentum=0.95, weight_decay=weight_decay)\n",
        "    elif args.opt == 'rmsprop':\n",
        "        optimizer = optim.RMSprop(filter_fn, lr=args.lr, weight_decay=weight_decay)\n",
        "    elif args.opt == 'adagrad':\n",
        "        optimizer = optim.Adagrad(filter_fn, lr=args.lr, weight_decay=weight_decay)\n",
        "    if args.opt_scheduler == 'none':\n",
        "        return None, optimizer\n",
        "    elif args.opt_scheduler == 'step':\n",
        "        scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=args.opt_decay_step, gamma=args.opt_decay_rate)\n",
        "    elif args.opt_scheduler == 'cos':\n",
        "        scheduler = optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max=args.opt_restart)\n",
        "    return scheduler, optimizer"
      ],
      "execution_count": 82,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "hBYdWFwYCkwY"
      },
      "source": [
        "## Training and Testing\n",
        "\n",
        "Here we provide you with the functions to train and test. **Please do not modify this part for grading purposes.**"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "_tZMWRc8CmGg"
      },
      "source": [
        "import time\n",
        "\n",
        "import networkx as nx\n",
        "import numpy as np\n",
        "import torch\n",
        "import torch.optim as optim\n",
        "from tqdm import trange\n",
        "import pandas as pd\n",
        "import copy\n",
        "\n",
        "from torch_geometric.datasets import TUDataset\n",
        "from torch_geometric.datasets import Planetoid\n",
        "from torch_geometric.data import DataLoader\n",
        "\n",
        "import torch_geometric.nn as pyg_nn\n",
        "\n",
        "import matplotlib.pyplot as plt\n",
        "\n",
        "\n",
        "def train(dataset, args):\n",
        "\n",
        "    print(\"Node task. test set size:\", np.sum(dataset[0]['test_mask'].numpy()))\n",
        "    print()\n",
        "    test_loader = loader = DataLoader(dataset, batch_size=args.batch_size, shuffle=False)\n",
        "\n",
        "    # build model\n",
        "    model = GNNStack(dataset.num_node_features, args.hidden_dim, dataset.num_classes,\n",
        "                            args)\n",
        "    scheduler, opt = build_optimizer(args, model.parameters())\n",
        "\n",
        "    # train\n",
        "    losses = []\n",
        "    test_accs = []\n",
        "    best_acc = 0\n",
        "    best_model = None\n",
        "    for epoch in trange(args.epochs, desc=\"Training\", unit=\"Epochs\"):\n",
        "        total_loss = 0\n",
        "        model.train()\n",
        "        for batch in loader:\n",
        "            opt.zero_grad()\n",
        "            pred = model(batch)\n",
        "            label = batch.y\n",
        "            pred = pred[batch.train_mask]\n",
        "            label = label[batch.train_mask]\n",
        "            loss = model.loss(pred, label)\n",
        "            loss.backward()\n",
        "            opt.step()\n",
        "            total_loss += loss.item() * batch.num_graphs\n",
        "        total_loss /= len(loader.dataset)\n",
        "        losses.append(total_loss)\n",
        "\n",
        "        if epoch % 10 == 0:\n",
        "          test_acc = test(test_loader, model)\n",
        "          test_accs.append(test_acc)\n",
        "          if test_acc > best_acc:\n",
        "            best_acc = test_acc\n",
        "            best_model = copy.deepcopy(model)\n",
        "        else:\n",
        "          test_accs.append(test_accs[-1])\n",
        "\n",
        "    return test_accs, losses, best_model, best_acc, test_loader\n",
        "\n",
        "def test(loader, test_model, is_validation=False, save_model_preds=False, model_type=None):\n",
        "    test_model.eval()\n",
        "\n",
        "    correct = 0\n",
        "    # Note that Cora is only one graph!\n",
        "    for data in loader:\n",
        "        with torch.no_grad():\n",
        "            # max(dim=1) returns values, indices tuple; only need indices\n",
        "            pred = test_model(data).max(dim=1)[1]\n",
        "            label = data.y\n",
        "\n",
        "        mask = data.val_mask if is_validation else data.test_mask\n",
        "        # node classification: only evaluate on nodes in test set\n",
        "        pred = pred[mask]\n",
        "        label = label[mask]\n",
        "\n",
        "        if save_model_preds:\n",
        "          print (\"Saving Model Predictions for Model Type\", model_type)\n",
        "\n",
        "          data = {}\n",
        "          data['pred'] = pred.view(-1).cpu().detach().numpy()\n",
        "          data['label'] = label.view(-1).cpu().detach().numpy()\n",
        "\n",
        "          df = pd.DataFrame(data=data)\n",
        "          # Save locally as csv\n",
        "          df.to_csv('CORA-Node-' + model_type + '.csv', sep=',', index=False)\n",
        "\n",
        "        correct += pred.eq(label).sum().item()\n",
        "\n",
        "    total = 0\n",
        "    for data in loader.dataset:\n",
        "        total += torch.sum(data.val_mask if is_validation else data.test_mask).item()\n",
        "\n",
        "    return correct / total\n",
        "\n",
        "class objectview(object):\n",
        "    def __init__(self, d):\n",
        "        self.__dict__ = d\n"
      ],
      "execution_count": 83,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "L7-h7jIsCns4"
      },
      "source": [
        "## Let's Start the Training!\n",
        "\n",
        "We will be working on the CORA dataset on node-level classification.\n",
        "\n",
        "This part is implemented for you. **For grading purposes, please do not modify the default parameters.** However, feel free to play with different configurations just for fun! Training with default parameters on the GPU runtime should take less than 5 minutes.\n",
        "\n",
        "**Submit your best accuracy on Gradescope.**"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "qe9B45l9Cpz2",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 1000
        },
        "outputId": "b5671424-9154-49d6-ee27-5fbf299834a6"
      },
      "source": [
        "if 'IS_GRADESCOPE_ENV' not in os.environ:\n",
        "    for args in [\n",
        "        {'model_type': 'GAT', 'dataset': 'cora', 'num_layers': 2, 'heads': 1, 'batch_size': 32, 'hidden_dim': 32, 'dropout': 0.5, 'epochs': 500, 'opt': 'adam', 'opt_scheduler': 'none', 'opt_restart': 0, 'weight_decay': 5e-3, 'lr': 0.01},\n",
        "    ]:\n",
        "        args = objectview(args)\n",
        "        for model in ['GAT']:\n",
        "            args.model_type = model\n",
        "\n",
        "            # Match the dimension.\n",
        "            if model == 'GAT':\n",
        "              args.heads = 2\n",
        "            else:\n",
        "              args.heads = 1\n",
        "\n",
        "            if args.dataset == 'cora':\n",
        "                dataset = Planetoid(root='/tmp/cora', name='Cora')\n",
        "            else:\n",
        "                raise NotImplementedError(\"Unknown dataset\")\n",
        "            test_accs, losses, best_model, best_acc, test_loader = train(dataset, args)\n",
        "\n",
        "            print(\"Maximum test set accuracy: {0}\".format(max(test_accs)))\n",
        "            print(\"Minimum loss: {0}\".format(min(losses)))\n",
        "\n",
        "            # Run test for our best model to save the predictions!\n",
        "            test(test_loader, best_model, is_validation=False, save_model_preds=True, model_type=model)\n",
        "            print()\n",
        "\n",
        "            plt.title(dataset.name)\n",
        "            plt.plot(losses, label=\"training loss\" + \" - \" + args.model_type)\n",
        "            plt.plot(test_accs, label=\"test accuracy\" + \" - \" + args.model_type)\n",
        "        plt.legend()\n",
        "        plt.show()\n"
      ],
      "execution_count": 84,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Node task. test set size: 1000\n",
            "\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   0%|          | 1/500 [00:00<01:33,  5.33Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   1%|          | 3/500 [00:00<01:01,  8.04Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   1%|          | 4/500 [00:00<01:03,  7.77Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   1%|          | 6/500 [00:00<01:03,  7.75Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   2%|▏         | 8/500 [00:01<01:04,  7.66Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   2%|▏         | 10/500 [00:01<01:01,  7.94Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   2%|▏         | 12/500 [00:01<01:08,  7.16Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   3%|▎         | 13/500 [00:01<01:06,  7.34Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   3%|▎         | 15/500 [00:02<01:04,  7.47Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   3%|▎         | 17/500 [00:02<01:06,  7.29Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   4%|▍         | 19/500 [00:02<01:04,  7.46Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   4%|▍         | 21/500 [00:02<01:13,  6.56Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   4%|▍         | 22/500 [00:03<01:09,  6.86Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   5%|▌         | 25/500 [00:03<01:01,  7.73Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   5%|▌         | 27/500 [00:03<00:51,  9.21Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   6%|▌         | 29/500 [00:03<00:45, 10.28Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   6%|▌         | 31/500 [00:03<00:47,  9.82Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   7%|▋         | 33/500 [00:04<00:45, 10.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   7%|▋         | 36/500 [00:04<00:53,  8.64Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   8%|▊         | 38/500 [00:04<00:55,  8.25Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   8%|▊         | 40/500 [00:05<00:56,  8.09Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   8%|▊         | 42/500 [00:05<01:05,  7.03Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:   9%|▊         | 43/500 [00:05<01:04,  7.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   9%|▉         | 45/500 [00:05<01:01,  7.45Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:   9%|▉         | 47/500 [00:06<00:59,  7.57Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  10%|▉         | 49/500 [00:06<00:59,  7.58Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  10%|█         | 51/500 [00:06<01:09,  6.46Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  10%|█         | 52/500 [00:06<01:05,  6.80Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  11%|█         | 54/500 [00:07<00:59,  7.49Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  11%|█         | 56/500 [00:07<00:48,  9.13Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  12%|█▏        | 60/500 [00:07<00:42, 10.41Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  12%|█▏        | 62/500 [00:07<00:43, 10.02Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  13%|█▎        | 64/500 [00:07<00:41, 10.59Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  14%|█▎        | 68/500 [00:08<00:38, 11.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  14%|█▍        | 70/500 [00:08<00:39, 10.95Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  14%|█▍        | 72/500 [00:08<00:40, 10.48Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  15%|█▌        | 76/500 [00:08<00:37, 11.23Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  16%|█▌        | 78/500 [00:09<00:36, 11.57Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  16%|█▋        | 82/500 [00:09<00:38, 10.86Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  17%|█▋        | 84/500 [00:09<00:37, 11.21Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  17%|█▋        | 86/500 [00:09<00:36, 11.46Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  18%|█▊        | 90/500 [00:10<00:34, 11.72Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  18%|█▊        | 92/500 [00:10<00:38, 10.48Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  19%|█▉        | 94/500 [00:10<00:37, 10.89Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  20%|█▉        | 98/500 [00:10<00:34, 11.67Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  20%|██        | 100/500 [00:11<00:33, 11.90Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  20%|██        | 102/500 [00:11<00:35, 11.20Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  21%|██        | 106/500 [00:11<00:35, 11.22Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  22%|██▏       | 108/500 [00:11<00:36, 10.62Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  22%|██▏       | 112/500 [00:12<00:36, 10.66Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  23%|██▎       | 114/500 [00:12<00:34, 11.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  23%|██▎       | 116/500 [00:12<00:34, 11.14Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  24%|██▍       | 120/500 [00:12<00:32, 11.73Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  24%|██▍       | 122/500 [00:13<00:34, 10.98Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  25%|██▍       | 124/500 [00:13<00:33, 11.07Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  25%|██▌       | 126/500 [00:13<00:35, 10.60Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  26%|██▌       | 129/500 [00:13<00:41,  9.02Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  26%|██▌       | 131/500 [00:14<00:50,  7.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  26%|██▋       | 132/500 [00:14<00:49,  7.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  27%|██▋       | 134/500 [00:14<00:50,  7.31Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  27%|██▋       | 136/500 [00:14<00:48,  7.47Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  28%|██▊       | 138/500 [00:15<00:50,  7.23Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  28%|██▊       | 140/500 [00:15<00:50,  7.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  28%|██▊       | 142/500 [00:15<00:55,  6.46Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  29%|██▊       | 143/500 [00:15<00:53,  6.69Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  29%|██▉       | 145/500 [00:16<00:49,  7.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  29%|██▉       | 147/500 [00:16<00:47,  7.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  30%|██▉       | 149/500 [00:16<00:38,  9.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  31%|███       | 153/500 [00:17<00:35,  9.76Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  31%|███       | 155/500 [00:17<00:33, 10.42Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  31%|███▏      | 157/500 [00:17<00:31, 10.94Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  32%|███▏      | 161/500 [00:17<00:33, 10.18Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  33%|███▎      | 163/500 [00:17<00:31, 10.64Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  33%|███▎      | 165/500 [00:18<00:31, 10.79Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  34%|███▍      | 169/500 [00:18<00:29, 11.11Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  34%|███▍      | 171/500 [00:18<00:31, 10.43Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  35%|███▍      | 173/500 [00:18<00:31, 10.44Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  35%|███▌      | 177/500 [00:19<00:28, 11.23Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  36%|███▌      | 179/500 [00:19<00:28, 11.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  36%|███▌      | 181/500 [00:19<00:31, 10.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  37%|███▋      | 183/500 [00:19<00:30, 10.52Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  37%|███▋      | 187/500 [00:20<00:27, 11.19Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  38%|███▊      | 189/500 [00:20<00:27, 11.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  38%|███▊      | 191/500 [00:20<00:30, 10.11Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  39%|███▊      | 193/500 [00:20<00:29, 10.41Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  39%|███▉      | 195/500 [00:20<00:29, 10.42Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  40%|███▉      | 199/500 [00:21<00:26, 11.26Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  40%|████      | 201/500 [00:21<00:28, 10.40Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  41%|████      | 203/500 [00:21<00:27, 10.87Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  41%|████▏     | 207/500 [00:21<00:26, 11.26Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  42%|████▏     | 209/500 [00:22<00:25, 11.49Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  42%|████▏     | 211/500 [00:22<00:26, 10.82Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  43%|████▎     | 215/500 [00:22<00:25, 11.14Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  43%|████▎     | 217/500 [00:22<00:25, 11.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  44%|████▍     | 219/500 [00:23<00:25, 11.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  45%|████▍     | 223/500 [00:23<00:25, 10.98Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  45%|████▌     | 225/500 [00:23<00:25, 10.95Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  45%|████▌     | 227/500 [00:23<00:25, 10.76Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  46%|████▌     | 229/500 [00:24<00:25, 10.74Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  47%|████▋     | 233/500 [00:24<00:25, 10.58Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  47%|████▋     | 235/500 [00:24<00:24, 10.91Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  47%|████▋     | 237/500 [00:24<00:23, 11.15Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  48%|████▊     | 239/500 [00:24<00:22, 11.46Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  49%|████▊     | 243/500 [00:25<00:23, 11.00Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  49%|████▉     | 245/500 [00:25<00:23, 11.07Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  49%|████▉     | 247/500 [00:25<00:22, 11.18Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  50%|█████     | 251/500 [00:26<00:23, 10.40Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  51%|█████     | 253/500 [00:26<00:22, 10.83Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  51%|█████     | 255/500 [00:26<00:22, 11.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  51%|█████▏    | 257/500 [00:26<00:23, 10.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  52%|█████▏    | 259/500 [00:26<00:25,  9.43Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  52%|█████▏    | 261/500 [00:27<00:31,  7.55Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  52%|█████▏    | 262/500 [00:27<00:31,  7.59Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  53%|█████▎    | 264/500 [00:27<00:31,  7.54Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  53%|█████▎    | 266/500 [00:27<00:30,  7.67Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  54%|█████▎    | 268/500 [00:28<00:30,  7.53Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  54%|█████▍    | 270/500 [00:28<00:31,  7.30Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  54%|█████▍    | 272/500 [00:28<00:34,  6.64Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  55%|█████▍    | 273/500 [00:28<00:34,  6.65Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  55%|█████▌    | 275/500 [00:29<00:32,  6.88Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  56%|█████▌    | 278/500 [00:29<00:28,  7.74Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  56%|█████▌    | 280/500 [00:29<00:23,  9.28Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  57%|█████▋    | 283/500 [00:30<00:22,  9.63Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  57%|█████▋    | 285/500 [00:30<00:21,  9.95Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  57%|█████▋    | 287/500 [00:30<00:20, 10.34Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  58%|█████▊    | 289/500 [00:30<00:19, 10.92Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  59%|█████▊    | 293/500 [00:30<00:19, 10.69Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  59%|█████▉    | 295/500 [00:31<00:18, 11.02Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  59%|█████▉    | 297/500 [00:31<00:18, 11.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  60%|██████    | 301/500 [00:31<00:18, 10.78Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  61%|██████    | 303/500 [00:31<00:17, 11.17Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  61%|██████    | 305/500 [00:32<00:17, 11.19Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  62%|██████▏   | 309/500 [00:32<00:16, 11.26Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  62%|██████▏   | 311/500 [00:32<00:18, 10.47Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  63%|██████▎   | 313/500 [00:32<00:17, 10.93Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  63%|██████▎   | 315/500 [00:32<00:16, 11.26Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  64%|██████▍   | 319/500 [00:33<00:15, 11.92Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  64%|██████▍   | 321/500 [00:33<00:16, 10.62Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  65%|██████▍   | 323/500 [00:33<00:15, 11.14Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  65%|██████▌   | 327/500 [00:33<00:14, 11.61Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  66%|██████▌   | 329/500 [00:34<00:14, 11.79Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  66%|██████▌   | 331/500 [00:34<00:15, 10.97Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  67%|██████▋   | 335/500 [00:34<00:14, 11.44Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  67%|██████▋   | 337/500 [00:34<00:14, 11.30Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  68%|██████▊   | 341/500 [00:35<00:14, 10.88Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  69%|██████▊   | 343/500 [00:35<00:14, 10.98Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  69%|██████▉   | 345/500 [00:35<00:13, 11.36Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  69%|██████▉   | 347/500 [00:35<00:13, 11.62Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  70%|███████   | 351/500 [00:36<00:13, 10.86Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  71%|███████   | 353/500 [00:36<00:12, 11.35Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  71%|███████   | 355/500 [00:36<00:12, 11.18Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  72%|███████▏  | 359/500 [00:36<00:11, 11.77Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  72%|███████▏  | 361/500 [00:37<00:12, 10.82Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  73%|███████▎  | 363/500 [00:37<00:12, 10.88Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  73%|███████▎  | 367/500 [00:37<00:11, 11.27Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  74%|███████▍  | 369/500 [00:37<00:11, 11.51Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  75%|███████▍  | 373/500 [00:38<00:11, 11.11Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  75%|███████▌  | 375/500 [00:38<00:11, 11.26Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  75%|███████▌  | 377/500 [00:38<00:10, 11.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  76%|███████▌  | 379/500 [00:38<00:10, 11.32Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  77%|███████▋  | 383/500 [00:38<00:10, 11.13Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  77%|███████▋  | 385/500 [00:39<00:10, 11.25Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  77%|███████▋  | 387/500 [00:39<00:09, 11.58Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  78%|███████▊  | 389/500 [00:39<00:09, 11.36Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  78%|███████▊  | 392/500 [00:39<00:12,  8.91Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  79%|███████▊  | 393/500 [00:40<00:12,  8.45Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  79%|███████▉  | 395/500 [00:40<00:12,  8.12Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  79%|███████▉  | 397/500 [00:40<00:12,  8.06Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  80%|███████▉  | 399/500 [00:40<00:12,  8.16Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  80%|████████  | 401/500 [00:41<00:14,  6.94Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  81%|████████  | 403/500 [00:41<00:13,  7.40Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  81%|████████  | 404/500 [00:41<00:12,  7.54Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  81%|████████  | 406/500 [00:41<00:12,  7.61Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  82%|████████▏ | 408/500 [00:42<00:12,  7.30Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  82%|████████▏ | 410/500 [00:42<00:11,  7.67Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  82%|████████▏ | 412/500 [00:42<00:12,  6.80Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  83%|████████▎ | 414/500 [00:42<00:10,  8.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  83%|████████▎ | 416/500 [00:43<00:08,  9.50Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  84%|████████▍ | 420/500 [00:43<00:07, 10.63Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  84%|████████▍ | 422/500 [00:43<00:07,  9.83Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  85%|████████▌ | 425/500 [00:43<00:07, 10.51Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  85%|████████▌ | 427/500 [00:44<00:06, 10.76Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  86%|████████▌ | 431/500 [00:44<00:06, 10.32Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  87%|████████▋ | 433/500 [00:44<00:06, 10.66Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  87%|████████▋ | 435/500 [00:44<00:06, 10.77Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  88%|████████▊ | 439/500 [00:45<00:05, 11.44Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  88%|████████▊ | 441/500 [00:45<00:05, 10.67Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  89%|████████▊ | 443/500 [00:45<00:05, 10.89Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  89%|████████▉ | 447/500 [00:45<00:04, 11.06Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  90%|████████▉ | 449/500 [00:46<00:04, 11.24Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  90%|█████████ | 451/500 [00:46<00:04, 10.37Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  91%|█████████ | 455/500 [00:46<00:04, 11.06Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  91%|█████████▏| 457/500 [00:46<00:03, 11.15Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  92%|█████████▏| 461/500 [00:47<00:03, 10.75Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  93%|█████████▎| 463/500 [00:47<00:03, 10.88Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  93%|█████████▎| 465/500 [00:47<00:03, 10.74Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  93%|█████████▎| 467/500 [00:47<00:02, 11.08Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  94%|█████████▍| 471/500 [00:48<00:02, 10.40Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  95%|█████████▍| 473/500 [00:48<00:02, 10.85Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  95%|█████████▌| 475/500 [00:48<00:02, 10.91Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  96%|█████████▌| 479/500 [00:48<00:01, 11.39Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  96%|█████████▌| 481/500 [00:49<00:01, 10.43Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  97%|█████████▋| 483/500 [00:49<00:01, 10.91Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  97%|█████████▋| 485/500 [00:49<00:01, 10.94Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  98%|█████████▊| 489/500 [00:49<00:00, 11.23Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  98%|█████████▊| 491/500 [00:49<00:00, 10.25Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining:  99%|█████████▊| 493/500 [00:50<00:00, 10.59Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Training:  99%|█████████▉| 497/500 [00:50<00:00, 10.54Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining: 100%|█████████▉| 499/500 [00:50<00:00, 10.73Epochs/s]"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "\rTraining: 100%|██████████| 500/500 [00:50<00:00,  9.85Epochs/s]\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Maximum test set accuracy: 0.795\n",
            "Minimum loss: 0.024975303560495377\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "shape source before: torch.Size([2708, 64])\n",
            "shape surce: torch.Size([2708, 2, 32])\n",
            "shape of alpha_l: torch.Size([2708, 2]) should be 2708,H\n",
            "shape H: 2\n",
            "shape C: 32\n",
            "shape of final_att: torch.Size([10556, 2])\n",
            "shape of x_j: torch.Size([10556, 2, 32])\n",
            "shape FINAL: torch.Size([10556, 2, 32])\n",
            "Saving Model Predictions for Model Type GAT\n",
            "\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<Figure size 640x480 with 1 Axes>"
            ],
            "image/png": "iVBORw0KGgoAAAANSUhEUgAAAiwAAAGzCAYAAAAMr0ziAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjcuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/bCgiHAAAACXBIWXMAAA9hAAAPYQGoP6dpAAB6eElEQVR4nO3deVzT9eMH8NcYbFxy34qAgrfgTXjkhZKVqV1qh0fnt7Q0tMx+pVl9s8xM08pvpWKHqZVapnnhlYp44i0qonhwI/cx2D6/P8Y+bLABY2ygvZ6Pxx7KPu999t6Hsc9r7+sjEQRBABEREVEzZtXUFSAiIiKqCwMLERERNXsMLERERNTsMbAQERFRs8fAQkRERM0eAwsRERE1ewwsRERE1OwxsBAREVGzx8BCREREzR4DCxERETV7DCxEZBFJSUl4+eWX0aZNG9ja2sLJyQn9+vXDkiVLUFJS0tTVI6JmzrqpK0BE974tW7bgiSeegFwux4QJE9ClSxcoFAocOHAAb775Js6dO4dvv/22qatJRM2YhBc/JCJzSk5ORmhoKFq1aoXdu3fD19dXZ/uVK1ewZcsWTJs2rcHPIQgCSktLYWdnZ2p1iaiZYpcQEZnVggULUFhYiBUrVtQIKwAQHBwshpWKigp8+OGHaNu2LeRyOQIDA/HOO++grKxM5zGBgYF4+OGHsX37dvTq1Qt2dnb43//+BwBYtWoVhgwZAi8vL8jlcnTq1AnffPON+V8oEZkVW1iIyKxatWoFuVyOpKSkOstOmjQJq1evxuOPP47BgwcjPj4eP/zwA0aPHo2NGzeK5QIDA2FjY4Ps7Gy8/PLLCAwMRPv27TFo0CD06dMHnTt3RlhYGKytrbF582bs2LEDy5Ytw5QpU8z5UonIjBhYiMhs8vPz4ezsjFGjRmHTpk21lj116hS6deuGF154Ad999514/5tvvomFCxdi9+7dGDx4MAB1YLl+/Tq2bduGqKgonf2UlJTU6Bp64IEHcPny5XqFJiJqntglRERmk5+fDwBo0aJFnWW3bt0KAIiOjta5f8aMGQDUA3e1BQUF1QgrAHTCSl5eHrKysjBw4EBcvXoVeXl5xr0AImo2OEuIiMzGyckJAFBQUFBn2evXr8PKygrBwcE69/v4+MDFxQXXr1/XuT8oKEjvfg4ePIi5c+ciLi4OxcXFOtvy8vLg7OxszEsgomaCgYWIzMbJyQl+fn44e/ZsvR8jkUjqVU7fjKCkpCQMHToUHTp0wKJFi+Dv7w+ZTIatW7fiiy++gEqlqnc9iKh5YZcQEZnVww8/jKSkJMTFxdVaLiAgACqVCpcvX9a5Pz09Hbm5uQgICKjzuTZv3oyysjL8+eefePnll/Hggw8iMjKS052J7gEMLERkVm+99RYcHBzwwgsvID09vcb2pKQkLFmyBA8++CAAYPHixTrbFy1aBAB46KGH6nwuqVQKQL0ui0ZeXh5WrVrV0OoTUTPBLiEiMqu2bdtizZo1GDt2LDp27Kiz0u2hQ4fw66+/YtKkSZg2bRomTpyIb7/9Frm5uRg4cCCOHDmC1atXY/To0eIModoMHz4cMpkMI0eOxMsvv4zCwkJ899138PLyQmpqqgVeLRGZC6c1E5FFXL58GZ999hl27tyJ27dvQy6XIzQ0FOPGjcOLL74IuVyOiooKfPzxx4iJicHNmzfh4+ODZ555BnPnzoVcLhf3FRgYiC5duuCvv/6q8TybN2/Gu+++i0uXLsHHxwevvPIKPD098dxzzyE5ORmBgYEWfNVE1FgYWIiIiKjZ4xgWIiIiavYYWIiIiKjZY2AhIiKiZo+BhYiIiJo9BhYiIiJq9hhYiIiIqNm7JxaOU6lUuH37Nlq0aFHv65AQERFR0xIEAQUFBfDz84OVVe1tKPdEYLl9+zb8/f2buhpERETUADdu3ECrVq1qLXNPBJYWLVoAUL9gzeXsiYiIqHnLz8+Hv7+/eB6vzT0RWDTdQE5OTgwsREREd5n6DOfgoFsiIiJq9owKLPPnz0fv3r3RokULeHl5YfTo0UhMTKzzcb/++is6dOgAW1tbdO3aFVu3btXZLggC5syZA19fX9jZ2SEyMhKXL1827pUQERHRPcuowLJv3z5MmTIFhw8fxs6dO1FeXo7hw4ejqKjI4GMOHTqE8ePH4/nnn8fJkycxevRojB49GmfPnhXLLFiwAF9++SWWL1+O+Ph4ODg4ICoqCqWlpQ1/ZURERHTPMOlqzZmZmfDy8sK+fftw//336y0zduxYFBUV6VwG/r777kO3bt2wfPlyCIIAPz8/zJgxAzNnzgQA5OXlwdvbGzExMRg3blyd9cjPz4ezszPy8vI4hoWI7kqCIKCiogJKpbKpq0LUqKRSKaytrfWOUzHm/G3SoNu8vDwAgJubm8EycXFxiI6O1rkvKioKmzZtAgAkJycjLS0NkZGR4nZnZ2eEh4cjLi5Ob2ApKytDWVmZ+HN+fr4pL4OIqEkpFAqkpqaiuLi4qatCZBb29vbw9fWFTCZr8D4aHFhUKhWmT5+Ofv36oUuXLgbLpaWlwdvbW+c+b29vpKWlids19xkqU938+fMxb968hladiKjZUKlUSE5OhlQqhZ+fH2QyGRfApHuGIAhQKBTIzMxEcnIyQkJC6lwgzpAGB5YpU6bg7NmzOHDgQEN30WCzZ8/WabXRzOMmIrrbKBQKqFQq+Pv7w97evqmrQ9To7OzsYGNjg+vXr0OhUMDW1rZB+2lQYJk6dSr++usv7N+/v86V6Xx8fJCenq5zX3p6Onx8fMTtmvt8fX11ynTr1k3vPuVyOeRyeUOqTkTULDX0WyfR3aAx3t9G7UEQBEydOhUbN27E7t27ERQUVOdjIiIiEBsbq3Pfzp07ERERAQAICgqCj4+PTpn8/HzEx8eLZYiIiOjfzagWlilTpmDNmjX4448/0KJFC3GMibOzM+zs7AAAEyZMQMuWLTF//nwAwLRp0zBw4EB8/vnneOihh7B27VocO3YM3377LQD16nbTp0/HRx99hJCQEAQFBeG9996Dn58fRo8e3YgvlYiIiO5WRrWwfPPNN8jLy8OgQYPg6+sr3tatWyeWSUlJQWpqqvhz3759sWbNGnz77bcICwvDb7/9hk2bNukM1H3rrbfw2muv4aWXXkLv3r1RWFiIbdu2Nbifi4iI7j6BgYFYvHhxvcvv3bsXEokEubm5ZqsTAMTExMDFxcWsz0F1M2kdluaC67AQ0d2qtLQUycnJCAoKuuu+pA0aNAjdunUzKmTUJjMzEw4ODvUefKxQKJCTkwNvb2+zzqyKiYnB9OnTzR6MTLVnzx58/vnniI+PR0FBAVq2bIlevXphypQpetdK69ChA5KTk3H9+nX4+Phg7969GDx4cJ3PMWjQIKPrZuh9bsz5m6O8alGhVGHVwWS8u+lMU1eFiOiupFkQrz48PT2Nmiklk8ng4+PDaeAAvv76awwdOhTu7u5Yt24dEhMTsXHjRvTt2xdvvPFGjfIHDhxASUkJHn/8caxevRqAukckNTVVvD355JN44IEHdO7r27evpV+aiIGlFpfSC/HBX+fx0+EUHLqS1dTVIaJ/CUEQUKyoaJJbfRvdJ02ahH379mHJkiWQSCSQSCS4du2a2E3z999/o2fPnpDL5Thw4ACSkpIwatQoeHt7w9HREb1798auXbt09lm9S0gikeD777/HmDFjYG9vj5CQEPz555/i9updQpqum+3bt6Njx45wdHQUT7gaFRUVeP311+Hi4gJ3d3fMmjULEydONHrM5DfffIO2bdtCJpOhffv2+PHHH3V+f++//z5at24NuVwOPz8/vP766+L2r7/+GiEhIbC1tYW3tzcef/xxo567upSUFEyfPh3Tp0/H6tWrMWTIEAQEBCA0NBTTpk3DsWPHajxmxYoVeOqpp/Dss89i5cqVAKoCoOZmZ2cHuVyuc58pC7+ZyqSVbu91nfyc8Ox9Afgh7jrmbT6P7W/ov/wAEVFjKilXotOc7U3y3Oc/iIK9rO5Tw5IlS3Dp0iV06dIFH3zwAQB1C8m1a9cAAG+//TYWLlyINm3awNXVFTdu3MCDDz6I//73v5DL5fjhhx8wcuRIJCYmonXr1gafZ968eViwYAE+++wzLF26FE8//TSuX79ucIX14uJiLFy4ED/++COsrKzwzDPPYObMmfj5558BAJ9++il+/vlnrFq1Ch07dsSSJUuwadOmOrtCtG3cuBHTpk3D4sWLERkZib/++guTJ09Gq1atMHjwYPz+++/44osvsHbtWnTu3BlpaWk4deoUAODYsWN4/fXX8eOPP6Jv377IycnBP//8U+/n1uf3339HeXk53nrrLb3bq7dAFRQU4Ndff0V8fDw6dOiAvLw8/PPPPxgwYIBJ9TA3trDUIXpYO0gkQGJ6AbILy+p+ABHRv4CzszNkMhns7e3Fb99SqVTc/sEHH2DYsGFo27Yt3NzcEBYWhpdffhldunRBSEgIPvzwQ7Rt21anxUSfSZMmYfz48QgODsbHH3+MwsJCHDlyxGD58vJyLF++HL169UKPHj0wdepUnWUzli5ditmzZ2PMmDHo0KEDli1bZvSA2oULF2LSpEl49dVX0a5dO0RHR+PRRx/FwoULAahbPHx8fBAZGYnWrVujT58+ePHFF8VtDg4OePjhhxEQEIDu3bvrtL40xKVLl+Dk5CSuawaoQ4yjo6N4O3OmamjD2rVrERISgs6dO0MqlWLcuHFYsWKFSXWwBLaw1MHFXoYAN3tcyy7GhdQC9A/hgnVEZF52NlKc/yCqyZ67MfTq1Uvn58LCQrz//vvYsmULUlNTUVFRgZKSEqSkpNS6n9DQUPH/Dg4OcHJyQkZGhsHy9vb2aNu2rfizr6+vWD4vLw/p6eno06ePuF0qlaJnz55QqVT1fm0XLlzASy+9pHNfv379sGTJEgDAE088gcWLF6NNmzZ44IEH8OCDD2LkyJGwtrbGsGHDEBAQIG574IEHxC4vfRwdHcX/P/PMM1i+fLnectVbUaKiopCQkIBbt25h0KBBOhfVXLlyJZ555hmd/Q4cOBBLly5FixYt6n0cLI2BpR46+jpVBpZ89A/xaOrqENE9TiKR1KtbpjlzcHDQ+XnmzJnYuXMnFi5ciODgYNjZ2eHxxx+HQqGodT82NjY6P0skklrDhb7ylp4M6+/vj8TEROzatQs7d+7Eq6++is8++wz79u1DixYtcOLECezduxc7duzAnDlz8P777+Po0aN6W3oSEhLE/xuaRRMSEoK8vDykpaWJrSyOjo4IDg6GtbXu++j8+fM4fPgwjhw5glmzZon3K5VKrF27VmwJao7YJVQPnXzVb5LzqbwqNBGRhkwm0/nmXpuDBw9i0qRJGDNmDLp27QofHx9xvIulODs7w9vbG0ePHhXvUyqVOHHihFH76dixIw4ePKhz38GDB9GpUyfxZzs7O4wcORJffvkl9u7di7i4OLFbxtraGpGRkViwYAFOnz6Na9euYffu3XqfKzg4WLx5eXnpLfP444/DxsYGn376aZ11X7FiBe6//36cOnUKCQkJ4i06OrrZdwvd3RHeQjpWBpYLDCxERKLAwEDEx8fj2rVrcHR0NDgQFlC3AmzYsAEjR46ERCLBe++9Z1Q3TGN57bXXMH/+fAQHB6NDhw5YunQp7ty5Y9TU6DfffBNPPvkkunfvjsjISGzevBkbNmwQZz3FxMRAqVQiPDwc9vb2+Omnn2BnZ4eAgAD89ddfuHr1Ku6//364urpi69atUKlUaN++fYNfU+vWrfH5559j2rRpyMnJwaRJkxAUFIScnBz89NNPANRdX+Xl5fjxxx/xwQcf6CzeCgAvvPACFi1ahHPnzqFz584Nros5sYWlHoK91H2I17KLLN60SETUXM2cORNSqRSdOnWCp6dnreNRFi1aBFdXV/Tt2xcjR45EVFQUevToYcHaqs2aNQvjx4/HhAkTEBERAUdHR0RFRRm1aN/o0aOxZMkSLFy4EJ07d8b//vc/rFq1SlxQzcXFBd999x369euH0NBQ7Nq1C5s3b4a7uztcXFywYcMGDBkyBB07dsTy5cvxyy+/mBwSXnvtNezYsQOZmZl4/PHHERISggcffBDJycnYtm0bunbtij///BPZ2dkYM2ZMjcd37NgRHTt2bNatLFzpth4UFSp0eO9vqATgyDtD4eV0d61GSUTN19280u29QKVSoWPHjnjyySfx4YcfNnV17lmNsdItu4TqQWZtBT8XO9y8U4LrOcUMLEREd6nr169jx44dGDhwIMrKyrBs2TIkJyfjqaeeauqqUR3YJVRPrd3UU85SsoubuCZERNRQVlZWiImJQe/evdGvXz+cOXMGu3btQseOHZu6alQHtrDUU4C7PQ4lZeN6DgMLEdHdyt/fv8YMH7o7sIWlnvwrW1huMLAQERFZHANLPQW4qRdBSs4qauKaEBER/fswsNRTO2/11ObL6QVQqe76iVVERER3FQaWegr0cIBMaoUihRK3ckuaujpERET/Kgws9WQjtULbygXkLqYVNHFtiIiI/l0YWIzQ0Ud9FcvENC7RT0REZEkMLEZoXxlYLqUXNnFNiIiI/l0YWIzgXbnCbXZRWRPXhIio6Q0aNAjTp09v1H1OmjQJo0ePbtR9UpXff/8dQ4YMgaurK+zs7NC+fXs899xzOHnyZI2yJSUlcHNzg4eHB8rK1Oe9mJgYSCSSWm/mugo3A4sRnO1tAAC5xeVNXBMiImpuysub97lh1qxZGDt2LLp164Y///wTiYmJWLNmDdq0aYPZs2fXKP/777+jc+fO6NChAzZt2gQAGDt2LFJTU8VbREQEXnzxRZ37/P39zVJ/BhYjuNgxsBCRBQgCoChqmls9r4c7adIk7Nu3D0uWLKnxzfrs2bMYMWIEHB0d4e3tjWeffRZZWVniY3/77Td07doVdnZ2cHd3R2RkJIqKivD+++9j9erV+OOPP8R97t27V+/zb9u2Df3794eLiwvc3d3x8MMPIykpSafMzZs3MX78eLi5ucHBwQG9evVCfHy8uH3z5s3o3bs3bG1t4eHhoXMVY4lEIp6kNVxcXBATEwMAuHbtGiQSCdatW4eBAwfC1tYWP//8M7KzszF+/Hi0bNkS9vb26Nq1K3755Red/ahUKixYsADBwcGQy+Vo3bo1/vvf/wIAhgwZgqlTp+qUz8zMhEwmQ2xsbJ2/F0MOHz6MBQsWYNGiRVi0aBEGDBiA1q1bo2fPnnj33Xfx999/13jMihUr8Mwzz+CZZ54Rr+JsZ2cHHx8f8SaTyWBvb69zn1QqbXA9a8Ol+Y3gai8DAOSVMLAQkRmVFwMf+zXNc79zG5A51FlsyZIluHTpErp06YIPPvgAAODp6Ync3FwMGTIEL7zwAr744guUlJRg1qxZePLJJ7F7926kpqZi/PjxWLBgAcaMGYOCggL8888/EAQBM2fOxIULF5Cfn49Vq1YBANzc3PQ+f1FREaKjoxEaGorCwkLMmTMHY8aMQUJCAqysrFBYWIiBAweiZcuW+PPPP+Hj44MTJ05ApVIBALZs2YIxY8bg//7v//DDDz9AoVBg69atRh+ut99+G59//jm6d+8OW1tblJaWomfPnpg1axacnJywZcsWPPvss2jbti369OkDAJg9eza+++47fPHFF+jfvz9SU1Nx8eJFAMALL7yAqVOn4vPPP4dcLgcA/PTTT2jZsiWGDBlidP00fvnlFzg6OuLVV1/Vu10ikej8nJSUhLi4OGzYsAGCIOCNN97A9evXERAQ0OA6mIqBxQgulV1ChWUVKFeqYCNlAxUR/Ts5OzvrfLvWWLZsGbp3746PP/5YvG/lypXw9/fHpUuXUFhYiIqKCjz66KPiya9r165iWTs7O5SVlensU5/HHntM5+eVK1fC09MT58+fR5cuXbBmzRpkZmbi6NGjYugJDg4Wy//3v//FuHHjMG/ePPG+sLAwo4/D9OnT8eijj+rcN3PmTPH/r732GrZv347169ejT58+KCgowJIlS7Bs2TJMnDgRANC2bVv0798fAPDoo49i6tSp+OOPP/Dkk08CUI8bmTRpUo1QYYxLly6hTZs2sLauOu0vWrQIc+bMEX++desWnJ2dAaiP54gRI+Dq6goAiIqKwqpVq/D+++83uA6mYmAxQgtbG0gk6hbTvJJyeDjKm7pKRHQvsrFXt3Q01XOb4NSpU9izZw8cHR1rbEtKSsLw4cMxdOhQdO3aFVFRURg+fDgef/xx8cRYX5cvX8acOXMQHx+PrKwsseUkJSUFXbp0QUJCArp3726whSYhIQEvvvii8S+wml69eun8rFQq8fHHH2P9+vW4desWFAoFysrKYG+vPq4XLlxAWVkZhg4dqnd/tra2ePbZZ7Fy5Uo8+eSTOHHiBM6ePYs///xTb/mUlBR06tRJ/Pmdd97BO++8U6+6P/fcc3jkkUcQHx+PZ555BkJld6BSqcTq1auxZMkSsewzzzyDmTNnYs6cObCyapov6wwsRpBaSeBka4O8knLkFjOwEJGZSCT16pZpjgoLCzFy5Eh8+umnNbb5+vpCKpVi586dOHToEHbs2IGlS5fi//7v/xAfH4+goKB6P8/IkSMREBCA7777Dn5+flCpVOjSpQsUCgUAdUtNberaLpFIxBO4hr5BtQ4Our+nzz77DEuWLMHixYvRtWtXODg4YPr06fWuF6DuFurWrRtu3ryJVatWYciQIQa7Yvz8/JCQkCD+bCighYSE4MCBAygvL4eNjbq3wMXFBS4uLrh586ZO2e3bt+PWrVsYO3aszv1KpRKxsbEYNmxYna/BHNinYSRNt1BeiaKJa0JE1LRkMhmUSqXOfT169MC5c+cQGBiI4OBgnZvm5C6RSNCvXz/MmzcPJ0+ehEwmw8aNGw3us7rs7GwkJibi3XffxdChQ9GxY0fcuXNHp0xoaCgSEhKQk5Ojdx+hoaG1DmL19PREamqq+PPly5dRXFxca70A4ODBgxg1ahSeeeYZhIWFoU2bNrh06ZK4PSQkBHZ2drU+d9euXdGrVy989913WLNmDZ577jmDZa2trXWOsaHAMn78eBQWFuLrr7+u8zWsWLEC48aNQ0JCgs5t3Lhx4uDbpsAWFiO52NngOjhTiIgoMDAQ8fHxuHbtGhwdHeHm5oYpU6bgu+++w/jx4/HWW2/Bzc0NV65cwdq1a/H999/j2LFjiI2NxfDhw+Hl5YX4+HhkZmaiY8eO4j63b9+OxMREuLu7w9nZWWwR0HB1dYW7uzu+/fZb+Pr6IiUlBW+//bZOmfHjx+Pjjz/G6NGjMX/+fPj6+uLkyZPw8/NDREQE5s6di6FDh6Jt27YYN24cKioqsHXrVsyaNQuAerbOsmXLEBERAaVSiVmzZtWohz4hISH47bffcOjQIbi6umLRokVIT08Xu21sbW0xa9YsvPXWW5DJZOjXrx8yMzNx7tw5PP/88+J+NINvHRwcdGYvNVRERARmzJiBGTNm4Pr163j00Ufh7++P1NRUrFixAhKJBFZWVsjMzMTmzZvx559/okuXLjr7mDBhAsaMGYOcnByDwcishHtAXl6eAEDIy8sz+3M9uyJeCJj1l/DbsRtmfy4iuveVlJQI58+fF0pKSpq6KkZLTEwU7rvvPsHOzk4AICQnJwuCIAiXLl0SxowZI7i4uAh2dnZChw4dhOnTpwsqlUo4f/68EBUVJXh6egpyuVxo166dsHTpUnGfGRkZwrBhwwRHR0cBgLBnzx69z71z506hY8eOglwuF0JDQ4W9e/cKAISNGzeKZa5duyY89thjgpOTk2Bvby/06tVLiI+PF7f//vvvQrdu3QSZTCZ4eHgIjz76qLjt1q1bwvDhwwUHBwchJCRE2Lp1q+Ds7CysWrVKEARBSE5OFgAIJ0+e1KlXdna2MGrUKMHR0VHw8vIS3n33XWHChAnCqFGjxDJKpVL46KOPhICAAMHGxkZo3bq18PHHH+vsp6CgQLC3txdeffXV+v9C6mHdunXCoEGDBGdnZ8HGxkZo1aqV8NRTTwmHDx8WBEEQFi5cKLi4uAgKhaLGY8vKygQXFxdhyZIl4n0DBw4Upk2bVufzGnqfG3P+lghCPSfdN2P5+flwdnZGXl4enJyczPpcr/9yEn+euo33Hu6E5/vXv7+ViEif0tJSJCcnIygoCLa2tk1dHWomrl27hrZt2+Lo0aPo0aNHU1fHZIbe58acv9klZCQXcbVbjmEhIqLGVV5ejuzsbLz77ru477777omw0lg46NZIznaaQbccw0JERI3r4MGD8PX1xdGjR7F8+fKmrk6zYnRg2b9/P0aOHAk/Pz+9SxdXp1nspvqtc+fOYpn333+/xvYOHToY/WIswdZGveRwaXnto9iJiIiMNWjQIAiCgMTERJ0F9agBgaWoqAhhYWH46quv6lV+yZIlOhdFunHjBtzc3PDEE0/olOvcubNOuQMHDhhbNYuwqwwsJeWqJq4JERHRv4fRY1hGjBiBESNG1Lu8s7OzuNQvAGzatAl37tzB5MmTdStibV3nUszNAVtYiMgc7oH5D0QGNcb72+JjWFasWIHIyMgaq/ZdvnwZfn5+aNOmDZ5++mmkpKQY3EdZWRny8/N1bpZia6M+ZAwsRNQYNGt71GdRMqK7leb9XZ+1bAyx6Cyh27dv4++//8aaNWt07g8PD0dMTAzat2+P1NRUzJs3DwMGDMDZs2fRokWLGvuZP3++zgWrLMmOLSxE1IikUilcXFyQkZEBALC3tzfpIndEzYkgCCguLkZGRgZcXFwglUobvC+LBpbVq1fDxcUFo0eP1rlfu4spNDQU4eHhCAgIwPr163VW/tOYPXs2oqOjxZ/z8/Ph7+9vtnprq+oS4hgWImocmu5wTWghute4uLiYPOzDYoFFEASsXLkSzz77LGQyWa1lXVxc0K5dO1y5ckXvdrlcDrm8aS48KGeXEBE1MolEAl9fX3h5eem9wB7R3czGxsaklhUNiwWWffv24cqVK3pbTKorLCxEUlISnn32WQvUzDhVs4QYWIiocUml0kb5YCe6Fxk96LawsFC8ciMAJCcnIyEhQRwkO3v2bEyYMKHG41asWIHw8PAaF1MCgJkzZ2Lfvn24du0aDh06hDFjxkAqlWL8+PHGVs/s2CVERERkeUa3sBw7dgyDBw8Wf9aMJZk4cSJiYmKQmppaY4ZPXl4efv/9dyxZskTvPm/evInx48cjOzsbnp6e6N+/Pw4fPgxPT09jq2d2msBSxhYWIiIiizE6sGhW4TMkJiamxn3Ozs61Ttlbu3atsdVoMuIsoQoGFiIiIkvhtYSMpFmHpVwpoELJbiEiIiJLYGAxkqZLCABKKxhYiIiILIGBxUhy66pDxqnNRERElsHAYiSJRCJ2C5UoGFiIiIgsgYGlAcSZQhx4S0REZBEMLA1gx7VYiIiILIqBpQFsudotERGRRTGwNIBm4C0H3RIREVkGA0sDcHl+IiIiy2JgaQBeAJGIiMiyGFgaQDOtmV1CRERElsHA0gC8ACIREZFlMbA0ALuEiIiILIuBpQHkHHRLRERkUQwsDcAxLERERJbFwNIA7BIiIiKyLAaWBuA6LERERJbFwNIAmi4hzhIiIiKyDAaWBmCXEBERkWUxsDRA1SwhBhYiIiJLYGBpAI5hISIisiwGlgZglxAREZFlMbA0ANdhISIisiwGlgYQryVUwS4hIiIiS2BgaQCxS0jBFhYiIiJLYGBpALFLqIKBhYiIyBIYWBpAbs1pzURERJbEwNIA2tOaBUFo4toQERHd+xhYGsBOJhX/z4G3RERE5sfA0gC21lWHjd1CRERE5sfA0gDWUitYW0kAcPE4IiIiS2BgaSA7Ls9PRERkMUYHlv3792PkyJHw8/ODRCLBpk2bai2/d+9eSCSSGre0tDSdcl999RUCAwNha2uL8PBwHDlyxNiqWRQvgEhERGQ5RgeWoqIihIWF4auvvjLqcYmJiUhNTRVvXl5e4rZ169YhOjoac+fOxYkTJxAWFoaoqChkZGQYWz2L0azFwi4hIiIi87M29gEjRozAiBEjjH4iLy8vuLi46N22aNEivPjii5g8eTIAYPny5diyZQtWrlyJt99+2+jnsgQ7trAQERFZjMXGsHTr1g2+vr4YNmwYDh48KN6vUChw/PhxREZGVlXKygqRkZGIi4vTu6+ysjLk5+fr3CxNvJ4Qx7AQERGZndkDi6+vL5YvX47ff/8dv//+O/z9/TFo0CCcOHECAJCVlQWlUglvb2+dx3l7e9cY56Ixf/58ODs7izd/f39zv4wa2CVERERkOUZ3CRmrffv2aN++vfhz3759kZSUhC+++AI//vhjg/Y5e/ZsREdHiz/n5+dbPLTYskuIiIjIYsweWPTp06cPDhw4AADw8PCAVCpFenq6Tpn09HT4+PjofbxcLodcLjd7PWtTdT0hdgkRERGZW5Osw5KQkABfX18AgEwmQ8+ePREbGytuV6lUiI2NRURERFNUr17klV1CCl6xmYiIyOyMbmEpLCzElStXxJ+Tk5ORkJAANzc3tG7dGrNnz8atW7fwww8/AAAWL16MoKAgdO7cGaWlpfj++++xe/du7NixQ9xHdHQ0Jk6ciF69eqFPnz5YvHgxioqKxFlDzZFcWhlYlGxhISIiMjejA8uxY8cwePBg8WfNWJKJEyciJiYGqampSElJEbcrFArMmDEDt27dgr29PUJDQ7Fr1y6dfYwdOxaZmZmYM2cO0tLS0K1bN2zbtq3GQNzmRNPCwllCRERE5icRBEFo6kqYKj8/H87OzsjLy4OTk5NFnnPuH2exOu46XhsSjBnD29f9ACIiItJhzPmb1xJqIM3S/GUVbGEhIiIyNwaWBpJpxrAwsBAREZkdA0sDya0rx7BwlhAREZHZMbA0kEwMLGxhISIiMjcGlgaSM7AQERFZDANLA8kqV7rlGBYiIiLzY2BpILawEBERWQ4DSwNpxrBwaX4iIiLzY2BpILawEBERWQ4DSwNVtbAwsBAREZkbA0sDya250i0REZGlMLA0kObih2xhISIiMj8GlgbSLM3PlW6JiIjMj4GlgWxtOOiWiIjIUhhYGkgm5cJxRERElsLA0kBytrAQERFZDANLA2nGsChVAiqUDC1ERETmxMDSQJoWFgBQMLAQERGZFQNLA2laWACOYyEiIjI3BpYGspZaQWolAcBxLERERObGwGICTSsLW1iIiIjMi4HFBFUzhbh4HBERkTkxsJigarVbtrAQERGZEwOLCbgWCxERkWUwsJiAY1iIiIgsg4HFBHJr9fL8bGEhIiIyLwYWE8is2cJCRERkCQwsJpBbc5YQERGRJTCwmIAtLERERJbBwGICjmEhIiKyDAYWE8jZwkJERGQRDCwm4BgWIiIiy2BgMQHHsBAREVmG0YFl//79GDlyJPz8/CCRSLBp06Zay2/YsAHDhg2Dp6cnnJycEBERge3bt+uUef/99yGRSHRuHTp0MLZqFlfVwsLAQkREZE5GB5aioiKEhYXhq6++qlf5/fv3Y9iwYdi6dSuOHz+OwYMHY+TIkTh58qROuc6dOyM1NVW8HThwwNiqWRxbWIiIiCzD2tgHjBgxAiNGjKh3+cWLF+v8/PHHH+OPP/7A5s2b0b1796qKWFvDx8fH2Oo0Kc4SIiIisgyLj2FRqVQoKCiAm5ubzv2XL1+Gn58f2rRpg6effhopKSkG91FWVob8/HydW1OQsUuIiIjIIiweWBYuXIjCwkI8+eST4n3h4eGIiYnBtm3b8M033yA5ORkDBgxAQUGB3n3Mnz8fzs7O4s3f399S1dfBWUJERESWYdHAsmbNGsybNw/r16+Hl5eXeP+IESPwxBNPIDQ0FFFRUdi6dStyc3Oxfv16vfuZPXs28vLyxNuNGzcs9RJ0cAwLERGRZRg9hqWh1q5dixdeeAG//vorIiMjay3r4uKCdu3a4cqVK3q3y+VyyOVyc1TTKBzDQkREZBkWaWH55ZdfMHnyZPzyyy946KGH6ixfWFiIpKQk+Pr6WqB2DccWFiIiIsswuoWlsLBQp+UjOTkZCQkJcHNzQ+vWrTF79mzcunULP/zwAwB1N9DEiROxZMkShIeHIy0tDQBgZ2cHZ2dnAMDMmTMxcuRIBAQE4Pbt25g7dy6kUinGjx/fGK/RbDiGhYiIyDKMbmE5duwYunfvLk5Jjo6ORvfu3TFnzhwAQGpqqs4Mn2+//RYVFRWYMmUKfH19xdu0adPEMjdv3sT48ePRvn17PPnkk3B3d8fhw4fh6elp6uszK7awEBERWYbRLSyDBg2CIAgGt8fExOj8vHfv3jr3uXbtWmOr0SxwpVsiIiLL4LWETMAWFiIiIstgYDEBZwkRERFZBgOLCeRsYSEiIrIIBhYTcJYQERGRZTCwmIBjWIiIiCyDgcUEHMNCRERkGQwsJtC0sFSoBChVhqd6ExERkWkYWEygGcMCsFuIiIjInBhYTCBjYCEiIrIIBhYTWFtJYCVR/58zhYiIiMyHgcUEEolEbGXhwFsiIiLzYWAxEWcKERERmR8Di4m4eBwREZH5MbCYyNZG3cJSWs4WFiIiInNhYDGRrU1lC0s5W1iIiIjMhYHFRHaVLSwlDCxERERmw8BiIlsGFiIiIrNjYDERx7AQERGZHwOLidglREREZH4MLCayk1W2sCgYWIiIiMyFgcVEmllCpWxhISIiMhsGFhNx0C0REZH5MbCYiGNYiIiIzI+BxUScJURERGR+DCwmshMDC1tYiIiIzIWBxUS2lbOESjhLiIiIyGwYWExkW3m15lJerZmIiMhsGFhMZMcWFiIiIrNjYDERx7AQERGZHwOLiThLiIiIyPwYWEzEheOIiIjMj4HFRJql+RlYiIiIzIeBxUQcw0JERGR+RgeW/fv3Y+TIkfDz84NEIsGmTZvqfMzevXvRo0cPyOVyBAcHIyYmpkaZr776CoGBgbC1tUV4eDiOHDlibNWahHi1ZgYWIiIiszE6sBQVFSEsLAxfffVVvconJyfjoYcewuDBg5GQkIDp06fjhRdewPbt28Uy69atQ3R0NObOnYsTJ04gLCwMUVFRyMjIMLZ6FmdrrQ4s5UoBFUoOvCUiIjIHiSAIQoMfLJFg48aNGD16tMEys2bNwpYtW3D27FnxvnHjxiE3Nxfbtm0DAISHh6N3795YtmwZAEClUsHf3x+vvfYa3n777Rr7LCsrQ1lZmfhzfn4+/P39kZeXBycnp4a+nAYpLVeiw3vq13F2XhQc5dYWfX4iIqK7VX5+Ppydnet1/jb7GJa4uDhERkbq3BcVFYW4uDgAgEKhwPHjx3XKWFlZITIyUixT3fz58+Hs7Cze/P39zfcC6iC3toJEov5/saKiyepBRER0LzN7YElLS4O3t7fOfd7e3sjPz0dJSQmysrKgVCr1lklLS9O7z9mzZyMvL0+83bhxw2z1r4tEIoF95cDb4jKOYyEiIjKHu7L/Qi6XQy6XN3U1RA5yaxQplCgsYwsLERGROZg9sPj4+CA9PV3nvvT0dDg5OcHOzg5SqRRSqVRvGR8fH3NXr1E4yq2RUVCGIgYWIiIiszB7l1BERARiY2N17tu5cyciIiIAADKZDD179tQpo1KpEBsbK5Zp7hwqB9oWcQwLERGRWRgdWAoLC5GQkICEhAQA6mnLCQkJSElJAaAeXzJhwgSx/H/+8x9cvXoVb731Fi5evIivv/4a69evxxtvvCGWiY6OxnfffYfVq1fjwoULeOWVV1BUVITJkyeb+PIsw0GuHsNSyDEsREREZmF0l9CxY8cwePBg8efo6GgAwMSJExETE4PU1FQxvABAUFAQtmzZgjfeeANLlixBq1at8P333yMqKkosM3bsWGRmZmLOnDlIS0tDt27dsG3bthoDcZsrzVRmdgkRERGZh0nrsDQXxszjNodpa0/ij4TbePehjnhhQBuLPz8REdHdqFmtw/JvoBnDwllCRERE5sHA0gjYJURERGReDCyNwEGmaWHhoFsiIiJzYGBpBJpZQmxhISIiMg8GlkbALiEiIiLzYmBpBI626sBSwMBCRERkFgwsjcCBLSxERERmxcDSCNglREREZF4MLI2As4SIiIjMi4GlEbCFhYiIyLwYWBqBfeW05pJyJVSqu/5KB0RERM0OA0sjkFlXHcZylaoJa0JERHRvYmBpBDKpVmBRsoWFiIiosTGwNAIbrcCiqGALCxERUWNjYGkEUisJpFYSAEC5koGFiIiosTGwNBJNtxBbWIiIiBofA0sjsZGqW1gUbGEhIiJqdAwsjURmrZ7azBYWIiKixsfA0khkUo5hISIiMhcGlkaiWYuFgYWIiKjxMbA0Es3U5jJ2CRERETU6BpZGogksXDiOiIio8TGwNBJNlxAH3RIRETU+BpZGIpNyDAsREZG5MLA0ErawEBERmQ8DSyPhwnFERETmw8DSSDitmYiIyHysm7oC9wobXkvIsMS/gX0LAFW54TI2DsCDnwG+oZarF5E57V8InN9Ue5kWfsBj3wO2ThapEtHdjIGlkTS7QbcluQBqmWJtYw9Yyy1Tl38WAbdP1F3u8NfAmOXmrw+RuZXmA3s+BgRl7eXSzgDn/wB6PGuZejWG8lIg9VTtr03mAHh3BazuokZ8QVD/PhSFhstIpOovVTZ2lqsXiRhYGkmzGnS78RXg1Jray8idgTHfACHD9W8XBODAIuDkT4Cqlg8meQvg8ZWATxf921VKIP2s+v+jvwEcvWuWyUwEts8Gkvaon1ciqb3utcm9AZz8ESgvNlxGKgd6TgJc/Bv+PI2pIB3YNkt9kjOFRAL0mAB0GtU49WpOBAFQVdRexsratPeORsYF4HZC7WW8OgJ+3QxvvxGvPqE7+wMjl+gvc3YDkPATkLzf9MCybwFwPEZ9nAyxcwXG/QS4tWn48wgCsO5p4MquussGDay9xdTGQf26nVs1vD6N6cRqYPO0uss5eAI+XQEYeK9ZSYE+LwEhw2rfj0pVd6CtD6mN6fu4SzCwNBKxS6ipF45TlgPnNtRdriwPWPuU6c9XACD+G2DUV/q3Zyepw4ONPRA6Vv3HXF1APyD2A6AwDdjxrjoE6VNeAty5Bgi1hMIb8UBhet31zrsBPPpt3eUsYf9nwLmNjbOvnKumB5byEmDX+0D+7drL+XUH2o8ArAx8YJblA2d+BfJu1r4f10CgZU9AYuDbuKII2L9A/dpqI3MEWvjA4IkEADzaAY+vMPwNuSgL+D6y9m/ZGo8sBVr31b8t8W/1v0H3A8FD9ZeRytSB5epe4Pyfhp/H1kn9N2LoxHTnOrD3k7pPfgW3gYQ1wJB3DZc58SMQt8zwlxRVBXAnWf07dw00vJ/cFCB5n/pWm7ivKn9nBljLgaiPgTYDDZdRKdXPU1bL70wqU/8uZPaGy1zdq/7XwQuwddZfpjADKMoEknYb3g8AZF4Epp02HKBvHgN+GFW/91ldek4yHIrvMQwsjaTZtLCknwUqSgFbF2DmZf1/MKoKYNMrdZ8kbeyByHlA63D929POAn+8ClzcCgy6pf+Ec+0f9b/enfWHFQCwsQUC+wNXdqo/LE3l0R5oF6V/W2E6cHodkHLY9OfJuAhsia7jQ0eibsmRGQhhgLpLAAAGv9vwb5slOcD2d4D81LpbqRRFtbfmHPkWiK9H19yFP4HYecbX1VwUhUD2ldrLZF8GNv4HaNVL//YbR9T7aeGrfs/qU5yj7uL887W66xTQz/C2Vr0Ba1ugKANYX1cLi8RwYFEp1WElcAAQ9V/9ZS7vAHZ/pG7NMaRCAeyco34v1WXoHKDf64a3p58Hzv6m/gJlSPI+dddSdkHtz7UlGpjwp+FAu+3tuscKAUCv54CHvzC8PfW0+t8x3wDBkfrLlJeoj6XCUAuuAGyZqQ5sV2INv4diP2icsAIAp9YBD34OSO/903mDXuFXX32Fzz77DGlpaQgLC8PSpUvRp08fvWUHDRqEfftqpuwHH3wQW7ZsAQBMmjQJq1ev1tkeFRWFbdu2NaR6TcKmrjEsBxar3+i1aeELDJoNOHoZLmMtr33syc1j6n9b9gSsZfrLSG2AJ2KAkV/W3lphbasOE4Z4dVa3iJTkAF90MlwOAHzqGEz7wHzgaDCgVBguY2UNuAaovy0ZYmMPdHzY8Dek0jzg9Hog97r627SDR+31qs2BL4DrB+sul5pQdxmvzsD9MxvepaEoVgeWihJ1y4ah15+dBCzvX3uXmUbf1wGX1vq3leWrPygL02rfh3dXoMNDtZ9sU+LqbhXz6QoMmFnLe18ACtKAkjuG95F+Dtg6U31yq+sEN+JTwy1Vygp1WEncUvs+nFurW6AMsbFVB4wzvxkuIwjqEFacVfffxpB3Ad8w/dttXdSB5eYx4Mh3+k/+OVfVf8uO3sDjqwy/F23sDT+PhncnwHtO7WWUFeq/DUOvS6UE1k9Qv/66Pl+srNUBUJ+KUuD2SXUX3IgF+t+LZQVATpL6/z61vDYbu7pbMJN2q1sWf36s9nISKfDKodpbmGolAF90BRQFQFai4XB0DzE6sKxbtw7R0dFYvnw5wsPDsXjxYkRFRSExMRFeXjVPtBs2bIBCUfWGzM7ORlhYGJ544gmdcg888ABWrVol/iyXW2hAaCOpdVpz+nlg19z67ehsLR9egPqDxtHH8LeN0lz1v4b+eLWZOjNBaq3+lrX3k9rHF8hbAF3q+OP1CAFGfGJaferD1hnwbK9uso1bBnh21F9OYqUOR4a6DlRK4GLlCevBhYabx5Xl6nBU28lGYqU+qZsy/kJmD8id1EGiMMNwYLl+qCqsWNXy59/hYWDYB7XXacCMhtdX233/aZz92LnWvj2grzoA3Dpeezn3YKDDSMPbpdbqb+GNofcL6lttlBV1Bzp5i9r/nl0D1O/RO9fUoa02oU8CgbW0DDUWqbXhli6N4R8CW9+s/e9H5gAM/6/hcUAqJfB5e3VXzuZp6jEo1RVlqf9t4Qc46tlujPBX1K0rZbW0YkqsgPteBbw6mPZcvqHqL02ppxhY9Fm0aBFefPFFTJ48GQCwfPlybNmyBStXrsTbb79do7ybm5vOz2vXroW9vX2NwCKXy+Hj09Ck2cR2zcOkkxsxUqaA03lr4Ea1VglNE2ubQer+Rn0EFRD/LXCjjq4KQaXui65LXQO+Gkv/N9S3u0mrXurAcqCW5uH6cvYHej3fPGZDOHqpPyQL0tQBUJ8719T/9noeeHiRxarWbIS/1NQ1MJ7UGnBuafp+HvocOL669lZVW2egbz0GnlpK92fUN1NYSYGOI4FjK4GEn2sv27KHac8FAK16ArOSTd9Pffh2UweWEz/WPlYsOLJxXlsTMyqwKBQKHD9+HLNnzxbvs7KyQmRkJOLi4uq1jxUrVmDcuHFwcHDQuX/v3r3w8vKCq6srhgwZgo8++gju7u5691FWVoaysjLx5/x8E2dXmOrgErgJSrhZASirvFUnkarHg9Q2s6DLY0CFvgdrKblT92BIBw/DTfkERLwGFN+pvVtEWa4+udfWcmRlre7Caw5hBVC3vGVfqf3b+J3KD9LaBkzSvSk40vDYjHvdoHcAe3f1GBRDpDZA97toejlQFUJSDqlvhhz+Boi+UHsX/13AqMCSlZUFpVIJb2/dqane3t64ePFinY8/cuQIzp49ixUrVujc/8ADD+DRRx9FUFAQkpKS8M4772DEiBGIi4uDVFpzoOb8+fMxb14zGewnCOLo/BcV0ejcxh/TI9vVLNfCF3BvW/f+6lobpYWPCX2eBEDdDDu+jmnfdyPN2KdaA8s19b8MLPRv4uhZ++you1XHR4CBs2r/m7+4Rd0dFr+89kHgnu0MdyU3ExYdVrxixQp07dq1xgDdcePGif/v2rUrQkND0bZtW+zduxdDh9acEjh79mxER0eLP+fn58Pfv4nW1NBa9+Coqj2s5MFAYB39skTmoAmyDCxE/w7WMmDwO7WXaeEL7J1f9zhK9xDg1bhmva6LUYHFw8MDUqkU6em6H4jp6el1jj8pKirC2rVr8cEHH9T5PG3atIGHhweuXLmiN7DI5fLmMyhXqz9YBUnTT2umfy9NC8vthKp1QLQpFUBxtvr/DCxE/w69X1CvMVOQarhMQZp6yv/+z9TrKxkisTK8ZIQFGBVYZDIZevbsidjYWIwePRoAoFKpEBsbi6lTp9b62F9//RVlZWV45pm6B1DdvHkT2dnZ8PX1NaZ6TUMrsAiwQnlTLxxH/14tKv9e6lqwy96d164h+rdw8ACeq2OJkH8+V68Ns+/T2stJ5cB7GY1XNyMZ3SUUHR2NiRMnolevXujTpw8WL16MoqIicdbQhAkT0LJlS8yfP1/ncStWrMDo0aNrDKQtLCzEvHnz8Nhjj8HHxwdJSUl46623EBwcjKiopkty9Va9haW5XEuI/n3aRQHtH6q9S0giAbo1wgrHRHTv6POSemp03q3ay9W2BpYFGB1Yxo4di8zMTMyZMwdpaWno1q0btm3bJg7ETUlJgVW1WROJiYk4cOAAduyouXCaVCrF6dOnsXr1auTm5sLPzw/Dhw/Hhx9+2Hy6fWrDLiFqLuxc783BxERkXvIWwJM/NHUt6iQRhNqulnV3yM/Ph7OzM/Ly8uDkZOGm7rJCYL56jYT2pTEI9vPAltcHWLYOREREdyFjzt/NZAGJu5jOGBa2sBAREZkDA4upqnUJGbyWEBERETUYA4updAKLFVtYiIiIzICBxVRaQ4BUkKCMgYWIiKjRMbCYSudCYhIUKWq59gwRERE1CAOLqSoDiyBRH8rSchUqOI6FiIioUTGwmErTwiKpOpRFCmUTVYaIiOjexMBiqsrAIpFYwUYqAQAUlbFbiIiIqDExsJhKq4XFQa5eOLiY41iIiIgaFQOLqbQDi0wdWArL2CVERETUmBhYTKXTwiIFwC4hIiKixsbAYio9XUKFDCxERESNioHFVJqF4yQSOFYGFrawEBERNS4GFlPpGcPCac1ERESNi4HFVFqBxZ5jWIiIiMyCgcVUWoGFXUJERETmwcBiKg66JSIiMjsGFlPpaWEp5josREREjYqBxVTaY1hk6jEshVzploiIqFExsJhKT5cQx7AQERE1LgYWU3EdFiIiIrNjYDGVnjEsBaUMLERERI2JgcVUWoHF3VEGAMgqVDRhhYiIiO49DCym0gosno5yAEBOURmUKqEJK0VERHRvYWAxlVZgcXOQQSIBVAKQU8RWFiIiosbCwGIqrcBiLbWCm72mW6isCStFRER0b2FgMZVWYAEAj8puocwCBhYiIqLGwsBiqmqBxbOFOrCwhYWIiKjxMLCYSgwsEgCAhyO7hIiIiBobA4upxIXjdFtY2CVERETUeBhYTGVgDAvXYiEiImo8DCymMhhY2MJCRETUWBhYTGVg0C27hIiIiBpPgwLLV199hcDAQNja2iI8PBxHjhwxWDYmJgYSiUTnZmtrq1NGEATMmTMHvr6+sLOzQ2RkJC5fvtyQqlkeW1iIiIjMzujAsm7dOkRHR2Pu3Lk4ceIEwsLCEBUVhYyMDIOPcXJyQmpqqni7fv26zvYFCxbgyy+/xPLlyxEfHw8HBwdERUWhtLTU+FdkadUDSwv1LKHsIgUqlKqmqhUREdE9xejAsmjRIrz44ouYPHkyOnXqhOXLl8Pe3h4rV640+BiJRAIfHx/x5u3tLW4TBAGLFy/Gu+++i1GjRiE0NBQ//PADbt++jU2bNjXoRVlUtcDi7iCHlUQ9eSinmANviYiIGoNRgUWhUOD48eOIjIys2oGVFSIjIxEXF2fwcYWFhQgICIC/vz9GjRqFc+fOiduSk5ORlpams09nZ2eEh4cb3GdZWRny8/N1bk2mWmCRWkng5lC5FksBAwsREVFjMCqwZGVlQalU6rSQAIC3tzfS0tL0PqZ9+/ZYuXIl/vjjD/z0009QqVTo27cvbt68CQDi44zZ5/z58+Hs7Cze/P39jXkZjavawnGA1vL8HMdCRETUKMw+SygiIgITJkxAt27dMHDgQGzYsAGenp743//+1+B9zp49G3l5eeLtxo0bjVhjI1VbOA7QWp6fM4WIiIgahVGBxcPDA1KpFOnp6Tr3p6enw8fHp177sLGxQffu3XHlyhUAEB9nzD7lcjmcnJx0bk2mWpcQwJlCREREjc2owCKTydCzZ0/ExsaK96lUKsTGxiIiIqJe+1AqlThz5gx8fX0BAEFBQfDx8dHZZ35+PuLj4+u9zyalJ7BwLRYiIqLGZW3sA6KjozFx4kT06tULffr0weLFi1FUVITJkycDACZMmICWLVti/vz5AIAPPvgA9913H4KDg5Gbm4vPPvsM169fxwsvvABAPYNo+vTp+OijjxASEoKgoCC899578PPzw+jRoxvvlZqLnsCiGXTLWUJERESNw+jAMnbsWGRmZmLOnDlIS0tDt27dsG3bNnHQbEpKCqysqk7ed+7cwYsvvoi0tDS4urqiZ8+eOHToEDp16iSWeeutt1BUVISXXnoJubm56N+/P7Zt21ZjgblmSU9gaWGrPqwFpRVNUSMiIqJ7jkQQNKNG7175+flwdnZGXl6e5cezHF8NbH4daP8gMP4XAMCfp27j9V9OIqKNO3556T7L1oeIiOguYcz5m9cSMpW+FhZ5ZQtLWXlT1IiIiOiew8BiKj3rsLBLiIiIqHExsJhK7xgWGwAMLERERI2FgcVUehaOc6xsYSlkYCEiImoUDCymqmWWkEKpQmm5silqRUREdE9hYDGVnsDiKLMWh7SwW4iIiMh0DCym0hNYrKwkcJRVdguVMbAQERGZioHFVHoCC1A1jqWglFObiYiITMXAYioDgYVTm4mIiBoPA4up9KzDAnBqMxERUWNiYDGVoS4hObuEiIiIGgsDi6n0rMMCsEuIiIioMTGwmMrgGBZ2CRERETUWBhZTGQgsvs62AIDr2UWWrhEREdE9h4HFVAYCSydf9WWyz6fmW7pGRERE9xwGFlMZCCydW6oDy+WMQi7PT0REZCIGFlMZCCw+TrZwtbeBUiXgUnpBE1SMiIjo3sHAYioDgUUikaCznzMA4AK7hYiIiEzCwGIqAwvHAUBLFzsAQHp+mSVrREREdM9hYDGVgRYWAHBzlAEAcooUlqwRERHRPYeBxVQGFo4DAHcHdWDJZmAhIiIyCQOLqWprYakMLHcYWIiIiEzCwGKqegQWtrAQERGZhoHFVLUEFncHOQAgp4iDbomIiEzBwGKqeg66FTRjXYiIiMhoDCymqrWFRR1YypUCCsp4EUQiIqKGYmAxVS3rsNjaSGEvkwIAcgo5joWIiKihGFhMVUsLCwC42nPgLRERkakYWExVyzosAODOxeOIiIhMxsBiqjpaWDRTmzlTiIiIqOEYWExVz8DCLiEiIqKGY2AxVR2BRTNTiINuiYiIGq5BgeWrr75CYGAgbG1tER4ejiNHjhgs+91332HAgAFwdXWFq6srIiMja5SfNGkSJBKJzu2BBx5oSNUsr84WFs3icQwsREREDWV0YFm3bh2io6Mxd+5cnDhxAmFhYYiKikJGRobe8nv37sX48eOxZ88exMXFwd/fH8OHD8etW7d0yj3wwANITU0Vb7/88kvDXpGl1beFpZiBhYiIqKGMDiyLFi3Ciy++iMmTJ6NTp05Yvnw57O3tsXLlSr3lf/75Z7z66qvo1q0bOnTogO+//x4qlQqxsbE65eRyOXx8fMSbq6trw16RpdV70K1uYLmVW4JZv53G6Zu55qwdERHRPcGowKJQKHD8+HFERkZW7cDKCpGRkYiLi6vXPoqLi1FeXg43Nzed+/fu3QsvLy+0b98er7zyCrKzsw3uo6ysDPn5+Tq3JlPLwnFA1fL82ZVjWARBwJr4FPT7ZDfWHbuBhTsuWaSaREREdzNrYwpnZWVBqVTC29tb535vb29cvHixXvuYNWsW/Pz8dELPAw88gEcffRRBQUFISkrCO++8gxEjRiAuLg5SqbTGPubPn4958+YZU3XzqW+XUJECP8Zdw8W0AvwcnyJuz8gvNXsViYiI7nZGBRZTffLJJ1i7di327t0LW1tb8f5x48aJ/+/atStCQ0PRtm1b7N27F0OHDq2xn9mzZyM6Olr8OT8/H/7+/uatvCF1LByn6RIqKVfivT/OGdxOREREhhnVJeTh4QGpVIr09HSd+9PT0+Hj41PrYxcuXIhPPvkEO3bsQGhoaK1l27RpAw8PD1y5ckXvdrlcDicnJ51bk6mjhcVRbg0bqf7uIgAoVijNUSsiIqJ7ilGBRSaToWfPnjoDZjUDaCMiIgw+bsGCBfjwww+xbds29OrVq87nuXnzJrKzs+Hr62tM9ZpGHYFFIpHobUWRSdXlS8sZWIiIiOpi9Cyh6OhofPfdd1i9ejUuXLiAV155BUVFRZg8eTIAYMKECZg9e7ZY/tNPP8V7772HlStXIjAwEGlpaUhLS0NhYSEAoLCwEG+++SYOHz6Ma9euITY2FqNGjUJwcDCioqIa6WWaUR2BBQAe7dGqxphcTYhhCwsREVHdjA4sY8eOxcKFCzFnzhx069YNCQkJ2LZtmzgQNyUlBampqWL5b775BgqFAo8//jh8fX3F28KFCwEAUqkUp0+fxiOPPIJ27drh+eefR8+ePfHPP/9ALpc30ss0o3oEllkPdMDh2UMht64qo7koYglbWIiIiOrUoEG3U6dOxdSpU/Vu27t3r87P165dq3VfdnZ22L59e0Oq0TzUI7AAgLeTLYI8HHAxrQAA4O6oDmMlbGEhIiKqE68lZKp6BhYAcJBX5UMPrdlDgmamEREREenFwGKqOhaO02Yvq1pTRtMlpFQJUChVZqkaERHRvYKBxVR1rMOiTTuwaC6KCLBbiIiIqC4MLKYypktIVtUl5GRXtT6LZuDt0tjLePnHY6hgiwsREZEOi650e08yIrDYabWw2MuksLORolxZgWKFehzL5zvV1xU6fDUH/UM8zFJdIiKiuxFbWEzVwEG3djbWYoApUSiRrXU1ZyUH4RIREelgC4upjAgs9tVaWOxl1gDKsOHELfx2/Ia4jWNaiIiIdDGwmMqEwGJro/555cFknXL5JeWNVz8iIqJ7ALuETGVUYLHW+b92gNGWX1p7YCktVxpshckoKMXui+lQqditRERE9w4GFlMZsQ6Lg7x6l5D+wJJX2cKSlleKcj0zhh775hB6frQTvx+/WWPbI0sP4rmYY/jz1O361J6IiOiuwMBiKiPWYbGz0W5hqeoSqi6vpBynbuTivvmxmLrmhM42RYUK527no1ihxIxfT+GO1mBdAEjLLwUA7DyfbsyraFQqlYDfjt9EclZRk9WBiIjuLQwspjKiS0hqVdUKY1dLC0t+STn+tz8JALD9nG7wyKs2vuWfK1l692FlVXeLj7n8duImZv56CoMX7m2yOhAR0b2FgcVURgQW7QhhL7OGnVYLi9zaCj5OtgDUoeROkf5xLHklui0q+xIzxf8rtcatSJsur+BwUnbTPTkREd2TGFhMZURg0Sa1kugsJPdkL3+8/0gnAEB+aQVytVpStC+OWL2FJS6pqoXlTnFVmJHUY0yNuRgz3DenSIG5f5xl9xEREdWKgcVURgSW3kFuAIA2Hg4AgAA3e3FbiLcjnOxsAADHr9/BhdR8cVvshQw8+b847DiXhtxidWDxaqG+FlF6QZk4Iyi7sCqwFJSWo0Kpwgebz+OHuGsNfHHmN3/rBayOu44hn+9t6qoQEdWLIAjILChr6mr863AdFlMZEVic7Wxw5v3hkFurW1YmRAQiwN0BafmlGNO9JZIyC/U+bnHsJZy9lY8jyTl4slcrAECQhwMyCsqgVAnIKymHq4MM2YVVf0B3isux6uA1cY2XZ+8LsFiri3aLUIVSBWup4WOjec2CAJRVKMVjQ0TUXH341wWsPJiM/z3bE1GdfZq6Ov8abGExlZFdQi1sbSCzVpe1spJgcAcvjO/TGrY2UjjZ2uh9zNlbVa0t64+ppzJ7tpDDyVadNzXL+mdpzRi6U6TA8n1J4s+aCyxagnaXUGFZRa1lPRyrrlp97NodM9WIiKjxaL4Ifvr3RQDq8YMv/nAMn1T+TObBwGKqBo5h0cfZXn9g0VvWzgbulSd7TcuKdgvL1awinesT3ansSlp/9AZC39+Ow1frNzD20JUsfLs/SafVxBCVSsCM9afwR0LVGjD5JTUDy4JtF/G/yjCVqVXn+HrWyZzuFCkwYeUR/JFwq8Y2lUpoNlfSzisux/HrOfX6vdztVh1M1gnfRM3NsWs52Hk+Hcv3JXHRTjNiYDGVuA6L6d0tjjJrtPF0gJOtNc68Pxx927qL2+xlUthoTf1xsbeBu4MMQFULi/YYluruFClQWq7EW7+fRn5pBeb+cc5g2QqlCmdu5kEQBDz1fTw+3noRO8+nI7OgDGdv5YmL2f156jYGLNgtjrc5kXIHv5/QXcyu+qq9t3NL8PXeJMz/+yIKSst1+oFv55XWenwaqkShxMSVR7Bwe2KdZf+3/yr2X8rEtLUJOvcLgoDHlh/C8MX79S7mZ2nProzHY9/EYU9iRlNXxawKyyowb/N5fPL3RZ1AbimX0wvQ/9Pd+OVIisWe806RAjlFhv+WzalcqUJeMS8NUm+VH8llFVWfCbm8tIrZMLDURhCA4pzab6rKFoRGaGGxspJg89T++OetIWhhawNXe5m4rWeAK7q3dhV/VrewVAWWlOxiLNtzxeC+80rK8adWy8eVzEKxu6awrAKPfXMIM9afQmm5El/vTcLIZQcQc+iaWH7/5UwM/GwPHl56AK/8dAKCIOD1X07iRk4JPt9xCR9sPo/Hl8fVeN7qgUU7VF1KL9QJLOn55gksMYeuYd+lzFqPj4b2QnzarRdFCiVOpuTiamYRrmTUHGt09FoOPvn7IsoqLNP1dvpmHgDg12M1Vzu+22w5nYrj1/V3B966UyL+v6Engpt3irHp5K0GtUa9v/kcbt4pwewNZxr03MY6fv0Oeny0Ez0+3ImfDl+v9+N+Onwdz8ccRbGi9i7Yusz98xx6frQTF9Py6y5MIu2u74wC83yOEQfd1k5ZDiwIqmfhxhnQ6iCv+pW4aHURhbVyAQAcSc5Rb7OTwc2hqkvo7Q2na93vnWIFTt7IFX9WqgS8+espRLR1x9Frd3D8uvp25lYuLqWrT8jzNp8Xyx+4nIXiyusXHb6ajcT0AnFbWYWyxgUcNZ76Lh6//ScCvQLVM6QyC6v+mI9fz9H5ZpKRb55v0HsuVrVCFCsqdK7pVJ2tTVXwzClSiN1uOVpBq0jPuJwnKsOak501Xh0UbHKda6PdwnO39wglphVgSuVqztc+eajG9pt3isX/X8koxPS1CXiiVytMiAis1/4FQcDAz/ZCqRLgILfGsE7eRtVPX5emOR1JzhF/p9vOpuGZ+wJwOb0AKTnFGNpRf90FQcC7m84CADaevIWnwwMa9NwqlYA18eqWpJiD1/DJY6EN2g8ArIlPwbLdl7Fqch+092nR4P00d5pPfe0lJTLyy9DBDONwBUFo0uUqmgO2sDQG92DAo12j71Y7vER28kavwKoWFic7G3hUtrAs3nUZhyoXa1v30n3wdbatsa87xeXih//g9p6wtpLg77NpmPPHOWzWuu6QJqxUdy276sRRWFaB1Yeqvv3FV4YoQx5fHofPtl9EabkSWQVVf9gHruiOWUk3wzeT/NJynEip+vZ+NbOo1nEoBaVVJ6jrOVWvObuoKkzVNp3x7K28hla1Vnkl5Zi94TSOJOfgdm5Vq8O2c2n4Of465vxxFqsMhMaGKi1X4pcjKQ2evpmRX1rj0hHVXdWaGaeoqPl7uaX1Wj/5+yLO3MrDnFq6M6uLu5otLqh4+Go2Fu28hCW7Ltf78dorRltivNCt3Kr33Jlb6m7ZSauO4vnVx7D1TKrex6RqdaWaErCuaq2FJLfWPTWUliuxJj4FafXstt1y5jZu55Vix7m0BtenudL3PtB+n2eYYbrzmvgUhM3bYbAl8t+CLSy1kdoAc+rxBpFIGmUMS3Xa/dihLZ1RqNXcK5FAHMOi8VR4a4S3cUdoK2fxQ8yzhRyZBWXIK1YgpfIE/MqgYLw2NARf70nCrgtVS/8vf6YH/vOT7rWLDFl7tKpPX9+Jprqv9iTBUW4DldYf+8HKywp4OMqRVViG3OJyzN96AQeTsvBWVAfc386zxn6UKkHnEgd1OXI1BxVag+AeXnoAk/sFYu7IznrLaw8CXhp7GWdv5+PVQW3h72qvt0x1ZeU1j0VRWYVO+GyIL3Zewi9HbuCXIzfw4/N9dLb938az4v8jO3pjwfZETOobgJ4BbuL9KdnFKFep4Otsiy9jr2BgO0/klSjwfxvPYtlTPRChNV5K45WfjmNPYiaWOl9GoIcD5j/aFQHuDvWqb1FZBSI+2Q17mRSn5w43+M1Qu4Utu6gMvs52OttvanUJ1XdxwcyCMrSwtYatjRQbTlQNnr6Ylo+DlSF5fLg/vFrUDPa1ySsph4u9rO6C9aSoUOG5mKM4mXIHY3q0xIejuuh0geWVlONyRqEY2pbuvoIHu/rW2I+mexAwrTvihNbJ8HZeKV756TgqVAKWP9MTX++5gi93X0GIlyN2Rg+s8ViVSsDquGvoE+SGzn7OYsi9rKf7VONGjrqr7vkBQbW2etaltFxp8Lps1W0+dRvrj93AknHd4ebQsN9lkaKq21fzvs7RWpncHF1C72xUd0m+s+EMtr9xf43tJQolfj9xEyPD/OBsV//JG3cbtrDURiIBrKzqvpmpme6FAUFwkEnxfw92hJWVBE62NhjawQtOttboHegmLjQHAE+Ht8bckeqVct96oIN4f68AdatMVqFC/DBs7WaPHq1d8f3EXjjx3jC0crVDaCtnDO/kg1Hd/OpVt4Z82fx020V8pjXwVfPNt62ng9gV87/9V3H2Vj4mrDxS4w//x7hr6Dx3m06LkD6/HruBYYv2ITGtQO9sqFUHrxkcyZ+l1fWzJzETmQVlmLf5PDacrBorklXtG5R2F432Cfj49TtYf+wGury/3eAsl9JyZb1mHp3UaiW6kVNisNzghXux+dRtvPVbVRehUiXg/s/2YOjn+/Dyj8exfF8SXvvlBLacSUN2kUJnYcHSciUUFSpcTMvHnsrLPtzOK8WhpGx8+NeFOuupcTmjEEqVgILSCp3QoZFbrMC1rCJk6cwSy8Horw4iVitEa3cJadPulissq8CNyjB+JaMA/T7ZjRnrTwEArmmFnINaLXpX9LQkJqYVYMOJmzrfoHO1mvpv5zbOiai0XIkjyTn4dNtFHLiShSKFEj8dTsGNnJIaz6E9W+1CqvrvIqZaS9qZW7ni//W9NyqUKlzJKKyzhejY9aqW0iPJOfj7bBp2nk9HXFI2tlS27lzOKNQ7KHdTwi3M23weD315AEBVK4NmvJdKJeC7/VfFFpfbuSV4YPF+fL7zEr7df7XWegHqlrg/EmqOQ9p9MR0d52yr0WqWmldSY0xcUVkFXvvlJP65nIU18TXHBx2/noNhi/Zh/6Wqy52cvplb4zOkQGtcnuZvt3qXkLG+2nMFQz/fi+//uYoNJwyPSysyMEbpnY1n8O6ms5ixPgHnb+fj5/jrNY5Vfmm5zn1Hr+Xgh7hrOpd0ae7YwtKMdfBxwtl5UTr3fT+xF8oqVLC1kaKDjxMAwNpKgrkjO4vru7T1dMSPz/dBVmEZ0vLK8PfZNHEArczaSlwlFwDcHGTYM3MQrK0kkEgk+PSxUPQJckNucblOuNDo29Zd7H6SW1vpnKAbqlegK9LyS3E9W/fkdOpGHoZ1qvoWvGjnJZSWq/DaLyfRzd8F/lorBWt7s/Jk/cIPRw2ubXMhLR+d/Zxr3G+o+2Prmaqm7czCMpRVKJGRXwY/Fztx9WGgar2bI8k5ePJ/VYOQP/n7Iryd5Hg41A82lQvpZeSXInLRPvQJcseDXX0wIMQTni3kKCqrQE6RQuf1abcSndIai1SdplxSZhEKStXf0P20Wi3+uaxu1coqVCAlW30yP3AlCxVKFS5nFOLZFUfgZGeNcb39a+zb0DfH5KwifLb9IqYNbSeOV9DuOriUXlDjdzVh5RGcvpmHyI5e4n3T1yUAAJ5ffUwcz3JLT9gB1N0gwV6OAIDJq47g6LU7CA9yQ1svRyiUKmw5k4oLn+/F1Uz9rTKJ6QXoG+yhc1/U4v0AADsbKUZ09YUgCDonvbT8EnTyc9K7P0D9LXfH+TQkphVg7dEbGNXNDzKpFV4dFIxnV8ajd6Ab3nu4Ez7ddhGrDl6r8fikrKrWlD5BbjiSnIPfj+tOr99/KRP7L2Wib7AHgj0dceZWHvZqXU/sRk7NgPf13iQs2nkJH4/pirG9/fW2UKpUgs5+tC8BsvHkLThq/R2FfbAD3z7bE8O1FkxLTKsa07YnMUP8m0jKVAfXv07fxn+3qgPvknHddGbh7bmYgemRhrvUr2UVYcjn+wAAPk62CG9T1Rr4+/FbEATgi12XMKCdBwLc7CGztkLE/N1wd5Dh8DtDYSO1wp6LGZgcc1R8XJaeGZWPfaP+e5294QwOvj0E5UoVHll2EABw6O0h8HNR/x1pdxtrBttqt4afvZWH//x4HM8PCELvwKpWTn2KyioQl5QtftZ+tEV9jHoHusHfzR55JeX49dgNsbyh1uWNJ9Xvk10XMrDrgnrMnrOdDexspHCQW8NGKsETy+Pw0v1t8fYI9RfaSSuPoEihRGm5Ei/d3xaCIOB8aj4C3R1wMa0Ay/cl4cNRXeDtJEdOkQKOttZNvrAnA0szV70pXSKRiM2fnfycsP7lCLTxdBDDisaAEHV3ytpq0zEdZNIaV3K20VqJ1tZGiqfDAypPQuo/ot6BrjhauajbiK6+YmDp5u+iM37lzPvD8dPhFHy6TXfxpNgZAzG08gNHw8XeRvxQeyo8AEeT79QILB/8dQ7WVhLczC1BaEtncS0ZANh+Lg0vDGgjfmP453IW8krKEdqqKoSov22qTwAD23lin9Y3p7ikbHT2c0ZecTmyisrQ0sUOL/5wTOcbvyG7LmRg+7l05BQpMKlvIJ4Oby1u05zU/zxVcx2XN9adwtFrd/DhqC54LuaoWJ9dF9Kx60I6Qls548+p/TFj/SnsvJCOP6b0Q5eWzhAEQedEVN+pzE99F48zt/IwPTJE7/ZTlV0JBaUVOHb9DmasP4WswjJkFZbht+M1v+Vpwp9KJaBCJYjvuRd/OIYrGYW4kFqAPTMHAdBtGUlML9AZMJpfWi52Y2g+XKtbE5+CsgqlwW6g1LwSBHs5IrdYIb4345NzdFpzDIUVQB2itGmHsf2XMzG4gxfyS8tRqtXFl1rH+I3ZG05jk9ZMPE0ouZ1XitM383D6Zh4e7dFSJ6x4OMrRo7ULdpxPx6kbueIJcGSYH44k5yAtX7drV2NJ7GX4u9rXaLm7cae4xuDMRTsvAVB/C//k7wvYOm0AWrnqBsizt/OQUVAGayuJTjgG1ONRHKt1ac789RROd/bBb8dvorRciQKtFq/Jq6qCQVmFCjfvFOPHuKoWjcXVWkNScmrWWZv2F6eLaQUI8nSAjZUVXB1kOi0bj359CFYS4I3K8JNdpG5VDvRwwF+ndcf/pFQLdtezq94rpZVfOrRXHj9zK08rsFR9DmnCi3Y9jlV2rW07lwYXexusnNQbPbRmeGpcySjE4l2XatQNUIceH2dbvPzjMRy+WnOMYLlShTd/PYXCsgq8NkT/33fshQwxyHTwaQGVACzfl4S3R3RAUVmF2LX1+Y5LGN29JaavTcChpGw8FOqLLZV1Ki1XooWtNbaeSUNLFzvEzhhY7+43c2Bgucv1Cao9wVfvz6xvv32guz18nW2RmleKp8MDENnRG/YyKZ7o2Qo37xQj9kIGJvcLgtRKgkNJ2ZgYEYAWtjYY19sfKw5cFb/BuNrboK2nI96Maq/zwfPOiI5YsD0RD3TxRksXO/i72ePINd0/zBs5JTrfirSdvpmHJ5fHIadYgYe6+mJJrPpD0EbPZaqdbK3RpaWTTmCJOXQNSZlF2JeYgdT8UowK8xNbH6qTSHS7wLRPHH8k3MLwzlUn4/S8sspv5vqDz5r4FLT1dNSpi/ZrKlZUYFtls/m3+6/iy/HdkZZfivxS7WmT6n3PebgTEtMKsE7rG5i2M5UDgKufIPSZ//dFnQGu+gZfZxcpoFQJeHDJPyhXqbDupQh88Nd5sdk/OasIU9acwAePdNYJDpfSdMPBmZt1D0zW9NkbkppbisS0ghrltF+DhkxqhXY+jjorRu+/lIVx38ahZ4ArbueWIsijamzOkeQcDFiwp0Zrm3ar0e3cEvznp+OQW1uhs58z+gV76IQVbdpdmOO+Payzrb2PI9p6OQLn08X3n7uDDANDdMdvvTSgDXacT8Ol9ELklZSLJxRxv739se7YDRQrlBj2xX6EtnTG50+G6SweCagvrPrP5SyM76MO2ZqgoAmOwzp542RKrhiUAKC0XIXScn37ycTMX9Vdb9rHr7r9l7PEkzhQNRbpyV6t8Ovxm7hTXI6tZ9Lg2UKOCqUKrg4yPLviCMb38ceM4e1x7nbV++Vkyh18sesSWthaY8+MQTUuZ6ISgM8rAxqgHjgf6OEgltO0EFcPwpoTO6AOWaXlSpy/XfV+OXMzT1yCX7uFpaxChevZRTrjiLTlFpdj5vpT2F0Z5DW2nU2tdbzg+dR8/HMlq0ZYySxQf74cuJwlvt8Mhf6jWp+nF9N0Z3VqL+5ZVqFCn//Gij9rv7cSUnLFbqhbuSU4cDkLkUbOtGtMDCz3OO1Bgl1bOiN6WP1mM0kkErwZ1R6bT93G0I5eaKHVJDx7REfMHtERgDq5H76ajcd7qq9x5Oogw+HZQ1GuFPDr8RsY0kHd5D9lcDCGd/LGsC/Uze6h/s449m6kuM8nerUSF53r6Oukc/FHbT5OtkjLL8WfWicBTVgBgHJlzf7YYC9HnfoD6sGc2ouBVT/ZTO4XiFUHr2HK4LbYdylT52QHAP5udridW4o7xeU606YVShVyihQ4mZKrt/4A8OFf5w1u00wrBdQL82m6BvSJaOuOMH/nGoHFzkZa66UYega41phtoOlm8nCU6W0uB9RjKH47fkOc0v7094drBJstp1Ph62SrE1g2JdyGi70MbT0dsO7YDZ0uqoa6nlOEJbGX9QaU6pzsbNDN30Xnd3grtwS3ckv0fntNMtAyo2mxmfPHWfyg1WJw9NodnTWLaqN9srO2kiB6WHtxppTmd9LKzR7+brrH6L427njx/jYAgFd/Pi52Ubawtca0oSF4Krw1Em7k4mJaAa5kFOJKRiGe6x+Ea9k1X8vVzELsTczA7dxSfLHrEgq0WpKGdPCClUQijlkZ3N5THMtU3bMrjoj/r21A9G8GAvWckZ1xLbsYR5JzxKnt2pbuvgIfZ1udGYqav9Pc4nLEJ+cY/GKgcfpGLrr5u4jH+MUBbcTA8mXsZZy/nY/W7vbYrfU3XFhWgQ7vbUNvrVmZy/ZcwZgeLdHW01HndwhA7K4ypLCsAtezi5BTpBDX0nr9l4Qa5dp4OmBiRCDm/nkOMQevoaCsosaXpWKFEot3Xdb5zDNE39gxQP1703ym2NpYQaVSf27pU1BtCYed59MZWMh87mvjhteHhiCslbPBdRwMebRHKzzao1WtZQI9HBBY7duVtdQK1lLUWCujracjPBzlKFeqEOCm+5jwIDcEutvjWnYxpg4O1vsB1s3fBR+N7oKHlx6osa29dwvczisRP0wmRgRgdeVJJdjLUVxDpj4e7OqD/3uwI+4P8UREW3fYy6xrBJb/ju6KX4/fxOZTt/HdP7qDIJfuvmKwa6m1m32N5mhtmj5sDc36GtZWEgzr5I2/z1aNpfF3s0eZnmDydHhrfH9A/xTnoR280C/YQzw5Otla67TefDG2GyasPCJ+SH77bE+8u+ms2Koz6/eqFg1DU+C3nEmt0bKnfUKvfiy1ffZ4KFq6qscFvfqz+j2gaenT9s3eJGj3XDwc6is2rbf1dECQh4P4zbNYUYFu/q746bDxq9Xa2lihf7AHdl3IwJYzqZCvS8CGkzW7+/Rp6WKHCpVKPKl+MTYMfs52uJ5TjKjOPpBJrWAnk0L36lvqkCCRSDC+T2v8ciQFrwxqi65aXZ1vRXVAclYx/JxtMWdkJ3Hm1pTBwXjtl5NiuUeWHUA7b/WYomfuaw0riQQ/xF3H9nPp+P5Ast6B84M7eEFqVRVYxvVprRNYZo/ogNZu9njlZ/2tA7/9J6LGApKarsd+we7iwOeuLZ3hKLfGR6O74M3fThscl6U9A666p7+PN7hN4/Odl8QWF4lEHfw0Fmm1xADqljig6uR9tNq1zYZ+vg9tPB1qdDVqD1qN7OitM/MSUI8rG//tYaTml2L9yxEIqRxnVZ1XCzk6+qrHSGmCwhM9W4nXj9PQDivagbL637Ih2l+ARoW1BADxS09EG3fE1XKJlNiL6UbP1GxMDCz3OIlEUu9WFXOzspJg75uDUF6hqvygriKRSLDx1X64mFZQOcW2BwB1l5K/mz28nOTigC/tE9gnj3ZFRFt3uDvKsfHkLby36SyevS8A48Nbi4GljacjegW44svYyzpjZwB1i8Lq5/pg+7l02NpYYVLfQNjZSCGRqC9MCQCjuvmJ3Vl9At0wsL0n7m/niRa21vjr9O0aH/yGvm1PGdwW04a2w5WMQjjKrRG1eL9OS4iVBNA3YF8mtcLXT/eAv5u9GFhc7G3gKLeGo9wac0d2wrHrd8Sm3Ojh7dDS1Q4XU3W7i957uBOe7x+E69lF+KCylcdRbo1ADwecvpmHx3q0woAQT0S0qRpYPbyzD4Z18kbQ7K16XxMALH+mJzxbyLF41yX8czkLqXml4u/n91f64vDVbL0DuDWtZdrCg9zR2t1eZ5xAiHeLGoFFJahPQHJrK9hYWeHNqPaoUAqQSID/jukKNwcZAt/eAkD9rbSbf9UJf3pkCL6MvVzjWPcP9sDUIcEY9+1hWEmAHW/cjwB3BxSXKRH2wQ4AEMNKazd7rJzUCw8uOaD35NPSxQ4H3x6ChdsTxRWWu7Z0QbCXo86gUc3r0x478nCoeqbeuw91xBO9WqG7v4tO+UAPB/w9bUCN53yoqy92X8zA9nNpKFYooRKqugJGhvpBJQA/xF03GJhbyK3h4SjHoPZVA6H9Xe3F8O/mIMPLA9sCAJaO764TjgB1y17PgJpjNTQe7d4Ksx7ogHVHb2BMd/WJsp13C/zwXB8M/2IfcooUeLynP4rKKnRaUAH158AdA5cM0B5jVxtrKwnsZFL0CnDV6aIS69ejJdYerdkaZCOViC23hsZFudjbYHS3lnh9aAje2XBG7NYFdAfkztt8Dt2q/T41yipU6OTnpPN8o7u1RIC7Q42/HweZFPe1ccfSp7pj+b6rCG3pjP4hHth/KRNKlYBpaxMMtppoc7S1xqS+gdh6JhUtXe2w4PFQ3P/ZnhqfaU/0bIU9iRno6OuEO8UKnYvWWhIDC1mUo9waMPBed3WQieuBPBRac70JjWlDQ/DfrRcgk1ohspO3+MfzTHhr9Al0Q4iXI5Raf3Gu9jboFeiG9S9HINDDXqe/9ti7wwBA74whjVau9pjcLxDnb+fjh+f7iMGpe2tXLH+mJ17+8TgA4Pn+QbiRU4wd59XfsF4cEITt59Lx7H0B6N7aBd1bu0JqJRFnmuyaMRByaytczy5CkIcjUnKK8dPh6yhXqvBC/zYorVDCydYGcmsrBHo46Ex/1v5WN7lfECb3C8LI0DSEeDvCXmaNyf3UKzT/czlTvEaTZkFB7bVUrKVW+GJsN5xMycXoyintH4/pike/OYQRXdR99nWtrnl/Ow/Yy6zx4/Ph+P6fq/jv1gsQBPU4qK4tndEzwBXrjt7QOVFOGxqCSX0DMWnVEfEbOACxK6SFrQ36B3sgPjkbrw5qi38uZ0IQgKmDg/FUeGsoKtRjHZxsrSEI6jC8/NmeOvUa3c0PmxJu44X+QWjj4SjeP653a4wM84O9TL1Oi6aL7Jn7AuBsZ4M1L4ajQikg2EvdOuFsX3P1hxFdfRDs1QLbpg/A4l2Xa5xgNeOaItq6i4ElwF3/rDYnWxv879meeO2Xk+ja0lmc/eQgt9Y7WNMQKysJvhjbDYIgYMWBZJ3Wul6BbjWuT9S1pTMm9g1Ee+8WmL3xtDhY1c1BhnmPdEZqXik6+rbAOw91hLujHIPaV42r0f6/Rv8QD0gkEoS1csapm3k1gkT31i5o4+mI0MpVuzWc7Wyw+bX+KC5Tiq21LvY2Ot1u9jJrMbCEtXLGudv5qFAJCA9StyBXb23RF2LsKgeLfvV0D1zNLMJ9bdxQrhSw7tgNONvZ4MEuPvBxttUZ8+XVQo7Ds4fi9K08PPt9PPzd7HFeT3f18XeHia0Oy5/tidkbTuOXIzXDz9lb+QZbGDv5OsFRbo2FT4Rh2toEtHSxQ3gbd0S0dUdUZx/supCOH+OuY8HjoeinNcNN+wupZubWmiMpBsfkAVVrX43o4gN/N3vseXMQbG2kcJRb49DbQ/BHwm2cvZUntlp28HXCR2O6NPksIYlwD1zuNT8/H87OzsjLy4OTk+Fph3TvUFSooBKEWkesT197EnFXs7F9+v06Y3m+2nMFn21PxJtR7TFlsGnL6FcoVQj+v78BqL/hvjKordhlFf/OUHg7Gbc4WV00rQaA/qXsq1uy6zK+2KVu+t7wal/xBHj6Zi5mbziDdx7sqPPhp6FSCTqzyd7+/bTOt88PRnXGop2X0LetO75+WjcoJGUW4nZuCXoHuom/n3O38zB7wxm8Oqgt3B3l6BXgColEAkEQUK4UMOePs+gf4iG2LgDqvv/cYgVaudrj7zOpsJVJMVjr239disoqcOBKFga284StjRSX0wtwp7i8zoHq+jyy7IA4sNLNQYaNr/YVg9+Zm3kY/fVBPNq9Jd4Y1g5bz6Ti2YgAyK2lEAQByyrHYjzRq+ZUcW2FZRWwtbaCtbRmQGqIbWfT8OrPx8WprIIgoP2726BQqtAnyA3rXrrPpKXe1x1NwYVU9VpH9jIpVk3qA2d7G2QWlOGv07fxSJgfen60C4B6UO7uGQPr/Xw5RQp8tOU8/j6ThpJyJWY90AEXUvNxIuUONrzSF7kl5XBzkIlfVtq+s1UM8SPD/DA9MkScmfjbfyIw6/fTeGNYO533lz5lFerBtmO+PgQAeCTMD1+O7w5A3bVoZyPFczFHdbrJWrna4cCsITr7KVEoseVMKn49dkOcSTmpb6DY+jplcFu42svw0ZYLCA9yQxtPB7wV1QGulYvZnb2VBxd7mxqzuerr7zOpYredvUyKh0N90aO1K97ecAaju/lh3iNdcDWrUOf6dNXdKVKg+4c7AUDn8iqNzZjzNwML3dOqn3gBdcg4dzsfXVs619jWEJpBmGtfug/3tXHHHwm3IJNaYYSeVUlN9dR3h3EoKRuPdm+JRWO71VleqRLw7qYzuJZVjJjnejf4G1JpuRLZRQpEr0vAtewi7IoeCJm1FWRSq3/F9U0upxfg2/1XMX1YO7R0qTlouEShhK1N8zsWucUKONnaiO/zPxJuIeFGLt6K6lCjW7ahNKcQfa/92RXxOHH9Dn57pa84PsMYhWUV2HMxQz3mx9rK4PTns7fy8OuxG5ge2U486R+7lgNXBxnaejrWKF+XZbvVrWYxk/uI05k1FBUqHLiSiX8uZ2H90Ru1vrbr2UXYdSEDA9t5oq2nA36KT0ELubW4QOeZW3no4ONUY1kKU5UrVRi59AAKyyqwK1o9FVmpEpBw4w7CWrnUOxRfTi/Audv5GNXNz2zvbaPO30IDLFu2TAgICBDkcrnQp08fIT4+vtby69evF9q3by/I5XKhS5cuwpYtW3S2q1Qq4b333hN8fHwEW1tbYejQocKlS5fqXZ+8vDwBgJCXl9eQl0NkkgqlSkjPK7HIc2UVlAorD1wVcosUFnm+6hQVSkFRoWyS56a7T1m5UrhTVNbU1TCb5vy3UFpeIZQ34/ppGHP+NjrWrVu3DtHR0Zg7dy5OnDiBsLAwREVFISND/1zwQ4cOYfz48Xj++edx8uRJjB49GqNHj8bZs1WjvxcsWIAvv/wSy5cvR3x8PBwcHBAVFYXSUl6mm5o/qZUEXo3c9WOIu6Mck/sFwdnepu7CZmAjtdJZaJCoNjJrq0a9/lJz05z/FuTW0kbrXmwujO4SCg8PR+/evbFs2TIAgEqlgr+/P1577TW8/fbbNcqPHTsWRUVF+Ouvv8T77rvvPnTr1g3Lly+HIAjw8/PDjBkzMHPmTABAXl4evL29ERMTg3HjxtVZJ3YJERER3X2MOX8bFb8UCgWOHz+OyMiqBb+srKwQGRmJuLg4vY+Ji4vTKQ8AUVFRYvnk5GSkpaXplHF2dkZ4eLjBfZaVlSE/P1/nRkRERPcuowJLVlYWlEolvL11FyDz9vZGWlqa3sekpaXVWl7zrzH7nD9/PpydncWbv3/to++JiIjo7nZXdnDNnj0beXl54u3GDf1LPxMREdG9wajA4uHhAalUivR03aWH09PT4ePjo/cxPj4+tZbX/GvMPuVyOZycnHRuREREdO8yKrDIZDL07NkTsbFVK4WqVCrExsYiIiJC72MiIiJ0ygPAzp07xfJBQUHw8fHRKZOfn4/4+HiD+yQiIqJ/F6OX5o+OjsbEiRPRq1cv9OnTB4sXL0ZRUREmT54MAJgwYQJatmyJ+fPnAwCmTZuGgQMH4vPPP8dDDz2EtWvX4tixY/j2228BqBcbmj59Oj766COEhIQgKCgI7733Hvz8/DB69OjGe6VERER01zI6sIwdOxaZmZmYM2cO0tLS0K1bN2zbtk0cNJuSkgIrq6qGm759+2LNmjV499138c477yAkJASbNm1Cly5dxDJvvfUWioqK8NJLLyE3Nxf9+/fHtm3bYGtrmbUtiIiIqHnj0vxERETUJMy2DgsRERFRU2BgISIiomaPgYWIiIiaPQYWIiIiavaMniXUHGnGDfOaQkRERHcPzXm7PvN/7onAUlBQAAC8phAREdFdqKCgAM7OzrWWuSemNatUKty+fRstWrSARCJp1H3n5+fD398fN27c4JRpM+Jxthwea8vgcbYMHmfLMcexFgQBBQUF8PPz01nDTZ97ooXFysoKrVq1Mutz8JpFlsHjbDk81pbB42wZPM6W09jHuq6WFQ0OuiUiIqJmj4GFiIiImj0GljrI5XLMnTsXcrm8qatyT+Nxthwea8vgcbYMHmfLaepjfU8MuiUiIqJ7G1tYiIiIqNljYCEiIqJmj4GFiIiImj0GFiIiImr2GFiIiIio2WNgqcNXX32FwMBA2NraIjw8HEeOHGnqKt1V9u/fj5EjR8LPzw8SiQSbNm3S2S4IAubMmQNfX1/Y2dkhMjISly9f1imTk5ODp59+Gk5OTnBxccHzzz+PwsJCC76K5m/+/Pno3bs3WrRoAS8vL4wePRqJiYk6ZUpLSzFlyhS4u7vD0dERjz32GNLT03XKpKSk4KGHHoK9vT28vLzw5ptvoqKiwpIvpVn75ptvEBoaKq70GRERgb///lvczmNsHp988gkkEgmmT58u3sdj3Tjef/99SCQSnVuHDh3E7c3qOAtk0Nq1awWZTCasXLlSOHfunPDiiy8KLi4uQnp6elNX7a6xdetW4f/+7/+EDRs2CACEjRs36mz/5JNPBGdnZ2HTpk3CqVOnhEceeUQICgoSSkpKxDIPPPCAEBYWJhw+fFj4559/hODgYGH8+PEWfiXNW1RUlLBq1Srh7NmzQkJCgvDggw8KrVu3FgoLC8Uy//nPfwR/f38hNjZWOHbsmHDfffcJffv2FbdXVFQIXbp0ESIjI4WTJ08KW7duFTw8PITZs2c3xUtqlv78809hy5YtwqVLl4TExEThnXfeEWxsbISzZ88KgsBjbA5HjhwRAgMDhdDQUGHatGni/TzWjWPu3LlC586dhdTUVPGWmZkpbm9Ox5mBpRZ9+vQRpkyZIv6sVCoFPz8/Yf78+U1Yq7tX9cCiUqkEHx8f4bPPPhPvy83NFeRyufDLL78IgiAI58+fFwAIR48eFcv8/fffgkQiEW7dumWxut9tMjIyBADCvn37BEFQH1cbGxvh119/FctcuHBBACDExcUJgqAOl1ZWVkJaWppY5ptvvhGcnJyEsrIyy76Au4irq6vw/fff8xibQUFBgRASEiLs3LlTGDhwoBhYeKwbz9y5c4WwsDC925rbcWaXkAEKhQLHjx9HZGSkeJ+VlRUiIyMRFxfXhDW7dyQnJyMtLU3nGDs7OyM8PFw8xnFxcXBxcUGvXr3EMpGRkbCyskJ8fLzF63y3yMvLAwC4ubkBAI4fP47y8nKdY92hQwe0bt1a51h37doV3t7eYpmoqCjk5+fj3LlzFqz93UGpVGLt2rUoKipCREQEj7EZTJkyBQ899JDOMQX4fm5sly9fhp+fH9q0aYOnn34aKSkpAJrfcb4nrtZsDllZWVAqlTq/BADw9vbGxYsXm6hW95a0tDQA0HuMNdvS0tLg5eWls93a2hpubm5iGdKlUqkwffp09OvXD126dAGgPo4ymQwuLi46Zasfa32/C802Ujtz5gwiIiJQWloKR0dHbNy4EZ06dUJCQgKPcSNau3YtTpw4gaNHj9bYxvdz4wkPD0dMTAzat2+P1NRUzJs3DwMGDMDZs2eb3XFmYCG6x0yZMgVnz57FgQMHmroq96T27dsjISEBeXl5+O233zBx4kTs27evqat1T7lx4wamTZuGnTt3wtbWtqmrc08bMWKE+P/Q0FCEh4cjICAA69evh52dXRPWrCZ2CRng4eEBqVRaYzR0eno6fHx8mqhW9xbNcaztGPv4+CAjI0Nne0VFBXJycvh70GPq1Kn466+/sGfPHrRq1Uq838fHBwqFArm5uTrlqx9rfb8LzTZSk8lkCA4ORs+ePTF//nyEhYVhyZIlPMaN6Pjx48jIyECPHj1gbW0Na2tr7Nu3D19++SWsra3h7e3NY20mLi4uaNeuHa5cudLs3tMMLAbIZDL07NkTsbGx4n0qlQqxsbGIiIhowprdO4KCguDj46NzjPPz8xEfHy8e44iICOTm5uL48eNimd27d0OlUiE8PNzidW6uBEHA1KlTsXHjRuzevRtBQUE623v27AkbGxudY52YmIiUlBSdY33mzBmdgLhz5044OTmhU6dOlnkhdyGVSoWysjIe40Y0dOhQnDlzBgkJCeKtV69eePrpp8X/81ibR2FhIZKSkuDr69v83tONOoT3HrN27VpBLpcLMTExwvnz54WXXnpJcHFx0RkNTbUrKCgQTp48KZw8eVIAICxatEg4efKkcP36dUEQ1NOaXVxchD/++EM4ffq0MGrUKL3Tmrt37y7Ex8cLBw4cEEJCQjituZpXXnlFcHZ2Fvbu3aszPbG4uFgs85///Edo3bq1sHv3buHYsWNCRESEEBERIW7XTE8cPny4kJCQIGzbtk3w9PTkNFAtb7/9trBv3z4hOTlZOH36tPD2228LEolE2LFjhyAIPMbmpD1LSBB4rBvLjBkzhL179wrJycnCwYMHhcjISMHDw0PIyMgQBKF5HWcGljosXbpUaN26tSCTyYQ+ffoIhw8fbuoq3VX27NkjAKhxmzhxoiAI6qnN7733nuDt7S3I5XJh6NChQmJios4+srOzhfHjxwuOjo6Ck5OTMHnyZKGgoKAJXk3zpe8YAxBWrVollikpKRFeffVVwdXVVbC3txfGjBkjpKam6uzn2rVrwogRIwQ7OzvBw8NDmDFjhlBeXm7hV9N8Pffcc0JAQIAgk8kET09PYejQoWJYEQQeY3OqHlh4rBvH2LFjBV9fX0EmkwktW7YUxo4dK1y5ckXc3pyOs0QQBKFx22yIiIiIGhfHsBAREVGzx8BCREREzR4DCxERETV7DCxERETU7DGwEBERUbPHwEJERETNHgMLERERNXsMLERERNTsMbAQERFRs8fAQkRERM0eAwsRERE1e/8PqxEcccfWgzAAAAAASUVORK5CYII=\n"
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "PlCtBEBLMBkR"
      },
      "source": [
        "## Question 1: What is the maximum accuracy obtained on test set for GAT? (10 points)\n",
        "\n",
        "\n",
        "Running the training cell above will also save your best GAT model predictions as *CORA-Node-GAT.csv*.  \n",
        "\n",
        "When you sumbit your assignment, you will have to download this file and attatch it to your submission. As with the other colabs, please zip this file `CS224W_Colab4.ipynb` and the *.csv* file that's generated!\n"
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "iL1EsPll3-kU"
      },
      "execution_count": 84,
      "outputs": []
    }
  ]
}