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

fixed the LSTM middleware initialization

This commit is contained in:
Itai Caspi
2017-12-30 15:18:09 +02:00
committed by Itai Caspi
parent b435c6d2d7
commit eeb3ec5497
4 changed files with 8 additions and 0 deletions

View File

@@ -37,6 +37,8 @@ class DDPGAgent(ActorCriticAgent):
self.q_values = Signal("Q")
self.signals.append(self.q_values)
self.reset_game(do_not_reset_env=True)
def learn_from_batch(self, batch):
current_states, next_states, actions, rewards, game_overs, _ = self.extract_batch(batch)