diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index 53a6e91de1..e054539c66 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -562,7 +562,7 @@ class RepBottleneck(nn.Module): def __init__(self, c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5): # ch_in, ch_out, shortcut, kernels, groups, expand super().__init__() c_ = int(c2 * e) # hidden channels - self.cv1 = RepConv(c1, c_, k[0], 1) + self.cv1 = Conv(c1, c_, k[0], 1) self.cv2 = Conv(c_, c2, k[1], 1, g=g) self.add = shortcut and c1 == c2