diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index ac619c5e3..e0500e337 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -408,7 +408,7 @@ class C3k(C3): def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): super().__init__(c1, c2, n, shortcut, g, e) c_ = int(c2 * e) # hidden channels - self.m = nn.Sequential(*(Bottleneck(c_, c_, shortcut, g, k=(3, 3), e=1.0) for _ in range(n))) + self.m = nn.Sequential(*(RepBottleneck(c_, c_, shortcut, g, k=(3, 3), e=1.0) for _ in range(n))) class C3K(C3):