config_aml.yml 387 Bytes
Newer Older
zhanggzh's avatar
zhanggzh committed
1
2
searchSpaceFile: search_space.json
trialCommand: python3 mnist.py
xuanbaby's avatar
DTK-x  
xuanbaby committed
3
trialConcurrency: 1
zhanggzh's avatar
zhanggzh committed
4
maxTrialNumber: 10
xuanbaby's avatar
DTK-x  
xuanbaby committed
5
tuner:
zhanggzh's avatar
zhanggzh committed
6
  name: TPE
xuanbaby's avatar
DTK-x  
xuanbaby committed
7
8
  classArgs:
    optimize_mode: maximize
zhanggzh's avatar
zhanggzh committed
9
10
11
12
13
14
15
trainingService:
  platform: aml
  dockerImage: msranni/nni
  subscriptionId: ${your subscription ID}
  resourceGroup: ${your resource group}
  workspaceName: ${your workspace name}
  computeTarget: ${your compute target}