From 532a88a030a0761f9c83279cd7c5f028bd5aa320 Mon Sep 17 00:00:00 2001
From: Brax Team <no-reply@google.com>
Date: Fri, 2 Feb 2024 18:34:46 +0000
Subject: [PATCH] Internal change

PiperOrigin-RevId: 603725125
Change-Id: Ic0e98adea3cd752f91813846fa8246af431aa009
---
 brax/training/agents/apg/train_test.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/brax/training/agents/apg/train_test.py b/brax/training/agents/apg/train_test.py
index 06c697e36..d9c47b4e1 100644
--- 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,