mnist-curvefitting.test.yml 558 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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:
  codeDir: ../../../examples/trials/mnist
  command: python3 mnist.py --batch_num 100
  gpuNum: 0

useAnnotation: false
multiPhase: false
multiThread: false

trainingServicePlatform: local