From 4bba371173c8f6bcc09c55cb5a34fa9fb704cb8f Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 20 Jun 2022 01:12:42 +0200
Subject: Add tags

---
 training/conf/experiment/conv_transformer_paragraphs.yaml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'training/conf/experiment')

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