nni_cgo.yaml 420 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
authorName: nni
experimentName: naive
trialConcurrency: 4
maxExecDuration: 1h
maxTrialNum: 10
trainingServicePlatform: local
searchSpacePath: fake_search_space.json
useAnnotation: false
advisor:
  codeDir: .
  classFileName: advisor_entry.py
  className: RetiariiAdvisor
  classArgs:
    strategy: debug_strategy.multi_model_cgo
trial:
  command: CGO=true python -m nni.retiarii.trial_entry
  codeDir: ../..
  gpuNum: 4