Functions — github.com/TongZhangTHU/sgr

CodeHub MCPcopy Create free account

hub / github.com/TongZhangTHU/sgr / functions

Functions316 in github.com/TongZhangTHU/sgr

⨍ Functions 316◇ Types & classes 54

Functioncheck_gimbal_lock

(pred_rot_and_grip, gt_rot_and_grip, resolution)

helpers/utils.py:71

Functioncorrect_rotation_instability

(disc, resolution)

helpers/utils.py:57

Functioncreate_agent

(cfg: DictConfig)

agents/peract_bc/launch_utils.py:362

Functioncreate_agent

(cfg: DictConfig)

agents/sgr/launch_utils.py:391

Functioncreate_agent

(cfg: DictConfig)

agents/semantic_bc_lang/launch_utils.py:350

Functioncreate_replay

(batch_size: int, timesteps: int, save_dir: str, cameras: list, voxel_size

agents/peract_bc/launch_utils.py:32

Functioncreate_replay

(batch_size: int, timesteps: int, save_dir: str, cameras

agents/sgr/launch_utils.py:30

Functioncreate_replay

(batch_size: int, timesteps: int, save_dir: str, cameras

agents/semantic_bc_lang/launch_utils.py:29

helpers/clip/core/clip.py:405

Methodencode_text

agents/peract_bc/perceiver_lang_io.py:324

Functionextract_obs

(obs: Observation, cameras, t: int = 0, prev_action=None,

helpers/utils.py:358

Functionfill_multi_task_replay

(cfg: DictConfig, obs_config: ObservationConfig, rank: i

agents/peract_bc/launch_utils.py:297

Functionfill_multi_task_replay

( cfg: DictConfig, obs_config: ObservationConfig, rank: int, replay: ReplayBuffer, tasks:

agents/sgr/launch_utils.py:336

Functionfill_multi_task_replay

( cfg: DictConfig, obs_config: ObservationConfig, rank: int, replay: ReplayBuffer, tasks:

agents/semantic_bc_lang/launch_utils.py:297

Functionfill_replay

(cfg: DictConfig, obs_config: ObservationConfig, rank: int, re

agents/peract_bc/launch_utils.py:207

Functionfill_replay

(cfg: DictConfig, obs_config: ObservationConfig, rank: int, re

agents/sgr/launch_utils.py:228

Functionfill_replay

(cfg: DictConfig, obs_config: ObservationConfig, rank: int, re

agents/semantic_bc_lang/launch_utils.py:213

Functionfind_minimum_difference

helpers/demo_loading_utils.py:66

(self, x, gamma, beta)

helpers/network_utils.py:58

helpers/network_utils.py:103

(self, x, gamma, beta)

helpers/network_utils.py:120

helpers/network_utils.py:166

helpers/network_utils.py:210

helpers/network_utils.py:233

helpers/network_utils.py:253

helpers/network_utils.py:285

helpers/network_utils.py:324

(self, observations, low_dim_ins)

helpers/network_utils.py:382

(self, observations, low_dim_ins, lang_goal_emb)

helpers/network_utils.py:453

(self, x, **kwargs)

helpers/network_utils.py:488

helpers/network_utils.py:502

helpers/network_utils.py:525

helpers/network_utils.py:548

helpers/network_utils.py:580

(self, observations, low_dim_ins, lang_goal_emb)

helpers/network_utils.py:654

helpers/network_utils.py:701

helpers/network_utils.py:736

helpers/network_utils.py:763

(self, feature)

helpers/network_utils.py:794

(self, x: torch.Tensor)

helpers/clip/core/clip.py:75

helpers/clip/core/clip.py:101

helpers/clip/core/clip.py:170

(self, x: torch.Tensor)

helpers/clip/core/clip.py:213

(self, x: torch.Tensor)

helpers/clip/core/clip.py:220

(self, x: torch.Tensor)

helpers/clip/core/clip.py:242

(self, x: torch.Tensor)

helpers/clip/core/clip.py:255

(self, x: torch.Tensor)

helpers/clip/core/clip.py:276

(self, x, **kwargs)

agents/peract_bc/perceiver_lang_io.py:59

agents/peract_bc/perceiver_lang_io.py:71

agents/peract_bc/perceiver_lang_io.py:85

(self, x, context=None, mask=None)

agents/peract_bc/perceiver_lang_io.py:103

( self, ins, proprio, lang_goal_emb, lang_token_em

agents/peract_bc/perceiver_lang_io.py:333

(self, rgb_pcd, proprio, pcd, lang_goal_emb, lang_token_embs, bounds=None, prev_bounds=None, p

agents/peract_bc/qattention_peract_bc_agent.py:72

(self, rgb_pcd, proprio, pcd, lang_goal_emb,

agents/sgr/sgr_agent.py:78

(self, pcd, rgb, bounds, proprio,

agents/sgr/sgr_network.py:424

(self, pcds, rgbs, bounds, low_dim_ins, lang_goal_emb)

agents/semantic_bc_lang/networks.py:276

(self, rgb_pcd, proprio, pcd, lang_goal_emb,

agents/semantic_bc_lang/semantic_bc_lang_agent.py:64

Methodforward_spatial

(self, x: torch.Tensor)

helpers/clip/core/clip.py:295

Functionfourier_encode

(x, max_freq, num_bands = 4)

agents/peract_bc/perceiver_lang_io.py:38

Functionget_device

helpers/utils.py:464

Functionkeypoint_discovery

(demo: Demo, stopping_delta=0.1, method='heuristic')

helpers/demo_loading_utils.py:21

helpers/custom_rlbench_env.py:283

helpers/custom_rlbench_env.py:477

Methodload_weights

(self, savedir: str)

helpers/preprocess_agent.py:80

Methodload_weights

(self, savedir: str)

agents/peract_bc/qattention_stack_agent.py:107

Methodload_weights

(self, savedir: str)

agents/peract_bc/qattention_peract_bc_agent.py:651

Methodload_weights

(self, savedir: str)

agents/sgr/sgr_agent.py:915

Methodload_weights

(self, savedir: str)

agents/sgr/stack_agent.py:122

Methodload_weights

(self, savedir: str)

agents/semantic_bc_lang/semantic_bc_lang_agent.py:472

Methodload_weights

(self, savedir: str)

agents/semantic_bc_lang/stack_agent.py:105

Functionloss_weights

(replay_sample, beta=1.0)

helpers/utils.py:32

Functionnormalize_quaternion

helpers/utils.py:53

Methodobservation_elements

helpers/custom_rlbench_env.py:59

Methodobservation_elements

helpers/custom_rlbench_env.py:246

Methodobservation_elements

helpers/custom_rlbench_env.py:440

Functionpoint_to_pixel_index

( point: np.ndarray, extrinsics: np.ndarray, intrinsics: np.ndarray)

helpers/utils.py:142

Functionpoint_to_voxel_index

( point: np.ndarray, voxel_size: np.ndarray, coord_bounds: np.ndarray)

helpers/utils.py:103

Functionpoint_to_voxel_index_no_clip

( point: np.ndarray, voxel_size: np.ndarray, coord_bounds: np.ndarray)

helpers/utils.py:117

Functionpreprocess

Pre-process input (subtract mean, divide by std).

helpers/utils.py:285

Functionquaternion_to_discrete_euler

(quaternion, resolution)

helpers/utils.py:82

Functionquaternion_to_soft_discrete_euler

(quaternion, resolution)

helpers/utils.py:89

Functionrand_discrete

(size, min=0, max=1)

helpers/utils.py:347

Functionrand_dist

(size, min=-1.0, max=1.0)

helpers/utils.py:343

Methodregister_callback

helpers/custom_rlbench_env.py:118

Methodregister_callback

helpers/custom_rlbench_env.py:305

Methodregister_callback

helpers/custom_rlbench_env.py:503

Methodremove_language_head

agents/semantic_bc_lang/networks.py:102

helpers/custom_rlbench_env.py:295

helpers/custom_rlbench_env.py:489

helpers/preprocess_agent.py:86

Methodreset_to_demo

helpers/custom_rlbench_env.py:372

Methodreset_to_demo

helpers/custom_rlbench_env.py:569

Functionrun_seed

(rank, cfg: DictConfig, obs_config: ObservationConfig, cams, multi_task, seed, world_size)

run_seed_fn.py:18

Methodsave_weights

(self, savedir: str)

helpers/preprocess_agent.py:83

Methodsave_weights

(self, savedir: str)

agents/peract_bc/qattention_stack_agent.py:111

Methodsave_weights

(self, savedir: str)

agents/peract_bc/qattention_peract_bc_agent.py:670

Methodsave_weights

(self, savedir: str)

agents/sgr/sgr_agent.py:943

Methodsave_weights

(self, savedir: str)

agents/sgr/stack_agent.py:126

Methodsave_weights

(self, savedir: str)

agents/semantic_bc_lang/semantic_bc_lang_agent.py:491

Methodsave_weights

(self, savedir: str)

agents/semantic_bc_lang/stack_agent.py:109

← previous next →201–300 of 316, ranked by callers