summaryrefslogtreecommitdiff
path: root/training/conf/experiment
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/experiment')
-rw-r--r--training/conf/experiment/vit_lines.yaml14
1 files changed, 6 insertions, 8 deletions
diff --git a/training/conf/experiment/vit_lines.yaml b/training/conf/experiment/vit_lines.yaml
index e2ddebf..f57eead 100644
--- a/training/conf/experiment/vit_lines.yaml
+++ b/training/conf/experiment/vit_lines.yaml
@@ -10,7 +10,7 @@ defaults:
- override /optimizer: null
tags: [lines, vit]
-epochs: &epochs 64
+epochs: &epochs 256
ignore_index: &ignore_index 3
# summary: [[1, 1, 56, 1024], [1, 89]]
@@ -56,7 +56,7 @@ lr_scheduler:
monitor: val/cer
datamodule:
- batch_size: 8
+ batch_size: 16
train_fraction: 0.95
network:
@@ -95,7 +95,7 @@ network:
dim: *dim
max_length: 89
use_l2: true
- tie_embeddings: false
+ tie_embeddings: true
pad_index: 3
model:
@@ -105,9 +105,7 @@ trainer:
fast_dev_run: false
gradient_clip_val: 1.0
max_epochs: *epochs
- accumulate_grad_batches: 1
- limit_val_batches: .02
- limit_test_batches: .02
+ accumulate_grad_batches: 4
limit_train_batches: 1.0
- # limit_val_batches: 1.0
- # limit_test_batches: 1.0
+ limit_val_batches: 1.0
+ limit_test_batches: 1.0