MCPcopy Create free account

hub / github.com/LeCAR-Lab/SoFTA / functions

Functions568 in github.com/LeCAR-Lab/SoFTA

Method_reward_waist_joint_angle_freeze
(self)
humanoidverse/envs/locomotion/locomotion_stand_ma_nouse.py:294
Method_rollout_step
(self, obs_dict)
humanoidverse/agents/ppo_ma/ppo_ma.py:221
Method_run_server
(self)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:149
Method_save_hidden_states
(self, hidden_states)
humanoidverse/agents/modules/data_utils.py:130
Method_setup_models_and_optimizer
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:89
Method_setup_simulator_control
(self)
humanoidverse/envs/locomotion/locomotion_ma_stand_gait_ee.py:105
Method_setup_simulator_control
(self)
humanoidverse/envs/locomotion/locomotion_ma.py:97
Method_setup_simulator_control
(self)
humanoidverse/envs/locomotion/locomotion_ma_stand_gait.py:26
Method_setup_storage
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:114
Method_train_mode
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:141
Method_training_step
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:346
Method_update_tasks_callback
Callback called before computing terminations, rewards, and observations Default behaviour: Compute ang vel command based on target and h
humanoidverse/envs/locomotion/locomotion_ma_stand.py:41
Method_update_tasks_callback
Callback called before computing terminations, rewards, and observations Default behaviour: Compute ang vel command based on target and h
humanoidverse/envs/locomotion/locomotion_ma.py:100
Method_update_tasks_callback
Callback called before computing terminations, rewards, and observations Default behaviour: Compute ang vel command based on target and h
humanoidverse/envs/locomotion/locomotion_stand_ma_nouse.py:54
Methodact
(self, obs_dict, **kwargs)
humanoidverse/agents/modules/ppo_modules.py:128
Methodact_inference
(self, actor_obs)
humanoidverse/agents/ppo_ma/ppo_ma.py:616
Methodaction_mean
(self)
humanoidverse/agents/modules/ppo_modules.py:53
Methodaction_mean
(self)
humanoidverse/agents/modules/ppo_modules.py:122
Methodaction_std
(self)
humanoidverse/agents/modules/ppo_modules.py:57
Methodadd
(self, data_dict)
humanoidverse/utils/average_meters.py:66
Methodadd
(self, key: str, value: Tensor)
humanoidverse/envs/env_utils/history_handler.py:36
Methodadd_limit_lines
(row, col, lower, upper, color=YELLOW)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:203
Methodadd_multi_transitions
(self, transitions: MultiTransitions)
humanoidverse/agents/modules/data_utils.py:99
Methodapply_torques_at_dof
Applies the specified torques to the robot's degrees of freedom (DOF). Args: torques (tensor): Tensor containing torques
humanoidverse/simulator/base_simulator/base_simulator.py:121
Functionaxis_angle_to_quaternion
Convert rotations given as axis/angle to quaternions. Args: axis_angle: Rotations given as a vector in axis angle form,
isaac_utils/isaac_utils/rotations.py:516
Functioncalc_heading_quat
(q, w_last)
isaac_utils/isaac_utils/rotations.py:265
Functioncalc_heading_quat_inv
(q, w_last)
isaac_utils/isaac_utils/rotations.py:278
Functioncapture_stdout_to_loguru
()
humanoidverse/utils/logging.py:32
Functionclass_to_dict
(obj)
humanoidverse/utils/helpers.py:12
Functionclass_to_dict
(obj)
humanoidverse/envs/env_utils/general.py:6
Methodclear
(self)
humanoidverse/utils/average_meters.py:24
Methodclear
(self)
humanoidverse/agents/modules/data_utils.py:35
Methodclear
(self)
humanoidverse/agents/modules/data_utils.py:44
Functioncompute_returns
(self, rewards, values, dones, last_values, gamma, lam)
humanoidverse/agents/modules/data_utils.py:5
Methodcreate_envs
Creates and initializes environments with specified configurations. Args: num_envs (int): Number of environments to crea
humanoidverse/simulator/base_simulator/base_simulator.py:70
Functiondistance_l2
(root_pos, wp_pos)
humanoidverse/utils/common.py:123
Methoddraw_sphere
(self, pos, radius, color, env_id, pos_id=None)
humanoidverse/simulator/isaacgym/isaacgym.py:836
Methodentropy
(self)
humanoidverse/agents/modules/ppo_modules.py:61
Methodenv_step
(self, actions, extra_info=None)
humanoidverse/agents/base_algo/base_algo.py:23
Methodevaluate_policy
(self)
humanoidverse/agents/base_algo/base_algo.py:28
Methodevaluate_policy
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:579
Functionexport_policy_as_jit
(actor_critic, path, exported_policy_name)
humanoidverse/utils/inference_helpers.py:6
Functionexport_policy_as_jit
(actor_critic, path)
humanoidverse/utils/helpers.py:89
Functionexport_policy_as_onnx
(inference_model, path, exported_policy_name, example_obs_dict)
humanoidverse/utils/inference_helpers.py:13
Methodfind_rigid_body_indice
Finds the index of a specified rigid body. Args: body_name (str): Name of the rigid body to locate. Returns:
humanoidverse/simulator/base_simulator/base_simulator.py:93
Methodflat_terrain_func
(self, terrain, difficulty)
humanoidverse/envs/env_utils/terrain.py:142
Methodflush
(self)
humanoidverse/utils/logging.py:28
Methodforward
Dynamically creates a dictionary from the input keys and args.
humanoidverse/utils/inference_helpers.py:28
Methodforward
Dynamically creates a dictionary from the input keys and args.
humanoidverse/utils/inference_helpers.py:63
Methodforward
(self, x)
humanoidverse/utils/helpers.py:111
Methodforward
(self)
humanoidverse/agents/modules/ppo_modules.py:49
Methodforward
(self, input)
humanoidverse/agents/modules/modules.py:72
Methodforward
(self, obs_history)
humanoidverse/agents/modules/encoder_modules.py:16
Methodgap_terrain_func
(self, terrain, difficulty)
humanoidverse/envs/env_utils/terrain.py:115
Methodget_actions_log_prob
(self, actions)
humanoidverse/agents/modules/ppo_modules.py:125
Functionget_basis_vector
(q, v)
humanoidverse/utils/torch_utils.py:130
Functionget_basis_vector
(q: Tensor, v: Tensor, w_last: bool)
isaac_utils/isaac_utils/rotations.py:168
Methodget_commands
(self, env_ids)
humanoidverse/envs/env_utils/command_generator.py:17
Methodget_dof_limits_properties
Retrieves the DOF (degrees of freedom) limits and properties. Returns: Tuple of tensors representing position li
humanoidverse/simulator/base_simulator/base_simulator.py:84
Methodget_env_dim
(self)
humanoidverse/agents/modules/world_models.py:139
Functionget_euler_xyz
(q)
humanoidverse/utils/torch_utils.py:153
Functionget_euler_xyz
(q: Tensor, w_last: bool)
isaac_utils/isaac_utils/rotations.py:299
Methodget_mean
(self)
humanoidverse/utils/average_meters.py:31
Methodget_mppi_buffers
Get buffers for MPPI MPPI algo need to store the buffers to replicate environments
humanoidverse/envs/legged_base_task/legged_robot_base_ma.py:461
Methodget_statistics
(self)
humanoidverse/agents/modules/data_utils.py:151
Methodget_subplot_titles
()
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:167
Functionget_time_stamp
()
humanoidverse/utils/common.py:81
Functionheading_to_quat
(h_theta, w_last: bool)
isaac_utils/isaac_utils/rotations.py:142
Methodinference_model
(self)
humanoidverse/agents/base_algo/base_algo.py:20
Methodinference_model
(self)
humanoidverse/agents/ppo/ppo.py:464
Methodinference_model
(self)
humanoidverse/agents/ppo_ma/ppo_ma.py:568
Methodinit_weights
(sequential, scales)
humanoidverse/agents/modules/ppo_modules.py:42
Functionis_valid_quat
(q)
isaac_utils/isaac_utils/rotations.py:366
Methodlearn
(self)
humanoidverse/agents/base_algo/base_algo.py:13
Methodload
(self, path)
humanoidverse/agents/base_algo/base_algo.py:16
Methodload
(self, ckpt_path)
humanoidverse/agents/ppo_ma/ppo_ma.py:147
Methodload_assets
Loads the robot assets into the simulation environment. save self.num_dofs, self.num_bodies, self.dof_names, self.body_names
humanoidverse/simulator/base_simulator/base_simulator.py:59
Methodlog_rewards
(self, dict, num_episodes)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:133
Methodlow_obst_terrain_func
(self, terrain, difficulty)
humanoidverse/envs/env_utils/terrain.py:165
Functionmatrix_to_quaternion
w x y z Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3,
isaac_utils/isaac_utils/rotations.py:550
Methodmaze_terrain_func
(self, terrain, difficulty)
humanoidverse/envs/env_utils/terrain.py:175
Methodmean_and_clear
(self)
humanoidverse/utils/average_meters.py:81
Methodnext
(self, obs, action)
humanoidverse/agents/modules/world_models.py:22
Methodnext
(self, states, actions)
humanoidverse/agents/modules/world_models.py:83
Methodnext
(self, states, actions)
humanoidverse/agents/modules/world_models.py:165
Methodnext_task
(self)
humanoidverse/simulator/isaacgym/isaacgym.py:829
Functionnormalize
Normalize a value from a given range to a target range.
humanoidverse/utils/common.py:146
Functionnormalize_angle
(x)
humanoidverse/utils/torch_utils.py:104
Methodnormalize_weights
(weights)
humanoidverse/envs/legged_base_task/legged_robot_base_ma.py:170
Methodon_post_eval_env_step
(self, actor_state)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:102
Methodon_post_evaluate_policy
(self)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:52
Methodon_pre_eval_env_step
(self, actor_state)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:55
Methodon_pre_evaluate_policy
(self)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:43
Functionparse_model_args
(model_args_path)
humanoidverse/utils/common.py:92
Methodpit_terrain_func
(self, terrain, difficulty)
humanoidverse/envs/env_utils/terrain.py:131
Methodprepare_sim
Prepares the simulation environment and refreshes any relevant tensors.
humanoidverse/simulator/base_simulator/base_simulator.py:107
Functionprint_error
(*message)
humanoidverse/utils/common.py:64
Functionprint_info
(*message)
humanoidverse/utils/common.py:77
Functionprint_ok
(*message)
humanoidverse/utils/common.py:69
Methodprint_rewards
(self)
humanoidverse/agents/callbacks/analysis_plot_locomotion.py:274
← previousnext →401–500 of 568, ranked by callers