fix hrdae

nnaakkaaii · Jun 27, 2024 · 74859c4 · 74859c4
1 parent bc6298a
commit 74859c4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/hrdae/conf b/hrdae/conf
diff --git a/hrdae/models/networks/hr_dae.py b/hrdae/models/networks/hr_dae.py
@@ -85,6 +85,7 @@ def create_hrdae3d(out_channels: int, opt: HRDAE3dOption) -> nn.Module:
             motion_encoder,
             opt.activation,
             opt.aggregator,
+            opt.connection_aggregation,
             opt.debug_show_dim,
         )
     return HRDAE3d(
@@ -96,6 +97,7 @@ def create_hrdae3d(out_channels: int, opt: HRDAE3dOption) -> nn.Module:
         motion_encoder,
         opt.activation,
         opt.aggregator,
+        opt.connection_aggregation,
         opt.debug_show_dim,
     )
 

diff --git a/test/models/networks/test_hr_dae.py b/test/models/networks/test_hr_dae.py
@@ -98,6 +98,7 @@ def test_hierarchical_decoder2d():
             }
         ]
         * 2,
+        aggregation="concatenation",
         debug_show_dim=False,
     )
     x = net(x, cs)
@@ -127,6 +128,7 @@ def test_hierarchical_decoder3d():
             }
         ]
         * 2,
+        aggregation="concatenation",
         debug_show_dim=False,
     )
     x = net(x, cs)