From c5b4fb0806c1418be36068f7897a3eb68b3d2dc6 Mon Sep 17 00:00:00 2001
From: Junhao Liang <43094337+darkliang@users.noreply.github.com>
Date: Tue, 15 Nov 2022 17:25:57 +0800
Subject: [PATCH] Update modules.py

fix  at least one dimension spans across two contiguous subspaces
---
 opengait/modeling/modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/opengait/modeling/modules.py b/opengait/modeling/modules.py
index a5326f0..194ed53 100644
--- 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()