summaryrefslogtreecommitdiff
path: root/training/conf/datamodule
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:11:11 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:11:11 +0200
commit0c578ab3e79c00389f6aab79329a7652f2ce6f22 (patch)
tree9b69c37a874e2e89fd4b2389f8c7a02bfc41cbb6 /training/conf/datamodule
parent7b8705f382b1642cf171cf7fcd01295104b9deef (diff)
Add new config files
Diffstat (limited to 'training/conf/datamodule')
-rw-r--r--training/conf/datamodule/iam_lines.yaml7
1 files changed, 7 insertions, 0 deletions
diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml
new file mode 100644
index 0000000..d3c2af8
--- /dev/null
+++ b/training/conf/datamodule/iam_lines.yaml
@@ -0,0 +1,7 @@
+_target_: text_recognizer.data.iam_lines.IAMLines
+batch_size: 8
+num_workers: 12
+train_fraction: 0.8
+augment: true
+pin_memory: false
+# word_pieces: true