diff --git a/demo/predictor.py b/demo/predictor.py index ad7c83e2f..49d76f81e 100644 --- a/demo/predictor.py +++ b/demo/predictor.py @@ -135,7 +135,7 @@ def __init__( show_mask_heatmaps=False, masks_per_dim=2, min_image_size=224, - weight_loading = None + weight_loading=None ): self.cfg = cfg.clone() self.model = build_detection_model(cfg)