From 8afa8e1c6e9623b0dea86236da04b2b4173e9443 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 7 Apr 2021 22:12:10 +0200 Subject: Fixed typing and typos, train script load config, reformatted --- text_recognizer/data/iam_extended_paragraphs.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) (limited to 'text_recognizer/data/iam_extended_paragraphs.py') diff --git a/text_recognizer/data/iam_extended_paragraphs.py b/text_recognizer/data/iam_extended_paragraphs.py index c144341..d2529b4 100644 --- a/text_recognizer/data/iam_extended_paragraphs.py +++ b/text_recognizer/data/iam_extended_paragraphs.py @@ -18,16 +18,10 @@ class IAMExtendedParagraphs(BaseDataModule): super().__init__(batch_size, num_workers) self.iam_paragraphs = IAMParagraphs( - batch_size, - num_workers, - train_fraction, - augment, + batch_size, num_workers, train_fraction, augment, ) self.iam_synthetic_paragraphs = IAMSyntheticParagraphs( - batch_size, - num_workers, - train_fraction, - augment, + batch_size, num_workers, train_fraction, augment, ) self.dims = self.iam_paragraphs.dims -- cgit v1.2.3-70-g09d2