diff --git a/openav/__init__.py b/openav/__init__.py
index 93482de..cdf7781 100644
--- a/openav/__init__.py
+++ b/openav/__init__.py
@@ -18,7 +18,7 @@
 __uri__ = "https://github.com/DmitryRyumin/openav"
 
 __version__ = "1.0"
-__release__ = __version__ + ".0-a19"
+__release__ = __version__ + ".0-a20"
 
 __author__ru__ = "Рюмин Дмитрий, Иванько Денис, Шилов Николай, Маркитантов Максим, Карпов Алексей"
 __author__en__ = "Dmitry Ryumin, Denis Ivanko, Nikolay Shilov, Maxim Markitantov, Alexey Karpov"
diff --git a/openav/modules/nn/models.py b/openav/modules/nn/models.py
index 5cb0fda..bbee71e 100644
--- a/openav/modules/nn/models.py
+++ b/openav/modules/nn/models.py
@@ -273,7 +273,7 @@ def __init__(self, shape_audio, shape_video, input_dim=512, h_u=512, h_f=64, n_c
         self.fc_audio = nn.Linear(shape_audio[1], h_f)
         self.feature_video = CNNLSTMPyTorch()
         self.fc_video = nn.Linear(shape_video[1], h_f)
-        self.fusion = Transformer(input_dim=input_dim, h_u=h_u, n_class=n_class, encoder_decoder=5)
+        self.fusion = Transformer(input_dim=input_dim, h_u=h_u, n_class=n_class, encoder_decoder=encoder_decoder)
 
     def forward(self, audio, video):
         audio_n = rearrange(audio, "b g c n l-> (b g) c n l")