config_hyperband.yml 763 Bytes
Newer Older
QuanluZhang's avatar
QuanluZhang committed
1
authorName: default
Shufan Huang's avatar
Shufan Huang committed
2
experimentName: example_mnist_hyperband
QuanluZhang's avatar
QuanluZhang committed
3
4
5
6
7
8
9
10
11
trialConcurrency: 2
maxExecDuration: 100h
maxTrialNum: 10000
#choice: local, remote, pai
trainingServicePlatform: local
searchSpacePath: search_space.json
#choice: true, false
useAnnotation: false
advisor:
Shufan Huang's avatar
Shufan Huang committed
12
  #choice: Hyperband, BOHB
QuanluZhang's avatar
QuanluZhang committed
13
14
  builtinAdvisorName: Hyperband
  classArgs:
Shufan Huang's avatar
Shufan Huang committed
15
16
    #R: the maximum trial budget (could be the number of mini-batches or epochs) can be
    #   allocated to a trial. Each trial should use trial budget to control how long it runs.
QuanluZhang's avatar
QuanluZhang committed
17
18
19
20
21
    R: 100
    #eta: proportion of discarded trials
    eta: 3
    #choice: maximize, minimize
    optimize_mode: maximize
J-shang's avatar
J-shang committed
22
23
    #choice: serial, parallelism
    exec_mode: parallelism
QuanluZhang's avatar
QuanluZhang committed
24
25
26
27
trial:
  command: python3 mnist.py
  codeDir: .
  gpuNum: 0