summaryrefslogtreecommitdiff
path: root/text_recognizer/data/iam_paragraphs.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-02 01:01:51 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-02 01:01:51 +0200
commite5eca28438cd17d436359f2c6eee0bb9e55d2a8b (patch)
treec8d77aa0a77d31f63dc2b81eee6d9c5f37e0e2f1 /text_recognizer/data/iam_paragraphs.py
parent04876c35a1eeea68794745f1ca113754c0d97757 (diff)
Fixed wandb callback bug, add word_pieces arg to iam paragraph
Diffstat (limited to 'text_recognizer/data/iam_paragraphs.py')
-rw-r--r--text_recognizer/data/iam_paragraphs.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py
index 7ba1077..0f3a2ce 100644
--- a/text_recognizer/data/iam_paragraphs.py
+++ b/text_recognizer/data/iam_paragraphs.py
@@ -39,6 +39,7 @@ class IAMParagraphs(BaseDataModule):
"""IAM handwriting database paragraphs."""
num_classes: int = attr.ib()
+ word_pieces: bool = attr.ib(default=False)
augment: bool = attr.ib(default=True)
train_fraction: float = attr.ib(default=0.8)
dims: Tuple[int, int, int] = attr.ib(