MCPcopy Create free account

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

Functions701 in github.com/YanjieZe/H-InDex

⨍ Functions 701◇ Types & classes 130
↓ 1 callersMethodscore_trajectory
(self, paths)
stage3_RL/mjrl/mjrl/algos/model_accel/model_learning_mpc.py:101
↓ 1 callersMethodscore_trajectory_ensemble
(self, paths, paths_list)
stage3_RL/mjrl/mjrl/algos/model_accel/model_learning_mpc.py:85
↓ 1 callersMethodset_env_state
(self, state)
stage3_RL/rrl/multicam.py:307
↓ 1 callersMethodset_env_state
(self, state_dict)
stage3_RL/mjrl/mjrl/utils/gym_env.py:140
↓ 1 callersMethodset_seed
(self, seed)
stage3_RL/rrl/multicam.py:292
↓ 1 callersMethodset_transformations
(self, in_shift=None, in_scale=None, out_shift=None, out_scale=None)
stage3_RL/mjrl/mjrl/utils/fc_network.py:28
↓ 1 callersMethodset_transformations
(self, in_shift=None, in_scale=None, out_shift=None, out_scale=None)
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:25
↓ 1 callersMethodshrink_to
(self, num_entries)
stage3_RL/mjrl/mjrl/utils/logger.py:70
↓ 1 callersFunctionstack_tensor_list
(tensor_list)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:63
↓ 1 callersMethodstep
(self, action)
stage3_RL/rrl/multicam.py:300
↓ 1 callersMethodtarget_reset
(self)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:64
↓ 1 callersMethodtarget_reset
(self)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:62
↓ 1 callersFunctiontest_time_training
(config, loader, dataset, model, model_state_dict, criterion, optimizer, optimizer_state_dict, outpu
stage2_adapt/lib/core/function.py:32
↓ 1 callersMethodtrain_from_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/dapg.py:57
↓ 1 callersMethodtrain_from_paths
(self, paths)
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:119
↓ 1 callersFunctiontrain_loop
(job_data: DictConfig)
stage3_RL/mjrl/tests/hydra/hydra_policy_opt_job_script.py:56
↓ 1 callersMethodtrain_step
(self, num_traj=10, **kwargs)
stage3_RL/mjrl/mjrl/algos/mbac.py:117
↓ 1 callersMethodtrain_step
(self, N, env=None, sample_mode='trajectories', horiz
stage3_RL/mjrl/mjrl/algos/model_accel/model_accel_npg.py:58
↓ 1 callersFunctiontransform_preds
(coords, center, scale, output_size)
stage2_adapt/lib/utils/transforms.py:56
↓ 1 callersFunctiontruncate_tensor_list
(tensor_list, truncated_len)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:139
FunctionPSNR
(img1, img2, max_val=1)
stage3_RL/rrl/utils.py:125
Method__call__
(self, x, params=None, return_params=False)
stage3_RL/rrl/adaptation_models/transforms.py:129
Method__call__
(self, img, params=None, return_params=False)
stage3_RL/rrl/adaptation_models/transforms.py:190
Method__call__
(self, x, params=None, return_params=False)
stage2_adapt/lib/utils/transforms.py:139
Method__call__
(self, img, params=None, return_params=False)
stage2_adapt/lib/utils/transforms.py:200
Method__getitem__
(self, index)
stage2_adapt/lib/dataset/robotic.py:66
Method__getitem__
(self, index)
stage2_adapt/lib/dataset/robotube.py:56
Method__getitem__
(self, idx)
stage2_adapt/lib/dataset/JointsDataset.py:116
Method__init__
(self, requires_grad=False)
stage3_RL/rrl/utils.py:134
Method__init__
(self)
stage3_RL/rrl/utils.py:172
Method__init__
(self, num_features, eps=1e-5, momentum=0.01, affine=True, track_running_stats=True)
stage3_RL/rrl/modules.py:6
Method__init__
(self, env, cameras, encoder_type="resnet34", encoder=None, latent_dim=512, hybrid_state=True, channels_first=
stage3_RL/rrl/multicam.py:52
Method__init__
(self, env, env_id, cameras, encoder_type="resnet34", encoder=None, latent_dim=512, hybrid_s
stage3_RL/rrl/multicam.py:211
Method__init__
(self, encoder_type, bn_momentum=0.01)
stage3_RL/rrl/encoder.py:62
Method__init__
(self, encoder_type, ckpt_path, bn_momentum=0.01)
stage3_RL/rrl/encoder.py:115
Method__init__
(self, encoder_type, ckpt_path, bn_momentum=0.01)
stage3_RL/rrl/encoder.py:194
Method__init__
(self, encoder_type, ckpt_path)
stage3_RL/rrl/encoder.py:274
Method__init__
(self, encoder_type, ckpt_path)
stage3_RL/rrl/encoder.py:358
Method__init__
(self, encoder_type, ckpt_path)
stage3_RL/rrl/encoder.py:398
Method__init__
(self, encoder_type, ckpt_path)
stage3_RL/rrl/encoder.py:461
Method__init__
(self, encoder_type, ckpt_path)
stage3_RL/rrl/encoder.py:529
Method__init__
(self)
stage3_RL/rrl/encoder.py:570
Method__init__
(self)
stage3_RL/rrl/encoder.py:585
Method__init__
(self, sigma=[.1, 2.])
stage3_RL/rrl/adaptation_models/transforms.py:123
Method__init__
(self, transforms, p)
stage3_RL/rrl/adaptation_models/transforms.py:138
Method__init__
(self, brightness, contrast, saturation, hue)
stage3_RL/rrl/adaptation_models/transforms.py:163
Method__init__
(self, p)
stage3_RL/rrl/adaptation_models/transforms.py:174
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:61
Method__init__
(self, d_model, nheads=1, ffn=False)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:101
Method__init__
(self, num_layers, d_model, nheads=1)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:135
Method__init__
(self, d_model, height, width)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:156
Method__init__
(self, block, layers, cfg, **kwargs)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:182
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_resnet.py:55
Method__init__
(self, block, layers, cfg, **kwargs)
stage3_RL/rrl/adaptation_models/pose_resnet.py:95
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:53
Method__init__
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels, fuse_method, multi_sca
stage3_RL/rrl/adaptation_models/pose_hrnet.py:92
Method__init__
(self, cfg, **kwargs)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:266
Method__init__
(self, in_channels)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:58
Method__init__
(self, cfg, is_train, is_finetune, freeze_bn)
stage3_RL/rrl/adaptation_models/imm_joint_model.py:99
Method__init__
(self, cfg)
stage3_RL/rrl/adaptation_models/perceptual_loss.py:44
Method__init__
(self, cfg)
stage3_RL/rrl/adaptation_models/perceptual_loss.py:67
Method__init__
(self, ckpt_path=None)
stage3_RL/rrl/representations/vc1_encoder.py:10
Method__init__
(self, cfg_path, ckpt_path, bn_momentum=0.01)
stage3_RL/rrl/representations/hindex_encoder.py:27
Method__init__
(self, ckpt_path)
stage3_RL/rrl/representations/mvp_encoder.py:13
Method__init__
(self, model_and_config_folder)
stage3_RL/rrl/representations/r3m_encoder.py:11
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:71
Method__init__
(self, d_model, nheads=1, ffn=False)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:111
Method__init__
(self, num_layers, d_model, nheads=1)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:145
Method__init__
(self, d_model, height, width)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:166
Method__init__
(self, block, layers, cfg, **kwargs)
stage3_RL/rrl/representations/TTP/pose_double_head_resnet.py:192
Method__init__
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:59
Method__init__
(self, cfg, is_train, is_finetune, freeze_bn)
stage3_RL/rrl/representations/TTP/imm_joint_model.py:100
Method__init__
(self, in_channels)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:84
Method__init__
(self, cfg, is_train, is_finetune, freeze_bn, freeze_encoder)
stage3_RL/rrl/representations/TTP/imm_joint_model_resnet.py:131
Method__init__
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:11
Method__init__
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:12
Method__init__
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:12
Method__init__
(self)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:12
Method__init__
(self, env_spec, inp_dim=None, inp='obs', reg_coeff=1e-5)
stage3_RL/mjrl/mjrl/baselines/linear_baseline.py:6
Method__init__
(self, env_spec, inp_dim=None, inp='obs', learn_rate=1e-3, reg_coef=0.0, batch_size=64, epoch
stage3_RL/mjrl/mjrl/baselines/mlp_baseline.py:11
Method__init__
(self, env_spec, inp_dim=None, inp='obs', reg_coeff=1e-3)
stage3_RL/mjrl/mjrl/baselines/quadratic_baseline.py:5
Method__init__
(self, env_spec, **kwargs)
stage3_RL/mjrl/mjrl/baselines/zero_baseline.py:5
Method__init__
(self, obs_dim, act_dim, hidden_sizes=(64,64), nonlinearity='tanh', # eith
stage3_RL/mjrl/mjrl/utils/fc_network.py:7
Method__init__
(self, use_wandb=False)
stage3_RL/mjrl/mjrl/utils/logger.py:13
Method__init__
(self, obs_dim, act_dim, horizon)
stage3_RL/mjrl/mjrl/utils/gym_env.py:10
Method__init__
(self, env, env_kwargs=None, obs_mask=None, act_repeat=1, *args, **kwargs)
stage3_RL/mjrl/mjrl/utils/gym_env.py:17
Method__init__
(self, env, H, paths_per_cpu, num_cpu=1, kappa=1.0, gamma=1
stage3_RL/mjrl/mjrl/policies/mpc_actor.py:6
Method__init__
:param env_spec: specifications of the env (see utils/gym_env.py) :param min_log_std: log_std is clamped at this value and can't go b
stage3_RL/mjrl/mjrl/policies/gaussian_linear.py:10
Method__init__
(self, obs_dim, act_dim, hidden_sizes=(64,64), in_shift = None, in_scale = N
stage3_RL/mjrl/mjrl/policies/gaussian_mlp.py:9
Method__init__
(self)
stage3_RL/mjrl/mjrl/envs/swimmer.py:7
Method__init__
(self)
stage3_RL/mjrl/mjrl/envs/point_mass.py:8
Method__init__
(self)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:8
Method__init__
(self)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:8
Method__init__
(self, model_path=None, frame_skip=1, sim=None)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:31
Method__init__
(self, expert_paths, policy, epochs = 5, batch_size = 64,
stage3_RL/mjrl/mjrl/algos/behavior_cloning.py:16
Method__init__
All inputs are expected in mjrl's format unless specified :param normalized_step_size: Normalized step size (under the KL metric). Tw
stage3_RL/mjrl/mjrl/algos/npg_cg.py:24
Method__init__
(self, env, policy, baseline, demo_paths=None, normalized_step_size=0.01,
stage3_RL/mjrl/mjrl/algos/dapg.py:27
Method__init__
(self, env_name, policy, expert_paths = None, # for the ini
stage3_RL/mjrl/mjrl/algos/mbac.py:16
Method__init__
(self, env, policy, baseline, clip_coef = 0.2, epochs = 10,
stage3_RL/mjrl/mjrl/algos/ppo_clip.py:24
Method__init__
(self, env, policy, baseline, learn_rate=0.01, seed=123, de
stage3_RL/mjrl/mjrl/algos/batch_reinforce.py:23
← previousnext →301–400 of 701, ranked by callers