diff --git a/src/super_gradients/training/models/detection_models/csp_darknet53.py b/src/super_gradients/training/models/detection_models/csp_darknet53.py index a773fdb36e..3223560702 100755 --- 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