Skip to content

Commit 203955e

Browse files
authored
Lowers the default number of environments for camera envs (#2287)
# Description Some example environments with rendering required an extensive amount of VRAM that's likely not available on mid/lower end GPUs. This PR reduces the default number of environments so that these examples can be more accessible for users to try. Policy can still train with the lowered number of environments. ## Type of change <!-- As you go through the list, delete the ones that are not applicable. --> - Bug fix (non-breaking change which fixes an issue) ## Checklist - [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with `./isaaclab.sh --format` - [x] I have made corresponding changes to the documentation - [x] My changes generate no new warnings - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] I have updated the changelog and the corresponding version in the extension's `config/extension.toml` file - [ ] I have added my name to the `CONTRIBUTORS.md` or my name already exists there <!-- As you go through the checklist above, you can mark something as done by putting an x character in it For example, - [x] I have done this task - [ ] I have not done this task -->
1 parent 359cdc9 commit 203955e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

source/isaaclab_tasks/isaaclab_tasks/direct/cartpole/cartpole_camera_env.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ class CartpoleRGBCameraEnvCfg(DirectRLEnvCfg):
5858
viewer = ViewerCfg(eye=(20.0, 20.0, 20.0))
5959

6060
# scene
61-
scene: InteractiveSceneCfg = InteractiveSceneCfg(num_envs=1024, env_spacing=20.0, replicate_physics=True)
61+
scene: InteractiveSceneCfg = InteractiveSceneCfg(num_envs=512, env_spacing=20.0, replicate_physics=True)
6262

6363
# reset
6464
max_cart_pos = 3.0 # the cart is reset if it exceeds that position [m]

source/isaaclab_tasks/isaaclab_tasks/manager_based/classic/cartpole/cartpole_camera_env_cfg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ class TheiaTinyFeaturesCameraPolicyCfg(ObsGroup):
134134
class CartpoleRGBCameraEnvCfg(CartpoleEnvCfg):
135135
"""Configuration for the cartpole environment with RGB camera."""
136136

137-
scene: CartpoleRGBCameraSceneCfg = CartpoleRGBCameraSceneCfg(num_envs=1024, env_spacing=20)
137+
scene: CartpoleRGBCameraSceneCfg = CartpoleRGBCameraSceneCfg(num_envs=512, env_spacing=20)
138138
observations: RGBObservationsCfg = RGBObservationsCfg()
139139

140140
def __post_init__(self):
@@ -150,7 +150,7 @@ def __post_init__(self):
150150
class CartpoleDepthCameraEnvCfg(CartpoleEnvCfg):
151151
"""Configuration for the cartpole environment with depth camera."""
152152

153-
scene: CartpoleDepthCameraSceneCfg = CartpoleDepthCameraSceneCfg(num_envs=1024, env_spacing=20)
153+
scene: CartpoleDepthCameraSceneCfg = CartpoleDepthCameraSceneCfg(num_envs=512, env_spacing=20)
154154
observations: DepthObservationsCfg = DepthObservationsCfg()
155155

156156
def __post_init__(self):

0 commit comments

Comments
 (0)