Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "e114c819dff7771b0fe387e9918fbb2ddd38f29e"
Unverified Commit 6aae16c5 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Fix aml cluster metadata empty bug (#3015)

parent 60ed8c3b
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
tools/nni_cmd/launcher.py
View file @ 6aae16c5
...@@ -371,6 +371,11 @@ def set_experiment(experiment_config, mode, port, config_file_name): ...@@ -371,6 +371,11 @@ def set_experiment(experiment_config, mode, port, config_file_name):
{'key': 'frameworkcontroller_config', 'value': experiment_config['frameworkcontrollerConfig']}) {'key': 'frameworkcontroller_config', 'value': experiment_config['frameworkcontrollerConfig']})
request_data['clusterMetaData'].append( request_data['clusterMetaData'].append(
{'key': 'trial_config', 'value': experiment_config['trial']}) {'key': 'trial_config', 'value': experiment_config['trial']})
elif experiment_config['trainingServicePlatform'] == 'aml':
request_data['clusterMetaData'].append(
{'key': 'aml_config', 'value': experiment_config['amlConfig']})
request_data['clusterMetaData'].append(
{'key': 'trial_config', 'value': experiment_config['trial']})
response = rest_post(experiment_url(port), json.dumps(request_data), REST_TIME_OUT, show_error=True) response = rest_post(experiment_url(port), json.dumps(request_data), REST_TIME_OUT, show_error=True)
if check_response(response): if check_response(response):
return response return response
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment