program: federatedscope/main.py
project: pFL-bench
name: fedAvg,FEMNIST-s02,FT
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"
  - "expname_tag"
  - "finetune"
  - "trainer.finetune.before_eval"
  - "True"
  - "trainer.finetune.steps"
  - "5"
  - ${args_no_hyphens}
parameters:
  optimizer.lr:
    values: [0.05, 0.005, 0.5, 0.01, 0.1]
  federate.local_update_steps:
    values: [1, 3]

early_terminate:
  type: hyperband
  min_iter: 5  # the first bucket indicates we called wandb at least 5 times