MCPcopy Create free account

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

Functions701 in github.com/YanjieZe/H-InDex

↓ 2 callersFunctionaffine_transform
(pt, t)
stage2_adapt/lib/utils/transforms.py:99
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:18
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage3_RL/rrl/adaptation_models/pose_resnet.py:12
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage3_RL/rrl/adaptation_models/pose_hrnet.py:12
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:28
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage2_adapt/lib/models/pose_double_head_resnet.py:28
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage2_adapt/lib/models/pose_resnet.py:22
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
stage2_adapt/lib/models/pose_hrnet.py:22
↓ 2 callersFunctiondiscount_sum
(x, gamma, terminal=0.0)
stage3_RL/mjrl/mjrl/utils/process_samples.py:37
↓ 2 callersFunctiondownload_file
(url, local_path)
stage2_adapt/lib/dino/model.py:9
↓ 2 callersFunctionenforce_tensor_bounds
Clamp the torch_tensor to Box[min_val, max_val] torch_tensor should have shape (A, B) min_val and max_val can either be scala
stage3_RL/mjrl/mjrl/algos/model_accel/sampling.py:286
↓ 2 callersMethodevaluate_policy
(self, policy, num_episodes=5, horizon=None,
stage3_RL/rrl/multicam.py:154
↓ 2 callersMethodevaluate_policy
(self, policy, num_episodes=5, horizon=None,
stage3_RL/mjrl/mjrl/utils/gym_env.py:171
↓ 2 callersFunctionfit_model
:param nn_model: pytorch model of form Y = f(*X) (class) :param X: tuple of necessary inputs to the function :param
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:344
↓ 2 callersFunctiongenerate_perturbed_actions
Generate perturbed actions around a base action sequence
stage3_RL/mjrl/mjrl/algos/model_accel/sampling.py:204
↓ 2 callersFunctionget_3rd_point
(a, b)
stage3_RL/rrl/adaptation_models/transforms.py:95
↓ 2 callersFunctionget_3rd_point
(a, b)
stage2_adapt/lib/utils/transforms.py:105
↓ 2 callersFunctionget_affine_transform
( center, scale, rot, output_size, shift=np.array([0, 0], dtype=np.float32), inv=0 )
stage3_RL/rrl/adaptation_models/transforms.py:54
↓ 2 callersFunctionget_double_head_pose_net
(cfg, is_train, **kwargs)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:440
↓ 2 callersMethodget_env_infos
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:124
↓ 2 callersMethodget_env_state
(self)
stage3_RL/rrl/multicam.py:151
↓ 2 callersMethodget_env_state
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:134
↓ 2 callersMethodget_env_state
(self)
stage3_RL/mjrl/mjrl/envs/point_mass.py:80
↓ 2 callersFunctionget_feature
feature for single step
stage3_RL/examples/train_dexmv.py:174
↓ 2 callersMethodget_obs
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:77
↓ 2 callersMethodget_obs
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:69
↓ 2 callersMethodget_obs
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:83
↓ 2 callersMethodget_obs
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:96
↓ 2 callersMethodget_obs
(self)
stage3_RL/mjrl/mjrl/envs/swimmer.py:24
↓ 2 callersMethodget_obs
(self)
stage3_RL/mjrl/mjrl/envs/point_mass.py:22
↓ 2 callersMethodget_obs
(self)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:23
↓ 2 callersMethodget_obs
(self)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:22
↓ 2 callersMethodget_params
(self)
stage3_RL/rrl/adaptation_models/transforms.py:126
↓ 2 callersMethodget_params
(self)
stage2_adapt/lib/utils/transforms.py:136
↓ 2 callersMethodget_reward
(self, obs, act=None)
stage3_RL/mjrl/mjrl/envs/point_mass.py:27
↓ 2 callersMethodget_reward
(self, obs, act=None)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:31
↓ 2 callersMethodget_reward
(self, obs, act=None)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:30
↓ 2 callersMethodget_specific_tokens
(self, x, layers_to_return=(11,))
stage2_adapt/lib/dino/vision_transformer_flexible.py:288
↓ 2 callersMethodlikelihood_ratio
(self, new_dist_info, old_dist_info)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:172
↓ 2 callersMethodload_model
(self)
stage2_adapt/lib/dino/model.py:60
↓ 2 callersMethodlog_ckpt
(self, ckpt, step=None)
stage3_RL/mjrl/mjrl/utils/logger.py:41
↓ 2 callersMethodlog_dict
(self, dict, step=None)
stage3_RL/mjrl/mjrl/utils/logger.py:33
↓ 2 callersFunctionmake_encoder
(encoder_type, device, is_eval=True, ckpt_path=None, test_time_momentum=0.0)
stage3_RL/rrl/multicam.py:23
↓ 2 callersMethodmean_kl
(self, new_dist_info, old_dist_info)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:178
↓ 2 callersMethodmj_render
(self)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:128
↓ 2 callersMethodpredict
(self, path)
stage3_RL/mjrl/mjrl/baselines/mlp_baseline.py:97
↓ 2 callersMethodpredict
(self, s, a)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:56
↓ 2 callersFunctionpreprocess_args
(args)
stage3_RL/rrl/utils.py:84
↓ 2 callersMethodprocess_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:176
↓ 2 callersMethodread_log
(self, log_path)
stage3_RL/mjrl/mjrl/utils/logger.py:78
↓ 2 callersFunctionrender
(env, img_size, cam)
stage3_RL/examples/createImgData_dexmv.py:31
↓ 2 callersFunctionrender
(env, img_size, cam)
stage3_RL/examples/train_dexmv.py:51
↓ 2 callersFunctionrender_obs
(env, img_size=224, camera_name="vil_camera", device=0)
stage3_RL/examples/createImgData_adroit.py:29
↓ 2 callersMethodset_transformations
(self, s_shift=None, s_scale=None, a_shift=None, a_scale=None,
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:193
↓ 2 callersFunctiontest_time_adaptation
unsupervised keypoint feature adaptation videos: shape (num_videos, timestep, num_frames, 256, 256, 3)
stage3_RL/rrl/adaptation.py:49
↓ 2 callersFunctiontrajectory_rollout
(actions, learned_model, init_states)
stage3_RL/mjrl/mjrl/algos/model_accel/sampling.py:96
↓ 2 callersFunctionupdate_config
(cfg, args)
stage2_adapt/lib/config/default.py:178
↓ 1 callersMethodHVP
(self, observations, actions, vector, regu_coef=None)
stage3_RL/mjrl/mjrl/algos/npg_cg.py:62
↓ 1 callersMethodPPO_surrogate
(self, observations, actions, advantages)
stage3_RL/mjrl/mjrl/algos/ppo_clip.py:48
↓ 1 callersFunctionPSNR
(img1, img2, max_val=1)
stage3_RL/rrl/adaptation.py:13
↓ 1 callersFunctionPSNR
(img1, img2, max_val=1)
stage2_adapt/lib/core/function.py:24
↓ 1 callersMethod__init__
:param env_spec: specifications of the env (see utils/gym_env.py) :param hidden_sizes: network hidden layer sizes (currently 2 layers
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:51
↓ 1 callersMethod__init__
(self, state_dim, act_dim, hidden_size=(64,64), s_shift = None,
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:263
↓ 1 callersMethod__init__
(self, requires_grad=False)
stage2_adapt/lib/core/perceptual_loss.py:16
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:109
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
stage2_adapt/lib/models/pose_hrnet.py:119
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:265
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
stage3_RL/rrl/adaptation_models/pose_resnet.py:143
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:277
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
stage2_adapt/lib/models/pose_double_head_resnet.py:275
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
stage2_adapt/lib/models/pose_resnet.py:153
↓ 1 callersFunction_load_latest_policy_and_logs
Loads the latest policy. Returns the next step number to begin with.
stage3_RL/mjrl/mjrl/utils/train_agent.py:15
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:167
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
stage2_adapt/lib/models/pose_hrnet.py:177
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:278
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
stage3_RL/rrl/adaptation_models/pose_resnet.py:156
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:290
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
stage2_adapt/lib/models/pose_double_head_resnet.py:288
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
stage2_adapt/lib/models/pose_resnet.py:166
↓ 1 callersMethod_make_first
(self, in_channels, out_channels)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:23
↓ 1 callersMethod_make_first
(self, in_channels, out_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:24
↓ 1 callersMethod_make_first
(self, in_channels, out_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:25
↓ 1 callersMethod_make_first
(self, in_channels, out_channels)
stage2_adapt/lib/models/imm_joint_model.py:27
↓ 1 callersMethod_make_fuse_layers
(self)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:177
↓ 1 callersMethod_make_fuse_layers
(self)
stage2_adapt/lib/models/pose_hrnet.py:187
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:364
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage2_adapt/lib/models/pose_hrnet.py:374
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:129
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
stage2_adapt/lib/models/pose_hrnet.py:139
↓ 1 callersFunction_try_multiprocess
(func, input_dict_list, num_cpu, max_process_time, max_timeouts)
stage3_RL/mjrl/mjrl/samplers/core.py:270
↓ 1 callersMethodadd_paths_to_buffer
(self, paths)
stage3_RL/mjrl/mjrl/algos/mbac.py:99
↓ 1 callersFunctionaffine_transform
(pt, t)
stage3_RL/rrl/adaptation_models/transforms.py:89
↓ 1 callersFunctionbuffer_size
(paths_list)
stage3_RL/mjrl/mjrl/algos/model_accel/run_experiments/run_model_accel_npg.py:75
↓ 1 callersFunctioncalc_dists
(preds, target, normalize)
stage2_adapt/lib/core/evaluate.py:16
↓ 1 callersMethodcollect_paths
(self, num_traj=10, mode='policy', horizon=None,
stage3_RL/mjrl/mjrl/algos/mbac.py:62
↓ 1 callersMethodcompute_loss
(self, s, a, s_next)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:79
↓ 1 callersFunctionconcat_tensor_list
(tensor_list)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:107
↓ 1 callersFunctionconcat_tensor_list_subsample
(tensor_list, f)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:89
↓ 1 callersFunctionconfig_tqdm
(range_inp, suppress_tqdm=False)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:145
↓ 1 callersFunctiondist_acc
Return percentage below threshold while ignoring values with a -1
stage2_adapt/lib/core/evaluate.py:31
← previousnext →101–200 of 701, ranked by callers