Functions — github.com/NVlabs/neural-robot-dynamics

hub / github.com/NVlabs/neural-robot-dynamics / functions

Functions452 in github.com/NVlabs/neural-robot-dynamics

⨍ Functions 452◇ Types & classes 49

Methodrender

(self, state=None)

envs/warp_sim_envs/environment.py:794

Functionresample_targets

( randomize_targets: bool, kernel_seed: int, env_to_resample: wp.array(dtype=wp.bool), env_off

envs/warp_sim_envs/env_franka_panda.py:135

Methodreset

(self)

envs/rlgames_env_wrapper.py:152

Methodreset

(self)

envs/warp_sim_envs/environment.py:820

Methodreset

(self)

integrators/integrator_neural_stateful.py:71

Methodreset

(self)

integrators/integrator_neural_rnn.py:35

Methodreset

(self)

integrators/integrator_neural.py:323

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_ant.py:228

Functionreset_anymal

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_anymal.py:337

Functionreset_anymal_dataset

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_anymal.py:239

Functionreset_cube

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_cube_toss.py:52

Methodreset_envs

Print distance for the envs to be reset.

envs/warp_sim_envs/env_ant.py:430

Methodreset_envs

(self, env_ids: wp.array = None)

envs/warp_sim_envs/env_cartpole.py:247

Methodreset_envs

(self, env_ids: wp.array = None)

envs/warp_sim_envs/env_pendulum_with_contact.py:212

Methodreset_envs

Print distance for the envs to be reset.

envs/warp_sim_envs/env_franka_panda.py:393

Methodreset_envs

Reset environments where env_ids buffer indicates True. Resets all envs if env_ids is None.

envs/warp_sim_envs/env_anymal.py:525

Methodreset_envs

Reset environments where env_ids buffer indicates True. Resets all envs if env_ids is None.

envs/warp_sim_envs/env_cube_toss.py:278

Functionreset_generalized_coords

( reset: wp.array(dtype=wp.bool), joint_q_init: wp.array(dtype=wp.float32), joint_qd_init: wp.arra

envs/warp_sim_envs/environment.py:148

Methodreset_hidden_states

(self, batch_indices=None, hidden_states=None)

models/base_models.py:79

Functionreset_init_state_franka

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_franka_panda.py:166

Functionreset_init_state_pendulum

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_pendulum_with_contact.py:25

Functionreset_init_state_single_cartpole

( reset: wp.array(dtype=wp.bool), seed: int, random_reset: bool, dof_q_per_env: int, dof_q

envs/warp_sim_envs/env_cartpole.py:93

Functionreset_maximal_coords

( reset: wp.array(dtype=wp.bool), body_q_init: wp.array(dtype=wp.transform), body_qd_init: wp.arra

envs/warp_sim_envs/environment.py:128

Methodreset_states_history

(self)

integrators/integrator_neural_transformer.py:35

Methodreset_timer

(self)

envs/abstract_contact_environment.py:361

Methodrnn_hidden_states_size

envs/neural_environment.py:220

Methodsample

( self, batch_size: int, low: Union[float, torch.Tensor], high: Union[float,

generate/simulation_sampler.py:91

Methodsample

( self, batch_size: int, low: Union[float, torch.Tensor], high: Union[float,

generate/simulation_sampler.py:110

Methodsample_initial_states

(self, num_envs, initial_states)

generate/trajectory_sampler_ant.py:37

Methodset_attr

Set attribute inside vectorized environments. :param attr_name: (str) The name of attribute to assign new value :param value

envs/rlgames_env_wrapper.py:251

Methodset_eval_collisions

(self, eval_collisions)

envs/abstract_contact_environment.py:302

Methodset_neural_model

(self, neural_model)

integrators/integrator_neural.py:315

Methodset_torch_state

( self, torch_state, state: wp.sim.State = None, eval_fk: bool = True, )

envs/warp_sim_envs/env_cube_toss.py:262

Methodshuffle

(self)

utils/datasets.py:104

Methodsim_states

(self)

envs/neural_environment.py:229

Functionsingle_cartpole_cost

( joint_q: wp.array(dtype=wp.float32), joint_qd: wp.array(dtype=wp.float32), joint_act: wp.array(d

envs/warp_sim_envs/env_cartpole.py:32

Methodstate_dim

(self)

envs/neural_environment.py:172

Methodstate_dim

(self)

generate/simulation_sampler.py:224

Methodstep

Step the environments with the given action :param actions: ([int] or [float]) the action :return: ([int] or [float], [float

envs/rlgames_env_wrapper.py:166

models/base_models.py:121

Methodto

(self, device)

models/base_models.py:184

Methodtorch_device

(self)

envs/neural_environment.py:216

Methodtorch_device

(self)

generate/simulation_sampler.py:236

Methodupdate

(self, obs: dict, batch_dim = False, time_dim = False)

utils/running_mean_std.py:107

Methodupdate_eval_horizon

(self, eval_horizon)

utils/evaluator.py:67

Methodupdate_grad

(self)

envs/abstract_contact_environment.py:356

Methodupdate_grad

(self)

envs/warp_sim_envs/environment.py:778

Methodupdate_nograd

(self)

envs/warp_sim_envs/environment.py:741

Methoduses_generalized_coordinates

(self)

envs/warp_sim_envs/environment.py:485

← previous401–452 of 452, ranked by callers