From b3a252d970c3b17ca994035e8b02a929aac9b893 Mon Sep 17 00:00:00 2001 From: Mark Date: Fri, 24 Feb 2023 14:21:48 -0800 Subject: [PATCH] Fixed a minor bug --- polecart/main/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/polecart/main/agent.py b/polecart/main/agent.py index dacc8fd..7fdaf75 100755 --- a/polecart/main/agent.py +++ b/polecart/main/agent.py @@ -374,13 +374,13 @@ class Agent: ) ) + state = next_state # Only train the network if we have enough # transitions in memory to do so. if len(self.memory) >= self.BATCH_SIZE: - state = next_state - + # Run optimizer self._optimize()