mnist-pytorch-gpu.yml

authorName: nni
experimentName: default_test
maxExecDuration: 15m
maxTrialNum: 2
trialConcurrency: 2
searchSpacePath: ./mnist_pytorch_search_space.json

tuner:
  builtinTunerName: Random
assessor:
  builtinAssessorName: Medianstop
  classArgs:
    optimize_mode: maximize
trial:
  codeDir: ../../../../examples/trials/mnist-pytorch
  command: python3 mnist.py  --epochs 1 --batch_num 10
  gpuNum: 1
localConfig:
  useActiveGpu: true
  maxTrialNumPerGpu: 2

useAnnotation: false
multiPhase: false
multiThread: false

trainingServicePlatform: local