MCPcopy Create free account

hub / github.com/SJTU-DENG-Lab/WLA / functions

Functions234 in github.com/SJTU-DENG-Lab/WLA

⨍ Functions 234◇ Types & classes 41
↓ 104 callersMethodget
(self, dataset_index, episode_index)
dataset.py:71
↓ 11 callersFunction_get_arg
(args: dict[str, Any], key: str, default: Any = None)
experiments/robotwin/robotwin_utils.py:15
↓ 11 callersFunction_get_arg
(args: dict[str, Any], key: str, default: Any = None)
experiments/rmbench/rmbench_utils.py:16
↓ 10 callersFunction_to_int
(value)
dataset.py:26
↓ 9 callersFunction_make_transform
(size)
utils/transforms.py:21
↓ 8 callersFunction_has_language_mode
(args)
dataset.py:38
↓ 7 callersFunction_to_list
(value: Any)
experiments/rmbench/rmbench_utils.py:98
↓ 5 callersFunction_to_list
(value: Any)
experiments/robotwin/robotwin_utils.py:97
↓ 5 callersFunctionnormalize_and_pad
(data, norm_stats_key, max_dim, add_eps=True)
utils/transforms.py:51
↓ 4 callersMethod__init__
(self, input_dim, hidden_dim=1024, output_dim=2048)
models/action_model/action_model.py:54
↓ 4 callersMethod__init__
( self, num_attention_heads: int = 32, attention_head_dim: int = 64, output_di
models/action_model/cross_attention_dit.py:195
↓ 4 callersFunction_aggregate_subtasks
(subtask_segments, dataset_index, episode_index, start_index, end_index)
dataset.py:153
↓ 4 callersFunction_required_arg
(args: dict[str, Any], key: str)
experiments/robotwin/robotwin_utils.py:101
↓ 4 callersFunction_required_arg
(args: dict[str, Any], key: str)
experiments/rmbench/rmbench_utils.py:102
↓ 4 callersMethodget_tokenize_fn
(self)
models/wla.py:130
↓ 4 callersMethodget_tokenizer
(self)
models/wla.py:127
↓ 3 callersMethod__init__
(self, config: Qwen2Config)
models/transformer_encoder.py:127
↓ 3 callersMethod_accumulate_loss_outputs
(self, outputs, prefix=None)
trainer.py:227
↓ 3 callersMethod_as_batch_list
(self, value, batch_size, default="")
models/wla.py:331
↓ 3 callersFunction_is_lerobot_repo
(path)
dataset.py:1263
↓ 3 callersMethod_normalize_input_images
(self, input_images)
models/wla.py:340
↓ 3 callersFunction_normalize_quaternion
(quaternion: torch.Tensor)
experiments/rmbench/rmbench_utils.py:116
↓ 3 callersFunction_normalize_subtask_text
(text)
dataset.py:42
↓ 3 callersFunctionlog_image_collection
(log_key, images)
train.py:243
↓ 3 callersMethodlog_images
(self, logs: Dict[str, float])
trainer.py:173
↓ 3 callersFunctionoptional_str
(value)
experiments/robotwin/run_robotwin_eval.py:19
↓ 3 callersFunctionoptional_str
(value)
experiments/rmbench/run_rmbench_eval.py:35
↓ 3 callersFunctionpad_to_dim
(x, target_dim, value: float = 0.0)
utils/transforms.py:25
↓ 3 callersMethodsample_actions
( self, caption="", input_images=None, num_images_per_prompt: int = 1,
models/wla.py:627
↓ 3 callersFunctionwrite_generated_file
(path: Path, content: str, *, overwrite: bool)
experiments/robotwin/run_robotwin_eval.py:33
↓ 3 callersFunctionwrite_generated_file
(path: Path, content: str, *, overwrite: bool)
experiments/rmbench/run_rmbench_eval.py:49
↓ 2 callersMethod_apply_layerwise_cross_attention
Apply layerwise cross-attention between state-action embeddings and vision-language embeddings. Args: saction_embs: Tens
models/action_model/action_model.py:157
↓ 2 callersFunction_as_bool
(value: Any, default: bool = False)
experiments/rmbench/rmbench_utils.py:21
↓ 2 callersFunction_build_subtask_segments
(dataset)
dataset.py:99
↓ 2 callersFunction_episode_start_index
(subtask_segments, dataset_index, episode_index)
dataset.py:174
↓ 2 callersFunction_is_pad
(value)
dataset.py:32
↓ 2 callersFunction_iter_lerobot_datasets
(dataset)
dataset.py:58
↓ 2 callersFunction_make_transform
(size)
experiments/libero/libero_utils.py:36
↓ 2 callersMethod_process_output
Process the output of the transformer blocks. Args: hidden_states: Tensor of shape (B, seq_length, embedding_dim)
models/action_model/action_model.py:188
↓ 2 callersFunction_resolve_file
(path_value: Any, wla_root: Path, *, default: Optional[str] = None)
experiments/robotwin/robotwin_utils.py:50
↓ 2 callersFunction_resolve_file
(path_value: Any, wla_root: Path, *, default: Optional[str] = None)
experiments/rmbench/rmbench_utils.py:51
↓ 2 callersFunction_safe_int
(value: Any, default: int = -1)
experiments/rmbench/rmbench_utils.py:124
↓ 2 callersFunctionadapter_options
(args)
experiments/rmbench/run_rmbench_eval.py:62
↓ 2 callersFunctiondefault_wla_root
()
experiments/robotwin/run_robotwin_eval.py:25
↓ 2 callersFunctiondefault_wla_root
()
experiments/rmbench/run_rmbench_eval.py:41
↓ 2 callersMethoddevice
(self)
models/action_model/action_model.py:312
↓ 2 callersMethodencode_condition
( self, input_ids, attention_mask, mllm_output, action_cond_features, current_image_index=None, **kwar
models/model.py:442
↓ 2 callersMethodencode_condition_action
( self, input_ids, mllm_output, **kwargs )
models/model.py:417
↓ 2 callersMethodgenerate_subtask
( self, caption="", input_images=None, history_subtask_text=None, max_
models/wla.py:353
↓ 2 callersFunctionload_robotwin_or_rmbench_dataset
( data_args, model_args, training_args, dataset_root_dir=None, include_actions=True, )
dataset.py:1293
↓ 2 callersFunctionpossible_override_args
(override_args, *args)
trainer_utils.py:17
↓ 2 callersMethodprint_params
(model, modules_to_stat)
trainer.py:380
↓ 2 callersMethodreset
(self)
experiments/rmbench/rmbench_utils.py:285
↓ 2 callersFunctionresize_with_pad
(img, size)
utils/transforms.py:7
↓ 2 callersFunctionswish
(x)
models/action_model/action_encoder.py:20
↓ 1 callersMethod__init__
(self, config, *args, **kwargs)
models/wla.py:61
↓ 1 callersMethod__init__
( self, config: MLLMInContextConfig, )
models/model.py:66
↓ 1 callersMethod__init__
(self, action_dim, hidden_size)
models/action_model/action_encoder.py:58
↓ 1 callersMethod_append_subtask_history
(self, subtask_text: Any)
experiments/rmbench/rmbench_utils.py:360
↓ 1 callersFunction_as_bool
(value: Any, default: bool = False)
experiments/robotwin/robotwin_utils.py:20
↓ 1 callersFunction_as_optional_path
(value: Any)
experiments/robotwin/robotwin_utils.py:30
↓ 1 callersFunction_as_optional_path
(value: Any)
experiments/rmbench/rmbench_utils.py:31
↓ 1 callersFunction_current_input_image
(sample, model_args)
dataset.py:835
↓ 1 callersFunction_discover_lerobot_repo_ids
(root_dir)
dataset.py:1271
↓ 1 callersMethod_encode_obs
(self, observation: dict[str, Any])
experiments/robotwin/robotwin_utils.py:229
↓ 1 callersMethod_encode_obs
(self, observation: dict[str, Any])
experiments/rmbench/rmbench_utils.py:299
↓ 1 callersFunction_force_inference_steps
(model: Any, num_inference_timesteps: int)
experiments/rmbench/rmbench_utils.py:144
↓ 1 callersFunction_import_wla_modules
(wla_root: Path)
experiments/robotwin/robotwin_utils.py:86
↓ 1 callersFunction_import_wla_modules
(wla_root: Path)
experiments/rmbench/rmbench_utils.py:87
↓ 1 callersFunction_is_video_raw_dataset
(path)
dataset.py:1267
↓ 1 callersFunction_join_subtask_texts
(texts)
dataset.py:49
↓ 1 callersMethod_locate_transition
(self, index)
dataset.py:550
↓ 1 callersMethod_locate_transition
(self, index)
dataset.py:652
↓ 1 callersFunction_normalize_subtask_text
(text: Any)
experiments/rmbench/rmbench_utils.py:131
↓ 1 callersMethod_postprocess_actions
(self, actions: Any)
experiments/robotwin/robotwin_utils.py:272
↓ 1 callersMethod_postprocess_actions
(self, actions: Any)
experiments/rmbench/rmbench_utils.py:402
↓ 1 callersMethod_predict_actions
(self, observation: dict[str, Any], instruction: str)
experiments/robotwin/robotwin_utils.py:301
↓ 1 callersMethod_predict_actions
(self, TASK_ENV, observation: dict[str, Any], instruction: str)
experiments/rmbench/rmbench_utils.py:437
↓ 1 callersMethod_record_subtask
(self, TASK_ENV, instruction: str, history_subtask_text: str, subtask_text: Any)
experiments/rmbench/rmbench_utils.py:377
↓ 1 callersFunction_resolve_model_id
(model_id: str, wla_root: Path)
experiments/robotwin/robotwin_utils.py:70
↓ 1 callersFunction_resolve_model_id
(model_id: str, wla_root: Path)
experiments/rmbench/rmbench_utils.py:71
↓ 1 callersMethod_resolve_subtask_log_path
(self, TASK_ENV)
experiments/rmbench/rmbench_utils.py:366
↓ 1 callersFunction_resolve_wla_root
(args: dict[str, Any])
experiments/robotwin/robotwin_utils.py:36
↓ 1 callersFunction_resolve_wla_root
(args: dict[str, Any])
experiments/rmbench/rmbench_utils.py:37
↓ 1 callersFunction_split_subtask_text
(text: Any)
experiments/rmbench/rmbench_utils.py:138
↓ 1 callersMethod_strip_unused_modules
(self)
experiments/robotwin/robotwin_utils.py:222
↓ 1 callersMethod_strip_unused_modules
(self)
experiments/rmbench/rmbench_utils.py:292
↓ 1 callersFunction_valid_subtask_episodes
(repo_id)
dataset.py:183
↓ 1 callersFunction_validate_control_mode
(control_mode: str)
experiments/robotwin/robotwin_utils.py:108
↓ 1 callersFunction_validate_control_mode
(control_mode: str)
experiments/rmbench/rmbench_utils.py:109
↓ 1 callersFunctionbuild_eval_command
(args)
experiments/robotwin/run_robotwin_eval.py:103
↓ 1 callersFunctionbuild_eval_command
(args)
experiments/rmbench/run_rmbench_eval.py:148
↓ 1 callersFunctioncompute_global_stats
(root_dir)
utils_dataset/get_norm_stats_lerobot_v3.py:8
↓ 1 callersMethoddecode_latents
(self, latents, normalize=True, return_tensor=False)
models/wla.py:299
↓ 1 callersFunctiondefault_rmbench_root
(wla_root: Path)
experiments/rmbench/run_rmbench_eval.py:45
↓ 1 callersFunctiondefault_robotwin_root
(wla_root: Path)
experiments/robotwin/run_robotwin_eval.py:29
↓ 1 callersFunctioneval_libero
(cfg: GenerateConfig)
experiments/libero/run_libero_eval.py:69
↓ 1 callersMethodeval_step
(self, TASK_ENV, observation: dict[str, Any])
experiments/robotwin/robotwin_utils.py:317
↓ 1 callersMethodeval_step
(self, TASK_ENV, observation: dict[str, Any])
experiments/rmbench/rmbench_utils.py:468
↓ 1 callersFunctionfind_newest_checkpoint
(checkpoint_path)
trainer_utils.py:46
next →1–100 of 234, ranked by callers