config_paiYarn_ppo.yml 647 Bytes
Newer Older
SparkSnail's avatar
SparkSnail committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
authorName: Unknown
experimentName: enas_macro
trialConcurrency: 20
maxExecDuration: 2400h
maxTrialNum: 20000
#choice: local, remote
trainingServicePlatform: paiYarn
#choice: true, false
useAnnotation: true
multiPhase: false
versionCheck: false
nniManagerIp: 0.0.0.0
tuner:
  builtinTunerName: PPOTuner
  classArgs:
    optimize_mode: maximize
    trials_per_update: 60
    epochs_per_update: 20
    minibatch_size: 6
trial:
  command: sh ./macro_cifar10_pai.sh
  codeDir: ./
  gpuNum: 1
  cpuNum: 1
  memoryMB: 8196
  image: msranni/nni:latest
  virtualCluster: nni
paiYarnConfig:
  userName: your_account
  passWord: your_passwd
  host: 0.0.0.0