Function save_on_master

utils.py:427–429

(*args, **kwargs)

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


def setup_for_distributed(is_master):