Fix nnictl multiThread option (#467)

microsoft · Dec 11, 2018 · 0425766 · 0425766
1 parent 84cb53a
commit 0425766
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/nni_cmd/launcher.py b/tools/nni_cmd/launcher.py
@@ -203,6 +203,8 @@ def set_experiment(experiment_config, mode, port, config_file_name):
         request_data['description'] = experiment_config['description']
     if experiment_config.get('multiPhase'):
         request_data['multiPhase'] = experiment_config.get('multiPhase')
+    if experiment_config.get('multiThread'):
+        request_data['multiThread'] = experiment_config.get('multiThread')
     if experiment_config.get('advisor'):
         request_data['advisor'] = experiment_config['advisor']
     else: