From 58ae7154aa945cfe5a46592cc1dfb28f0a4e51b3 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 1 May 2021 23:53:50 +0200
Subject: Working on new attention module

---
 notebooks/03-look-at-iam-paragraphs.ipynb | 35 +++++++------------------------
 1 file changed, 7 insertions(+), 28 deletions(-)

(limited to 'notebooks/03-look-at-iam-paragraphs.ipynb')

diff --git a/notebooks/03-look-at-iam-paragraphs.ipynb b/notebooks/03-look-at-iam-paragraphs.ipynb
index add0b80..67456f9 100644
--- a/notebooks/03-look-at-iam-paragraphs.ipynb
+++ b/notebooks/03-look-at-iam-paragraphs.ipynb
@@ -56,8 +56,8 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "2021-04-25 23:17:44.177 | INFO     | text_recognizer.data.iam_paragraphs:setup:107 - Loading IAM paragraph regions and lines for None...\n",
-      "2021-04-25 23:18:00.750 | INFO     | text_recognizer.data.iam_synthetic_paragraphs:setup:79 - IAM Synthetic dataset steup for stage None\n"
+      "2021-05-01 20:11:32.991 | INFO     | text_recognizer.data.iam_paragraphs:setup:107 - Loading IAM paragraph regions and lines for None...\n",
+      "2021-05-01 20:11:49.406 | INFO     | text_recognizer.data.iam_synthetic_paragraphs:setup:79 - IAM Synthetic dataset steup for stage None\n"
      ]
     },
     {
@@ -68,9 +68,9 @@
       "Num classes: 84\n",
       "Dims: (1, 576, 640)\n",
       "Output dims: (682, 1)\n",
-      "Train/val/test sizes: 19948, 262, 231\n",
-      "Train Batch x stats: (torch.Size([1, 1, 576, 640]), torch.float32, tensor(0.), tensor(0.0109), tensor(0.0499), tensor(0.8314))\n",
-      "Train Batch y stats: (torch.Size([1, 682]), torch.int64, tensor(1), tensor(83))\n",
+      "Train/val/test sizes: 19925, 262, 231\n",
+      "Train Batch x stats: (torch.Size([1, 1, 576, 640]), torch.float32, tensor(0.), tensor(0.0064), tensor(0.0435), tensor(0.9176))\n",
+      "Train Batch y stats: (torch.Size([1, 682]), torch.int64, tensor(1), tensor(78))\n",
       "Test Batch x stats: (torch.Size([1, 1, 576, 640]), torch.float32, tensor(0.), tensor(0.0372), tensor(0.0767), tensor(0.8118))\n",
       "Test Batch y stats: (torch.Size([1, 682]), torch.int64, tensor(1), tensor(83))\n",
       "\n"
@@ -123,7 +123,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 4,
    "id": "0cf22683",
    "metadata": {},
    "outputs": [],
@@ -131,27 +131,6 @@
     "x, y = dataset.data_train[1]"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "id": "af7747a8",
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "torch.Size([682])"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "y.shape"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 7,
@@ -424,7 +403,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "25a074df",
+   "id": "4150722e",
    "metadata": {},
    "outputs": [],
    "source": []
-- 
cgit v1.2.3-70-g09d2