prepared scaffold for TF2.17

LeoDuda · Jul 31, 2024 · 979bac5 · 979bac5
1 parent e69cfc1
commit 979bac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvflare/app_opt/tf/scaffold.py b/nvflare/app_opt/tf/scaffold.py
@@ -155,4 +155,4 @@ def __init__(self, scaffold_helper):
     def on_epoch_end(self, epoch, logs=None):
         curr_lr = self.model.optimizer.learning_rate
         self.scaffold_helper.model_update(self.model, curr_lr, self.c_global_para, self.c_local_para)
-        print(f"SCAFFOLD model updated at end of epoch {epoch + 1}")
+        print(f"SCAFFOLD model updated at end of epoch {epoch + 1}")