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

allow visualizing the observation + bug fixes to coach summary

Browse Source
This commit is contained in:
Itai Caspi
2018-02-13 18:47:24 +02:00
committed by Itai Caspi
parent 5d1a2bc392
commit 55c8c87afc
4 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
logger.py
View File

@@ -241,7 +241,7 @@ class Logger(BaseLogger):
pil_images[0].save(output_path, save_all=True, append_images=pil_images[1:], duration=1.0 / fps, loop=0)
def remove_experiment_dir(self):
os.removedirs(self.experiments_path)
shutil.rmtree(self.experiments_path)
def print_summary(self):
screen.separator()
@@ -252,7 +252,7 @@ class Logger(BaseLogger):
screen.separator()
if screen.ask_yes_no("Do you want to discard the experiment results (Warning: this cannot be undone)?", False):
self.remove_experiment_dir()
if 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_path = self.get_experiment_path(new_name, create_path=False)
shutil.move(self.experiments_path, new_path)