config_assessor.yml 736 Bytes
Newer Older
1
2
authorName: NNI Example
experimentName: MNIST TF v2.x with assessor
3
4
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:
Guoxin's avatar
Guoxin committed
12
  #choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner, GPTuner
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