Minor fixes

2023-02-24 17:46:33 -08:00
parent 589f41c205
commit c372ef8cc7
2 changed files with 6 additions and 8 deletions
--- a/polecart/basic/main.py
+++ b/polecart/basic/main.py
@ -200,13 +200,12 @@ for ep in range(num_episodes):



-
+		state = next_state
+		
+		
 		# Only train the network if we have enough
 		# transitions in memory to do so.
 		if len(memory) >= BATCH_SIZE:
-
-			state = next_state
-
 			# Run optimizer
 			optimize.optimize_model(
 				memory,