Function save_on_master

util/misc.py:538–540

(*args, **kwargs)

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

def init_distributed_mode_ssc(args):
    init_dist('pytorch')