MCPcopy Create free account

hub / github.com/RoboVerseOrg/RoboVerse / functions

Functions4,056 in github.com/RoboVerseOrg/RoboVerse

Method__init__
(self, dim)
roboverse_learn/il/policies/dp/models/diffusion/conv1d_components.py:9
Method__init__
(self, inp_channels, out_channels, kernel_size, n_groups=8)
roboverse_learn/il/policies/dp/models/diffusion/conv1d_components.py:31
Method__init__
(self, *models, no_grad=False)
roboverse_learn/il/policies/dp/models/bet/utils.py:29
Method__init__
Constructor. Args: alpha (Tensor, optional): Weights for each class. Defaults to None. gamma (float, optional): A cons
roboverse_learn/il/policies/dp/models/bet/libraries/loss_fn.py:61
Method__init__
(self, **kwargs)
roboverse_learn/il/policies/dp/models/bet/libraries/mingpt/trainer.py:35
Method__init__
(self, model, train_dataset, test_dataset, config)
roboverse_learn/il/policies/dp/models/bet/libraries/mingpt/trainer.py:41
Method__init__
(self, vocab_size, block_size, **kwargs)
roboverse_learn/il/policies/dp/models/bet/libraries/mingpt/model.py:31
Method__init__
(self, config)
roboverse_learn/il/policies/dp/models/bet/libraries/mingpt/model.py:53
Method__init__
(self, config)
roboverse_learn/il/policies/dp/models/bet/libraries/mingpt/model.py:102
Method__init__
( self, action_dim: int, num_bins: int = 100, predict_offsets: bool = False,
roboverse_learn/il/policies/dp/models/bet/action_ae/discretizers/k_means.py:14
Method__init__
( self, input_dim: int, n_layer: int = 12, n_head: int = 12, n_embd: i
roboverse_learn/il/policies/dp/models/bet/latent_generators/mingpt.py:14
Method__init__
( self, input_dim: int, num_bins: int, action_dim: int, horizon: int,
roboverse_learn/il/policies/dp/models/bet/latent_generators/transformer.py:15
Method__init__
( self, shm_manager: SharedMemoryManager, array_specs: List[ArraySpec], buffer
roboverse_learn/il/policies/dp/shared_memory/shared_memory_queue.py:20
Method__init__
Initialize a SharedNDArray object from existing shared memory, object shape, and dtype. To initialize a SharedNDArray object from a memory man
roboverse_learn/il/policies/dp/shared_memory/shared_ndarray.py:78
Method__init__
(self, shm_manager: SharedMemoryManager, size: int = 8)
roboverse_learn/il/policies/dp/shared_memory/shared_memory_util.py:17
Method__init__
shm_manager: Manages the life cycle of share memories across processes. Remember to run .start() before passing. array_sp
roboverse_learn/il/policies/dp/shared_memory/shared_memory_ring_buffer.py:21
Method__init__
( self, zarr_path, horizon=1, pad_before=0, pad_after=0, seed=
roboverse_learn/il/datasets/robot_image_dataset.py:41
Method__init__
(self, num_envs, device, max_episodes=1000)
roboverse_learn/rl/episode_tracker.py:9
Method__init__
(self, env)
roboverse_learn/rl/clean_rl/td3.py:55
Method__init__
(self, env)
roboverse_learn/rl/clean_rl/sac.py:55
Method__init__
(self, envs)
roboverse_learn/rl/clean_rl/ppo.py:61
Method__init__
( self, buffer_size: int, observation_space: spaces.Space, action_space: space
roboverse_learn/rl/clean_rl/buffer.py:277
Method__init__
( self, buffer_size: int, observation_space: spaces.Space, action_space: space
roboverse_learn/rl/clean_rl/buffer.py:465
Method__init__
Initialize the environment.
roboverse_learn/rl/sb3/ppo_train.py:49
Method__init__
A simple replay buffer that stores transitions in a circular buffer. Supports n-step returns and asymmetric observations. Wh
roboverse_learn/rl/fast_td3/fttd3_module.py:8
Method__init__
Initialize EmpiricalNormalization module. Args: shape (int or tuple of int): Shape of input values except batch axis.
roboverse_learn/rl/fast_td3/fttd3_module.py:294
Method__init__
( self, n_obs: int, n_act: int, num_atoms: int, v_min: float,
roboverse_learn/rl/fast_td3/fttd3_module.py:371
Method__init__
( self, n_obs: int, n_act: int, num_atoms: int, v_min: float,
roboverse_learn/rl/fast_td3/fttd3_module.py:439
Method__init__
(self, env: AgentTask, train_cfg: dict | object = None)
roboverse_learn/rl/rsl_rl/env_wrapper.py:20
Method__init__
( self, scenario: ScenarioCfg, cfg: ManagerBasedRVEnvCfg, device: str | torch.
roboverse_learn/managers/manager_based_rv_env.py:116
Method__init__
( self, env, scenario, num_envs: int, robot_name: str, solver:
roboverse_learn/vla/pi0/pi_eval.py:50
Method__init__
( self, env, scenario, num_envs: int, checkpoint_path: str, ta
roboverse_learn/vla/SmolVLA/smolvla_eval.py:55
Method__init__
(self, use_first_action=True)
roboverse_learn/vla/SmolVLA/roboverse_smolvla_policy.py:104
Method__init__
(self, *args, **kwargs)
roboverse_learn/vla/rlds_utils/roboverse/roboverse.py:20
Method__init__
( self, env, scenario, num_envs: int, checkpoint_path: str, ta
roboverse_learn/vla/OpenVLA/vla_eval.py:45
Method__init__
(self, chunk: int = CHUNK, hidden: int = 512)
scripts/policy/bc_model.py:42
Method__init__
(self, file_path)
scripts/unitree_deploy/config.py:10
Method__init__
(self, config: G1Config)
scripts/unitree_deploy/deploy_real.py:47
Method__init__
(self)
scripts/unitree_deploy/common/remote_controller.py:28
Method__init__
(self, asset_root: str)
scripts/conversion/asset_converter.py:2
Method__init__
(self, input_file, success, error_msg=None)
scripts/conversion/batch_mjcf2urdf.py:10
Method__init__
(self, handler, scenario, args)
scripts/advanced/replay_state.py:110
Method__init__
(self, image_dir: str | None = None, video_path: str | None = None)
scripts/advanced/replay_state.py:298
Method__init__
(self, scenario: ScenarioCfg, num_envs: int = 1, sim_type: SimType = SimType.ISAACLAB)
scripts/advanced/train_ppo.py:21
Method__init__
(self, env: MetaSimVecEnv)
scripts/advanced/train_ppo_vec.py:40
Method__init__
(self, handler, robot_cfg, task_desc="", demo_start_idx=0)
scripts/advanced/collect_demo.py:272
Method__init__
(self, save_root_dir: str, start_idx: int, end_idx: int, pbar: tqdm)
scripts/advanced/collect_demo.py:378
Method__init__
Initialize the ObsSaver.
scripts/advanced/replay_demo.py:98
Method__init__
(self, file_name, flip_visual_attachments=False)
scripts/advanced/isaacgym_animate.py:37
Method__init__
(self, handler, scenario, args)
scripts/advanced/replay_real_asset.py:236
Method__init__
(self, image_dir: str | None = None, video_path: str | None = None)
scripts/advanced/replay_real_asset.py:325
Method__init__
( self, physics, eef_site: str, arm_joint_qvel_index, arm_actuator_ind
scripts/osc/osc_pose_controller.py:49
Method__init__
(self, in_dim, hidden, out_dim)
tests/test_fusion_bc_warmstart.py:33
Method__init__
(self, maxlen: int | None, keys: tuple[str] | list[str] | str, name: str | None = None)
roboverse_pack/callback_funcs/humanoid/step_funcs.py:114
Method__init__
(self)
roboverse_pack/utils/humanoid_utils.py:234
Method__init__
(self, sensor_name: str)
roboverse_pack/queries/sensor.py:26
Method__init__
(self, site_name: str)
roboverse_pack/queries/site.py:74
Method__init__
(self, geom1_name: str, geom2_name: str)
roboverse_pack/queries/contact.py:12
Method__init__
( self, link_name: str = "mid360_link", sensor_type: str = "mid360", apply_opt
roboverse_pack/queries/lidar.py:46
Method__init__
( self, obj_name: str, body_names: list[str] | str | None = None, static_frict
roboverse_pack/randomization/humanoid.py:29
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/task_template.py:50
Method__init__
( self, scenario: ScenarioCfg, config: BaseEnvCfg, device: str | torch.device
roboverse_pack/tasks/humanoid/base/base_legged_robot.py:27
Method__init__
( self, scenario: ScenarioCfg, config: Any | BaseEnvCfg, device: str | torch.d
roboverse_pack/tasks/humanoid/base/base_agent.py:21
Method__init__
( self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None,
roboverse_pack/tasks/humanoid/locomotion/walk_g1_dof12.py:214
Method__init__
( self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None,
roboverse_pack/tasks/humanoid/locomotion/walk_g1_dof29.py:211
Method__init__
( self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None,
roboverse_pack/tasks/humanoid/locomotion/walk_agibot_a2_dof12.py:220
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/libero_90/libero_90_base.py:33
Method__init__
(self, robot_name: str = "h1", move_speed: float | None = None)
roboverse_pack/tasks/humanoid_bench/humanoid_env.py:60
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/humanoid_bench/humanoid_env.py:106
Method__init__
(self, robot_name: str = "h1")
roboverse_pack/tasks/humanoid_bench/crawl.py:26
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/humanoid_bench/crawl.py:102
Method__init__
Initialize the slide reward.
roboverse_pack/tasks/humanoid_bench/slide.py:29
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/humanoid_bench/slide.py:100
Method__init__
Initialize the stair reward.
roboverse_pack/tasks/humanoid_bench/stair.py:29
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/humanoid_bench/stair.py:99
Method__init__
(self, robot_name: str = "h1")
roboverse_pack/tasks/humanoid_bench/sit.py:27
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/humanoid_bench/sit.py:124
Method__init__
(self, scenario, device=None)
roboverse_pack/tasks/maniskill/_dense_rl.py:38
Method__init__
(self, scenario: ScenarioCfg, device: str | torch.device | None = None)
roboverse_pack/tasks/maniskill/maniskill_base.py:24
Method__init__
(self)
roboverse_pack/tasks/maniskill/_native/control.py:59
Method__init__
(self, scenario, device=None)
roboverse_pack/tasks/maniskill/_native/base.py:35
Method__init__
(self, scenario, device=None)
roboverse_pack/tasks/maniskill/_native/multi_robot.py:36
Method__init__
Generate the cached handover trajectory before constructing the env.
roboverse_pack/tasks/bimanual/franka_handover.py:163
Method__init__
(self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None)
roboverse_pack/tasks/mjlab/cartpole_v2.py:210
Method__init__
(self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None)
roboverse_pack/tasks/mjlab/velocity_go1_v2.py:553
Method__init__
( self, task_id: str, num_envs: int = 1, play: bool = False, motion_fi
roboverse_pack/tasks/mjlab/_passthrough.py:55
Method__init__
(self, scenario: ScenarioCfg | None = None, device=None)
roboverse_pack/tasks/mjlab/lift_cube_yam_v2.py:521
Method__init__
(self, scenario: ScenarioCfg | None = None, device=None)
roboverse_pack/tasks/mjlab/lift_cube_yam_v2.py:539
Method__init__
(self, scenario: ScenarioCfg | None = None, device=None)
roboverse_pack/tasks/mjlab/lift_cube_yam_v2.py:557
Method__init__
(self, scenario: ScenarioCfg | None = None, device: str | torch.device | None = None)
roboverse_pack/tasks/mjlab/velocity_g1_v2.py:782
Method__init__
(self, scenario=None, device=None)
roboverse_pack/tasks/mjlab/cartpole_train.py:90
Method__init__
(self, env, cfg: VelocityCommandCfg | None = None)
roboverse_pack/tasks/mjlab/mdp/commands.py:60
Method__init__
(self, env, cfg: LiftingCommandCfg | None = None)
roboverse_pack/tasks/mjlab/mdp/commands.py:217
Method__init__
(self, motion_file: str | None, n_bodies: int, device)
roboverse_pack/tasks/mjlab/mdp/commands.py:384
Method__init__
(self, env, cfg: MotionCommandCfg)
roboverse_pack/tasks/mjlab/mdp/commands.py:443
Method__init__
(self, env, cfg: ContactSensorCfg)
roboverse_pack/tasks/mjlab/mdp/sensors.py:112
Method__init__
(self, env, cfg: TerrainHeightSensorCfg)
roboverse_pack/tasks/mjlab/mdp/sensors.py:425
Method__init__
(self, env, cfg: TerrainGridScanSensorCfg)
roboverse_pack/tasks/mjlab/mdp/sensors.py:609
Method__init__
(self, env, cfg: BuiltinSensorCfg)
roboverse_pack/tasks/mjlab/mdp/sensors.py:717
Method__init__
( self, env, cfg: JointPositionActionCfg, joint_names: Iterable[str],
roboverse_pack/tasks/mjlab/mdp/actions.py:167
← previousnext →2,301–2,400 of 4,056, ranked by callers