Update audiovisual.py

DmitryRyumin · Apr 26, 2024 · 2fc2430 · 2fc2430
1 parent c6d1804
commit 2fc2430
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/openav/modules/lab/audiovisual.py b/openav/modules/lab/audiovisual.py
@@ -260,10 +260,10 @@ def train_audiovisual(
             shape_video = (
                 None,
                 max_segment,
-                shape_video[SHAPE_AUDIO[0]],
-                shape_video[SHAPE_AUDIO[1]],
-                shape_video[SHAPE_AUDIO[3]],
-                shape_video[SHAPE_AUDIO[2]],
+                shape_video[SHAPE_VIDEO[0]],
+                shape_video[SHAPE_VIDEO[1]],
+                shape_video[SHAPE_VIDEO[3]],
+                shape_video[SHAPE_VIDEO[2]],
             )
 
             # Информационное сообщение