fix(framework): Fix incorrect evaluation flags

KarhouTam · Dec 15, 2024 · acfd5e3 · acfd5e3
1 parent 90977b4
commit acfd5e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/server/fedavg.py b/src/server/fedavg.py
@@ -908,12 +908,12 @@ def run(self):
                     for split, flag in [
                         (
                             "train",
-                            self.args.common.eval_train or self.args.common.test_test,
+                            self.args.common.eval_train or self.args.common.test_train,
                         ),
                         ("val", self.args.common.eval_val or self.args.common.test_val),
                         (
                             "test",
-                            self.args.common.eval_test or self.args.common.test_train,
+                            self.args.common.eval_test or self.args.common.test_test,
                         ),
                     ]
                     if flag