Summary

DmitryRyumin · Jan 31, 2025 · db818d2 · db818d2
1 parent d6310d6
commit db818d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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")