Function save_on_master

utils/utils.py:310–312

(*args, **kwargs)

def save_on_master(*args, **kwargs):
    if is_main_process():
        torch.save(*args, **kwargs)


def init_distributed_mode(args):