Refactor examples' default image (#449)

* update * remove kubernetsServer in nnictl * update document * update * add warning to expand relative path * update doc * update * update * fix doc * update * update * update * refactor image * update * update * update * update
microsoft · Dec 5, 2018 · ddb3c14 · ddb3c14
1 parent 13e5b25
commit ddb3c14
Show file tree

Hide file tree

Showing 11 changed files with 81 additions and 9 deletions.
diff --git a/examples/trials/auto-gbdt/config_pai.yml b/examples/trials/auto-gbdt/config_pai.yml
@@ -22,7 +22,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/cifar10_pytorch/config_pai.yml b/examples/trials/cifar10_pytorch/config_pai.yml
@@ -22,7 +22,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/ga_squad/config_pai.yml b/examples/trials/ga_squad/config_pai.yml
@@ -20,7 +20,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist-annotation/config_pai.yml b/examples/trials/mnist-annotation/config_pai.yml
@@ -21,7 +21,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist-batch-tune-keras/config_pai.yml b/examples/trials/mnist-batch-tune-keras/config_pai.yml
@@ -22,7 +22,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist-hyperband/config_pai.yml b/examples/trials/mnist-hyperband/config_pai.yml
@@ -25,7 +25,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist-keras/config_pai.yml b/examples/trials/mnist-keras/config_pai.yml
@@ -22,7 +22,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist-smartparam/config_pai.yml b/examples/trials/mnist-smartparam/config_pai.yml
@@ -21,7 +21,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/mnist/config_pai.yml b/examples/trials/mnist/config_pai.yml
@@ -22,7 +22,7 @@ trial:
   cpuNum: 1
   memoryMB: 8196
   #The docker image to run nni job on pai
-  image: openpai/pai.example.tensorflow
+  image: msranni/nni:latest
   #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
   dataDir: hdfs://10.10.10.10:9000/username/nni
   #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'

diff --git a/examples/trials/sklearn/classification/config_pai.yml b/examples/trials/sklearn/classification/config_pai.yml
@@ -0,0 +1,36 @@
+authorName: default
+experimentName: example_sklearn
+trialConcurrency: 1
+maxExecDuration: 1h
+maxTrialNum: 100
+#choice: local, remote, pai
+trainingServicePlatform: pai
+searchSpacePath: search_space.json
+#choice: true, false
+useAnnotation: false
+tuner:
+  #choice: TPE, Random, Anneal, Evolution, BatchTuner
+  #SMAC (SMAC should be installed through nnictl)
+  builtinTunerName: TPE
+  classArgs:
+    #choice: maximize, minimize
+    optimize_mode: maximize
+trial:
+  command: python3 main.py
+  codeDir: .
+  gpuNum: 0
+  cpuNum: 1
+  memoryMB: 8196
+  #The docker image to run nni job on pai
+  image: msranni/nni:latest
+  #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
+  dataDir: hdfs://10.10.10.10:9000/username/nni
+  #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'
+  outputDir: hdfs://10.10.10.10:9000/username/nni
+paiConfig:
+  #The username to login pai
+  userName: username
+  #The password to login pai
+  passWord: password
+  #The host of restful server of pai
+  host: 10.10.10.10
diff --git a/examples/trials/sklearn/regression/config_pai.yml b/examples/trials/sklearn/regression/config_pai.yml
@@ -0,0 +1,36 @@
+authorName: default
+experimentName: example_sklearn
+trialConcurrency: 1
+maxExecDuration: 1h
+maxTrialNum: 100
+#choice: local, remote, pai
+trainingServicePlatform: pai
+searchSpacePath: search_space.json
+#choice: true, false
+useAnnotation: false
+tuner:
+  #choice: TPE, Random, Anneal, Evolution, BatchTuner
+  #SMAC (SMAC should be installed through nnictl)
+  builtinTunerName: TPE
+  classArgs:
+    #choice: maximize, minimize
+    optimize_mode: maximize
+trial:
+  command: python3 main.py
+  codeDir: .
+  gpuNum: 0
+  cpuNum: 1
+  memoryMB: 8196
+  #The docker image to run nni job on pai
+  image: msranni/nni:latest
+  #The hdfs directory to store data on pai, format 'hdfs://host:port/directory'
+  dataDir: hdfs://10.10.10.10:9000/username/nni
+  #The hdfs directory to store output data generated by nni, format 'hdfs://host:port/directory'
+  outputDir: hdfs://10.10.10.10:9000/username/nni
+paiConfig:
+  #The username to login pai
+  userName: username
+  #The password to login pai
+  passWord: password
+  #The host of restful server of pai
+  host: 10.10.10.10