We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
Conv2()
1 parent 882dbe6 commit a2bb42dCopy full SHA for a2bb42d
ultralytics/nn/modules/conv.py
@@ -59,7 +59,7 @@ def fuse_convs(self):
59
"""Fuse parallel convolutions."""
60
w = torch.zeros_like(self.conv.weight.data)
61
i = [x // 2 for x in w.shape[2:]]
62
- 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()
63
self.conv.weight.data += w
64
self.__delattr__('cv2')
65
0 commit comments