Update modules.py

fix at least one dimension spans across two contiguous subspaces
cocoJennie · Nov 15, 2022 · c5b4fb0 · c5b4fb0
1 parent 6e843db
commit c5b4fb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opengait/modeling/modules.py b/opengait/modeling/modules.py
@@ -44,7 +44,7 @@ def forward(self, x, *args, **kwargs):
         """
         n, c, s, h, w = x.size()
         x = self.forward_block(x.transpose(
-            1, 2).view(-1, c, h, w), *args, **kwargs)
+            1, 2).reshape(-1, c, h, w), *args, **kwargs)
         output_size = x.size()
         return x.reshape(n, s, *output_size[1:]).transpose(1, 2).contiguous()