fix (#1564)

(cherry picked from commit 24798b0)
Deci-AI · Oct 26, 2023 · d2d9c18 · d2d9c18
1 parent 5879c46
commit d2d9c18
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/super_gradients/training/models/detection_models/csp_darknet53.py b/src/super_gradients/training/models/detection_models/csp_darknet53.py
@@ -210,4 +210,6 @@ def __init__(self, arch_params: HpmStruct):
             self._modules_list.append(nn.Linear(1024, self.num_classes))
 
     def forward(self, x):
-        return self._modules_list(x)
+        for module in self._modules_list:
+            x = module(x)
+        return x