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