polish(pu): polish visit_count_to_action_dist method, polish reset_ka… · opendilab/LightZero@bfeeaf5