From 191aa5a5c080ed7bf7f6d0408aa1cac4295c57d2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 13 Sep 2022 18:44:24 +0200 Subject: Fix emnist datamodule --- text_recognizer/data/emnist_lines.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/data/emnist_lines.py') diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py index 2107d74..b33c9e1 100644 --- a/text_recognizer/data/emnist_lines.py +++ b/text_recognizer/data/emnist_lines.py @@ -71,7 +71,7 @@ class EMNISTLines(BaseDataModule): self.num_val = num_val self.num_test = num_test - self.emnist = EMNIST(mapping=self.mapping) + self.emnist = EMNIST() max_width = ( int(self.emnist.dims[2] * (self.max_length + 1) * (1 - self.min_overlap)) + IMAGE_X_PADDING -- cgit v1.2.3-70-g09d2