Skip to content

Commit fa17000

Browse files
committed
renamed model_weights to global_model_weights
1 parent 2954c38 commit fa17000

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

fed_learn/fed_server.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def __init__(self, model_fn: Callable,
1717

1818
self.model_fn = model_fn
1919
model = self.model_fn()
20-
self.model_weights = model.get_weights()
20+
self.global_model_weights = model.get_weights()
2121
fed_learn.get_rid_of_the_models(model)
2222

2323
(x_train, y_train), _ = datasets.cifar10.load_data()
@@ -64,7 +64,7 @@ def send_train_data(self, client):
6464
return x, y
6565

6666
def send_model(self, client):
67-
client.receive_and_init_model(self.model_fn, self.model_weights)
67+
client.receive_and_init_model(self.model_fn, self.global_model_weights)
6868

6969
def init_for_new_epoch(self):
7070
self._generate_data_indices()
@@ -85,7 +85,7 @@ def create_clients(self):
8585

8686
def summarize_weights(self):
8787
new_weights = self.weight_summarizer.process(self.client_model_weights)
88-
self.model_weights = new_weights
88+
self.global_model_weights = new_weights
8989

9090
def get_client_train_param_dict(self):
9191
return self.train_dict

0 commit comments

Comments
 (0)