From dca7cabc2715962607d55902c0903665015b7482 Mon Sep 17 00:00:00 2001
From: Harald Scheidl <harald@newpc.com>
Date: Mon, 24 May 2021 18:24:38 +0200
Subject: [PATCH] use correct batch size

---
 src/dataloader_iam.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dataloader_iam.py b/src/dataloader_iam.py
index 5ce2f17..1ef6aa5 100644
--- a/src/dataloader_iam.py
+++ b/src/dataloader_iam.py
@@ -120,4 +120,4 @@ class DataLoaderIAM:
         gt_texts = [self.samples[i].gt_text for i in batch_range]
 
         self.curr_idx += self.batch_size
-        return Batch(imgs, gt_texts, self.batch_size)
+        return Batch(imgs, gt_texts, len(imgs))
-- 
GitLab