diff --git a/docs/en_US/TrainingService/FrameworkControllerMode.md b/docs/en_US/TrainingService/FrameworkControllerMode.md index dfa4260eea..d936ba3011 100644 --- a/docs/en_US/TrainingService/FrameworkControllerMode.md +++ b/docs/en_US/TrainingService/FrameworkControllerMode.md @@ -57,7 +57,6 @@ assessor: builtinAssessorName: Medianstop classArgs: optimize_mode: maximize - gpuNum: 0 trial: codeDir: ~/nni/examples/trials/mnist-tfv1 taskRoles: diff --git a/docs/en_US/TrainingService/KubeflowMode.md b/docs/en_US/TrainingService/KubeflowMode.md index 130dd9b3c6..9097ab7a1f 100644 --- a/docs/en_US/TrainingService/KubeflowMode.md +++ b/docs/en_US/TrainingService/KubeflowMode.md @@ -107,7 +107,6 @@ assessor: builtinAssessorName: Medianstop classArgs: optimize_mode: maximize - gpuNum: 0 trial: codeDir: . worker: diff --git a/examples/trials/mnist-distributed/config_kubeflow.yml b/examples/trials/mnist-distributed/config_kubeflow.yml index 87120319ea..96d32bf59a 100644 --- a/examples/trials/mnist-distributed/config_kubeflow.yml +++ b/examples/trials/mnist-distributed/config_kubeflow.yml @@ -18,7 +18,6 @@ assessor: builtinAssessorName: Medianstop classArgs: optimize_mode: maximize - gpuNum: 0 trial: codeDir: . worker: diff --git a/examples/trials/mnist-pytorch/config_frameworkcontroller.yml b/examples/trials/mnist-pytorch/config_frameworkcontroller.yml index e01900a8ef..6047f4daf6 100644 --- a/examples/trials/mnist-pytorch/config_frameworkcontroller.yml +++ b/examples/trials/mnist-pytorch/config_frameworkcontroller.yml @@ -18,7 +18,6 @@ assessor: builtinAssessorName: Medianstop classArgs: optimize_mode: maximize - gpuNum: 0 trial: codeDir: . taskRoles: diff --git a/examples/trials/mnist-tfv1/config_frameworkcontroller.yml b/examples/trials/mnist-tfv1/config_frameworkcontroller.yml index bf230ea872..63cdb34585 100644 --- a/examples/trials/mnist-tfv1/config_frameworkcontroller.yml +++ b/examples/trials/mnist-tfv1/config_frameworkcontroller.yml @@ -18,7 +18,6 @@ assessor: builtinAssessorName: Medianstop classArgs: optimize_mode: maximize - gpuNum: 0 trial: codeDir: . taskRoles: diff --git a/src/nni_manager/core/test/nnimanager.test.ts b/src/nni_manager/core/test/nnimanager.test.ts index a04cb8bad5..27432f3e77 100644 --- a/src/nni_manager/core/test/nnimanager.test.ts +++ b/src/nni_manager/core/test/nnimanager.test.ts @@ -49,12 +49,10 @@ describe('Unit test for nnimanager', function () { optimize_mode: 'maximize' }, checkpointDir: '', - gpuNum: 0 }, assessor: { builtinAssessorName: 'Medianstop', checkpointDir: '', - gpuNum: 1 } }