Internal change

PiperOrigin-RevId: 603725125 Change-Id: Ic0e98adea3cd752f91813846fa8246af431aa009
google · Feb 2, 2024 · 532a88a · 532a88a
1 parent 543d7c3
commit 532a88a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/brax/training/agents/apg/train_test.py b/brax/training/agents/apg/train_test.py
@@ -86,9 +86,9 @@ def get_offset(rng):
 
     _, _, _ = apg.train(
         envs.get_environment('inverted_pendulum', backend='spring'),
-        episode_length=128,
-        num_envs=64,
-        num_evals=200,
+        episode_length=100,
+        num_envs=8,
+        num_evals=10,
         learning_rate=3e-3,
         normalize_observations=True,
         randomization_fn=rand_fn,