diff --git a/flexnlp/notebooks/Federated SS with SentenceTransformers using FLEXible.ipynb b/flexnlp/notebooks/Federated SS with SentenceTransformers using FLEXible.ipynb index f785638..0908342 100644 --- a/flexnlp/notebooks/Federated SS with SentenceTransformers using FLEXible.ipynb +++ b/flexnlp/notebooks/Federated SS with SentenceTransformers using FLEXible.ipynb @@ -382,7 +382,7 @@ " pool.aggregators.map(fed_avg)\n", " # The aggregator send its aggregated weights to the server\n", " pool.aggregators.map(set_aggregated_weights_pt, pool.servers)\n", - " servers.map(evaluate_global_model, test_data=test_dataset)" + " pool.servers.map(evaluate_global_model, test_data=test_dataset)" ] }, {