From f8614ff231f351fd6903fabe35e61a155e8686ea Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 18 May 2026 15:58:18 +0100 Subject: [PATCH] Update [ghstack-poisoned] --- examples/collectors/isaaclab_rnn_ppo_memory.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/examples/collectors/isaaclab_rnn_ppo_memory.py b/examples/collectors/isaaclab_rnn_ppo_memory.py index 49dd5ac9484..d9f6ad0a3f6 100644 --- a/examples/collectors/isaaclab_rnn_ppo_memory.py +++ b/examples/collectors/isaaclab_rnn_ppo_memory.py @@ -253,6 +253,12 @@ def parse_args() -> argparse.Namespace: "'compact' for the compile-friendly compact shifted path." ), ) + parser.add_argument( + "--gae-compact-cat-dim", + choices=["batch", "time"], + default="batch", + help="Concatenation dimension used by GAE shifted='compact'.", + ) parser.add_argument( "--deactivate-vmap", action=argparse.BooleanOptionalAction, @@ -342,6 +348,7 @@ def main() -> None: value_network=full_value, average_gae=False, shifted=gae_shifted, + compact_cat_dim=args.gae_compact_cat_dim, deactivate_vmap=args.deactivate_vmap, device=train_device, )