[Hotfix] Skip cfg check when not use (alibaba#266)

Schichael · Sep 7, 2022 · 0936751 · 0936751
1 parent e2b4c59
commit 0936751
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/federatedscope/core/configs/cfg_asyn.py b/federatedscope/core/configs/cfg_asyn.py
@@ -32,6 +32,8 @@ def extend_asyn_cfg(cfg):
 
 
 def assert_asyn_cfg(cfg):
+    if not cfg.asyn.use:
+        return True
     # to ensure a valid time budget
     assert isinstance(cfg.asyn.time_budget, int) or isinstance(
         cfg.asyn.time_budget, float

diff --git a/federatedscope/core/trainers/torch_trainer.py b/federatedscope/core/trainers/torch_trainer.py
@@ -141,7 +141,8 @@ def _hook_on_fit_start_init(self, ctx):
             # across different routines
             ctx.optimizer = get_optimizer(ctx.model,
                                           **ctx.cfg[ctx.cur_mode].optimizer)
-            ctx.scheduler = get_scheduler(ctx.optimizer, **ctx.cfg[ctx.cur_mode].scheduler)
+            ctx.scheduler = get_scheduler(ctx.optimizer,
+                                          **ctx.cfg[ctx.cur_mode].scheduler)
 
         # prepare statistics
         setattr(ctx, "loss_batch_total_{}".format(ctx.cur_data_split), 0)