diff --git a/mmengine/visualization/visualizer.py b/mmengine/visualization/visualizer.py index 0e4c1b50df..3b31c67727 100644 --- a/mmengine/visualization/visualizer.py +++ b/mmengine/visualization/visualizer.py @@ -961,7 +961,7 @@ def draw_featmap(featmap: torch.Tensor, if topk <= 0, tensor_chw is assert to be one or three. Defaults to 20. arrangement (Tuple[int, int]): The arrangement of featmap when - channel_reduction is not None and topk > 0. Defaults to (4, 5). + channel_reduction is None and topk > 0. Defaults to (4, 5). resize_shape (tuple, optional): The shape to scale the feature map. Defaults to None. alpha (Union[int, List[int]]): The transparency of featmap.