MCPcopy Create free account

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

Functions701 in github.com/YanjieZe/H-InDex

⨍ Functions 701◇ Types & classes 130
↓ 79 callersMethodlog_kv
(self, key, value, step=None)
stage3_RL/mjrl/mjrl/utils/logger.py:22
↓ 45 callersMethodto
(self, device)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:40
↓ 26 callersMethodforward
(self, x)
stage3_RL/mjrl/mjrl/utils/fc_network.py:41
↓ 24 callersMethodeval
(v)
stage3_RL/mjrl/mjrl/algos/npg_cg.py:84
↓ 21 callersMethodget_action
(self, observation)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:134
↓ 17 callersMethodreset
(self, seed=None)
stage3_RL/mjrl/mjrl/utils/gym_env.py:80
↓ 17 callersMethodseed
(self, seed=None)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:67
↓ 14 callersMethodrender
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:105
↓ 14 callersMethodset_state
(self, qpos, qvel)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:108
↓ 13 callersMethodset_param_values
(self, new_params, set_new=True, set_old=True)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:108
↓ 12 callersMethodCPI_surrogate
(self, observations, actions, advantages)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:41
↓ 12 callersMethodset_seed
(self, seed=123)
stage3_RL/mjrl/mjrl/utils/gym_env.py:112
↓ 11 callersMethodstep
(self, action)
stage3_RL/mjrl/mjrl/utils/gym_env.py:93
↓ 10 callersMethod__init__
(self, encoder_type)
stage3_RL/rrl/encoder.py:28
↓ 9 callersMethodevaluate_success
Log various success metrics calculated based on input paths into the logger
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:94
↓ 9 callersMethodforward
(self, s, a, sp)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:313
↓ 9 callersMethodget_param_values
(self)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:103
↓ 9 callersFunctiontrain_agent
(job_name, agent, seed = 0, niter = 101, gamma = 0.995,
stage3_RL/mjrl/mjrl/utils/train_agent.py:62
↓ 8 callersMethoddo_simulation
(self, ctrl, n_frames)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:120
↓ 8 callersFunctionget_max_preds
get predictions from score maps heatmaps: numpy.ndarray([batch_size, num_joints, height, width])
stage2_adapt/lib/core/inference.py:18
↓ 8 callersMethodreset
(self)
stage3_RL/rrl/multicam.py:138
↓ 8 callersMethodstep
(self, action)
stage3_RL/rrl/multicam.py:143
↓ 7 callersMethodfit
(self, data, suppress_fit_tqdm=False, **kwargs)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:107
↓ 7 callersMethodkl_old_new
(self, observations, actions)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:49
↓ 7 callersMethodset_seed
(self, seed)
stage3_RL/rrl/multicam.py:135
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.0)
stage2_adapt/lib/dino/vision_transformer_flexible.py:53
↓ 6 callersMethodreset
(self)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:102
↓ 6 callersFunctionsave_batch_image_with_joints
batch_image: [batch_size, channel, height, width] batch_joints: [batch_size, num_joints, 3], batch_joints_vis: [batch_size, num_joints, 1
stage2_adapt/lib/utils/vis.py:20
↓ 6 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:281
↓ 5 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:29
↓ 5 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:39
↓ 5 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_double_head_resnet.py:39
↓ 5 callersMethod_make_layer
(self, in_channels, is_last=False)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:98
↓ 5 callersFunctionevaluate_policy
(e, policy, learned_model, noise_level=0.0, real_step=False, num_episodes=10, visualize=Fa
stage3_RL/mjrl/mjrl/algos/model_accel/sampling.py:235
↓ 5 callersMethodget_obs
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:118
↓ 5 callersMethodlog_rollout_statistics
(self, paths)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:198
↓ 5 callersFunctionmake_env
(env, cam_list=[], encoder=None, from_pixels=False, encoder_type=None, hybrid_state=None, episode_length=None,
stage3_RL/rrl/utils.py:30
↓ 5 callersFunctionmake_train_plots
(log = None, log_path = None, keys = None, save
stage3_RL/mjrl/mjrl/utils/make_train_plots.py:10
↓ 5 callersMethodnew_dist_info
(self, observations, actions)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:168
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:248
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage3_RL/rrl/adaptation_models/pose_resnet.py:126
↓ 4 callersMethod_make_layer
(self, in_channels, is_last=False)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:68
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:260
↓ 4 callersMethod_make_layer
(self, in_channels, is_last=False)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:69
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage2_adapt/lib/models/pose_double_head_resnet.py:258
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
stage2_adapt/lib/models/pose_resnet.py:136
↓ 4 callersMethod_make_layer
(self, in_channels, is_last=False)
stage2_adapt/lib/models/imm_joint_model.py:72
↓ 4 callersMethodcompute_path_rewards
(self, paths)
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:149
↓ 4 callersMethodflat_vpg
(self, observations, actions, advantages)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:55
↓ 4 callersMethodget_action
(self, obs)
stage3_RL/mjrl/mjrl/algos/model_accel/model_learning_mpc.py:42
↓ 4 callersMethodget_obs
(self, state=None, img_size=None)
stage3_RL/rrl/multicam.py:92
↓ 4 callersMethodget_transform
(self)
stage3_RL/rrl/encoder.py:49
↓ 4 callersMethodold_dist_info
(self, observations, actions)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:164
↓ 4 callersMethodprepare_tokens
(self, x)
stage2_adapt/lib/dino/vision_transformer_flexible.py:253
↓ 4 callersMethodsave_log
(self, save_path)
stage3_RL/mjrl/mjrl/utils/logger.py:46
↓ 4 callersMethodset_transformations
(self, in_shift=None, in_scale=None, out_shift=None, out_scale=None)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:63
↓ 4 callersMethodstep
(self, a)
stage3_RL/mjrl/mjrl/envs/swimmer.py:11
↓ 4 callersMethodupdate
(self, val, n=1)
stage2_adapt/lib/core/function.py:216
↓ 3 callersMethod__init__
(self, transforms)
stage3_RL/rrl/adaptation_models/transforms.py:187
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:21
↓ 3 callersMethod__init__
(self, transforms)
stage2_adapt/lib/utils/transforms.py:197
↓ 3 callersMethod__init__
(self, cfg)
stage2_adapt/lib/core/loss.py:107
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_hrnet.py:31
↓ 3 callersMethod_make_layer
(self, in_channels)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:35
↓ 3 callersMethod_make_layer
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:36
↓ 3 callersMethod_make_layer
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:37
↓ 3 callersMethod_make_layer
(self, in_channels)
stage2_adapt/lib/models/imm_joint_model.py:39
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:383
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
stage2_adapt/lib/models/pose_hrnet.py:393
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:323
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
stage2_adapt/lib/models/pose_hrnet.py:333
↓ 3 callersMethodbuild_Hvp_eval
(self, inputs, regu_coef=None)
stage3_RL/mjrl/mjrl/algos/npg_cg.py:83
↓ 3 callersFunctioncg_solve
(f_Ax, b, x_0=None, cg_iters=10, residual_tol=1e-10)
stage3_RL/mjrl/mjrl/utils/cg_solve.py:3
↓ 3 callersMethodcompute_transformations
(self)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:52
↓ 3 callersFunctionget_affine_transform
( center, scale, rot, output_size, shift=np.array([0, 0], dtype=np.float32), inv=0 )
stage2_adapt/lib/utils/transforms.py:64
↓ 3 callersMethodget_current_log
(self)
stage3_RL/mjrl/mjrl/utils/logger.py:63
↓ 3 callersMethodget_features
(self, x)
stage3_RL/rrl/encoder.py:56
↓ 3 callersMethodget_obs
(self, state=None, img_size=None)
stage3_RL/rrl/multicam.py:250
↓ 3 callersMethodloss
(self, data, idx=None)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:74
↓ 3 callersMethodmean_LL
(self, observations, actions, model=None, log_std=None)
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:99
↓ 3 callersMethodmean_LL
(self, observations, actions, model=None, log_std=None)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:142
↓ 3 callersMethodrender
(self, *args, **kwargs)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:137
↓ 3 callersFunctionsample_paths
( num_traj, env, policy, eval_mode = False, horizon = 1e6, bas
stage3_RL/mjrl/mjrl/samplers/core.py:178
↓ 3 callersFunctionsave_batch_heatmaps
batch_image: [batch_size, channel, height, width] batch_heatmaps: ['batch_size, num_joints, height, width] file_name: saved file name
stage2_adapt/lib/utils/vis.py:57
↓ 3 callersMethodset_env_state
(self, state)
stage3_RL/rrl/multicam.py:148
↓ 3 callersMethodset_variance_with_data
(self, out_scale)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:68
↓ 3 callersMethodtrain
(self, **kwargs)
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:138
↓ 3 callersMethodvisualize_policy
(self, policy, horizon=1000, num_episodes=1, mode='exploration')
stage3_RL/mjrl/mjrl/utils/gym_env.py:154
↓ 2 callersMethod__init__
(self, env=None, env_name="relocate-mug-1")
stage3_RL/rrl/utils.py:20
↓ 2 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_resnet.py:23
↓ 2 callersMethod__init__
(self, in_channels)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:16
↓ 2 callersMethod__init__
(self, requires_grad=False)
stage3_RL/rrl/adaptation_models/perceptual_loss.py:6
↓ 2 callersMethod__init__
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:17
↓ 2 callersMethod__init__
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:18
↓ 2 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_resnet.py:33
↓ 2 callersMethod__init__
(self, in_channels)
stage2_adapt/lib/models/imm_joint_model.py:20
↓ 2 callersMethod_features
(self, paths)
stage3_RL/mjrl/mjrl/baselines/linear_baseline.py:11
↓ 2 callersMethod_features
(self, paths)
stage3_RL/mjrl/mjrl/baselines/mlp_baseline.py:36
↓ 2 callersMethod_features
(self, paths)
stage3_RL/mjrl/mjrl/baselines/quadratic_baseline.py:11
↓ 2 callersFunctionadd_path
(path)
stage2_adapt/tools/_init_paths.py:5
next →1–100 of 701, ranked by callers