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