diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-04 15:15:26 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-04 15:15:26 +0200 |
| commit | 04c40f790e405ced6e6b90cf0a8aea268b9345c4 (patch) | |
| tree | d5e05ee09fa99ee8d56d5373bde18626274a1fdd /training/conf/experiment | |
| parent | d3afa310f77f47553586eeee58e3d3345a754e2c (diff) | |
Add char htr experiment, rename from ocr to htr, vqvae loss collapses
Diffstat (limited to 'training/conf/experiment')
| -rw-r--r-- | training/conf/experiment/htr_char.yaml | 12 | ||||
| -rw-r--r-- | training/conf/experiment/vqvae.yaml | 3 |
2 files changed, 14 insertions, 1 deletions
diff --git a/training/conf/experiment/htr_char.yaml b/training/conf/experiment/htr_char.yaml new file mode 100644 index 0000000..77126ae --- /dev/null +++ b/training/conf/experiment/htr_char.yaml @@ -0,0 +1,12 @@ +# @package _global_ + +defaults: + - override /mapping: characters + +criterion: + ignore_index: 3 + +network: + num_classes: 89 + pad_index: 3 + max_output_len: 682 diff --git a/training/conf/experiment/vqvae.yaml b/training/conf/experiment/vqvae.yaml index 13e5f34..699612e 100644 --- a/training/conf/experiment/vqvae.yaml +++ b/training/conf/experiment/vqvae.yaml @@ -8,6 +8,7 @@ defaults: trainer: max_epochs: 64 + gradient_clip_val: 0.25 datamodule: batch_size: 32 @@ -17,4 +18,4 @@ lr_scheduler: steps_per_epoch: 624 optimizer: - lr: 1.0e-2 + lr: 1.0e-3 |