Update pgmorl_halfcheetah.py example

LucasAlegre · Dec 1, 2023 · 72161ed · 72161ed
1 parent 26e7c45
commit 72161ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/pgmorl_halfcheetah.py b/examples/pgmorl_halfcheetah.py
@@ -14,6 +14,7 @@
         warmup_iterations=80,
         evolutionary_iterations=20,
         num_weight_candidates=7,
+        origin=np.array([0.0, -5.0]),
     )
     algo.train(
         total_timesteps=int(5e6),