Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix cuda bug

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11968695
This commit is contained in:
yuanzhi.zyz
2023-03-13 17:49:45 +08:00
committed by yuze.zyz
parent 7659b64cdc
commit 352f670bec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/models/cv/ocr_recognition/modules/vitstr.py
View File

@@ -41,7 +41,7 @@ class ViTSTR(VisionTransformer):
x = self.forward_features(x)
ap = x.view(x.shape[0] // 3, 3, 75, x.shape[2])
features_1d_concat = torch.ones(x.shape[0] // 3, 201,
x.shape[2]).cuda()
x.shape[2]).type_as(x)
features_1d_concat[:, :69, :] = ap[:, 0, :69, :]
features_1d_concat[:, 69:69 + 63, :] = ap[:, 1, 6:-6, :]
features_1d_concat[:, 69 + 63:, :] = ap[:, 2, 6:, :]