diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-02 01:53:55 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-02 01:53:55 +0200 |
| commit | cf42f987a34d4de0db10f733fd532f949dd9c278 (patch) | |
| tree | 1420cc62d2792a3921272f4a5aa39e40ef7f7ce9 /training/conf/experiment | |
| parent | abc2d60d69d115cdb34615d8bcb6c03ab6357141 (diff) | |
Update configs
Diffstat (limited to 'training/conf/experiment')
| -rw-r--r-- | training/conf/experiment/vit_lines.yaml | 14 |
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 |