MCPcopy Index your code

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

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

⨍ Functions 452◇ Types & classes 49
↓ 17 callersFunctionprint_info
(*message)
utils/python_utils.py:134
↓ 16 callersMethodadd_scalar
(self, name, value, step)
utils/logger.py:37
↓ 14 callersMethodto
(self, device)
models/base_models.py:45
↓ 13 callersMethodnormalize
(self, arr:torch.tensor, un_norm = False)
utils/running_mean_std.py:81
↓ 12 callersMethodreset
(self)
utils/time_report.py:34
↓ 11 callersFunctionset_random_seed
(seed)
utils/python_utils.py:114
↓ 10 callersMethodsample
( self, batch_size: int, low: Union[float, torch.Tensor], high: Union[float,
generate/simulation_sampler.py:71
↓ 10 callersMethodset_eval_collisions
(self, eval_collisions)
envs/neural_environment.py:329
↓ 8 callersFunctionTimeProfiler
(time_report: TimeReport, timer_name: str)
utils/time_report.py:100
↓ 8 callersMethodclose
(self)
envs/neural_environment.py:300
↓ 8 callersMethodset_env_mode
(self, env_mode)
envs/neural_environment.py:314
↓ 7 callersMethodstep
( self, actions: torch.Tensor, env_mode = None )
envs/neural_environment.py:358
↓ 6 callersMethodend_video_export
(self)
envs/neural_environment.py:489
↓ 6 callersMethodflush
(self)
utils/logger.py:43
↓ 6 callersFunctionquat_mul
(a, b)
utils/torch_utils.py:20
↓ 6 callersMethodrender
(self)
envs/neural_environment.py:509
↓ 6 callersMethodstart_video_export
(self, video_export_filename)
envs/neural_environment.py:476
↓ 5 callersMethod_update_states
(self, states: Optional[torch.Tensor] = None)
envs/neural_environment.py:338
↓ 5 callersFunctionformat_dict
(dict, precision = 8)
utils/python_utils.py:82
↓ 5 callersMethodstate
Shortcut to the current state
envs/warp_sim_envs/environment.py:529
↓ 5 callersMethodwrap2PI
(self, states, is_continuous_dof = None)
integrators/integrator_neural.py:515
↓ 4 callersMethod__init__
(self, config)
models/model_transformer.py:90
↓ 4 callersFunctionangle_normalize
(x: float)
envs/warp_sim_envs/env_cartpole.py:28
↓ 4 callersMethodeval
(self, epoch)
algorithms/vanilla_trainer.py:617
↓ 4 callersMethodget_abstract_contacts
(self, model)
integrators/integrator_neural.py:408
↓ 4 callersMethodinit_rnn
(self, batch_size)
models/models.py:268
↓ 4 callersFunctionprint_ok
(*message)
utils/python_utils.py:126
↓ 4 callersMethodprocess_neural_model_inputs
(self, model_inputs)
integrators/integrator_neural.py:443
↓ 4 callersFunctionquat_inv
(a)
utils/torch_utils.py:45
↓ 4 callersFunctionquat_rotate
(q, v)
utils/torch_utils.py:51
↓ 4 callersFunctionquat_rotate_inverse
(q, v)
utils/torch_utils.py:63
↓ 4 callersMethodreset
( self, initial_states: Optional[torch.Tensor] = None )
envs/neural_environment.py:447
↓ 4 callersMethodsave_model
(self, filename = None)
algorithms/vanilla_trainer.py:741
↓ 4 callersMethodto
(self, device)
utils/running_mean_std.py:37
↓ 4 callersMethodupdate
( self, arr: torch.tensor, batch_dim = False, time_dim = False )
utils/running_mean_std.py:45
↓ 3 callersMethodadd_timers
(self, timer_names)
utils/time_report.py:57
↓ 3 callersMethodafter_update
(self)
envs/warp_sim_envs/environment.py:522
↓ 3 callersMethodbefore_update
(self)
envs/warp_sim_envs/environment.py:519
↓ 3 callersMethodcustomize_state
(self, state: wp.sim.State)
envs/warp_sim_envs/environment.py:640
↓ 3 callersMethodevaluate_action_mode
( self, num_traj = 10, eval_mode = 'rollout', env_mode = 'neural', pas
utils/evaluator.py:79
↓ 3 callersMethodone_epoch
( self, train: bool, dataloader, dataloader_iter, num_batches,
algorithms/vanilla_trainer.py:355
↓ 3 callersMethodprint
(self, string_mode = False, in_second = True)
utils/time_report.py:38
↓ 3 callersFunctionprint_warning
(*message)
utils/python_utils.py:130
↓ 3 callersMethodsample_trajectories_joint_act_mode
( self, num_transitions: int, trajectory_length: int, passive: bool = False,
generate/trajectory_sampler.py:67
↓ 3 callersMethodto
(self, device)
models/models.py:261
↓ 3 callersMethodupdate
(self)
envs/abstract_contact_environment.py:334
↓ 3 callersMethodwrap2PI
(self, states)
envs/neural_environment.py:307
↓ 2 callersMethod__init__
(self, in_features, network_cfg, device='cuda:0')
models/base_models.py:22
↓ 2 callersMethod_convert_states_w2b
( self, root_body_q, # (B, T, 7) states, # (B, T, dof_states) translation_only
integrators/integrator_neural.py:899
↓ 2 callersMethodacquire_states_to_torch
(self, warp_states: State, torch_states: torch.Tensor)
integrators/integrator_neural.py:526
↓ 2 callersMethodassign_states_from_torch
(self, warp_state: State, torch_states: torch.Tensor)
integrators/integrator_neural.py:540
↓ 2 callersFunctionconstruct_env
(env_specs, device, args)
eval/eval_rl/run_rl.py:161
↓ 2 callersFunctionconstruct_rlg_config
(rl_cfg)
eval/eval_rl/run_rl.py:213
↓ 2 callersMethodconvert_next_states_to_prediction_orientation_dofs
( self, states, next_states, prediction )
integrators/integrator_neural.py:829
↓ 2 callersMethodconvert_next_states_to_prediction_regular_dofs
( self, states, next_states, is_continuous_dof, prediction )
integrators/integrator_neural.py:795
↓ 2 callersMethodconvert_prediction_to_next_states
(self, states, prediction, dt = None)
integrators/integrator_neural.py:566
↓ 2 callersMethodconvert_prediction_to_next_states_orientation_dofs
( self, states, prediction, next_states )
integrators/integrator_neural.py:671
↓ 2 callersMethodconvert_prediction_to_next_states_regular_dofs
( self, states, prediction, next_states )
integrators/integrator_neural.py:639
↓ 2 callersFunctiondelta_quat_world
(quat0, quat1)
utils/torch_utils.py:82
↓ 2 callersMethodembed_states
(self, states, states_embedding=None)
integrators/integrator_neural.py:1077
↓ 2 callersFunctionevaluate_policy
(runner, policy_path)
eval/eval_rl/run_rl.py:254
↓ 2 callersMethodextract_input_features
(self, input_dict)
models/models.py:182
↓ 2 callersMethodget_contact_masks
( self, contact_depths, # (num_envs, (T), num_contacts_per_env) contact_thickness # (
integrators/integrator_neural.py:388
↓ 2 callersMethodget_neural_model_inputs
(self)
integrators/integrator_neural.py:494
↓ 2 callersMethodget_num_params
Return the number of parameters in the model. For non-embedding count (default), the position embeddings get subtracted. The
models/model_transformer.py:163
↓ 2 callersFunctionget_time_stamp
()
utils/python_utils.py:104
↓ 2 callersFunctiongrad_norm
(params)
utils/torch_utils.py:159
↓ 2 callersMethodmodel
(self)
envs/neural_environment.py:246
↓ 2 callersFunctionnum_params_torch_model
(model)
utils/torch_utils.py:156
↓ 2 callersMethodpreprocess_data_batch
(self, data)
algorithms/vanilla_trainer.py:278
↓ 2 callersFunctionquat_rotate
(q, v)
envs/warp_sim_envs/env_cube_toss.py:145
↓ 2 callersMethodreset_states_history
(self)
integrators/integrator_neural_stateful.py:39
↓ 2 callersMethodreset_timer
(self, timer_name = None)
utils/time_report.py:61
↓ 2 callersMethodsave_usd
(self)
envs/neural_environment.py:532
↓ 2 callersMethodseed
Sets the random seeds for all environments, based on a given seed. Each individual environment will still get its own seed, by increm
envs/rlgames_env_wrapper.py:274
↓ 2 callersMethodstep
( self, state: wp.sim.State, next_state: wp.sim.State, control: wp.sim.Control
envs/warp_sim_envs/environment.py:710
↓ 2 callersMethodstep_with_joint_act
( self, joint_acts: torch.Tensor, env_mode = None )
envs/neural_environment.py:411
↓ 2 callersMethodtrain
(self)
algorithms/vanilla_trainer.py:439
↓ 2 callersMethodupdate_sample_sequence_length
(self, sample_sequence_length)
utils/datasets.py:158
↓ 1 callersMethod__init__
( self, input_sample, output_dim, input_cfg, network_cfg, devi
models/models.py:58
↓ 1 callersMethod__setattr__
(self, name, value)
envs/abstract_contact_environment.py:226
↓ 1 callersMethod_build_dof_types
(self)
integrators/integrator_neural.py:237
↓ 1 callersMethod_convert_contacts_w2b
( self, root_body_q, # (B, T, num_contacts, 7) contact_points_1, # (B, T, num_contacts
integrators/integrator_neural.py:863
↓ 1 callersMethod_convert_gravity_w2b
( self, root_body_q, # (B, T, 7) gravity_dir, # (B, T, 3) translation_only
integrators/integrator_neural.py:935
↓ 1 callersMethod_init_prediction
(self)
integrators/integrator_neural.py:293
↓ 1 callersMethod_init_state_embedding
(self)
integrators/integrator_neural.py:280
↓ 1 callersMethod_update_states
(self, model: Model, warp_states: State, joint_act)
integrators/integrator_neural.py:377
↓ 1 callersFunctionadd_additional_params
(parser)
train/train.py:33
↓ 1 callersMethodadd_timer
(self, timer_name)
utils/time_report.py:54
↓ 1 callersMethodafter_init
(self)
envs/warp_sim_envs/env_ant.py:418
↓ 1 callersMethodafter_init
(self)
envs/warp_sim_envs/env_anymal.py:513
↓ 1 callersMethodafter_step
( self, state: wp.sim.State, next_state: wp.sim.State, control: wp.sim.Control
envs/warp_sim_envs/environment.py:510
↓ 1 callersMethodapply_pd_control
( self, control_out: wp.array = None, body_f: wp.array = None, target_q: wp.ar
envs/warp_sim_envs/environment.py:581
↓ 1 callersMethodassign_control
( self, actions: wp.array, control: wp.sim.Control, state: wp.sim.State, )
envs/warp_sim_envs/environment.py:556
↓ 1 callersMethodbefore_model_forward
(self)
integrators/integrator_neural.py:326
↓ 1 callersMethodbefore_step
( self, state: wp.sim.State, next_state: wp.sim.State, control: wp.sim.Control
envs/warp_sim_envs/environment.py:501
↓ 1 callersMethodbuild_index
(self)
utils/datasets.py:162
↓ 1 callersMethodcalculate_error
( self, target_next_states, # (T - 1, B, state_dim) rollout_states # (T - 1, B, state
utils/evaluator.py:273
↓ 1 callersMethodclamp_actions
(self, actions, action_limits)
envs/rlgames_env_wrapper.py:162
↓ 1 callersFunctioncollect_dataset
( env_name, num_envs, initial_states_source, num_transitions, trajectory_length, d
generate/generate_dataset_contact_free.py:36
next →1–100 of 452, ranked by callers