fix (#1564)

Deci-AI · Oct 23, 2023 · 24798b0 · 24798b0
1 parent f153c61
commit 24798b0
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