Function save_on_master

SwissArmyTransformer/examples/yolos/util/misc.py:401–403

(*args, **kwargs)

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


def init_distributed_mode(args):