changed default ppo2 lr schedule to linear RUN BENCHMARKS

2018-08-01 16:24:44 -07:00
parent ad43fd9a35
commit 974c15756e
2 changed files with 2 additions and 2 deletions
--- a/.benchmark_pattern
+++ b/.benchmark_pattern
@@ -1 +1 @@
-ppo2_mpi
+ppo2
--- a/baselines/ppo2/defaults.py
+++ b/baselines/ppo2/defaults.py
@@ -7,7 +7,7 @@ def mujoco():
        noptepochs=10,
        log_interval=1,
        ent_coef=0.0,
-        lr=3e-4,
+        lr=lambda f: 3e-4 * f,
        cliprange=0.2,
        value_network='copy'
    )