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