UltralyticsAssistant 9 months ago
parent 15a92f214b
commit d4e45dcaeb
  1. 4
      ultralytics/nn/modules/block.py

@ -559,7 +559,9 @@ class RepBottleneck(nn.Module):
"""Rep bottleneck."""
def __init__(self, c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5):
"""Initializes a RepBottleneck module with customizable in/out channels, shortcut option, groups and expansion ratio."""
"""Initializes a RepBottleneck module with customizable in/out channels, shortcut option, groups and expansion
ratio.
"""
super().__init__()
c_ = int(c2 * e) # hidden channels
self.cv1 = RepConv(c1, c_, k[0], 1)

Loading…
Cancel
Save