FIX

GreenWizard2015 · GreenWizard2015 · commit ac8c9f388896 · 2021-01-02T20:05:14.000+02:00
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,
         {

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ def testModel(EXPLORE_RATE):`
`87`	`87`	`)`
`88`	`88`	`print('Avg. train loss: %.4f' % trainLoss)`
`89`	`89`
`90`		`- if params['batchSize'] < len(doomMemory):`
	`90`	`+ if BATCH_SIZE < len(doomMemory):`
`91`	`91`	`trainLoss = fit_stage.train(`
`92`	`92`	`model, doomMemory,`
`93`	`93`	`{`