From 653f1c80eb9c27df164ad18b0dab0d06c00f7528 Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 12 Sep 2018 15:08:37 +0800 Subject: [PATCH 1/8] support install from venv and travis CI --- Makefile | 8 +++++--- src/nni_manager/common/utils.ts | 4 ++++ test/naive/run.py | 1 + 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/Makefile b/Makefile index 344f4b8b84..2f0e26046d 100644 --- a/Makefile +++ b/Makefile @@ -1,10 +1,12 @@ BIN_PATH ?= ${HOME}/.local/bin INSTALL_PREFIX ?= ${HOME}/.local -PIP_MODE ?= --user EXAMPLES_PATH ?= ${HOME}/nni/examples WHOAMI := $(shell whoami) -.PHONY: build install uninstall dev-install YARN := $(INSTALL_PREFIX)/yarn/bin/yarn +ifndef TRAVIS +PIP_MODE ?= --user +endif +.PHONY: build install uninstall dev-install build: ### Building NNI Manager ### @@ -63,7 +65,7 @@ install: pip-install: ifneq ('$(HOME)', '/root') ifeq (${WHOAMI}, root) - ### Sorry, sudo make install is not supported ### + ### Sorry, sudo pip install is not supported ### exit 1 endif endif diff --git a/src/nni_manager/common/utils.ts b/src/nni_manager/common/utils.ts index 4708aea7f6..98beed927a 100644 --- a/src/nni_manager/common/utils.ts +++ b/src/nni_manager/common/utils.ts @@ -154,6 +154,10 @@ function parseArg(names: string[]): string { function getMsgDispatcherCommand(tuner: any, assessor: any): string { let command: string = `python3 -m nni --tuner_class_name ${tuner.className}`; + if (process.env.VIRTUAL_ENV) { + command = path.join(process.env.VIRTUAL_ENV, 'bin/') +command; + } + if (tuner.classArgs !== undefined) { command += ` --tuner_args ${JSON.stringify(JSON.stringify(tuner.classArgs))}`; } diff --git a/test/naive/run.py b/test/naive/run.py index dd1db2b20c..387bafa89e 100644 --- a/test/naive/run.py +++ b/test/naive/run.py @@ -80,5 +80,6 @@ def run(): print(RED + 'FAIL' + CLEAR) print('%r' % e) traceback.print_exc() + raise e subprocess.run(['nnictl', 'stop']) From eb8cbd7200ee8900610aea9c6950322d1c150915 Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 12 Sep 2018 15:15:10 +0800 Subject: [PATCH 2/8] support install from venv and travis CI --- Makefile | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/Makefile b/Makefile index 2f0e26046d..d042b02fe5 100644 --- a/Makefile +++ b/Makefile @@ -3,8 +3,12 @@ INSTALL_PREFIX ?= ${HOME}/.local EXAMPLES_PATH ?= ${HOME}/nni/examples WHOAMI := $(shell whoami) YARN := $(INSTALL_PREFIX)/yarn/bin/yarn -ifndef TRAVIS PIP_MODE ?= --user +ifdef TRAVIS +undefine PIP_MODE +endif +ifdef VIRTUAL_ENV +undefine PIP_MODE endif .PHONY: build install uninstall dev-install From f078736b0245fcc5a000059fad3a993f58b7b988 Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 12 Sep 2018 15:52:25 +0800 Subject: [PATCH 3/8] support install from venv and travis CI --- Makefile | 3 --- test/naive/run.py | 6 +++--- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/Makefile b/Makefile index d042b02fe5..279d6c24f5 100644 --- a/Makefile +++ b/Makefile @@ -4,9 +4,6 @@ EXAMPLES_PATH ?= ${HOME}/nni/examples WHOAMI := $(shell whoami) YARN := $(INSTALL_PREFIX)/yarn/bin/yarn PIP_MODE ?= --user -ifdef TRAVIS -undefine PIP_MODE -endif ifdef VIRTUAL_ENV undefine PIP_MODE endif diff --git a/test/naive/run.py b/test/naive/run.py index 387bafa89e..239bedcd2c 100644 --- a/test/naive/run.py +++ b/test/naive/run.py @@ -76,10 +76,10 @@ def run(): run() # TODO: check the output of rest server print(GREEN + 'PASS' + CLEAR) - except Exception as e: + except Exception as error: print(RED + 'FAIL' + CLEAR) - print('%r' % e) + print('%r' % error) traceback.print_exc() - raise e + raise error subprocess.run(['nnictl', 'stop']) From e882866b6f68ca9ea3eca4efcc5f36962c2e7980 Mon Sep 17 00:00:00 2001 From: Crysple Date: Mon, 17 Sep 2018 15:13:25 +0800 Subject: [PATCH 4/8] support conda --- Makefile | 10 ++++++---- src/nni_manager/common/utils.ts | 6 +++--- src/nni_manager/core/nnimanager.ts | 12 +++++++----- 3 files changed, 16 insertions(+), 12 deletions(-) diff --git a/Makefile b/Makefile index 6e8513e06c..c8edb5d650 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,8 @@ # Setting variables SHELL := /bin/bash +PIP_INSTALL := python3 -m pip install +PIP_UNINSTALL := python3 -m pip uninstall ## Install directories ifeq ($(shell id -u), 0) # is root @@ -146,8 +148,8 @@ dev-install: .PHONY: uninstall uninstall: - -pip3 uninstall -y nni - -pip3 uninstall -y nnictl + -$(PIP_UNINSTALL) -y nni + -$(PIP_UNINSTALL) -y nnictl -rm -rf $(INSTALL_PREFIX)/nni -rm -f $(BIN_PATH)/nnimanager -rm -f $(BIN_PATH)/nnictl @@ -222,10 +224,10 @@ install-node-modules: .PHONY: install-dev-modules install-dev-modules: #$(_INFO) Installing Python SDK $(_END) - cd src/sdk/pynni && pip3 install $(PIP_MODE) -e . + cd src/sdk/pynni && $(PIP_INSTALL) $(PIP_MODE) -e . #$(_INFO) Installing nnictl $(_END) - cd tools && pip3 install $(PIP_MODE) -e . + cd tools && $(PIP_INSTALL) $(PIP_MODE) -e . mkdir -p $(INSTALL_PREFIX)/nni diff --git a/src/nni_manager/common/utils.ts b/src/nni_manager/common/utils.ts index ba0650ef28..c31ad9b8b8 100644 --- a/src/nni_manager/common/utils.ts +++ b/src/nni_manager/common/utils.ts @@ -161,9 +161,9 @@ function parseArg(names: string[]): string { function getMsgDispatcherCommand(tuner: any, assessor: any): string { let command: string = `python3 -m nni --tuner_class_name ${tuner.className}`; - if (process.env.VIRTUAL_ENV) { - command = path.join(process.env.VIRTUAL_ENV, 'bin/') +command; - } + // if (process.env.VIRTUAL_ENV) { + // command = path.join(process.env.VIRTUAL_ENV, 'bin/') +command; + // } if (tuner.classArgs !== undefined) { command += ` --tuner_args ${JSON.stringify(JSON.stringify(tuner.classArgs))}`; diff --git a/src/nni_manager/core/nnimanager.ts b/src/nni_manager/core/nnimanager.ts index 48d9fa3c83..f562598b11 100644 --- a/src/nni_manager/core/nnimanager.ts +++ b/src/nni_manager/core/nnimanager.ts @@ -236,14 +236,16 @@ class NNIManager implements Manager { newCwd = cwd; } // TO DO: add CUDA_VISIBLE_DEVICES + let nniEnv = { + NNI_MODE: mode, + NNI_CHECKPOINT_DIRECTORY: dataDirectory, + NNI_LOG_DIRECTORY: getLogDir() + }; + let newEnv = Object.assign({}, process.env, nniEnv); const tunerProc: ChildProcess = spawn(command, [], { stdio, cwd: newCwd, - env: { - NNI_MODE: mode, - NNI_CHECKPOINT_DIRECTORY: dataDirectory, - NNI_LOG_DIRECTORY: getLogDir() - }, + env: newEnv, shell: true }); this.dispatcherPid = tunerProc.pid; From 8d1826a34596f86a12b9746bb42e02b8120ee636 Mon Sep 17 00:00:00 2001 From: Crysple Date: Tue, 18 Sep 2018 14:59:48 +0800 Subject: [PATCH 5/8] support conda --- src/nni_manager/common/utils.ts | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/nni_manager/common/utils.ts b/src/nni_manager/common/utils.ts index c31ad9b8b8..f272ea22b1 100644 --- a/src/nni_manager/common/utils.ts +++ b/src/nni_manager/common/utils.ts @@ -161,10 +161,6 @@ function parseArg(names: string[]): string { function getMsgDispatcherCommand(tuner: any, assessor: any): string { let command: string = `python3 -m nni --tuner_class_name ${tuner.className}`; - // if (process.env.VIRTUAL_ENV) { - // command = path.join(process.env.VIRTUAL_ENV, 'bin/') +command; - // } - if (tuner.classArgs !== undefined) { command += ` --tuner_args ${JSON.stringify(JSON.stringify(tuner.classArgs))}`; } From 0aecf813406f46d64727647ff063d6b10d039605 Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 19 Sep 2018 11:39:34 +0800 Subject: [PATCH 6/8] modify example config path to relative path and increase maxTrialNum --- examples/trials/auto-gbdt/config.yml | 4 ++-- examples/trials/ga_squad/config.yml | 6 +++--- examples/trials/mnist-annotation/config.yml | 4 ++-- examples/trials/mnist-batch-tune-keras/config.yml | 6 +++--- examples/trials/mnist-keras/config.yml | 6 +++--- examples/trials/mnist-smartparam/config.yml | 4 ++-- examples/trials/mnist/config.yml | 6 +++--- examples/trials/pytorch_cifar10/config.yml | 6 +++--- 8 files changed, 21 insertions(+), 21 deletions(-) diff --git a/examples/trials/auto-gbdt/config.yml b/examples/trials/auto-gbdt/config.yml index 205b6f3a76..3107b53cfc 100644 --- a/examples/trials/auto-gbdt/config.yml +++ b/examples/trials/auto-gbdt/config.yml @@ -5,7 +5,7 @@ maxExecDuration: 10h maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: ~/nni/examples/trials/auto-gbdt/search_space.json +searchSpacePath: search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: minimize trial: command: python3 main.py - codeDir: ~/nni/examples/trials/auto-gbdt/ + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/ga_squad/config.yml b/examples/trials/ga_squad/config.yml index a094a71f79..a0b1480992 100644 --- a/examples/trials/ga_squad/config.yml +++ b/examples/trials/ga_squad/config.yml @@ -2,18 +2,18 @@ authorName: default experimentName: example_ga_squad trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local #choice: true, false useAnnotation: false tuner: - codeDir: ~/nni/examples/tuners/ga_customer_tuner + codeDir: ../tuners/ga_customer_tuner classFileName: customer_tuner.py className: CustomerTuner classArgs: optimize_mode: maximize trial: command: python3 trial.py - codeDir: ~/nni/examples/trials/ga_squad + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-annotation/config.yml b/examples/trials/mnist-annotation/config.yml index f7a0b06da8..34f2f5a321 100644 --- a/examples/trials/mnist-annotation/config.yml +++ b/examples/trials/mnist-annotation/config.yml @@ -2,7 +2,7 @@ authorName: default experimentName: example_mnist trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local #choice: true, false @@ -15,5 +15,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: ~/nni/examples/trials/mnist-annotation + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-batch-tune-keras/config.yml b/examples/trials/mnist-batch-tune-keras/config.yml index 4b750a2394..eacee55a9f 100644 --- a/examples/trials/mnist-batch-tune-keras/config.yml +++ b/examples/trials/mnist-batch-tune-keras/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_mnist-keras trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 6 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: ~/nni/examples/trials/mnist-batch-tune-keras/search_space.json +searchSpacePath: search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist-keras.py - codeDir: ~/nni/examples/trials/mnist-batch-tune-keras + codeDir: . gpuNum: 0 diff --git a/examples/trials/mnist-keras/config.yml b/examples/trials/mnist-keras/config.yml index 83baf364ad..483bb6837a 100644 --- a/examples/trials/mnist-keras/config.yml +++ b/examples/trials/mnist-keras/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_mnist-keras trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: ~/nni/examples/trials/mnist-keras/search_space.json +searchSpacePath: search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist-keras.py - codeDir: ~/nni/examples/trials/mnist-keras + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-smartparam/config.yml b/examples/trials/mnist-smartparam/config.yml index 1343937742..78d08978e1 100644 --- a/examples/trials/mnist-smartparam/config.yml +++ b/examples/trials/mnist-smartparam/config.yml @@ -2,7 +2,7 @@ authorName: default experimentName: example_mnist-smartparam trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local #choice: true, false @@ -15,5 +15,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: ~/nni/examples/trials/mnist-smartparam + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist/config.yml b/examples/trials/mnist/config.yml index 331afab2b9..bc56c95dec 100644 --- a/examples/trials/mnist/config.yml +++ b/examples/trials/mnist/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_mnist trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: ~/nni/examples/trials/mnist/search_space.json +searchSpacePath: search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: ~/nni/examples/trials/mnist + codeDir: . gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/pytorch_cifar10/config.yml b/examples/trials/pytorch_cifar10/config.yml index 102191315c..d813e1ce7c 100644 --- a/examples/trials/pytorch_cifar10/config.yml +++ b/examples/trials/pytorch_cifar10/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_pytorch_cifar10 trialConcurrency: 1 maxExecDuration: 100h -maxTrialNum: 1 +maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: ~/nni/examples/trials/pytorch_cifar10/search_space.json +searchSpacePath: search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 main.py - codeDir: ~/nni/examples/trials/pytorch_cifar10 + codeDir: . gpuNum: 1 From c55871b3d4a3d29902882120e5d9e329e418c71d Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 19 Sep 2018 11:44:13 +0800 Subject: [PATCH 7/8] undo messy commit --- examples/trials/auto-gbdt/config.yml | 2 +- examples/trials/ga_squad/config.yml | 6 +++--- examples/trials/mnist-annotation/config.yml | 4 ++-- examples/trials/mnist-batch-tune-keras/config.yml | 6 +++--- examples/trials/mnist-keras/config.yml | 6 +++--- examples/trials/mnist-smartparam/config.yml | 4 ++-- examples/trials/mnist/config.yml | 2 +- examples/trials/pytorch_cifar10/config.yml | 6 +++--- 8 files changed, 18 insertions(+), 18 deletions(-) diff --git a/examples/trials/auto-gbdt/config.yml b/examples/trials/auto-gbdt/config.yml index 3107b53cfc..a6e1f98c29 100644 --- a/examples/trials/auto-gbdt/config.yml +++ b/examples/trials/auto-gbdt/config.yml @@ -5,7 +5,7 @@ maxExecDuration: 10h maxTrialNum: 10 #choice: local, remote trainingServicePlatform: local -searchSpacePath: search_space.json +searchSpacePath: ~/nni/examples/trials/auto-gbdt/search_space.json #choice: true, false useAnnotation: false tuner: diff --git a/examples/trials/ga_squad/config.yml b/examples/trials/ga_squad/config.yml index a0b1480992..a094a71f79 100644 --- a/examples/trials/ga_squad/config.yml +++ b/examples/trials/ga_squad/config.yml @@ -2,18 +2,18 @@ authorName: default experimentName: example_ga_squad trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local #choice: true, false useAnnotation: false tuner: - codeDir: ../tuners/ga_customer_tuner + codeDir: ~/nni/examples/tuners/ga_customer_tuner classFileName: customer_tuner.py className: CustomerTuner classArgs: optimize_mode: maximize trial: command: python3 trial.py - codeDir: . + codeDir: ~/nni/examples/trials/ga_squad gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-annotation/config.yml b/examples/trials/mnist-annotation/config.yml index 34f2f5a321..f7a0b06da8 100644 --- a/examples/trials/mnist-annotation/config.yml +++ b/examples/trials/mnist-annotation/config.yml @@ -2,7 +2,7 @@ authorName: default experimentName: example_mnist trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local #choice: true, false @@ -15,5 +15,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: . + codeDir: ~/nni/examples/trials/mnist-annotation gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-batch-tune-keras/config.yml b/examples/trials/mnist-batch-tune-keras/config.yml index eacee55a9f..4b750a2394 100644 --- a/examples/trials/mnist-batch-tune-keras/config.yml +++ b/examples/trials/mnist-batch-tune-keras/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_mnist-keras trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 6 #choice: local, remote trainingServicePlatform: local -searchSpacePath: search_space.json +searchSpacePath: ~/nni/examples/trials/mnist-batch-tune-keras/search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist-keras.py - codeDir: . + codeDir: ~/nni/examples/trials/mnist-batch-tune-keras gpuNum: 0 diff --git a/examples/trials/mnist-keras/config.yml b/examples/trials/mnist-keras/config.yml index 483bb6837a..83baf364ad 100644 --- a/examples/trials/mnist-keras/config.yml +++ b/examples/trials/mnist-keras/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_mnist-keras trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local -searchSpacePath: search_space.json +searchSpacePath: ~/nni/examples/trials/mnist-keras/search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist-keras.py - codeDir: . + codeDir: ~/nni/examples/trials/mnist-keras gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist-smartparam/config.yml b/examples/trials/mnist-smartparam/config.yml index 78d08978e1..1343937742 100644 --- a/examples/trials/mnist-smartparam/config.yml +++ b/examples/trials/mnist-smartparam/config.yml @@ -2,7 +2,7 @@ authorName: default experimentName: example_mnist-smartparam trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local #choice: true, false @@ -15,5 +15,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: . + codeDir: ~/nni/examples/trials/mnist-smartparam gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist/config.yml b/examples/trials/mnist/config.yml index bc56c95dec..1ef0cfddbf 100644 --- a/examples/trials/mnist/config.yml +++ b/examples/trials/mnist/config.yml @@ -2,7 +2,7 @@ authorName: default experimentName: example_mnist trialConcurrency: 1 maxExecDuration: 1h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local searchSpacePath: search_space.json diff --git a/examples/trials/pytorch_cifar10/config.yml b/examples/trials/pytorch_cifar10/config.yml index d813e1ce7c..102191315c 100644 --- a/examples/trials/pytorch_cifar10/config.yml +++ b/examples/trials/pytorch_cifar10/config.yml @@ -2,10 +2,10 @@ authorName: default experimentName: example_pytorch_cifar10 trialConcurrency: 1 maxExecDuration: 100h -maxTrialNum: 10 +maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local -searchSpacePath: search_space.json +searchSpacePath: ~/nni/examples/trials/pytorch_cifar10/search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 main.py - codeDir: . + codeDir: ~/nni/examples/trials/pytorch_cifar10 gpuNum: 1 From 6c36d2edd5dfc72c7425ef48f23f93814df3d700 Mon Sep 17 00:00:00 2001 From: Crysple Date: Wed, 19 Sep 2018 11:46:21 +0800 Subject: [PATCH 8/8] undo messy commit --- examples/trials/auto-gbdt/config.yml | 2 +- examples/trials/mnist/config.yml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/trials/auto-gbdt/config.yml b/examples/trials/auto-gbdt/config.yml index a6e1f98c29..205b6f3a76 100644 --- a/examples/trials/auto-gbdt/config.yml +++ b/examples/trials/auto-gbdt/config.yml @@ -16,5 +16,5 @@ tuner: optimize_mode: minimize trial: command: python3 main.py - codeDir: . + codeDir: ~/nni/examples/trials/auto-gbdt/ gpuNum: 0 \ No newline at end of file diff --git a/examples/trials/mnist/config.yml b/examples/trials/mnist/config.yml index 1ef0cfddbf..331afab2b9 100644 --- a/examples/trials/mnist/config.yml +++ b/examples/trials/mnist/config.yml @@ -5,7 +5,7 @@ maxExecDuration: 1h maxTrialNum: 1 #choice: local, remote trainingServicePlatform: local -searchSpacePath: search_space.json +searchSpacePath: ~/nni/examples/trials/mnist/search_space.json #choice: true, false useAnnotation: false tuner: @@ -16,5 +16,5 @@ tuner: optimize_mode: maximize trial: command: python3 mnist.py - codeDir: . + codeDir: ~/nni/examples/trials/mnist gpuNum: 0 \ No newline at end of file