Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
95f4c863
Unverified
Commit
95f4c863
authored
Jun 11, 2021
by
gxiaotian
Committed by
GitHub
Jun 11, 2021
Browse files
Adapt system auto tuning examples to NNI V2 (#3784)
parent
9e70639a
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
28 additions
and
37 deletions
+28
-37
docs/en_US/ResearchPublications.rst
docs/en_US/ResearchPublications.rst
+8
-5
examples/trials/systems_auto_tuning/rocksdb-fillrandom/config_smac.yml
...ls/systems_auto_tuning/rocksdb-fillrandom/config_smac.yml
+10
-16
examples/trials/systems_auto_tuning/rocksdb-fillrandom/config_tpe.yml
...als/systems_auto_tuning/rocksdb-fillrandom/config_tpe.yml
+10
-16
No files found.
docs/en_US/ResearchPublications.rst
View file @
95f4c863
...
@@ -95,11 +95,14 @@ New Algorithms
...
@@ -95,11 +95,14 @@ New Algorithms
.. code-block:: bibtex
.. code-block:: bibtex
@article{gao2020opevo,
@article{Gao2021opevo,
title={OpEvo: An Evolutionary Method for Tensor Operator Optimization},
title={OpEvo: An Evolutionary Method for Tensor Operator Optimization},
author={Gao, Xiaotian and Wei, Cui and Zhang, Lintao and Yang, Mao},
volume={35},
journal={arXiv preprint arXiv:2006.05664},
url={https://ojs.aaai.org/index.php/AAAI/article/view/17462},
year={2020}
number={14},
journal={Proceedings of the AAAI Conference on Artificial Intelligence},
author={Gao, Xiaotian and Cui, Wei and Zhang, Lintao and Yang, Mao},
year={2021}, month={May}, pages={12320-12327}
}
}
Measurement and Understanding
Measurement and Understanding
...
...
examples/trials/systems_auto_tuning/rocksdb-fillrandom/config_smac.yml
View file @
95f4c863
authorName
:
default
experimentName
:
auto_rocksdb_SMAC
experimentName
:
auto_rocksdb_SMAC
searchSpaceFile
:
search_space.json
trialCommand
:
python3 main.py
trialCodeDirectory
:
.
trialGpuNumber
:
0
trialConcurrency
:
1
trialConcurrency
:
1
maxExecDuration
:
12h
maxExperimentDuration
:
24h
maxTrialNum
:
256
maxTrialNumber
:
256
#choice: local, remote, pai
trainingServicePlatform
:
local
searchSpacePath
:
search_space.json
#choice: true, false
useAnnotation
:
false
tuner
:
tuner
:
#choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
name
:
SMAC
#SMAC (SMAC should be installed through nnictl)
builtinTunerName
:
SMAC
classArgs
:
classArgs
:
#choice: maximize, minimize
optimize_mode
:
maximize
optimize_mode
:
maximize
trial
:
trainingService
:
command
:
python3 main.py
platform
:
local
codeDir
:
.
useActiveGpu
:
False
gpuNum
:
0
examples/trials/systems_auto_tuning/rocksdb-fillrandom/config_tpe.yml
View file @
95f4c863
authorName
:
default
experimentName
:
auto_rocksdb_TPE
experimentName
:
auto_rocksdb_TPE
searchSpaceFile
:
search_space.json
trialCommand
:
python3 main.py
trialCodeDirectory
:
.
trialGpuNumber
:
0
trialConcurrency
:
1
trialConcurrency
:
1
maxExecDuration
:
12h
maxExperimentDuration
:
24h
maxTrialNum
:
256
maxTrialNumber
:
256
#choice: local, remote, pai
trainingServicePlatform
:
local
searchSpacePath
:
search_space.json
#choice: true, false
useAnnotation
:
false
tuner
:
tuner
:
#choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
name
:
TPE
#SMAC (SMAC should be installed through nnictl)
builtinTunerName
:
TPE
classArgs
:
classArgs
:
#choice: maximize, minimize
optimize_mode
:
maximize
optimize_mode
:
maximize
trial
:
trainingService
:
command
:
python3 main.py
platform
:
local
codeDir
:
.
useActiveGpu
:
False
gpuNum
:
0
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment