xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Conv2() fusing (#2885)

Browse Source
This commit is contained in:
Glenn Jocher 2023-05-29 02:08:45 +02:00 committed by GitHub
parent 882dbe62ad
commit a2bb42dfe9
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
ultralytics/nn/modules/conv.py
View File

@ -59,7 +59,7 @@ class Conv2(Conv):
"""Fuse parallel convolutions.""" """Fuse parallel convolutions."""
w = torch.zeros_like(self.conv.weight.data) w = torch.zeros_like(self.conv.weight.data)
i = [x // 2 for x in w.shape[2:]] i = [x // 2 for x in w.shape[2:]]
w[:, :, i[0] - 1:i[0], i[1] - 1:i[1]] = self.cv2.weight.data.clone() w[:, :, i[0]:i[0] + 1, i[1]:i[1] + 1] = self.cv2.weight.data.clone()
self.conv.weight.data += w self.conv.weight.data += w
self.__delattr__('cv2') self.__delattr__('cv2')