diff --git a/lib/core/loss.py b/lib/core/loss.py index 7dbeaa925f..945eba8cd0 100644 --- a/lib/core/loss.py +++ b/lib/core/loss.py @@ -139,7 +139,7 @@ def _forward_impl(self, predictions, targets, shapes, model): liou_ll *= cfg.LOSS.LL_IOU_GAIN * self.lambdas[5] - if cfg.TRAIN.DET_ONLY or cfg.TRAIN.ENC_DET_ONLY or cfg.TRAIN.DET_ONLY: + if cfg.TRAIN.DET_ONLY or cfg.TRAIN.ENC_DET_ONLY: lseg_da = 0 * lseg_da lseg_ll = 0 * lseg_ll liou_ll = 0 * liou_ll