Function save_on_master

SwissArmyTransformer/examples/mae/util/misc.py:211–213

(*args, **kwargs)

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


def init_distributed_mode(args):