diff --git a/ultralytics/nn/modules/conv.py b/ultralytics/nn/modules/conv.py index aaa70f57..94c5b6c6 100644 --- a/ultralytics/nn/modules/conv.py +++ b/ultralytics/nn/modules/conv.py @@ -50,7 +50,7 @@ class Conv(nn.Module): return self.act(self.bn(self.conv(x))) def forward_fuse(self, x): - """Perform transposed convolution of 2D data.""" + """Apply convolution and activation without batch normalization.""" return self.act(self.conv(x))