Archived
1
0

Fixed a minor bug

This commit is contained in:
2023-02-24 14:21:48 -08:00
parent 2b1140e85f
commit b3a252d970

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()