diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:24:20 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 23:24:20 +0200 |
| commit | dedf8deb025ac9efdad5e9baf9165ef63d6829ff (patch) | |
| tree | 56b10fcaef479d8abe9b0e6c05e07ad5e02b9ab0 /text_recognizer/data/iam_extended_paragraphs.py | |
| parent | 532286b516b17d279c321358bf03dddc8adc8029 (diff) | |
Pre-commit fixes, optimizer loading fix
Diffstat (limited to 'text_recognizer/data/iam_extended_paragraphs.py')
| -rw-r--r-- | text_recognizer/data/iam_extended_paragraphs.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/text_recognizer/data/iam_extended_paragraphs.py b/text_recognizer/data/iam_extended_paragraphs.py index d2529b4..c144341 100644 --- a/text_recognizer/data/iam_extended_paragraphs.py +++ b/text_recognizer/data/iam_extended_paragraphs.py @@ -18,10 +18,16 @@ 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 |