Merge branch 'master' into feature/SG-1179_unify_recipe_script

Deci-AI · Oct 23, 2023 · eaadf73 · eaadf73
2 parents 3ea4d29 + 24798b0
commit eaadf73
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