1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +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

@@ -45,6 +45,8 @@ class PPOAgent(ActorCriticAgent):
self.unclipped_grads = Signal('Grads (unclipped)')
self.signals.append(self.unclipped_grads)
self.reset_game(do_not_reset_env=True)
def fill_advantages(self, batch):
current_states, next_states, actions, rewards, game_overs, total_return = self.extract_batch(batch)