Function _load_world

accessory/model/multi_gpu_wrapper.py:138–141

(load_dict, load_world)

    return save_dict, save_world

def _load_world(load_dict, load_world):
    for key, val in load_dict.items():
        setattr(dist.distributed_c10d, key, val)
    dist.distributed_c10d._world = load_world

class MultiGpuWrapper:
    """