def is_master()

in train.py [0:0]


def is_master():
    return dist.get_rank() == 0 if is_dist() else True