config_assessor.yml 697 Bytes
Newer Older
1
2
3
4
authorName: default
experimentName: example_mnist
trialConcurrency: 1
maxExecDuration: 1h
Yan Ni's avatar
Yan Ni committed
5
maxTrialNum: 50
6
7
#choice: local, remote
trainingServicePlatform: local
QuanluZhang's avatar
QuanluZhang committed
8
searchSpacePath: search_space.json
9
10
11
#choice: true, false
useAnnotation: false
tuner:
xuehui's avatar
xuehui committed
12
  #choice: TPE, Random, Anneal, Evolution, BatchTuner
fishyds's avatar
fishyds committed
13
  #SMAC (SMAC should be installed through nnictl)
14
15
16
17
  builtinTunerName: TPE
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
18
assessor:
19
  #choice: Medianstop, Curvefitting
Yan Ni's avatar
Yan Ni committed
20
  builtinAssessorName: Curvefitting
21
22
23
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
Yan Ni's avatar
Yan Ni committed
24
25
    epoch_num: 20
    threshold: 0.9
26
trial:
27
  command: python3 mnist.py
QuanluZhang's avatar
QuanluZhang committed
28
  codeDir: .
fishyds's avatar
fishyds committed
29
  gpuNum: 0