summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/target_transform
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-25 22:32:10 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-25 22:32:10 +0200
commit41c3e99fe57874ba1855c893bf47087d474ec6b8 (patch)
tree2ba286209251f7ec8d7ab4cf061cffe5de86f398 /training/conf/datamodule/target_transform
parentda29c1cf4d062087f1b29dc9402ee6384203b690 (diff)
Updates to configs
Diffstat (limited to 'training/conf/datamodule/target_transform')
-rw-r--r--training/conf/datamodule/target_transform/word_piece.yaml3
-rw-r--r--training/conf/datamodule/target_transform/word_piece_iam.yaml3
-rw-r--r--training/conf/datamodule/target_transform/word_pieces.yaml3
3 files changed, 6 insertions, 3 deletions
diff --git a/training/conf/datamodule/target_transform/word_piece.yaml b/training/conf/datamodule/target_transform/word_piece.yaml
new file mode 100644
index 0000000..bf284fb
--- /dev/null
+++ b/training/conf/datamodule/target_transform/word_piece.yaml
@@ -0,0 +1,3 @@
+word_piece:
+ _target_: text_recognizer.data.transforms.word_piece.WordPiece
+ max_len: 451
diff --git a/training/conf/datamodule/target_transform/word_piece_iam.yaml b/training/conf/datamodule/target_transform/word_piece_iam.yaml
new file mode 100644
index 0000000..478987c
--- /dev/null
+++ b/training/conf/datamodule/target_transform/word_piece_iam.yaml
@@ -0,0 +1,3 @@
+word_piece:
+ _target_: text_recognizer.data.transforms.word_piece.WordPiece
+ max_len: 72
diff --git a/training/conf/datamodule/target_transform/word_pieces.yaml b/training/conf/datamodule/target_transform/word_pieces.yaml
deleted file mode 100644
index 8ace2af..0000000
--- a/training/conf/datamodule/target_transform/word_pieces.yaml
+++ /dev/null
@@ -1,3 +0,0 @@
-word_pieces:
- _target_: text_recognizer.data.transforms.word_pieces.WordPieces
- max_len: 451