Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AgibotTech/agibot_x1_train
/ functions
Functions
205 in github.com/AgibotTech/agibot_x1_train
⨍
Functions
205
◇
Types & classes
62
↓ 7 callers
Function
get_euler_xyz_tensor
(quat)
humanoid/envs/base/legged_robot.py:77
↓ 6 callers
Function
class_to_dict
(obj)
humanoid/utils/helpers.py:44
↓ 6 callers
Function
get_args
()
humanoid/utils/helpers.py:165
↓ 6 callers
Method
get_cfgs
(self, name)
humanoid/utils/task_registry.py:59
↓ 5 callers
Method
_get_stance_mask
(self)
humanoid/envs/x1/x1_dh_stand_env.py:149
↓ 5 callers
Method
clear
(self)
humanoid/algo/ppo/rollout_storage.py:125
↓ 4 callers
Method
load
(self, path, load_optimizer=True)
humanoid/algo/ppo/dh_on_policy_runner.py:301
↓ 4 callers
Method
save
(self, path, infos=None)
humanoid/algo/ppo/dh_on_policy_runner.py:289
↓ 3 callers
Method
_get_phase
(self)
humanoid/envs/x1/x1_dh_stand_env.py:136
↓ 3 callers
Method
_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 callers
Method
add_terrain_to_map
(self, terrain, row, col)
humanoid/utils/terrain.py:216
↓ 3 callers
Method
evaluate
(self, critic_observations, **kwargs)
humanoid/algo/ppo/actor_critic_dh.py:183
↓ 3 callers
Method
plot_states
(self)
humanoid/utils/logger.py:63
↓ 3 callers
Function
quat_apply_yaw
(quat, vec)
humanoid/utils/math.py:40
↓ 3 callers
Method
randomize_dof_props
(self, env_ids)
humanoid/envs/base/legged_robot.py:411
↓ 3 callers
Method
step
(self, actions: torch.Tensor)
humanoid/algo/vec_env.py:53
↓ 2 callers
Method
_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 callers
Method
_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 callers
Method
_create_heightfield
Adds a heightfield terrain to the simulation, sets parameters based on the cfg.
humanoid/envs/base/legged_robot.py:1010
↓ 2 callers
Method
_create_trimesh
Adds a triangle mesh terrain to the simulation, sets parameters based on the cfg. #
humanoid/envs/base/legged_robot.py:1029
↓ 2 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
act
(self, obs, critic_obs)
humanoid/algo/ppo/dh_ppo.py:98
↓ 2 callers
Method
export
(self, path)
humanoid/scripts/export_policy_dh.py:64
↓ 2 callers
Method
get_actions_log_prob
(self, actions)
humanoid/algo/ppo/actor_critic_dh.py:172
↓ 2 callers
Function
get_euler_xyz_tensor
(quat)
humanoid/envs/x1/x1_dh_stand_env.py:72
↓ 2 callers
Function
get_load_path
(root, load_run=-1, checkpoint=-1)
humanoid/utils/helpers.py:114
↓ 2 callers
Method
get_observations
(self)
humanoid/algo/vec_env.py:59
↓ 2 callers
Method
log
(self, locs, width=80, pad=35)
humanoid/algo/ppo/dh_on_policy_runner.py:181
↓ 2 callers
Method
log_states
(self, dict)
humanoid/utils/logger.py:49
↓ 2 callers
Method
make_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 callers
Method
make_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 callers
Method
make_terrain
(self, choice, difficulty)
humanoid/utils/terrain.py:128
↓ 2 callers
Method
randomize_lag_props
random add lag
humanoid/envs/base/legged_robot.py:293
↓ 2 callers
Method
render
(self, sync_frame_time=True)
humanoid/envs/base/base_task.py:155
↓ 2 callers
Method
reset
(self, dones=None)
humanoid/algo/ppo/actor_critic_dh.py:142
↓ 2 callers
Method
reset_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 callers
Method
set_camera
Set camera position and direction
humanoid/envs/base/legged_robot.py:365
↓ 2 callers
Function
update_cfg_from_args
(env_cfg, cfg_train, args)
humanoid/utils/helpers.py:139
↓ 2 callers
Method
update_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 callers
Function
wrap_to_pi
(angles)
humanoid/utils/math.py:47
↓ 1 callers
Method
__init__
(self)
humanoid/algo/ppo/rollout_storage.py:37
↓ 1 callers
Method
_add_ext_force
Apply force and torque on fix body
humanoid/envs/base/legged_robot.py:665
↓ 1 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_init_buffers
Initialize torch tensors which will contain simulation states and processed quantities
humanoid/envs/base/legged_robot.py:835
↓ 1 callers
Method
_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 callers
Method
_parse_cfg
Parse simulation, reward, command, terrain, random configuration
humanoid/envs/base/legged_robot.py:1253
↓ 1 callers
Method
_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 callers
Method
_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 callers
Method
_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 callers
Method
_process_rigid_body_props
(self, props, env_id)
humanoid/envs/base/legged_robot.py:535
↓ 1 callers
Method
_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 callers
Method
_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 callers
Method
_push_robots
Random pushes the robots. Emulates an impulse by setting a randomized base velocity.
humanoid/envs/base/legged_robot.py:791
↓ 1 callers
Method
_reward_termination
(self)
humanoid/envs/x1/x1_dh_stand_env.py:870
↓ 1 callers
Method
_save_hidden_states
(self, hidden_states)
humanoid/algo/ppo/rollout_storage.py:108
↓ 1 callers
Method
add_transitions
(self, transition: Transition)
humanoid/algo/ppo/rollout_storage.py:91
↓ 1 callers
Method
check_termination
Check if environments need to be reset
humanoid/envs/base/legged_robot.py:206
↓ 1 callers
Method
compute_observations
(self)
humanoid/envs/x1/x1_dh_stand_env.py:358
↓ 1 callers
Method
compute_ref_state
(self)
humanoid/envs/x1/x1_dh_stand_env.py:274
↓ 1 callers
Method
compute_returns
(self, last_critic_obs)
humanoid/algo/ppo/dh_ppo.py:122
↓ 1 callers
Method
compute_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 callers
Function
copysign_new
(a, b)
humanoid/envs/x1/x1_dh_stand_env.py:45
↓ 1 callers
Function
copysign_new
(a, b)
humanoid/envs/base/legged_robot.py:50
↓ 1 callers
Method
create_sim
Creates simulation, terrain and evironments
humanoid/envs/base/legged_robot.py:1045
↓ 1 callers
Method
curiculum
(self)
humanoid/utils/terrain.py:103
↓ 1 callers
Function
export_onnx
(args)
humanoid/scripts/export_onnx_dh.py:61
↓ 1 callers
Function
export_policy
(args)
humanoid/scripts/export_policy_dh.py:70
↓ 1 callers
Function
export_policy_as_jit
(actor_critic, path)
humanoid/utils/helpers.py:246
↓ 1 callers
Function
gap_terrain
(terrain, gap_size, platform_size=1.)
humanoid/utils/terrain.py:236
↓ 1 callers
Method
generate_gait_time
(self,envs)
humanoid/envs/x1/x1_dh_stand_env.py:165
↓ 1 callers
Function
get_euler_rpy
(q)
humanoid/envs/x1/x1_dh_stand_env.py:51
↓ 1 callers
Function
get_euler_rpy
(q)
humanoid/envs/base/legged_robot.py:56
↓ 1 callers
Method
get_inference_policy
(self, device=None)
humanoid/algo/ppo/dh_on_policy_runner.py:310
↓ 1 callers
Function
get_load_path
(root, load_run=-1, checkpoint=-1)
humanoid/scripts/export_onnx_dh.py:40
↓ 1 callers
Function
get_obs
Extracts an observation from the mujoco data structure
humanoid/scripts/sim2sim.py:108
↓ 1 callers
Method
get_privileged_observations
(self)
humanoid/algo/vec_env.py:62
↓ 1 callers
Method
get_task_class
(self, name: str)
humanoid/utils/task_registry.py:56
↓ 1 callers
Method
init_member_classes
(obj)
humanoid/envs/base/base_config.py:41
↓ 1 callers
Method
init_randomize_props
Initialize torch tensors for random properties
humanoid/envs/base/legged_robot.py:1169
↓ 1 callers
Method
init_storage
(self, num_envs, num_transitions_per_env, actor_obs_shape, critic_obs_shape, action_shape)
humanoid/algo/ppo/dh_ppo.py:89
↓ 1 callers
Method
learn
(self, num_learning_iterations, init_at_random_ep_len=False)
humanoid/algo/ppo/dh_on_policy_runner.py:97
↓ 1 callers
Method
log_rewards
(self, dict, num_episodes)
humanoid/utils/logger.py:53
↓ 1 callers
Method
log_state
(self, key, value)
humanoid/utils/logger.py:46
↓ 1 callers
Method
mini_batch_generator
(self, num_mini_batches, num_epochs=8)
humanoid/algo/ppo/rollout_storage.py:152
↓ 1 callers
Function
parse_sim_params
(args, cfg)
humanoid/utils/helpers.py:89
↓ 1 callers
Function
pd_control
Calculates torques from position commands
humanoid/scripts/sim2sim.py:129
↓ 1 callers
Function
pit_terrain
(terrain, depth, platform_size=1.)
humanoid/utils/terrain.py:250
↓ 1 callers
Function
play
(args)
humanoid/scripts/play.py:85
↓ 1 callers
Method
post_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 callers
Method
process_env_step
(self, rewards, dones, infos)
humanoid/algo/ppo/dh_ppo.py:110
↓ 1 callers
Function
quaternion_to_euler_array
(quat)
humanoid/scripts/sim2sim.py:86
↓ 1 callers
Method
randomize_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