use repconv

7 months ago · 2eb7f5a1fc
parent 0f99a40993
commit 2eb7f5a1fc
1 changed files with 1 additions and 1 deletions
--- 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):