Mark
/
celeste-ai
Archived
1
0
Fork 0

Fixed a minor bug

master
Mark 2023-02-24 14:21:48 -08:00
parent 2b1140e85f
commit b3a252d970
Signed by: Mark
GPG Key ID: AD62BB059C2AAEE4
1 changed files with 2 additions and 2 deletions

View File

@ -374,13 +374,13 @@ class Agent:
) )
) )
state = next_state
# Only train the network if we have enough # Only train the network if we have enough
# transitions in memory to do so. # transitions in memory to do so.
if len(self.memory) >= self.BATCH_SIZE: if len(self.memory) >= self.BATCH_SIZE:
state = next_state
# Run optimizer # Run optimizer
self._optimize() self._optimize()