FS-TFP/scripts/example_configs/fedex_flat_search_space.yaml

25 lines
520 B
YAML

arm0:
train.optimizer.lr: 0.01
train.optimizer.momentum: 0.1
arm1:
train.optimizer.lr: 0.01
train.optimizer.momentum: 0.9
arm2:
train.optimizer.lr: 0.02
train.optimizer.momentum: 0.2
arm3:
train.optimizer.lr: 0.02
train.optimizer.momentum: 0.8
arm4:
train.optimizer.lr: 0.04
train.optimizer.momentum: 0.1
arm5:
train.optimizer.lr: 0.04
train.optimizer.momentum: 0.7
arm6:
train.optimizer.lr: 0.08
train.optimizer.momentum: 0.3
arm7:
train.optimizer.lr: 0.08
train.optimizer.momentum: 0.8