Unverified Commit b1ae52ea authored by Zeya Wang's avatar Zeya Wang Committed by GitHub
Browse files

Update the Example Configuration File in AdaptDL Mode (#3381)

parent 57d2a0e5
......@@ -21,13 +21,19 @@ trial:
command: python3 /cifar10/main_adl.py
codeDir: /cifar10
gpuNum: 1
image: {image_built_by_adl.Dockerfile}
# optional
imagePullSecrets:
- name: {secret}
# 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/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
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
cpuNum: 1
memorySize: 1Gi
......@@ -3,7 +3,7 @@ experimentName: example_mnist_pytorch
trialConcurrency: 1
maxExecDuration: 1h
maxTrialNum: 10
nniManagerIp: {replace_with_your_ip}
logCollection: http
trainingServicePlatform: adl
......@@ -15,7 +15,14 @@ tuner:
optimize_mode: maximize
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
codeDir: .
gpuNum: 1
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment