update: 适配昇腾

ultralytics/nn/modules/block.py

@@ -185,7 +185,10 @@ class SPPF(nn.Module):
     def forward(self, x):
         """Forward pass through Ghost Convolution block."""
         y = [self.cv1(x)]
-        y.extend(self.m(y[-1]) for _ in range(3))
+        # y.extend(self.m(y[-1]) for _ in range(3))
+        for _ in range(3):
+            o1 = self.m(y[-1])
+            y.extend(o1.unsqueeze(0))
         return self.cv2(torch.cat(y, 1))
 
 
@@ -236,7 +239,11 @@ class C2f(nn.Module):
     def forward(self, x):
         """Forward pass through C2f layer."""
         y = list(self.cv1(x).chunk(2, 1))
-        y.extend(m(y[-1]) for m in self.m)
+        # y.extend(m(y[-1]) for m in self.m)
+        # 该条代码在torch和dynamo中存在逻辑分歧，改为外部循环表示
+        for m in self.m:
+            o1 = m(y[-1])
+            y.extend(o1.unsqueeze(0))
         return self.cv2(torch.cat(y, 1))
 
     def forward_split(self, x):