fix misc

GreenWizard2015 · Jan 2, 2021 · 6f3979c · 6f3979c
1 parent e9550bd
commit 6f3979c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 13 deletions.
diff --git a/fit_stage.py b/fit_stage.py
@@ -2,8 +2,6 @@
 import numpy as np
 
 def train(model, memory, params):
-  if len(memory) < params['batchSize']: return np.Inf
-
   modelClone = tf.keras.models.clone_model(model)
   modelClone.set_weights(model.get_weights()) # use clone model for stability
 

diff --git a/learn_environment.py b/learn_environment.py
@@ -87,17 +87,18 @@ def testModel(EXPLORE_RATE):
     )
     print('Avg. train loss: %.4f' % trainLoss)
 
-    trainLoss = fit_stage.train(
-      model, doomMemory,
-      {
-        'gamma': GAMMA,
-        'batchSize': BATCH_SIZE,
-        'steps': BOOTSTRAPPED_STEPS,
-        'episodes': params['train doom episodes'](epoch),
-        'alpha': params.get('doom alpha', lambda _: alpha)(epoch)
-      }
-    )
-    print('Avg. train doom loss: %.4f' % trainLoss)
+    if params['batchSize'] < len(doomMemory):
+      trainLoss = fit_stage.train(
+        model, doomMemory,
+        {
+          'gamma': GAMMA,
+          'batchSize': BATCH_SIZE,
+          'steps': BOOTSTRAPPED_STEPS,
+          'episodes': params['train doom episodes'](epoch),
+          'alpha': params.get('doom alpha', lambda _: alpha)(epoch)
+        }
+      )
+      print('Avg. train doom loss: %.4f' % trainLoss)
     ##################
     # test
     print('Testing...')