From 04c40f790e405ced6e6b90cf0a8aea268b9345c4 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 4 Aug 2021 15:15:26 +0200
Subject: Add char htr experiment, rename from ocr to htr, vqvae loss collapses

---
 training/conf/experiment/htr_char.yaml | 12 ++++++++++++
 training/conf/experiment/vqvae.yaml    |  3 ++-
 2 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 training/conf/experiment/htr_char.yaml

(limited to 'training/conf/experiment')

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
-- 
cgit v1.2.3-70-g09d2