gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-01-06 22:04:15 +01:00
Code Issues Projects Releases Wiki Activity

Added missing imports, correct usages

Browse Source
This commit is contained in:
Roman Dobosz
2018-04-24 13:33:10 +02:00
parent 42a9ec132d
commit 5c53f9be02
4 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
agents/human_agent.py
View File

@@ -31,14 +31,14 @@ class HumanAgent(agent.Agent):
self.clock = pygame.time.Clock()
self.max_fps = int(self.tp.visualization.max_fps_for_human_control)
utils.screen.log_title("Human Control Mode")
logger.screen.log_title("Human Control Mode")
available_keys = self.env.get_available_keys()
if available_keys:
utils.screen.log("Use keyboard keys to move. Press escape to quit. Available keys:")
utils.screen.log("")
logger.screen.log("Use keyboard keys to move. Press escape to quit. Available keys:")
logger.screen.log("")
for action, key in self.env.get_available_keys():
utils.screen.log("\t- {}: {}".format(action, key))
utils.screen.separator()
logger.screen.log("\t- {}: {}".format(action, key))
logger.screen.separator()
def train(self):
return 0
@@ -58,12 +58,12 @@ class HumanAgent(agent.Agent):
replay_buffer_path = os.path.join(logger.logger.experiments_path, 'replay_buffer.p')
self.memory.tp = None
pickle.to_pickle(self.memory, replay_buffer_path)
utils.screen.log_title("Replay buffer was stored in {}".format(replay_buffer_path))
logger.screen.log_title("Replay buffer was stored in {}".format(replay_buffer_path))
exit()
def log_to_screen(self, phase):
# log to utils.screen
utils.screen.log_dict(
# log to logger.screen
logger.screen.log_dict(
collections.OrderedDict([
("Episode", self.current_episode),
("total reward", self.total_reward_in_current_episode),