FIX

GreenWizard2015 · Jan 2, 2021 · ac8c9f3 · ac8c9f3
1 parent 6f3979c
commit ac8c9f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/learn_environment.py b/learn_environment.py
@@ -87,7 +87,7 @@ def testModel(EXPLORE_RATE):
     )
     print('Avg. train loss: %.4f' % trainLoss)
 
-    if params['batchSize'] < len(doomMemory):
+    if BATCH_SIZE < len(doomMemory):
       trainLoss = fit_stage.train(
         model, doomMemory,
         {