From 521f3bbbba9f04f48e81d78033c6e1c29a08e515 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 20 Jun 2022 00:09:20 +0200
Subject: Update eff net config

---
 training/conf/network/conv_transformer.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'training/conf/network')

diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml
index ccdf960..90c2cb8 100644
--- a/training/conf/network/conv_transformer.yaml
+++ b/training/conf/network/conv_transformer.yaml
@@ -9,7 +9,7 @@ encoder:
   stochastic_dropout_rate: 0.2
   bn_momentum: 0.99
   bn_eps: 1.0e-3
-  depth: 3
+  depth: 5
   out_channels: *hidden_dim
   stride: [2, 1]
 decoder:
@@ -47,4 +47,4 @@ decoder:
 pixel_embedding:
   _target_: text_recognizer.networks.transformer.AxialPositionalEmbedding
   dim: *hidden_dim
-  shape: [17, 78]
+  shape: [18, 78]
-- 
cgit v1.2.3-70-g09d2