nni_config.yaml

authorName: HaoHuang
experimentName: trace
trialConcurrency: 8
maxExecDuration: 1000h
maxTrialNum: 1000
#choice: local, remote, pai
trainingServicePlatform: local
searchSpacePath: search_space.json
#choice: true, false
useAnnotation: false

tuner:
  #choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
  #SMAC (SMAC should be installed through nnictl)
  builtinTunerName: TPE
  gpuIndices: 0,1,2,3,4,5,6,7
  
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
trial:
  command: python3 universal_attack_nni.py
  codeDir: .
  gpuNum: 1
  
localConfig:
  maxTrialNumPerGpu: 1
  useActiveGpu: true
  gpuIndices: 0,1,2,3,4,5,6,7