Mark/celeste-ai
Archived
1
0
Fork 0
Code Releases Activity

Fixed a minor bug

Browse Source
This commit is contained in:
Mark
2023-02-24 14:21:48 -08:00
parent 2b1140e85f
commit b3a252d970
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
polecart/main/agent.py
View File

@ -374,12 +374,12 @@ 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()