summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--training/conf/experiment/conv_transformer_paragraphs.yaml7
1 files changed, 6 insertions, 1 deletions
diff --git a/training/conf/experiment/conv_transformer_paragraphs.yaml b/training/conf/experiment/conv_transformer_paragraphs.yaml
index d3311fd..e0d572a 100644
--- a/training/conf/experiment/conv_transformer_paragraphs.yaml
+++ b/training/conf/experiment/conv_transformer_paragraphs.yaml
@@ -9,12 +9,17 @@ defaults:
- override /lr_scheduler: null
- override /optimizer: null
+tags: [paragraphs]
epochs: &epochs 600
num_classes: &num_classes 58
ignore_index: &ignore_index 3
max_output_len: &max_output_len 682
summary: [[1, 1, 576, 640], [1, 682]]
+logger:
+ wandb:
+ tags: ${tags}
+
criterion:
ignore_index: *ignore_index
label_smoothing: 0.05
@@ -65,7 +70,7 @@ network:
encoder:
depth: 6
decoder:
- depth: 5
+ depth: 4
pixel_embedding:
shape: [18, 80]