39 lines
839 B
YAML
39 lines
839 B
YAML
|
|
program: federatedscope/main.py
|
|
project: pFL-bench
|
|
name: ditto,FEMNIST-s02
|
|
method: grid
|
|
#method: bayes
|
|
metric:
|
|
goal: maximize
|
|
name: best_client_summarized_weighted_avg/val_acc
|
|
command:
|
|
- ${env}
|
|
- ${interpreter}
|
|
- ${program}
|
|
- "--cfg"
|
|
- "scripts/personalization_exp_scripts/pfl_bench/FEMNIST-s02/fedavg_convnet2_on_femnist.yaml"
|
|
- "outdir"
|
|
- "exp_pfl_bench"
|
|
- "wandb.use"
|
|
- "True"
|
|
- "wandb.name_project"
|
|
- "pFL-bench"
|
|
- "wandb.name_user"
|
|
- "daoyuan"
|
|
- "federate.method"
|
|
- "Ditto"
|
|
- ${args_no_hyphens}
|
|
parameters:
|
|
optimizer.lr:
|
|
values: [0.05, 0.005, 0.5, 0.01, 0.1]
|
|
federate.local_update_steps:
|
|
values: [1, 3]
|
|
personalization.regular_weight:
|
|
values: [0.05, 0.1, 0.5, 0.8]
|
|
|
|
early_terminate:
|
|
type: hyperband
|
|
min_iter: 5 # the first bucket indicates we called wandb at least 5 times
|
|
|