import flwr as fl

import sys

import numpy as np

class SaveModelStrategy(fl.server.strategy.FedAvg):

return aggregated_weights

# Create strategy and run server

strategy = SaveModelStrategy()

# Start Flower server for three rounds of federated learning with 1Gb of data

fl.server.start_server(

server_address = 'localhost:'+str(sys.argv[1]) ,

config={"num_rounds": 3} ,

grpc_max_message_length = 1024*1024*1024,

strategy = strategy

)