config_adl.yml 804 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
authorName: default
experimentName: example_pytorch_cifar10
trialConcurrency: 1
maxExecDuration: 100h
maxTrialNum: 10
nniManagerIp: {replace_with_your_ip}
trainingServicePlatform: adl
searchSpacePath: search_space_adl.json
logCollection: http
#choice: true, false
useAnnotation: false
tuner:
  #choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
  #SMAC (SMAC should be installed through nnictl)
  builtinTunerName: TPE
  classArgs:
    #choice: maximize, minimize
    optimize_mode: maximize
trial:
20
  namespace: default
21
22
  command: python3 /cifar10/main_adl.py
  codeDir: /cifar10
23
  gpuNum: 1
24
  image: {image_built_by_adl.Dockerfile}
25
26
27
  # optional
  imagePullSecrets:
    - name: {secret}
28
29
30
31
32
33
  adaptive: true
  checkpoint:
    storageClass: dfs
    storageSize: 1Gi
  cpuNum: 1
  memorySize: 1Gi