mnist-curvefitting.test.yml 563 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
authorName: nni
experimentName: default_test
maxExecDuration: 5m
maxTrialNum: 2
trialConcurrency: 1
searchSpacePath: search_space.json

tuner:
  builtinTunerName: TPE
  classArgs:
    optimize_mode: maximize
assessor:
    builtinAssessorName: Curvefitting
    classArgs:
      epoch_num: 20
      optimize_mode: maximize
      start_step: 6
      threshold: 0.95
trial:
chicm-ms's avatar
chicm-ms committed
20
  codeDir: ../../../examples/trials/mnist-tfv1
21
22
23
24
25
26
27
28
  command: python3 mnist.py --batch_num 100
  gpuNum: 0

useAnnotation: false
multiPhase: false
multiThread: false

trainingServicePlatform: local