add a comment in tf/fedopt_ctl

LeoDuda · Nov 22, 2024 · 37f5bb9 · 37f5bb9
1 parent a183869
commit 37f5bb9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nvflare/app_opt/tf/fedopt_ctl.py b/nvflare/app_opt/tf/fedopt_ctl.py
@@ -118,6 +118,7 @@ def update_model(self, global_model: FLModel, aggr_result: FLModel):
         specific set of Variables.
 
         """
+        # Get the Keras model stored in memory in persistor.
         global_model_tf = self.persistor.model
         global_params = global_model_tf.trainable_weights
         num_trainable_weights = len(global_params)