diff --git a/examples/trials/auto-gbdt/config_pai.yml b/examples/trials/auto-gbdt/config_pai.yml index 34bf5e83c4..d04cef6b91 100644 --- 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 index 3c9424f1fb..1f07062872 100644 --- 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 index d4435e5657..239d20ce74 100644 --- 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 index c48bb89a46..e4dc0b6d9e 100644 --- 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 index c7c0de7a14..11f9afa2ff 100644 --- 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 index f8032502a7..82fc4d5ebb 100644 --- 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 index 45c0bd56b0..e97d0930c4 100644 --- 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 index b605341deb..6c448391dc 100644 --- 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 index 6f2479c860..baa575b1dd 100644 --- 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 new file mode 100644 index 0000000000..664b58dbb0 --- /dev/null +++ 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 \ No newline at end of file diff --git a/examples/trials/sklearn/regression/config_pai.yml b/examples/trials/sklearn/regression/config_pai.yml new file mode 100644 index 0000000000..664b58dbb0 --- /dev/null +++ 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 \ No newline at end of file