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
b1ae52ea
"...composable_kernel.git" did not exist on "b5f1f3ebcb49c70e02d770648140be89b0733e65"
Unverified
Commit
b1ae52ea
authored
Feb 19, 2021
by
Zeya Wang
Committed by
GitHub
Feb 19, 2021
Browse files
Update the Example Configuration File in AdaptDL Mode (#3381)
parent
57d2a0e5
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
20 additions
and
7 deletions
+20
-7
examples/trials/cifar10_pytorch/config_adl.yml
examples/trials/cifar10_pytorch/config_adl.yml
+11
-5
examples/trials/mnist-pytorch/config_adl.yml
examples/trials/mnist-pytorch/config_adl.yml
+9
-2
No files found.
examples/trials/cifar10_pytorch/config_adl.yml
View file @
b1ae52ea
...
@@ -21,13 +21,19 @@ trial:
...
@@ -21,13 +21,19 @@ trial:
command
:
python3 /cifar10/main_adl.py
command
:
python3 /cifar10/main_adl.py
codeDir
:
/cifar10
codeDir
:
/cifar10
gpuNum
:
1
gpuNum
:
1
image
:
{
image_built_by_adl.Dockerfile
}
# the user needs to have a docker image built by the adl.Dockerfile
# optional
# the docker image should be pushed to a registry for the cluster to pull
imagePullSecrets
:
# in our example we provide a docker image from our public docker hub
-
name
:
{
secret
}
image
:
petuum/nni:cifar-example
# optional:
# the user needs to provide the secret if the image is pulled from a private registry
# imagePullSecrets:
# - name: {secret}
adaptive
:
true
adaptive
:
true
checkpoint
:
checkpoint
:
storageClass
:
dfs
# the user needs to determine the storageClass in Kubenetes
# For example, for MicroK8s, ‘microk8s-hostpath’ can be used
storageClass
:
microk8s-hostpath
storageSize
:
1Gi
storageSize
:
1Gi
cpuNum
:
1
cpuNum
:
1
memorySize
:
1Gi
memorySize
:
1Gi
examples/trials/mnist-pytorch/config_adl.yml
View file @
b1ae52ea
...
@@ -3,7 +3,7 @@ experimentName: example_mnist_pytorch
...
@@ -3,7 +3,7 @@ experimentName: example_mnist_pytorch
trialConcurrency
:
1
trialConcurrency
:
1
maxExecDuration
:
1h
maxExecDuration
:
1h
maxTrialNum
:
10
maxTrialNum
:
10
nniManagerIp
:
{
replace_with_your_ip
}
logCollection
:
http
logCollection
:
http
trainingServicePlatform
:
adl
trainingServicePlatform
:
adl
...
@@ -15,7 +15,14 @@ tuner:
...
@@ -15,7 +15,14 @@ tuner:
optimize_mode
:
maximize
optimize_mode
:
maximize
trial
:
trial
:
image
:
{
replace_to_your_image_tag
}
# the user needs to have a docker image built by the adl.Dockerfile
# the docker image should be pushed to a registry for the cluster to pull
# in our example we provide a docker image from our public docker hub
image
:
petuum/train-demo:mnist-pytorch-adl
# optional:
# the user needs to provide the secret if the image is pulled from a private registry
# imagePullSecrets:
# - name: {secret}
command
:
python3 mnist.py
command
:
python3 mnist.py
codeDir
:
.
codeDir
:
.
gpuNum
:
1
gpuNum
:
1
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