Conversation
acrlw
force-pushed
the
enhance/rl-batch-controls
branch
from
September 16, 2026 02:17
6b9ffbf to
c502be1
Compare
yuecideng
approved these changes
Sep 16, 2026
acrlw
force-pushed
the
enhance/rl-batch-controls
branch
from
September 17, 2026 01:30
c502be1 to
a74af62
Compare
7 tasks
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
This PR fixes active-joint list sharing between environments and exposes the root-velocity and scene-capacity controls needed for batched locomotion training.
Changes
control_partscan change the first environment's joint list from[0, 2]to[0, 2, 0, 2].EmbodiedEnv._setup_robot()now builds a fresh instance list on each setup and copies explicitly configuredactive_joint_ids.Robot.set_root_velocity()previously raisedAttributeError. AddArticulation.set_root_velocity(velocity, env_ids=None), inherited byRobot, and the corresponding view interface. The input is an(N, 6)tensor containing world-frame linear velocity followed by angular velocity.SceneArticulationViewvalidates the shape, selects one batch for both writes, preserves row order, and leaves unselected environments unchanged.DefaultPhysicsCfg.cache_material, defaulting toFalse, and forward it throughto_dexsim_args(). Training configurations can enable DexSim's existing material reuse withphysics_config: {cache_material: true}for thedefaultphysics backend.SimulationManagerCfg.scene_node_capacityand forward the training configuration throughconfig_to_cfg()toWorldConfig.scene_node_capacity.Nonepreserves the native default; an explicit value such as262144reaches World creation.Public API usage, configuration documentation, and the affected project context are updated.
Dependencies: no new dependencies. The changes use existing DexSim batch and configuration APIs.
Validation
Validated on the combined branch based on
main@7bc7271c, using Python 3.11 and a local RTX 5090 D v2:[2, 0], zero maximum readback error, and finite values after three physics stepsgit diff --checkpassedPending on this branch: 2,000-update training runs and a full Sphinx build.
Type of change
Screenshots
Not applicable; this PR changes simulation APIs and configuration.
Checklist
black .command to format the code base.python docs/scripts/check_api_docs.py), if applicable