[visualizer.draw_featmap] RuntimeError: "topk_cpu" not implemented for 'Half'

[vansin]

when I visualize FPN feature map，failed

from mmengine.visualization import Visualizer
visualizer = Visualizer()
drawn_img = visualizer.draw_featmap(x[0][0], channel_reduction='select_max')

[vansin]

.float() can save this problem

drawn_img = visualizer.draw_featmap(x[0][0].float(), channel_reduction='select_max')