cifar10-pytorch.yml 478 Bytes
Newer Older
1
2
3
authorName: nni
experimentName: default_test
maxExecDuration: 15m
chicm-ms's avatar
chicm-ms committed
4
maxTrialNum: 2
5
6
7
8
9
10
11
12
13
14
15
trialConcurrency: 2
searchSpacePath: ./cifar10_search_space.json

tuner:
  builtinTunerName: Random
assessor:
  builtinAssessorName: Medianstop
  classArgs:
    optimize_mode: maximize
trial:
  codeDir: ../../../examples/trials/cifar10_pytorch
Scarlett Li's avatar
Scarlett Li committed
16
  command: python3 main.py --epochs 1 --batches 1
17
18
19
20
21
22
23
  gpuNum: 1

useAnnotation: false
multiPhase: false
multiThread: false

trainingServicePlatform: local