summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-20 01:12:42 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-20 01:12:42 +0200
commit4bba371173c8f6bcc09c55cb5a34fa9fb704cb8f (patch)
treec9045f52b61742ac166f1f899b3b4cf6c9007ab4 /training
parent687e3c829d1b50a2f3bf96bafc0b29cff02e30d4 (diff)
Add tags
Diffstat (limited to 'training')
-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]