MCPcopy Create free account

hub / github.com/YanjieZe/H-InDex / functions

Functions701 in github.com/YanjieZe/H-InDex

⨍ Functions 701◇ Types & classes 130
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:259
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:343
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:384
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:447
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:515
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:555
Methodget_transform
(self)
stage3_RL/rrl/encoder.py:576
Methodget_transform
(self)
stage3_RL/rrl/representations/vc1_encoder.py:27
Methodget_transform
(self)
stage3_RL/rrl/representations/hindex_encoder.py:93
Methodget_transform
(self)
stage3_RL/rrl/representations/mvp_encoder.py:27
Methodget_transform
(self)
stage3_RL/rrl/representations/r3m_encoder.py:24
Functionhigh_res_normalize
(probs)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:59
Functionhook
(module, input, output)
stage2_adapt/lib/utils/utils.py:104
Methodhorizon
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:77
Functionimread
(filename, flags=cv2.IMREAD_COLOR)
stage2_adapt/lib/utils/zipreader.py:23
Functioninfo
(type, value, tb)
stage2_adapt/tools/visualize_keypoint.py:3
Methodis_cuda
(self)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:44
Methodis_cuda
(self)
stage3_RL/mjrl/mjrl/algos/model_accel/model_accel_npg.py:51
Methodlikelihood_ratio
(self, new_dist_info, old_dist_info)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:123
Methodlog_likelihood
(self, observations, actions, model=None, log_std=None)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:111
Methodlog_likelihood
(self, observations, actions, model=None, log_std=None)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:160
Functionmake_dir
(dir_path)
stage3_RL/rrl/utils.py:79
Functionmat2quat
Convert Rotation Matrix to Quaternion. See rotation.py for notes
stage3_RL/mj_envs/mj_envs/utils/quatmath.py:99
Methodmean_kl
(self, new_dist_info, old_dist_info)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:129
Methodmj_viewer_setup
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:122
Methodmj_viewer_setup
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:117
Methodmj_viewer_setup
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:124
Methodmj_viewer_setup
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:137
Methodmj_viewer_setup
(self)
stage3_RL/mjrl/mjrl/envs/swimmer.py:58
Methodmj_viewer_setup
(self)
stage3_RL/mjrl/mjrl/envs/point_mass.py:101
Methodmj_viewer_setup
(self)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:113
Methodmj_viewer_setup
(self)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:104
Methodnew_dist_info
(self, observations, actions)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:119
Methodobservation_dim
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:65
Methodobservation_space
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:69
Methodold_dist_info
(self, observations, actions)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:115
Functionpad_tensor_dict
(tensor_dict, max_len, mode='zero')
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:36
Functionpad_tensor_n
(xs, max_len)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:29
Methodpredict
(self, path)
stage3_RL/mjrl/mjrl/baselines/linear_baseline.py:62
Methodpredict
(self, path)
stage3_RL/mjrl/mjrl/baselines/quadratic_baseline.py:71
Methodpredict
(self, path)
stage3_RL/mjrl/mjrl/baselines/zero_baseline.py:13
Methodpredict_keypoint
(self, img)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:134
Functionpreprocess
Preprocesses an image before extraction. :param image_path: path to image to be extracted. :param load_size: optional. Size to resize ima
stage2_adapt/lib/dino/vision_transformer_flexible.py:393
FunctionquatDiff2Vel
(quat1, quat2, dt)
stage3_RL/mj_envs/mj_envs/utils/quatmath.py:25
Functionread
(fname)
stage3_RL/setup.py:14
Methodread_img
(img_path)
stage2_adapt/lib/dataset/robotic.py:80
Methodread_img
(img_path)
stage2_adapt/lib/dataset/robotube.py:74
Methodreal_env_step
(self, bool_val)
stage3_RL/mjrl/mjrl/utils/gym_env.py:146
Functionreplace_bn_recursively
(module, momentum=0.01)
stage3_RL/rrl/representations/hindex_encoder.py:12
Methodreset_model
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:79
Methodreset_model
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:96
Methodreset_model
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:106
Methodreset_model
(self)
stage3_RL/mjrl/mjrl/envs/swimmer.py:30
Methodreset_model
(self)
stage3_RL/mjrl/mjrl/envs/point_mass.py:53
Methodreset_model
(self, seed=None)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:75
Methodreset_model
(self, seed=None)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:70
Functionsample_data_batch
( num_samples, env, policy, eval_mode = False, horizon = 1e6,
stage3_RL/mjrl/mjrl/samplers/core.py:232
Functionsave_checkpoint
(states, is_best, output_dir, filename='checkpoint.pth')
stage2_adapt/lib/utils/utils.py:79
Functionsave_debug_images
(config, input, meta, target, joints_pred, output, prefix, ref_images=None, pred_images=
stage2_adapt/lib/utils/vis.py:123
Methodselect_data
(self, db)
stage2_adapt/lib/dataset/JointsDataset.py:204
Methodset_env_state
Set the state which includes hand as well as objects and targets in the scene
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:112
Methodset_env_state
Set the state which includes hand as well as objects and targets in the scene
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:104
Methodset_env_state
Set the state which includes hand as well as objects and targets in the scene
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:113
Methodset_env_state
Set the state which includes hand as well as objects and targets in the scene
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:126
Methodset_env_state
(self, state)
stage3_RL/mjrl/mjrl/envs/swimmer.py:44
Methodset_env_state
(self, state)
stage3_RL/mjrl/mjrl/envs/point_mass.py:85
Methodset_env_state
(self, state)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:92
Methodset_env_state
(self, state)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:87
Methodset_param_values
(self, new_params, set_new=True, set_old=True)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:65
Methodset_params
(self, new_params)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:254
Methodset_params
(self, new_params)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:336
Functionsplit_tensor_dict_list
(tensor_dict)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:124
Functionstack_tensor_dict_list
Stack a list of dictionaries of {tensors or dictionary of tensors}. :param tensor_dict_list: a list of dictionaries of {tensors or dictionary
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:71
Methodstate_vector
(self)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:145
Methodstep
(self, a)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:36
Methodstep
(self, a)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:34
Methodstep
(self, a)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:38
Methodstep
(self, a)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:51
Methodstep
(self, a)
stage3_RL/mjrl/mjrl/envs/point_mass.py:16
Methodstep
(self, a)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:17
Methodstep
(self, a)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:16
Methodto
(self, device)
stage3_RL/mjrl/mjrl/algos/model_accel/model_accel_npg.py:43
Methodtrain
(self, mode=True)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:195
Methodtrain
(self, mode=True)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:170
Methodtrain
(self, mode=True)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:196
Methodtrain
(self, mode=True)
stage2_adapt/lib/models/imm_joint_model.py:199
Methodtrain_from_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/npg_cg.py:91
Methodtrain_from_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/ppo_clip.py:58
Methodtrain_from_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/trpo.py:56
Methodtrain_step
(self, N, env=None, sample_mode='trajectories', horiz
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:62
Functiontransform_preds
(coords, center, scale, output_size)
stage3_RL/rrl/adaptation_models/transforms.py:46
Functiontruncate_tensor_dict
(tensor_dict, truncated_len)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:143
Functionunflatten_tensors
(flattened, tensor_shapes)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:13
Methodviewer_setup
Does not work. Use mj_viewer_setup() instead
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:88
Functionvisualize_feature_map_CHW
(feature_map, save_path)
stage3_RL/rrl/encoder.py:19
Methodvisualize_policy
(self, policy, horizon=1000, num_episodes=1, mode='exploration')
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:152
Methodvisualize_policy_offscreen
(self, policy, horizon=1000, num_episodes=1,
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:167
Functionvit_base
(patch_size=16, **kwargs)
stage2_adapt/lib/dino/vision_transformer_flexible.py:340
Functionvit_small
(patch_size=16, **kwargs)
stage2_adapt/lib/dino/vision_transformer_flexible.py:326
Functionvit_tiny
(patch_size=16, **kwargs)
stage2_adapt/lib/dino/vision_transformer_flexible.py:312
← previousnext →601–700 of 701, ranked by callers