From 2eb7f5a1fc8233782cd71b9b733365c1cbd4187b Mon Sep 17 00:00:00 2001 From: Laughing-q <1185102784@qq.com> Date: Tue, 7 May 2024 15:41:22 +0800 Subject: [PATCH] use repconv --- ultralytics/nn/modules/block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):