Browse Source

Update block.py

Mengqi Lei 2 months ago
parent
commit
fa04b90884
1 changed files with 1 additions and 2 deletions
  1. 1 2
      ultralytics/nn/modules/block.py

+ 1 - 2
ultralytics/nn/modules/block.py

@@ -1604,7 +1604,6 @@ class HyperACE(nn.Module):
                     
                     
     def forward(self, X):
     def forward(self, X):
         x = self.fuse(X)
         x = self.fuse(X)
-        """Forward pass through C2f layer."""
         y = list(self.cv1(x).chunk(3, 1))
         y = list(self.cv1(x).chunk(3, 1))
         out1 = self.branch1(y[1])
         out1 = self.branch1(y[1])
         out2 = self.branch2(y[1])
         out2 = self.branch2(y[1])
@@ -1631,4 +1630,4 @@ class FullPAD_Tunnel(nn.Module):
         self.gate = nn.Parameter(torch.tensor(0.0))
         self.gate = nn.Parameter(torch.tensor(0.0))
     def forward(self, x):
     def forward(self, x):
         out = x[0] + self.gate * x[1]
         out = x[0] + self.gate * x[1]
-        return out
+        return out