27 lines
488 B
YAML
27 lines
488 B
YAML
outdir: 'fedex_test'
|
|
use_gpu: True
|
|
device: 1
|
|
federate:
|
|
mode: 'standalone'
|
|
total_round_num: 20
|
|
make_global_eval: False
|
|
client_num: 5
|
|
share_local_model: True
|
|
online_aggr: True
|
|
save_to: 'fedex_test/lr.pth'
|
|
use_diff: True
|
|
trainer:
|
|
type: 'general'
|
|
eval:
|
|
freq: 10
|
|
data:
|
|
type: 'toy'
|
|
model:
|
|
type: 'lr'
|
|
hpo:
|
|
fedex:
|
|
use: True
|
|
# ss: 'scripts/example_configs/fedex_flat_search_space.yaml'
|
|
ss: 'scripts/example_configs/fedex_grid_search_space.yaml'
|
|
diff: True
|