1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00

adding support in tensorboard (#52)

* bug-fix in architecture.py where additional fetches would acquire more entries than it should
* change in run_test to allow ignoring some test(s)
This commit is contained in:
Gal Leibovich
2018-02-05 15:21:49 +02:00
committed by GitHub
parent a8d5fb7bdf
commit 7c8962c991
10 changed files with 107 additions and 36 deletions

View File

@@ -91,7 +91,7 @@ class Agent(object):
self.total_steps_counter = 0
self.running_reward = None
self.training_iteration = 0
self.current_episode = 0
self.current_episode = self.tp.current_episode = 0
self.curr_state = []
self.current_episode_steps_counter = 0
self.episode_running_info = {}
@@ -406,6 +406,7 @@ class Agent(object):
self.reset_game()
self.current_episode += 1
self.tp.current_episode = self.current_episode
# return episode really ended
return result['done']