Coggle requires JavaScript to display documents.
elif args.training_type == FEDML_TRAINING_PLATFORM_CROSS_SILO:
def _init_cross_silo_runner(self, args, device, dataset, model, client_trainer=None, server_aggregator=None):
if args.scenario == "horizontal":
from .cross_silo import Client from .cross_silo import Server
if args.role == "client":
runner = Client(args, device, dataset, model, client_trainer)
if args.federated_optimizer == "FedAvg":
client_initializer.init_client(args,device,args.comm,args.rank,args.worker_num,model,train_data_num,train_data_local_num_dict,train_data_local_dict,test_data_local_dict,model_trainer,)
trainer_dist_adapter = get_trainer_dist_adapter(
if args.scenario == FEDML_CROSS_SILO_SCENARIO_HORIZONTAL:
client_manager.run()
elif args.role == "server":
runner = Server(args, device, dataset, model, server_aggregator)
if server_aggregator is None: