Function printlog

PATH/core/utils.py:740–741

(*args, **kwargs)

#                      preserve_rng_state=preserve)

def printlog(*args, **kwargs):
    print(f"[rank {dist.get_rank()}]", *args, **kwargs)

def _max_by_axis(the_list):
    # type: (List[List[int]]) -> List[int]