mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

vit contiguous (#992)

Browse Source
This commit is contained in:
Yuan Haobo 2021-10-25 18:11:48 +08:00 committed by GitHub
parent e27017efa7
commit 93f8b142a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/models/backbones/vit.py
View File

@ -395,7 +395,7 @@ class VisionTransformer(BaseModule):
out = x out = x
B, _, C = out.shape B, _, C = out.shape
out = out.reshape(B, hw_shape[0], hw_shape[1], 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: if self.output_cls_token:
out = [out, x[:, 0]] out = [out, x[:, 0]]
outs.append(out) outs.append(out)