From 6968572c1a21394b88a29f675b17b9698784a898 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 25 Aug 2023 23:19:39 +0200
Subject: Update training stuff

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

(limited to 'training/conf/network/conv_transformer.yaml')

diff --git a/training/conf/network/conv_transformer.yaml b/training/conf/network/conv_transformer.yaml
index 016adbb..1e03946 100644
--- a/training/conf/network/conv_transformer.yaml
+++ b/training/conf/network/conv_transformer.yaml
@@ -1,58 +1,58 @@
-_target_: text_recognizer.networks.ConvTransformer
+_target_: text_recognizer.network.ConvTransformer
 encoder:
-  _target_: text_recognizer.networks.image_encoder.ImageEncoder
+  _target_: text_recognizer.network.image_encoder.ImageEncoder
   encoder:
-    _target_: text_recognizer.networks.convnext.ConvNext
+    _target_: text_recognizer.network.convnext.ConvNext
     dim: 16
     dim_mults: [2, 4, 8]
     depths: [3, 3, 6]
     downsampling_factors: [[2, 2], [2, 2], [2, 2]]
   pixel_embedding:
-    _target_: "text_recognizer.networks.transformer.embeddings.axial.\
+    _target_: "text_recognizer.network.transformer.embeddings.axial.\
       AxialPositionalEmbeddingImage"
     dim: &hidden_dim 128
     axial_shape: [7, 128]
     axial_dims: [64, 64]
 decoder:
-  _target_: text_recognizer.networks.text_decoder.TextDecoder
+  _target_: text_recognizer.network.text_decoder.TextDecoder
   hidden_dim: *hidden_dim
   num_classes: 58
   pad_index: 3
   decoder:
-    _target_: text_recognizer.networks.transformer.Decoder
+    _target_: text_recognizer.network.transformer.Decoder
     dim: *hidden_dim
     depth: 10
     block:
-      _target_: text_recognizer.networks.transformer.decoder_block.DecoderBlock
+      _target_: text_recognizer.network.transformer.decoder_block.DecoderBlock
       self_attn:
-        _target_: text_recognizer.networks.transformer.Attention
+        _target_: text_recognizer.network.transformer.Attention
         dim: *hidden_dim
         num_heads: 12
         dim_head: 64
         dropout_rate: &dropout_rate 0.2
         causal: true
       cross_attn:
-        _target_: text_recognizer.networks.transformer.Attention
+        _target_: text_recognizer.network.transformer.Attention
         dim: *hidden_dim
         num_heads: 12
         dim_head: 64
         dropout_rate: *dropout_rate
         causal: false
       norm:
-        _target_: text_recognizer.networks.transformer.RMSNorm
+        _target_: text_recognizer.network.transformer.RMSNorm
         dim: *hidden_dim
       ff:
-        _target_: text_recognizer.networks.transformer.FeedForward
+        _target_: text_recognizer.network.transformer.FeedForward
         dim: *hidden_dim
         dim_out: null
         expansion_factor: 2
         glu: true
         dropout_rate: *dropout_rate
     rotary_embedding:
-      _target_: text_recognizer.networks.transformer.RotaryEmbedding
+      _target_: text_recognizer.network.transformer.RotaryEmbedding
       dim: 64
   token_pos_embedding:
-    _target_: "text_recognizer.networks.transformer.embeddings.fourier.\
+    _target_: "text_recognizer.network.transformer.embeddings.fourier.\
       PositionalEncoding"
     dim: *hidden_dim
     dropout_rate: 0.1
-- 
cgit v1.2.3-70-g09d2