gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

cleanup debugging code

Browse Source
This commit is contained in:
Zach Dwiel
2018-02-20 11:43:56 -05:00
parent 86362683b1
commit f9f92a42fd
2 changed files with 7 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
coach.py
View File

@@ -160,38 +160,6 @@ def run_dict_to_json(_run_dict, task_id=''):
return json_path return json_path
def enqueue_output(out, queue):
for line in iter(out.readline, b''):
queue.put(line)
out.close()
def merge_streams(processes, output_stream=sys.stdout):
q = Queue()
threads = []
for p in processes:
threads.append(Thread(target=enqueue_output, args=(p.stdout, q)))
threads.append(Thread(target=enqueue_output, args=(p.stderr, q)))
for t in threads:
t.daemon = True
t.start()
while True:
try:
line = q.get_nowait()
except Empty:
# break when all processes are done and q is empty
if all(p.poll() is not None for p in processes):
break
else:
# sys.stdout.write(line)
output_stream.write(line.decode(output_stream.encoding))
output_stream.flush()
print('All processes done')
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-p', '--preset', parser.add_argument('-p', '--preset',
@@ -292,8 +260,6 @@ if __name__ == "__main__":
if not args.no_summary: if not args.no_summary:
atexit.register(logger.print_summary) atexit.register(logger.print_summary)
set_cpu()
# Single-threaded runs # Single-threaded runs
if run_dict['num_threads'] == 1: if run_dict['num_threads'] == 1:
# set tuning parameters # set tuning parameters
@@ -368,8 +334,6 @@ if __name__ == "__main__":
else: else:
evaluation_worker = p evaluation_worker = p
merge_streams(workers + [parameter_server])
# wait for all workers # wait for all workers
[w.wait() for w in workers] [w.wait() for w in workers]
evaluation_worker.kill() evaluation_worker.kill()

14
logger.py
View File

@@ -250,13 +250,13 @@ class Logger(BaseLogger):
if 'Training Reward' in self.data.keys() and 'Evaluation Reward' in self.data.keys(): if 'Training Reward' in self.data.keys() and 'Evaluation Reward' in self.data.keys():
screen.log_title("Max training reward: {}, max evaluation reward: {}".format(self.data['Training Reward'].max(), self.data['Evaluation Reward'].max())) screen.log_title("Max training reward: {}, max evaluation reward: {}".format(self.data['Training Reward'].max(), self.data['Evaluation Reward'].max()))
screen.separator() screen.separator()
# if screen.ask_yes_no("Do you want to discard the experiment results (Warning: this cannot be undone)?", False): if screen.ask_yes_no("Do you want to discard the experiment results (Warning: this cannot be undone)?", False):
# self.remove_experiment_dir() self.remove_experiment_dir()
# elif screen.ask_yes_no("Do you want to specify a different experiment name to save to?", False): elif screen.ask_yes_no("Do you want to specify a different experiment name to save to?", False):
# new_name = self.get_experiment_name() new_name = self.get_experiment_name()
# new_path = self.get_experiment_path(new_name, create_path=False) new_path = self.get_experiment_path(new_name, create_path=False)
# shutil.move(self.experiments_path, new_path) shutil.move(self.experiments_path, new_path)
# screen.log_title("Results moved to: {}".format(new_path)) screen.log_title("Results moved to: {}".format(new_path))
def get_experiment_name(self, initial_experiment_name=''): def get_experiment_name(self, initial_experiment_name=''):
match = None match = None