Fix config read part (NVIDIA#2232)

Co-authored-by: Chester Chen <[email protected]>
YuanTingHsieh · Dec 20, 2023 · 79b253a · 79b253a
1 parent 3ae2eb1
commit 79b253a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvflare/app_opt/lightning/api.py b/nvflare/app_opt/lightning/api.py
@@ -72,7 +72,7 @@ def __init__(self, rank: int = 0, load_state_dict_strict: bool = True):
         """
         super(FLCallback, self).__init__()
         init(rank=str(rank))
-        self.train_with_evaluation = get_config().get(ConfigKey.TRAIN_WITH_EVAL, False)
+        self.train_with_evaluation = get_config().get(ConfigKey.TASK_EXCHANGE, {}).get(ConfigKey.TRAIN_WITH_EVAL, False)
         self.current_round = None
         self.metrics = None
         self.total_local_epochs = 0