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:
@@ -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']
|
||||
|
||||
Reference in New Issue
Block a user