Fixed a minor bug
parent
2b1140e85f
commit
b3a252d970
|
@ -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()
|
||||||
|
|
Reference in New Issue