diff --git a/nvflare/app_opt/tf/scaffold.py b/nvflare/app_opt/tf/scaffold.py index b7c62b8c94..957f3e4aef 100644 --- 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}") \ No newline at end of file