Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GuyTevet/CLoSD
/ functions
Functions
2,247 in github.com/GuyTevet/CLoSD
⨍
Functions
2,247
◇
Types & classes
226
↓ 1 callers
Method
__init__
(self, task: "BasePlotterTask")
closd/utils/poselib/poselib/visualization/plt_plotter.py:31
↓ 1 callers
Method
__init__
( self, task_name: str, skeleton_state, joints_color: str = "red", lin
closd/utils/poselib/poselib/visualization/skeleton_plotter_tasks.py:22
↓ 1 callers
Method
__init__
:param node_names: a list of names for each tree node :type node_names: List[str] :param parent_indices: an int32-typed tenso
closd/utils/poselib/poselib/skeleton/skeleton3d.py:76
↓ 1 callers
Method
__init__
(self, model, timestep_map, rescale_timesteps, original_num_steps)
closd/diffusion_planner/diffusion/respace.py:119
↓ 1 callers
Method
__init__
(self, args, sample_fn, required_frames=196)
closd/diffusion_planner/utils/sampler_util.py:39
↓ 1 callers
Method
__init__
(self, save_dir, *args, **kwargs)
closd/diffusion_planner/train/train_platforms.py:56
↓ 1 callers
Method
__init__
(self, **kwargs)
closd/learning/amp_network_builder.py:10
↓ 1 callers
Method
__init__
(self, config)
closd/learning/amp_players.py:167
↓ 1 callers
Method
__init__
(self, base_name, config)
closd/learning/common_agent.py:25
↓ 1 callers
Method
__init__
(self, config)
closd/learning/common_player.py:14
↓ 1 callers
Method
__init__
(self, network)
closd/learning/amp_models.py:6
↓ 1 callers
Method
_amp_debug
(self, info)
closd/learning/amp_players.py:118
↓ 1 callers
Method
_amp_debug
(self, info)
closd/learning/amp_players.py:271
↓ 1 callers
Method
_anneal_lr
(self)
closd/diffusion_planner/train/training_loop.py:356
↓ 1 callers
Method
_assemble_train_info
(self, train_info, frame)
closd/learning/common_agent.py:599
↓ 1 callers
Method
_assemble_train_info
(self, train_info, frame)
closd/learning/common_agent.py:1081
↓ 1 callers
Function
_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. A
closd/utils/pytorch3d_transforms.py:271
↓ 1 callers
Function
_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. A
closd/utils/smpllib/utils/rotation_conversions.py:148
↓ 1 callers
Method
_build_actor
(self, output_dim, log_std_bounds, **mlp_args)
closd/learning/network_builder.py:864
↓ 1 callers
Method
_build_amp_buffers
(self)
closd/learning/amp_agent.py:873
↓ 1 callers
Method
_build_amp_obs_demo_buf
(self, num_samples)
closd/env/tasks/humanoid_amp.py:268
↓ 1 callers
Method
_build_cnn1d
(self, input_shape, convs, activation, norm_func_name=None)
closd/learning/network_builder.py:160
↓ 1 callers
Method
_build_contact_body_ids_tensor
(self, contact_body_names)
closd/env/tasks/humanoid.py:1361
↓ 1 callers
Method
_build_disc
(self, input_shape)
closd/learning/amp_network_builder.py:227
↓ 1 callers
Method
_build_env
(self, env_id, env_ptr, humanoid_asset)
closd/env/tasks/humanoid.py:901
↓ 1 callers
Method
_build_impala
(self, input_shape, depths)
closd/learning/network_builder.py:732
↓ 1 callers
Method
_build_marker
(self, env_id, env_ptr)
closd/env/tasks/closd_task.py:91
↓ 1 callers
Method
_build_marker_state_tensors
(self)
closd/env/tasks/humanoid_im.py:617
↓ 1 callers
Method
_build_net
(self, config)
closd/learning/amp_players.py:224
↓ 1 callers
Method
_build_net
(self, config)
closd/learning/common_player.py:175
↓ 1 callers
Method
_build_net
(self, config)
closd/learning/common_player.py:371
↓ 1 callers
Method
_build_net_config
(self)
closd/learning/amp_players.py:249
↓ 1 callers
Method
_build_net_config
(self)
closd/learning/common_agent.py:524
↓ 1 callers
Method
_build_net_config
(self)
closd/learning/common_agent.py:1001
↓ 1 callers
Method
_build_net_config
(self)
closd/learning/common_player.py:198
↓ 1 callers
Method
_build_net_config
(self)
closd/learning/common_player.py:391
↓ 1 callers
Method
_build_pd_action_offset_scale
(self)
closd/env/tasks/humanoid.py:1020
↓ 1 callers
Method
_build_proj
(self, env_id, env_ptr)
closd/env/tasks/humanoid.py:153
↓ 1 callers
Method
_build_res_mlp
(self, input_size, units, activation, dense_func, norm_only_first_layer=False, norm_func_name=None)
closd/learning/network_builder.py:101
↓ 1 callers
Method
_build_strike_body_ids_tensor
(self, env_ptr, actor_handle, body_names)
closd/env/tasks/closd_multitask.py:145
↓ 1 callers
Method
_build_target
(self, env_id, env_ptr)
closd/env/tasks/closd_multitask.py:208
↓ 1 callers
Method
_build_target_tensors
(self)
closd/env/tasks/closd_multitask.py:334
↓ 1 callers
Method
_build_termination_heights
(self)
closd/env/tasks/humanoid.py:653
↓ 1 callers
Method
_calc_advs
(self, batch_dict)
closd/learning/common_agent.py:584
↓ 1 callers
Method
_calc_advs
(self, batch_dict)
closd/learning/common_agent.py:1066
↓ 1 callers
Method
_calc_disc_rewards
(self, amp_obs)
closd/learning/amp_players.py:157
↓ 1 callers
Method
_calc_disc_rewards
(self, amp_obs)
closd/learning/amp_players.py:309
↓ 1 callers
Method
_calc_disc_rewards
(self, amp_obs)
closd/learning/amp_agent.py:925
↓ 1 callers
Method
_clear_recorded_states
(self)
closd/env/tasks/base_task.py:235
↓ 1 callers
Method
_compute_aux_reward
(self, actions)
closd/env/tasks/closd_task.py:219
↓ 1 callers
Method
_compute_disc_acc
(self, disc_agent_logit, disc_demo_logit)
closd/learning/amp_agent.py:862
↓ 1 callers
Method
_compute_humanoid_obs
(self, env_ids=None)
closd/env/tasks/humanoid_amp.py:501
↓ 1 callers
Method
_compute_reset
(self)
closd/env/tasks/humanoid.py:1100
↓ 1 callers
Method
_compute_reward
(self, actions)
closd/env/tasks/humanoid.py:1096
↓ 1 callers
Function
_compute_rotation_matrix
(angle_axis, theta2, eps=1e-6)
closd/utils/smpllib/utils/torch_geometry_transforms.py:138
↓ 1 callers
Function
_compute_rotation_matrix_taylor
(angle_axis)
closd/utils/smpllib/utils/torch_geometry_transforms.py:162
↓ 1 callers
Method
_compute_task_obs
(self, env_ids=None, save_buffer = True)
closd/env/tasks/humanoid_im.py:682
↓ 1 callers
Method
_compute_task_obs
(self, env_ids=None)
closd/env/tasks/humanoid_amp_task.py:115
↓ 1 callers
Method
_compute_task_obs_demo
(self, env_ids=None, set_goal=None)
closd/env/tasks/closd.py:577
↓ 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
closd/env/tasks/humanoid.py:1242
↓ 1 callers
Function
_configure_default_logger
()
closd/diffusion_planner/diffusion/logger.py:474
↓ 1 callers
Function
_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
closd/utils/pytorch3d_transforms.py:101
↓ 1 callers
Function
_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
closd/utils/smpllib/utils/rotation_conversions.py:70
↓ 1 callers
Method
_create_envs
(self, num_envs, spacing, num_per_row)
closd/env/tasks/humanoid.py:711
↓ 1 callers
Method
_create_full_overlay
(self)
closd/utils/smpllib/khrylib/rl/envs/common/mjviewer.py:250
↓ 1 callers
Method
_create_ground_plane
(self)
closd/env/tasks/humanoid.py:589
↓ 1 callers
Method
_create_impl
(self, task_list: List[BasePlotterTask])
closd/utils/poselib/poselib/visualization/core.py:77
↓ 1 callers
Method
_disc_loss
disc_agent_logit: replay and current episode logit (fake examples) disc_demo_logit: disc_demo_logit logit obs_demo: gradient
closd/learning/amp_agent.py:793
↓ 1 callers
Method
_disc_loss_neg
(self, disc_logits)
closd/learning/amp_agent.py:852
↓ 1 callers
Method
_disc_loss_pos
(self, disc_logits)
closd/learning/amp_agent.py:857
↓ 1 callers
Method
_do_log
(self, args)
closd/diffusion_planner/diffusion/logger.py:397
↓ 1 callers
Method
_draw_task
(self)
closd/env/tasks/humanoid_amp_task.py:121
↓ 1 callers
Method
_eval_critic
(self, obs_dict)
closd/learning/common_agent.py:1029
↓ 1 callers
Method
_eval_disc
(self, amp_obs)
closd/learning/amp_players.py:126
↓ 1 callers
Method
_eval_disc
(self, amp_obs)
closd/learning/amp_players.py:279
↓ 1 callers
Function
_gaussian_kernel1d
Computes a 1-D Gaussian convolution kernel.
closd/utils/filter_utils.py:5
↓ 1 callers
Function
_get_animation_curve
(joint, fbx_scene)
closd/utils/poselib/poselib/skeleton/backend/fbx/fbx_py27_backend.py:205
↓ 1 callers
Method
_get_fixed_smpl_state_from_motionlib
(self, motion_ids, motion_times, curr_gender_betas)
closd/env/tasks/humanoid_amp.py:363
↓ 1 callers
Function
_get_frame_count
(fbx_scene)
closd/utils/poselib/poselib/skeleton/backend/fbx/fbx_py27_backend.py:180
↓ 1 callers
Method
_get_mean_rewards
(self)
closd/learning/common_agent.py:517
↓ 1 callers
Method
_get_mean_rewards
(self)
closd/learning/common_agent.py:994
↓ 1 callers
Function
_get_skeleton
(root_joint)
closd/utils/poselib/poselib/skeleton/backend/fbx/fbx_py27_backend.py:261
↓ 1 callers
Method
_hack_motion_sync
(self)
closd/env/tasks/humanoid_amp.py:672
↓ 1 callers
Method
_hack_output_motion
(self)
closd/env/tasks/humanoid_amp.py:780
↓ 1 callers
Function
_import_module
Try import all public attributes from module into global namespace. Existing attributes with name clashes are renamed with prefix. Attributes
closd/utils/smpllib/utils/transformation.py:2018
↓ 1 callers
Function
_import_module
Try import all public attributes from module into global namespace. Existing attributes with name clashes are renamed with prefix. Attributes
closd/utils/smpllib/khrylib/utils/transformation.py:2005
↓ 1 callers
Method
_init_amp_demo_buf
(self)
closd/learning/amp_agent.py:886
↓ 1 callers
Method
_init_amp_obs
(self, env_ids)
closd/env/tasks/humanoid_amp.py:514
↓ 1 callers
Method
_init_amp_obs_default
(self, env_ids)
closd/env/tasks/humanoid_amp.py:525
↓ 1 callers
Method
_init_amp_obs_ref
(self, env_ids, motion_ids, motion_times)
closd/env/tasks/humanoid_amp.py:530
↓ 1 callers
Method
_init_camera
(self)
closd/env/tasks/humanoid.py:1378
↓ 1 callers
Method
_init_data_buf
(self, data_dict)
closd/learning/replay_buffer.py:77
↓ 1 callers
Method
_init_tensor_history
(self, env_ids)
closd/env/tasks/humanoid.py:1284
↓ 1 callers
Method
_init_train
(self)
closd/learning/common_agent.py:544
↓ 1 callers
Method
_init_train
(self)
closd/learning/common_agent.py:1026
↓ 1 callers
Method
_load_amass_gender_betas
(self, cfg)
closd/env/tasks/humanoid.py:702
↓ 1 callers
Method
_load_and_sync_parameters
(self)
closd/diffusion_planner/train/training_loop.py:129
↓ 1 callers
Method
_load_config_params
(self, config)
closd/learning/common_agent.py:520
↓ 1 callers
Method
_load_config_params
(self, config)
closd/learning/common_agent.py:997
↓ 1 callers
Method
_load_marker_asset
(self)
closd/env/tasks/humanoid_im.py:535
← previous
next →
501–600 of 2,247, ranked by callers