From 407815cf9ec9607cb9e4d17a5efb3bfb7b49fd9b Mon Sep 17 00:00:00 2001 From: Arun Date: Tue, 26 Nov 2024 15:33:23 +0530 Subject: [PATCH] Fix `forward_fuse` comment (#17714) Co-authored-by: Glenn Jocher --- ultralytics/nn/modules/conv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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))