config_assessor.yml 634 Bytes
Newer Older
1
2
3
4
5
6
7
authorName: default
experimentName: example_mnist
trialConcurrency: 1
maxExecDuration: 1h
maxTrialNum: 1
#choice: local, remote
trainingServicePlatform: local
8
searchSpacePath: ~/nni/examples/trials/mnist/search_space.json
9
10
11
12
#choice: true, false
useAnnotation: false
tuner:
  #choice: TPE, Random, Anneal, Evolution
13
14
15
16
  builtinTunerName: TPE
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
17
18
assessor:
  #choice: Medianstop
19
20
21
22
  builtinAssessorName: Medianstop
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
23
trial:
24
25
26
  command: python3 mnist.py
  codeDir: ~/nni/examples/trials/mnist
  gpuNum: 0