diff --git a/object-locator/losses.py b/object-locator/losses.py index 27cd65d..eb08c82 100644 --- a/object-locator/losses.py +++ b/object-locator/losses.py @@ -226,7 +226,6 @@ def forward(self, prob_map, gt, orig_sizes): if self.return_2_terms: res = terms_1.mean(), terms_2.mean() - print(res) else: res = terms_1.mean() + terms_2.mean()