MCPcopy Create free account

hub / github.com/Improbable-AI/walk-these-ways / functions

Functions411 in github.com/Improbable-AI/walk-these-ways

↓ 15 callersMethodsave
(self, filename)
go1_gym_deploy/utils/logger.py:38
↓ 12 callersMethod_call_train_eval
(self, func, env_ids)
go1_gym/envs/base/legged_robot.py:531
↓ 10 callersFunctionget_scale_shift
(range)
go1_gym/utils/math_utils.py:35
↓ 10 callersMethodlog
Args: slots: ids for the array **key_vals:
go1_gym_learn/ppo/metrics_caches.py:14
↓ 8 callersMethodreset
(self)
go1_gym_deploy/utils/logger.py:56
↓ 8 callersMethodstep
(self, actions: torch.Tensor)
go1_gym_learn/env/vec_env.py:25
↓ 7 callersMethod__init__
(self, dt, max_time_s=10.)
go1_gym_deploy/utils/command_profile.py:5
↓ 7 callersMethoddecode
(data)
go1_gym_deploy/lcm_types/rc_command_lcmt.py:49
↓ 7 callersMethodget_buttons
(self)
go1_gym_deploy/utils/command_profile.py:16
↓ 6 callersMethodcalibrate
(self, wait=True, low=False)
go1_gym_deploy/utils/deployment_runner.py:65
↓ 6 callersMethodget_command
(self, t)
go1_gym_deploy/utils/command_profile.py:11
↓ 6 callersFunctionsplit_and_pad_trajectories
Splits trajectories at done indices. Then concatenates them and padds with zeros up to the length og the longest trajectory. Returns masks corres
go1_gym_learn/utils/utils.py:5
↓ 5 callersMethodinit_log_filename
(self)
go1_gym_deploy/utils/deployment_runner.py:31
↓ 5 callersMethodlog
Args: slots: ids for the array **key_vals:
go1_gym_learn/ppo_cse/metrics_caches.py:16
↓ 4 callersMethodclear
(self)
go1_gym_learn/ppo_cse/rollout_storage.py:73
↓ 4 callersMethodclear
(self)
go1_gym_learn/ppo/rollout_storage.py:73
↓ 4 callersMethodget_obs
(self)
go1_gym_deploy/envs/lcm_agent.py:125
↓ 4 callersMethodsample
(self, batch_size, low=None, high=None)
go1_gym/envs/base/curriculum.py:87
↓ 3 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/rc_command_lcmt.py:81
↓ 3 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/state_estimator_lcmt.py:93
↓ 3 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/pd_tau_targets_lcmt.py:79
↓ 3 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/leg_control_data_lcmt.py:76
↓ 3 callersMethod_randomize_gravity
(self, external_force = None)
go1_gym/envs/base/legged_robot.py:546
↓ 3 callersMethodadd_terrain_to_map
(self, cfg, terrain, row, col)
go1_gym/utils/terrain.py:161
↓ 3 callersMethodevaluate
(self, observation_history, privileged_observations, **kwargs)
go1_gym_learn/ppo_cse/actor_critic.py:142
↓ 3 callersMethodevaluate
(self, critic_observations, privileged_observations, **kwargs)
go1_gym_learn/ppo/actor_critic.py:170
↓ 3 callersMethodget_privileged_observations
(self)
go1_gym_deploy/envs/lcm_agent.py:186
↓ 3 callersMethodget_summary
(self)
go1_gym_learn/ppo_cse/metrics_caches.py:28
↓ 3 callersMethodget_summary
(self)
go1_gym_learn/ppo/metrics_caches.py:26
↓ 3 callersFunctionquat_apply_yaw
(quat, vec)
go1_gym/utils/math_utils.py:12
↓ 3 callersMethodupdate
(self, **kwargs)
go1_gym/envs/base/curriculum.py:63
↓ 3 callersMethodupdate
(self)
go1_gym_learn/ppo/ppo.py:94
↓ 2 callersMethod__init__
(self, seed, **key_ranges)
go1_gym/envs/base/curriculum.py:28
↓ 2 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/camera_message_rect_wide.py:52
↓ 2 callersMethod_get_packed_fingerprint
()
go1_gym_deploy/lcm_types/camera_message_lcmt.py:56
↓ 2 callersMethod_initialize_terrain
(self, cfg)
go1_gym/utils/terrain.py:72
↓ 2 callersMethod_load_cfg
(self, cfg)
go1_gym/utils/terrain.py:54
↓ 2 callersMethod_parse_cfg
(self, cfg)
go1_gym/envs/base/legged_robot.py:1716
↓ 2 callersMethod_resample_commands
(self, env_ids)
go1_gym/envs/base/legged_robot.py:710
↓ 2 callersMethodact
(self, obs, privileged_obs, obs_history)
go1_gym_learn/ppo_cse/ppo.py:65
↓ 2 callersMethodact
(self, obs, privileged_obs, obs_history)
go1_gym_learn/ppo/ppo.py:62
↓ 2 callersMethodact_student
(self, observation_history, policy_info={})
go1_gym_learn/ppo_cse/actor_critic.py:131
↓ 2 callersMethodact_student
(self, observations, observation_history, policy_info={})
go1_gym_learn/ppo/actor_critic.py:158
↓ 2 callersMethodact_teacher
(self, observation_history, privileged_info, policy_info={})
go1_gym_learn/ppo_cse/actor_critic.py:137
↓ 2 callersMethodact_teacher
(self, observations, privileged_info, policy_info={})
go1_gym_learn/ppo/actor_critic.py:164
↓ 2 callersMethodadd_robot
(self, name, cfg)
go1_gym_deploy/utils/logger.py:31
↓ 2 callersFunctionconfig_go1
(Cnfg: Union[Cfg, Meta])
go1_gym/envs/go1/go1_config.py:8
↓ 2 callersMethodencode
(self)
go1_gym_deploy/lcm_types/rc_command_lcmt.py:34
↓ 2 callersMethodforward
(self, input)
scripts/actuator_net/utils.py:34
↓ 2 callersMethodforward
(self)
go1_gym_learn/ppo/actor_critic.py:122
↓ 2 callersMethodget_actions_log_prob
(self, actions)
go1_gym_learn/ppo_cse/actor_critic.py:122
↓ 2 callersMethodget_actions_log_prob
(self, actions)
go1_gym_learn/ppo/actor_critic.py:146
↓ 2 callersMethodget_complete_frames
(self)
go1_gym/envs/base/legged_robot.py:1665
↓ 2 callersMethodget_complete_frames_eval
(self)
go1_gym/envs/base/legged_robot.py:1670
↓ 2 callersMethodget_local_bins
(self, bin_inds, ranges=0.1)
go1_gym/envs/base/curriculum.py:123
↓ 2 callersMethodget_observations
(self)
go1_gym_learn/env/vec_env.py:34
↓ 2 callersMethodget_privileged_observations
(self)
go1_gym/envs/base/base_task.py:91
↓ 2 callersMethodget_rpy
(self)
go1_gym_deploy/utils/cheetah_state_estimator.py:145
↓ 2 callersMethodlog
(self, name, info)
go1_gym_deploy/utils/logger.py:35
↓ 2 callersMethodmake_terrain
(self, cfg, choice, difficulty, proportions)
go1_gym/utils/terrain.py:114
↓ 2 callersMethodpause_recording
(self)
go1_gym/envs/base/legged_robot.py:1655
↓ 2 callersMethodpause_recording_eval
(self)
go1_gym/envs/base/legged_robot.py:1660
↓ 2 callersMethodrender
(self, mode="rgb_array")
go1_gym/envs/base/legged_robot.py:1611
↓ 2 callersMethodreset
(self, reset_time)
go1_gym_deploy/utils/command_profile.py:19
↓ 2 callersMethodreset
(self, dones=None)
go1_gym_learn/ppo_cse/actor_critic.py:95
↓ 2 callersMethodreset
(self, dones=None)
go1_gym_learn/ppo/actor_critic.py:119
↓ 2 callersMethodreset_idx
Reset selected robots
go1_gym/envs/base/base_task.py:94
↓ 2 callersMethodset_probing
(self, is_currently_probing)
go1_gym_deploy/envs/lcm_agent.py:122
↓ 2 callersMethodstart_recording
(self)
go1_gym/envs/base/legged_robot.py:1647
↓ 2 callersMethodstart_recording_eval
(self)
go1_gym/envs/base/legged_robot.py:1651
↓ 2 callersMethodstep
(self, actions, hard_reset=False)
go1_gym_deploy/envs/lcm_agent.py:228
↓ 2 callersMethodstep
(self, actions)
go1_gym/envs/go1/velocity_tracking/__init__.py:22
↓ 2 callersFunctiontrain_actuator_network_and_plot_predictions
(log_dir_root, log_dir, actuator_network_path, load_pretrained_model=False)
scripts/actuator_net/utils.py:148
↓ 1 callersMethod__init__
(self)
go1_gym_learn/ppo_cse/rollout_storage.py:9
↓ 1 callersMethod__init__
(self, env, device='cpu')
go1_gym_learn/ppo/__init__.py:67
↓ 1 callersMethod__init__
(self)
go1_gym_learn/ppo/rollout_storage.py:9
↓ 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
go1_gym/envs/base/legged_robot.py:907
↓ 1 callersMethod_create_envs
Creates environments: 1. loads the robot URDF/MJCF asset, 2. For each environment 2.1 creates the environme
go1_gym/envs/base/legged_robot.py:1481
↓ 1 callersMethod_create_ground_plane
Adds a ground plane to the simulation, sets friction and restitution based on the cfg.
go1_gym/envs/base/legged_robot.py:1431
↓ 1 callersMethod_create_heightfield
Adds a heightfield terrain to the simulation, sets parameters based on the cfg.
go1_gym/envs/base/legged_robot.py:1441
↓ 1 callersMethod_create_trimesh
Adds a triangle mesh terrain to the simulation, sets parameters based on the cfg. #
go1_gym/envs/base/legged_robot.py:1463
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/rc_command_lcmt.py:60
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/camera_message_rect_wide.py:37
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/camera_message_lcmt.py:41
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/state_estimator_lcmt.py:67
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/pd_tau_targets_lcmt.py:57
↓ 1 callersMethod_decode_one
(buf)
go1_gym_deploy/lcm_types/leg_control_data_lcmt.py:55
↓ 1 callersMethod_draw_debug_vis
Draws visualizations for dubugging (slows down simulation a lot). Default behaviour: draws height measurement points
go1_gym/envs/base/legged_robot.py:1734
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/rc_command_lcmt.py:40
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/camera_message_rect_wide.py:24
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/camera_message_lcmt.py:28
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/state_estimator_lcmt.py:43
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/pd_tau_targets_lcmt.py:37
↓ 1 callersMethod_encode_one
(self, buf)
go1_gym_deploy/lcm_types/leg_control_data_lcmt.py:36
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/rc_command_lcmt.py:72
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/camera_message_rect_wide.py:44
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/camera_message_lcmt.py:48
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/state_estimator_lcmt.py:84
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/pd_tau_targets_lcmt.py:70
↓ 1 callersMethod_get_hash_recursive
(parents)
go1_gym_deploy/lcm_types/leg_control_data_lcmt.py:67
next →1–100 of 411, ranked by callers