Skip to content

GitLab

Unverified Commit b83e3b3b authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

fix trial keeper (#1199)

parent 97829ccd
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tools/nni_trial_tool/trial_keeper.py
View file @ b83e3b3b
...@@ -51,7 +51,7 @@ def get_hdfs_client(args): ...@@ -51,7 +51,7 @@ def get_hdfs_client(args):
return _hdfs_client return _hdfs_client
# backward compatibility # backward compatibility
hdfs_host = None hdfs_host = None
hdfs_output_dir = None
if args.hdfs_host: if args.hdfs_host:
hdfs_host = args.hdfs_host hdfs_host = args.hdfs_host
elif args.pai_hdfs_host: elif args.pai_hdfs_host:
...@@ -83,6 +83,8 @@ def main_loop(args): ...@@ -83,6 +83,8 @@ def main_loop(args):
# redirect trial keeper's stdout and stderr to syslog # redirect trial keeper's stdout and stderr to syslog
trial_syslogger_stdout = RemoteLogger(args.nnimanager_ip, args.nnimanager_port, 'trial', StdOutputType.Stdout, args.log_collection) trial_syslogger_stdout = RemoteLogger(args.nnimanager_ip, args.nnimanager_port, 'trial', StdOutputType.Stdout, args.log_collection)
sys.stdout = sys.stderr = trial_keeper_syslogger sys.stdout = sys.stderr = trial_keeper_syslogger
hdfs_output_dir = None
if args.hdfs_output_dir: if args.hdfs_output_dir:
hdfs_output_dir = args.hdfs_output_dir hdfs_output_dir = args.hdfs_output_dir
elif args.pai_hdfs_output_dir: elif args.pai_hdfs_output_dir:
......
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