From 2c9066b685d41ef0ab5ea94e938b8a30b4123656 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 15 Jun 2022 00:15:40 +0200 Subject: Update configs --- training/conf/network/conv_transformer.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (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 54eb028..39c5c46 100644 --- a/training/conf/network/conv_transformer.yaml +++ b/training/conf/network/conv_transformer.yaml @@ -10,7 +10,7 @@ encoder: bn_momentum: 0.99 bn_eps: 1.0e-3 depth: 3 - out_channels: 128 + out_channels: *hidden_dim decoder: _target_: text_recognizer.networks.transformer.Decoder depth: 6 -- cgit v1.2.3-70-g09d2