config_aml.yml 387 Bytes
Newer Older
1
2
searchSpaceFile: search_space.json
trialCommand: python3 mnist.py
SparkSnail's avatar
SparkSnail committed
3
trialConcurrency: 1
4
maxTrialNumber: 10
SparkSnail's avatar
SparkSnail committed
5
tuner:
6
  name: TPE
SparkSnail's avatar
SparkSnail committed
7
8
  classArgs:
    optimize_mode: maximize
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}