-
Notifications
You must be signed in to change notification settings - Fork 3
/
nni_config.yml
41 lines (40 loc) · 1.21 KB
/
nni_config.yml
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
32
33
34
35
36
37
38
39
40
41
authorName: default
experimentName: Data Augmentation Policy
trialConcurrency: 1
maxExecDuration: 100h
maxTrialNum: 60
#choice: local, remote, pai
trainingServicePlatform: local
searchSpacePath: search_space.json
#choice: true, false
useAnnotation: false
#tuner:
# #choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner, GPTuner
# #SMAC (SMAC should be installed through nnictl)
# builtinTunerName: TPE
# classArgs:
# #choice: maximize, minimize
# optimize_mode: minimize
tuner:
codeDir: {Your_Tuun_Code_Parent_Directory}/tuun/tuun
classFileName: nni_tuner.py
className: TuunTuner
# Any parameter need to pass to your tuner class __init__ constructor
# can be specified in this optional classArgs field, for example
classArgs:
optimize_mode: maximize
tuun_config: {
'seed': 1,
'model_config': {'name': 'standistmatgp'},
'acqfunction_config': {'name': 'default', 'acq_str': 'ei', 'n_gen': 500},
'acqoptimizer_config': {'n_init_rs': 5, 'jitter': True},
'probo_config': {'normalize_real': True},
}
gpuIndices: '1'
trial:
command: python3 main.py --extra 10 --dataset svhn
codeDir: .
gpuNum: 1
localConfig:
maxTrialNumPerGpu: 1
useActiveGpu: true