MCPcopy Create free account

hub / github.com/AgibotTech/agibot_x1_train / functions

Functions205 in github.com/AgibotTech/agibot_x1_train

↓ 7 callersFunctionget_euler_xyz_tensor
(quat)
humanoid/envs/base/legged_robot.py:77
↓ 6 callersFunctionclass_to_dict
(obj)
humanoid/utils/helpers.py:44
↓ 6 callersFunctionget_args
()
humanoid/utils/helpers.py:165
↓ 6 callersMethodget_cfgs
(self, name)
humanoid/utils/task_registry.py:59
↓ 5 callersMethod_get_stance_mask
(self)
humanoid/envs/x1/x1_dh_stand_env.py:149
↓ 5 callersMethodclear
(self)
humanoid/algo/ppo/rollout_storage.py:125
↓ 4 callersMethodload
(self, path, load_optimizer=True)
humanoid/algo/ppo/dh_on_policy_runner.py:301
↓ 4 callersMethodsave
(self, path, infos=None)
humanoid/algo/ppo/dh_on_policy_runner.py:289
↓ 3 callersMethod_get_phase
(self)
humanoid/envs/x1/x1_dh_stand_env.py:136
↓ 3 callersMethod_refresh_actor_dof_props
Refresh the dof properties of the actor in the given environments, i.e. dof friction, damping, armature
humanoid/envs/base/legged_robot.py:606
↓ 3 callersMethodadd_terrain_to_map
(self, terrain, row, col)
humanoid/utils/terrain.py:216
↓ 3 callersMethodevaluate
(self, critic_observations, **kwargs)
humanoid/algo/ppo/actor_critic_dh.py:183
↓ 3 callersMethodplot_states
(self)
humanoid/utils/logger.py:63
↓ 3 callersFunctionquat_apply_yaw
(quat, vec)
humanoid/utils/math.py:40
↓ 3 callersMethodrandomize_dof_props
(self, env_ids)
humanoid/envs/base/legged_robot.py:411
↓ 3 callersMethodstep
(self, actions: torch.Tensor)
humanoid/algo/vec_env.py:53
↓ 2 callersMethod_create_envs
Creates environments: 1. loads the robot URDF/MJCF asset, 2. For each environment 2.1 creates the environme
humanoid/envs/base/legged_robot.py:1065
↓ 2 callersMethod_create_ground_plane
Adds a ground plane to the simulation, sets friction and restitution based on the cfg.
humanoid/envs/base/legged_robot.py:1000
↓ 2 callersMethod_create_heightfield
Adds a heightfield terrain to the simulation, sets parameters based on the cfg.
humanoid/envs/base/legged_robot.py:1010
↓ 2 callersMethod_create_trimesh
Adds a triangle mesh terrain to the simulation, sets parameters based on the cfg. #
humanoid/envs/base/legged_robot.py:1029
↓ 2 callersMethod_get_heights
Samples heights of the terrain at required points around each robot. The points are offset by the base's position and rotated by the base
humanoid/envs/base/legged_robot.py:1306
↓ 2 callersMethod_resample_commands
Randommly select commands of some environments Args: env_ids (List[int]): Environments ids for which new commands are needed
humanoid/envs/x1/x1_dh_stand_env.py:190
↓ 2 callersMethod_resample_commands
Randommly select commands of some environments Args: env_ids (List[int]): Environments ids for which new commands are needed
humanoid/envs/base/legged_robot.py:679
↓ 2 callersMethod_reset_dofs
Resets DOF position and velocities of selected environmments Positions are randomly selected within 0.5:1.5 x default positions. Velo
humanoid/envs/base/legged_robot.py:744
↓ 2 callersMethod_reset_root_states
Resets ROOT states position and velocities of selected environmments Sets base position based on the curriculum Selects rando
humanoid/envs/base/legged_robot.py:761
↓ 2 callersMethod_update_terrain_curriculum
Implements the game-inspired curriculum. Args: env_ids (List[int]): ids of environments being reset
humanoid/envs/base/legged_robot.py:801
↓ 2 callersMethodact
(self, obs, critic_obs)
humanoid/algo/ppo/dh_ppo.py:98
↓ 2 callersMethodexport
(self, path)
humanoid/scripts/export_policy_dh.py:64
↓ 2 callersMethodget_actions_log_prob
(self, actions)
humanoid/algo/ppo/actor_critic_dh.py:172
↓ 2 callersFunctionget_euler_xyz_tensor
(quat)
humanoid/envs/x1/x1_dh_stand_env.py:72
↓ 2 callersFunctionget_load_path
(root, load_run=-1, checkpoint=-1)
humanoid/utils/helpers.py:114
↓ 2 callersMethodget_observations
(self)
humanoid/algo/vec_env.py:59
↓ 2 callersMethodlog
(self, locs, width=80, pad=35)
humanoid/algo/ppo/dh_on_policy_runner.py:181
↓ 2 callersMethodlog_states
(self, dict)
humanoid/utils/logger.py:49
↓ 2 callersMethodmake_alg_runner
Creates the training algorithm either from a registered namme or from the provided config file. Args: env (isaacgym.VecTaskPyth
humanoid/utils/task_registry.py:106
↓ 2 callersMethodmake_env
Creates an environment either from a registered namme or from the provided config file. Args: name (string): Name of a registere
humanoid/utils/task_registry.py:66
↓ 2 callersMethodmake_terrain
(self, choice, difficulty)
humanoid/utils/terrain.py:128
↓ 2 callersMethodrandomize_lag_props
random add lag
humanoid/envs/base/legged_robot.py:293
↓ 2 callersMethodrender
(self, sync_frame_time=True)
humanoid/envs/base/base_task.py:155
↓ 2 callersMethodreset
(self, dones=None)
humanoid/algo/ppo/actor_critic_dh.py:142
↓ 2 callersMethodreset_idx
Reset some environments. Calls self._reset_dofs(env_ids), self._reset_root_states(env_ids), and self._resample_commands(env_ids)
humanoid/envs/base/legged_robot.py:219
↓ 2 callersMethodset_camera
Set camera position and direction
humanoid/envs/base/legged_robot.py:365
↓ 2 callersFunctionupdate_cfg_from_args
(env_cfg, cfg_train, args)
humanoid/utils/helpers.py:139
↓ 2 callersMethodupdate_command_curriculum
Implements a curriculum of increasing commands Args: env_ids (List[int]): ids of environments being reset
humanoid/envs/base/legged_robot.py:823
↓ 2 callersFunctionwrap_to_pi
(angles)
humanoid/utils/math.py:47
↓ 1 callersMethod__init__
(self)
humanoid/algo/ppo/rollout_storage.py:37
↓ 1 callersMethod_add_ext_force
Apply force and torque on fix body
humanoid/envs/base/legged_robot.py:665
↓ 1 callersMethod_compute_torques
Compute torques from actions. Actions can be interpreted as position or velocity targets given to a PD controller, or directly as scaled
humanoid/envs/base/legged_robot.py:696
↓ 1 callersMethod_draw_debug_vis
Draws visualizations for dubugging (slows down simulation a lot). Default behaviour: draws height measurement points
humanoid/envs/base/legged_robot.py:1268
↓ 1 callersMethod_get_env_origins
Sets environment origins. On rough terrain the origins are defined by the terrain platforms. Otherwise create a grid.
humanoid/envs/base/legged_robot.py:1226
↓ 1 callersMethod_get_noise_scale_vec
Sets a vector used to scale the noise added to the observations. [NOTE]: Must be adapted when changing the observations structure
humanoid/envs/x1/x1_dh_stand_env.py:329
↓ 1 callersMethod_init_buffers
Initialize torch tensors which will contain simulation states and processed quantities
humanoid/envs/base/legged_robot.py:835
↓ 1 callersMethod_init_height_points
Returns points at which the height measurments are sampled (in base frame) Returns: [torch.Tensor]: Tensor of shape (num_envs, s
humanoid/envs/base/legged_robot.py:1289
↓ 1 callersMethod_parse_cfg
Parse simulation, reward, command, terrain, random configuration
humanoid/envs/base/legged_robot.py:1253
↓ 1 callersMethod_post_physics_step_callback
Callback called before computing terminations, rewards, and observations Default behaviour: Compute ang vel command based on target and h
humanoid/envs/base/legged_robot.py:632
↓ 1 callersMethod_prepare_reward_function
Prepares a list of reward functions, which will be called to compute the total reward. Looks for self._reward_<REWARD_NAME>, where <REWAR
humanoid/envs/base/legged_robot.py:975
↓ 1 callersMethod_process_dof_props
Callback allowing to store/change/randomize the DOF properties of each environment. Called During environment creation. Base
humanoid/envs/base/legged_robot.py:505
↓ 1 callersMethod_process_rigid_body_props
(self, props, env_id)
humanoid/envs/base/legged_robot.py:535
↓ 1 callersMethod_process_rigid_shape_props
Callback allowing to store/change/randomize the rigid shape properties of each environment. Called During environment creation.
humanoid/envs/base/legged_robot.py:472
↓ 1 callersMethod_push_robots
Random pushes the robots. Emulates an impulse by setting a randomized base velocity.
humanoid/envs/x1/x1_dh_stand_env.py:120
↓ 1 callersMethod_push_robots
Random pushes the robots. Emulates an impulse by setting a randomized base velocity.
humanoid/envs/base/legged_robot.py:791
↓ 1 callersMethod_reward_termination
(self)
humanoid/envs/x1/x1_dh_stand_env.py:870
↓ 1 callersMethod_save_hidden_states
(self, hidden_states)
humanoid/algo/ppo/rollout_storage.py:108
↓ 1 callersMethodadd_transitions
(self, transition: Transition)
humanoid/algo/ppo/rollout_storage.py:91
↓ 1 callersMethodcheck_termination
Check if environments need to be reset
humanoid/envs/base/legged_robot.py:206
↓ 1 callersMethodcompute_observations
(self)
humanoid/envs/x1/x1_dh_stand_env.py:358
↓ 1 callersMethodcompute_ref_state
(self)
humanoid/envs/x1/x1_dh_stand_env.py:274
↓ 1 callersMethodcompute_returns
(self, last_critic_obs)
humanoid/algo/ppo/dh_ppo.py:122
↓ 1 callersMethodcompute_reward
Compute rewards Calls each reward function which had a non-zero scale (processed in self._prepare_reward_function()) adds eac
humanoid/envs/base/legged_robot.py:345
↓ 1 callersFunctioncopysign_new
(a, b)
humanoid/envs/x1/x1_dh_stand_env.py:45
↓ 1 callersFunctioncopysign_new
(a, b)
humanoid/envs/base/legged_robot.py:50
↓ 1 callersMethodcreate_sim
Creates simulation, terrain and evironments
humanoid/envs/base/legged_robot.py:1045
↓ 1 callersMethodcuriculum
(self)
humanoid/utils/terrain.py:103
↓ 1 callersFunctionexport_onnx
(args)
humanoid/scripts/export_onnx_dh.py:61
↓ 1 callersFunctionexport_policy
(args)
humanoid/scripts/export_policy_dh.py:70
↓ 1 callersFunctionexport_policy_as_jit
(actor_critic, path)
humanoid/utils/helpers.py:246
↓ 1 callersFunctiongap_terrain
(terrain, gap_size, platform_size=1.)
humanoid/utils/terrain.py:236
↓ 1 callersMethodgenerate_gait_time
(self,envs)
humanoid/envs/x1/x1_dh_stand_env.py:165
↓ 1 callersFunctionget_euler_rpy
(q)
humanoid/envs/x1/x1_dh_stand_env.py:51
↓ 1 callersFunctionget_euler_rpy
(q)
humanoid/envs/base/legged_robot.py:56
↓ 1 callersMethodget_inference_policy
(self, device=None)
humanoid/algo/ppo/dh_on_policy_runner.py:310
↓ 1 callersFunctionget_load_path
(root, load_run=-1, checkpoint=-1)
humanoid/scripts/export_onnx_dh.py:40
↓ 1 callersFunctionget_obs
Extracts an observation from the mujoco data structure
humanoid/scripts/sim2sim.py:108
↓ 1 callersMethodget_privileged_observations
(self)
humanoid/algo/vec_env.py:62
↓ 1 callersMethodget_task_class
(self, name: str)
humanoid/utils/task_registry.py:56
↓ 1 callersMethodinit_member_classes
(obj)
humanoid/envs/base/base_config.py:41
↓ 1 callersMethodinit_randomize_props
Initialize torch tensors for random properties
humanoid/envs/base/legged_robot.py:1169
↓ 1 callersMethodinit_storage
(self, num_envs, num_transitions_per_env, actor_obs_shape, critic_obs_shape, action_shape)
humanoid/algo/ppo/dh_ppo.py:89
↓ 1 callersMethodlearn
(self, num_learning_iterations, init_at_random_ep_len=False)
humanoid/algo/ppo/dh_on_policy_runner.py:97
↓ 1 callersMethodlog_rewards
(self, dict, num_episodes)
humanoid/utils/logger.py:53
↓ 1 callersMethodlog_state
(self, key, value)
humanoid/utils/logger.py:46
↓ 1 callersMethodmini_batch_generator
(self, num_mini_batches, num_epochs=8)
humanoid/algo/ppo/rollout_storage.py:152
↓ 1 callersFunctionparse_sim_params
(args, cfg)
humanoid/utils/helpers.py:89
↓ 1 callersFunctionpd_control
Calculates torques from position commands
humanoid/scripts/sim2sim.py:129
↓ 1 callersFunctionpit_terrain
(terrain, depth, platform_size=1.)
humanoid/utils/terrain.py:250
↓ 1 callersFunctionplay
(args)
humanoid/scripts/play.py:85
↓ 1 callersMethodpost_physics_step
check terminations, compute observations and rewards calls self._post_physics_step_callback() for common computations calls
humanoid/envs/base/legged_robot.py:164
↓ 1 callersMethodprocess_env_step
(self, rewards, dones, infos)
humanoid/algo/ppo/dh_ppo.py:110
↓ 1 callersFunctionquaternion_to_euler_array
(quat)
humanoid/scripts/sim2sim.py:86
↓ 1 callersMethodrandomize_rigid_body_props
Randomise some of the rigid body properties of the actor in the given environments, i.e. sample the mass, centre of mass position, fricti
humanoid/envs/base/legged_robot.py:373
next →1–100 of 205, ranked by callers