From a33cbbf039f20aa75ad6d77a2e6f85b5e6d6a535 Mon Sep 17 00:00:00 2001
From: Yuan Haobo <yuanhaobo@whu.edu.cn>
Date: Mon, 25 Oct 2021 18:11:48 +0800
Subject: [PATCH] vit contiguous (#992)

---
 mmseg/models/backbones/vit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmseg/models/backbones/vit.py b/mmseg/models/backbones/vit.py
index 9c099d2e..5cd3ff24 100644
--- a/mmseg/models/backbones/vit.py
+++ b/mmseg/models/backbones/vit.py
@@ -395,7 +395,7 @@ class VisionTransformer(BaseModule):
                     out = x
                 B, _, C = out.shape
                 out = out.reshape(B, hw_shape[0], hw_shape[1],
-                                  C).permute(0, 3, 1, 2)
+                                  C).permute(0, 3, 1, 2).contiguous()
                 if self.output_cls_token:
                     out = [out, x[:, 0]]
                 outs.append(out)
-- 
GitLab