diff --git a/examples/trials/mnist/config_frameworkcontroller.yml b/examples/trials/mnist/config_frameworkcontroller.yml new file mode 100644 index 0000000000..f634630754 --- /dev/null +++ b/examples/trials/mnist/config_frameworkcontroller.yml @@ -0,0 +1,41 @@ +authorName: default +experimentName: example_mnist +trialConcurrency: 1 +maxExecDuration: 1h +maxTrialNum: 10 +#choice: local, remote, pai, kubeflow +trainingServicePlatform: frameworkcontroller +searchSpacePath: search_space.json +#choice: true, false +useAnnotation: false +tuner: + #choice: TPE, Random, Anneal, Evolution + builtinTunerName: TPE + classArgs: + #choice: maximize, minimize + optimize_mode: maximize +assessor: + builtinAssessorName: Medianstop + classArgs: + optimize_mode: maximize + gpuNum: 0 +trial: + codeDir: . + taskRoles: + - name: worker + taskNum: 1 + command: python3 mnist.py + gpuNum: 1 + cpuNum: 1 + memoryMB: 8192 + image: msranni/nni:latest + frameworkAttemptCompletionPolicy: + minFailedTaskCount: 1 + minSucceededTaskCount: 1 +frameworkcontrollerConfig: + storage: nfs + nfs: + # Your NFS server IP, like 10.10.10.10 + server: {your_nfs_server_ip} + # Your NFS server export path, like /var/nfs/nni + path: {your_nfs_server_export_path} \ No newline at end of file diff --git a/src/sdk/pynni/nni/platform/local.py b/src/sdk/pynni/nni/platform/local.py index afc7a878b0..783b0d02a0 100644 --- a/src/sdk/pynni/nni/platform/local.py +++ b/src/sdk/pynni/nni/platform/local.py @@ -36,7 +36,7 @@ os.makedirs(_outputdir) _nni_platform = os.environ['NNI_PLATFORM'] -if _nni_platform not in ['pai', 'kubeflow']: +if _nni_platform not in ['pai', 'kubeflow', 'frameworkcontroller']: _log_file_path = os.path.join(_outputdir, 'trial.log') init_logger(_log_file_path) @@ -77,7 +77,7 @@ def get_next_parameter(): return params def send_metric(string): - if _nni_platform in ['pai', 'kubeflow']: + if _nni_platform in ['pai', 'kubeflow', 'frameworkcontroller']: data = (string).encode('utf8') assert len(data) < 1000000, 'Metric too long' print('NNISDK_ME%s' % (data))