MCPcopy Create free account

hub / github.com/Marine-RL/MarineGym / functions

Functions434 in github.com/Marine-RL/MarineGym

Method__init__
( self, input_dim: int, output_dims: Union[List[int], torch.Tensor], )
marinegym/learning/modules/distributions.py:344
Method__init__
( self, logits: torch.Tensor = None, probs: torch.Tensor = None, unimix: float
marinegym/learning/modules/distributions.py:364
Method__init__
( self, input_shape: Union[int, Iterable], subtract_mean: bool = True, )
marinegym/learning/utils/valuenorm.py:107
Method__init__
(self, gamma, lmbda)
marinegym/learning/ppo/common.py:29
Method__init__
( self, cfg: PPOConfig, observation_spec: CompositeSpec, action_spec: Composit
marinegym/learning/ppo/ppo.py:86
Method__init_subclass__
(cls, **kwargs)
marinegym/robots/robot.py:89
Method__init_subclass__
(cls, **kwargs)
marinegym/controllers/controller.py:13
Method__init_subclass__
(cls, **kwargs)
marinegym/envs/isaac_env.py:191
Method__len__
(self)
marinegym/utils/torchrl/env.py:126
Method__setitem__
(self, k: str, v: AgentSpec)
marinegym/envs/isaac_env.py:417
Method__str__
(self)
marinegym/robots/drone/tiltRotor.py:563
Method__str__
(self)
marinegym/robots/drone/underwaterVehicle.py:346
Method__str__
(self)
marinegym/robots/drone/underwaterVehicleFin.py:401
Function_agent_spec
(self: TransformedEnv)
marinegym/utils/torchrl/transforms.py:61
Method_apply_transform
(self, obs: torch.Tensor)
marinegym/utils/torchrl/transforms.py:170
Method_call
(self, tensordict: TensorDictBase)
marinegym/utils/torchrl/transforms.py:302
Method_compute_reward_and_done
(self)
marinegym/envs/single/track.py:233
Method_compute_reward_and_done
(self)
marinegym/envs/single/hover.py:214
Method_compute_state_and_obs
(self)
marinegym/envs/single/track.py:203
Method_compute_state_and_obs
(self)
marinegym/envs/single/hover.py:190
Method_design_scene
(self)
marinegym/envs/single/track.py:91
Method_design_scene
(self)
marinegym/envs/single/hover.py:75
Function_func
(*args, **kwargs)
marinegym/views/__init__.py:45
Function_get_devices
(self: TensorDict)
marinegym/__init__.py:60
Function_get_shapes
(self: TensorDict)
marinegym/__init__.py:54
Method_inv_apply_transform
(self, action: torch.Tensor)
marinegym/utils/torchrl/transforms.py:106
Method_inv_apply_transform
(self, action: torch.Tensor)
marinegym/utils/torchrl/transforms.py:149
Method_inv_call
(self, tensordict: TensorDictBase)
marinegym/utils/torchrl/transforms.py:153
Method_inv_call
(self, tensordict: TensorDictBase)
marinegym/utils/torchrl/transforms.py:260
Method_pre_sim_step
(self, tensordict: TensorDictBase)
marinegym/envs/single/track.py:192
Method_pre_sim_step
(self, tensordict: TensorDictBase)
marinegym/envs/single/hover.py:186
Method_reset
(self, tensordict: TensorDictBase, **kwargs)
marinegym/envs/isaac_env.py:246
Method_reset_idx
(self, env_ids: torch.Tensor, train: bool=True)
marinegym/robots/robot.py:205
Method_reset_idx
(self, env_ids: torch.Tensor, train: bool=True)
marinegym/robots/drone/tiltRotor.py:455
Method_reset_idx
(self, env_ids: torch.Tensor, train: bool=True)
marinegym/robots/drone/underwaterVehicle.py:313
Method_reset_idx
(self, env_ids: torch.Tensor, train: bool=True)
marinegym/robots/drone/underwaterVehicleFin.py:368
Method_reset_idx
(self, env_ids: torch.Tensor)
marinegym/envs/single/track.py:146
Method_reset_idx
(self, env_ids: torch.Tensor)
marinegym/envs/single/hover.py:152
Method_set_seed
(self, seed: Optional[int] = -1)
marinegym/envs/isaac_env.py:289
Method_set_specs
(self)
marinegym/envs/single/track.py:106
Method_set_specs
(self)
marinegym/envs/single/hover.py:108
Method_setup_backbone
(self)
marinegym/learning/modules/networks.py:412
Method_step
(self, tensordict: TensorDictBase)
marinegym/utils/torchrl/transforms.py:310
Method_step
(self, tensordict: TensorDictBase)
marinegym/envs/isaac_env.py:264
Function_transform_agent_spec
(self: Transform, agent_spec: AgentSpec)
marinegym/utils/torchrl/transforms.py:49
Methodaction_spec
(self)
marinegym/utils/torchrl/env.py:52
Methodagent_spec
(self)
marinegym/envs/isaac_env.py:200
Methodapply_action
(self, actions: torch.Tensor)
marinegym/robots/drone/underwaterVehicle.py:172
Functioncircle
(t, c)
marinegym/envs/utils/__init__.py:176
Functionclass_to_dict
Convert an object into dictionary recursively. Note: Ignores all names starting with "__" (i.e. built-in methods). Args: obj
marinegym/sensors/config.py:121
Functionclear_stage
Deletes all prims in the stage without populating the undo command buffer Args: predicate (typing.Optional[typing.Callable[[str], bool]],
marinegym/envs/utils/stage.py:92
Functionclip_grad_norm_
( grads: _tensor_or_tensors, max_norm: float, norm_type: float = 2.0, error_if_nonfinite: bool
marinegym/learning/utils/clip_grad.py:33
Functionclose_stage
Closes the current opened USD stage. Args: callback_fn (typing.Callable, optional): Callback function to call while closing. Defaults to
marinegym/envs/utils/stage.py:263
Methodcompute
( self, root_state: torch.Tensor, target_pos: torch.Tensor=None, target_vel: t
marinegym/controllers/lee_position_controller.py:117
Methodcompute
(self, *args, **kwargs)
marinegym/controllers/controller.py:21
Functioncompute_gae
( reward: torch.Tensor, # [N, T, k] done: torch.Tensor, # [N, T, 1] value: torch.Tensor, # [N,
marinegym/learning/utils/gae.py:27
Functioncompute_gae_
( reward: torch.Tensor, # [T, N, k] done: torch.Tensor, # [T, N, 1] value: torch.Tensor, # [T,
marinegym/learning/utils/gae.py:54
Functioncpos
(p1: torch.Tensor, p2: torch.Tensor)
marinegym/utils/torch.py:74
Functioncpos
(p1: torch.Tensor, p2: torch.Tensor)
marinegym/envs/utils/helpers.py:37
Functioncreate_bar
( prim_path: str, length: float, translation=(0, 0, 0), from_prim: str = None, to_prim: st
marinegym/utils/scene.py:175
Functioncreate_bar
( prim_path: str, length: float, from_prim: Union[str, Usd.Prim]=None, to_prim: Union[str, Usd
marinegym/envs/utils/__init__.py:87
Functioncreate_ground_plane
Spawns a ground plane into the scene. This method spawns the default ground plane (grid plane) from Isaac Sim into the scene. It applies a ph
marinegym/utils/kit.py:20
Functioncreate_new_stage
Create a new stage. Returns: Usd.Stage: The created USD stage.
marinegym/envs/utils/stage.py:180
Functioncreate_new_stage_async
Create a new stage (asynchronous version).
marinegym/envs/utils/stage.py:189
Functioncreate_obstacle
( prim_path: str, prim_type: str, translation: Sequence[float], attributes: Dict, )
marinegym/envs/utils/__init__.py:57
Functioncreate_prim
Create a prim into current USD stage. The method applies specified transforms, the semantic label and set specified attributes. Args:
marinegym/envs/utils/prims.py:366
Functioncreate_rope
( xform_path: str = "/World/rope", translation=(0, 0, 0), from_prim: Union[str, Usd.Prim] = None,
marinegym/utils/scene.py:58
Functiondecorator
(func)
marinegym/learning/modules/networks.py:37
Functiondefault_predicate
(prim_path: str)
marinegym/envs/utils/stage.py:112
Functiondelete_prim
Remove the USD Prim and its decendants from the scene if able Args: prim_path (str): path of the prim in the stage
marinegym/envs/utils/prims.py:429
Methoddenormalize
(self, input_vector: torch.Tensor)
marinegym/learning/utils/valuenorm.py:99
Methoddenormalize
(self, input_vector: torch.Tensor)
marinegym/learning/utils/valuenorm.py:157
Functiondesign_scene
()
marinegym/utils/scene.py:37
Methoddone_spec
(self)
marinegym/utils/torchrl/env.py:76
Methoddownwash
A highly simplified downwash effect model. References: https://arxiv.org/pdf/2207.09645.pdf https://ieeexpl
marinegym/robots/drone/tiltRotor.py:574
Methodencode
(self, t)
marinegym/envs/utils/__init__.py:207
Methodentropy
(self)
marinegym/learning/modules/distributions.py:213
Functioneuler_rotate
(rpy: torch.Tensor, v: torch.Tensor)
marinegym/utils/torch.py:208
Functioneuler_to_quaternion
(euler: torch.Tensor)
marinegym/utils/math.py:77
Functionexpln
A smooth, continuous positive mapping presented in "State-Dependent Exploration for Policy Gradient Methods". https://people.idsia.ch/~juergen/ec
marinegym/learning/modules/distributions.py:45
Functionfind_matching_prim_paths
Find all the matching prim paths in the stage based on Regex expression. Args: prim_path_regex (str): The Regex expression for prim path.
marinegym/envs/utils/prims.py:200
Methodforward
( self, root_state: torch.Tensor, target_thrust: torch.Tensor, target_yaw_rate
marinegym/controllers/lee_position_controller.py:239
Methodforward
( self, root_state: torch.Tensor, target_rate: torch.Tensor, target_thrust: to
marinegym/controllers/lee_position_controller.py:330
Methodforward
(self, cmds: torch.Tensor)
marinegym/actuators/t200.py:34
Methodforward
(self, cmds: torch.Tensor)
marinegym/actuators/rotor_group.py:54
Methodforward
(self, cmds: torch.Tensor)
marinegym/actuators/m600.py:32
Methodforward
(self, cmds: torch.Tensor, flow_vels: torch.Tensor, fin_pos, fin_rot, fin_vel)
marinegym/actuators/fin200.py:30
Methodforward
(self, x: torch.Tensor)
marinegym/learning/modules/networks.py:70
Methodforward
(self, x: Tensor)
marinegym/learning/modules/networks.py:116
Methodforward
(self, tensordict: TensorDict)
marinegym/learning/modules/networks.py:156
Methodforward
(self, x: torch.Tensor, mask: torch.Tensor = None)
marinegym/learning/modules/networks.py:195
Methodforward
(self, x: torch.Tensor, mask: torch.Tensor = None)
marinegym/learning/modules/networks.py:234
Methodforward
Args: x: (batch, N, dim) padding_mask: (batch, N)
marinegym/learning/modules/networks.py:282
Methodforward
(self, x: TensorDict)
marinegym/learning/modules/networks.py:348
Methodforward
(self, x: torch.Tensor)
marinegym/learning/modules/networks.py:401
Methodforward
input: [N, L, H_in] or [N, H_in]
marinegym/learning/modules/rnn.py:44
Methodforward
(self, x)
marinegym/learning/modules/distributions.py:78
Methodforward
(self, tensor: torch.Tensor)
marinegym/learning/modules/distributions.py:200
Methodforward
(self, tensor: torch.Tensor)
marinegym/learning/modules/distributions.py:247
Methodforward
(self, tensor: torch.Tensor)
marinegym/learning/modules/distributions.py:279
← previousnext →201–300 of 434, ranked by callers