The train_n_rounds was not working because it was evaluating with the…

… servers and not pool.servers. Now the function works as it should.
FLEXible-FL · Jan 23, 2024 · b0cb028 · b0cb028
1 parent ee594f9
commit b0cb028
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)"
    ]
   },
   {