23 lines
569 B
YAML
23 lines
569 B
YAML
|
authorName: default
|
||
|
|
experimentName: example_impax_pytorch
|
||
|
|
trialConcurrency: 1
|
||
|
|
maxExecDuration: 1h
|
||
|
|
maxTrialNum: 100
|
||
|
|
#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: maximize
|
||
|
|
optimize_mode: minimize
|
||
|
|
trial:
|
||
|
|
command: python3 main.py
|
||
|
|
codeDir: .
|
||
|
|
gpuNum: 1
|