Function init

python/fedml/fa/__init__.py:8–31

(args=None)

def init(args=None):
    print(f"args={args}")
    if args is None:
        args = load_arguments(training_type=None, comm_backend=None)

    """Initialize FedML Engine."""
    collect_env(args)
    fedml._global_training_type = args.training_type
    fedml._global_comm_backend = args.backend

    if args.training_type == FEDML_TRAINING_PLATFORM_SIMULATION and hasattr(args, "backend") and args.backend == "sp":
        args = init_simulation_sp(args)
    elif args.training_type == FEDML_TRAINING_PLATFORM_CROSS_SILO:
        args = init_cross_silo(args)
    else:
        raise Exception("no such setting: training_type = {}, backend = {}".format(args.training_type, args.backend))

    fedml._update_client_id_list(args)
    mlops.init(args)
    logging.info("args.rank = {}, args.worker_num = {}".format(args.rank, args.worker_num))
    fedml._update_client_specific_args(args)
    fedml._print_args(args)

    return args

def manage_mpi_args(args):
    if hasattr(args, "backend") and args.backend == "MPI":