From 67ad629cf9ee52521330a350d8f66b80b6bb7b8b Mon Sep 17 00:00:00 2001 From: dnth Date: Fri, 28 Jan 2022 12:42:40 +0800 Subject: [PATCH] remove redundant statement --- icevision/models/mmdet/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/icevision/models/mmdet/utils.py b/icevision/models/mmdet/utils.py index d1392ff22..099acec55 100644 --- a/icevision/models/mmdet/utils.py +++ b/icevision/models/mmdet/utils.py @@ -81,9 +81,9 @@ def param_groups(model): layers += [model.bbox_head] # YOLACT has mask_head and segm_head - if getattr(model, "mask_head", False): + if getattr(model, "mask_head"): layers += [model.mask_head] - if getattr(model, "segm_head", False): + if getattr(model, "segm_head"): layers += [model.segm_head] elif isinstance(model, TwoStageDetector):