fix bug in linux -- add config in configs/cfg_model

2024-11-28 10:23:44 +08:00 · 2024-11-28 10:23:44 +08:00 · 1b25920188
parent fc6ed02806
commit 1b25920188
1 changed files with 2 additions and 0 deletions
--- a/federatedscope/core/configs/cfg_model.py
+++ b/federatedscope/core/configs/cfg_model.py
@ -62,6 +62,8 @@ def extend_model_cfg(cfg):
    cfg.model.cheb_order = 1 # A tuple, e.g., (in_channel, h, w)
    cfg.model.use_day = True
    cfg.model.use_week = True
+    cfg.model.minigraph_size = 5
+    cfg.model.use_minigraph = False


    # ---------------------------------------------------------------------- #