config_assessor.yml 686 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
#choice: true, false
useAnnotation: false
tuner:
QuanluZhang's avatar
QuanluZhang committed
12
13
  #choice: TPE, Random, Anneal, Evolution,
  #SMAC (SMAC should be installed through nnictl)
14
15
16
17
  builtinTunerName: TPE
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
18
19
assessor:
  #choice: Medianstop
20
21
22
23
  builtinAssessorName: Medianstop
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
24
trial:
25
26
  command: python3 mnist.py
  codeDir: ~/nni/examples/trials/mnist
QuanluZhang's avatar
QuanluZhang committed
27
  gpuNum: 0