forgot to update max

lazyprogrammer · lazyprogrammer · commit 5f98a8e61849 · 2018-02-03T18:51:40.000-05:00
diff --git a/rl2/mountaincar/pg_tf_random.py b/rl2/mountaincar/pg_tf_random.py
@@ -220,6 +220,7 @@ def random_search(env, pmodel, gamma):
 
     if avg_totalrewards > best_avg_totalreward:
       best_pmodel = tmp_pmodel
+      best_avg_totalreward = avg_totalrewards
   return totalrewards, best_pmodel
 
 
diff --git a/rl2/mountaincar/pg_theano_random.py b/rl2/mountaincar/pg_theano_random.py
@@ -191,6 +191,7 @@ def random_search(env, pmodel, gamma):
 
     if avg_totalrewards > best_avg_totalreward:
       best_pmodel = tmp_pmodel
+      best_avg_totalreward = avg_totalrewards
   return totalrewards, best_pmodel
 
 
