MCPcopy Create free account

hub / github.com/RLinf/LaWAM / functions

Functions1,214 in github.com/RLinf/LaWAM

⨍ Functions 1,214◇ Types & classes 176↳ Endpoints 1
↓ 3 callersMethod_maybe_data_dependent_init
(self, nodes_code: Tensor)
latent_action_model/core/vq.py:382
↓ 3 callersMethod_module_param_dtype
(self, module: Optional[nn.Module], default: torch.dtype = torch.float32)
latent_action_model/core/vjepa_encoder.py:498
↓ 3 callersMethod_normalize_framework_name
(name: str)
examples/LIBERO/eval_files/model2libero_interface.py:76
↓ 3 callersFunction_recv_image
(conn: socket.socket)
deployment/model_server/server_policy_tcp_fold_towel.py:92
↓ 3 callersMethod_resolve_log_dir
(self, trainer)
latent_action_model/core/lam_lightinng.py:999
↓ 3 callersMethod_run
( self, videos: torch.Tensor, #[B, T, C,H,W] states: Optional[torch.Tensor], #[B,T,
latent_action_model/core/lam_model.py:326
↓ 3 callersMethod_sanitize_recording_run_id
(value: str | None)
deployment/model_server/server_policy_tcp_fold_towel.py:166
↓ 3 callersMethod_sanitize_recording_run_id
(value: str | None)
deployment/model_server/server_policy_http_franka.py:350
↓ 3 callersMethod_state14_from_payload
(payload: dict[str, Any])
deployment/model_server/server_policy_tcp_fold_towel.py:174
↓ 3 callersMethod_update_slot_redundancy_metrics
(self, min_indices: Tensor)
latent_action_model/core/vq.py:279
↓ 3 callersFunction_validate_vision_tokens_hw
(vision_tokens_hw: tuple[int, int])
examples/eval_utils/similarity_video.py:29
↓ 3 callersMethod_write_cache_result_to_memmap
( self, memmap: np.memmap, result: dict[str, Any], dataset_name: str,
starVLA/dataloader/gr00t_lerobot/datasets.py:2138
↓ 3 callersFunctionalign_feature_dim_with_mask
(seq: torch.Tensor, target_dim: int)
starVLA/model/framework/latent_world/batch_utils.py:15
↓ 3 callersFunctionbuild_policy_server_metadata
( policy, *, ckpt_path: str | Path, server_type: str, env: str = "generic", supported_
deployment/model_server/server_policy.py:57
↓ 3 callersMethodcheck_input
(self, data: dict[str, Any])
starVLA/dataloader/gr00t_lerobot/transform/video.py:331
↓ 3 callersFunctioncount_completed_tasks
( run_dir: str | os.PathLike[str], task_names: list[str], *, expected_num_episodes: int | None
examples/Robotwin/eval_files/robotwin_eval_common.py:381
↓ 3 callersFunctiondetach_slot
(slot_id: int, *, mode: str)
examples/Robotwin/eval_files/robotwin_batch_bridge.py:652
↓ 3 callersFunctiondetect_gpu_ids
()
examples/Robotwin/eval_files/robotwin_eval_common.py:124
↓ 3 callersMethodentropy_loss
( self, affinity: Tensor, temperature: Optional[Union[float, Tensor]] = None, )
latent_action_model/core/vq.py:389
↓ 3 callersMethodforward
(self, context: torch.Tensor)
starVLA/model/framework/vlas/lawam.py:128
↓ 3 callersFunctiongenerate_action_mask_for_used_keys
Generate mask based on action modalities, but only for used keys. Only action dimensions normalized as `binary` are marked False. Ar
starVLA/dataloader/gr00t_lerobot/datasets.py:2334
↓ 3 callersMethodget_step_data
Get the RAW data for a single step in a trajectory. No transforms are applied. Args: trajectory_id (int): The name of the traject
starVLA/dataloader/gr00t_lerobot/datasets.py:1275
↓ 3 callersFunctionget_vla_dataset
Get a LeRobotMixtureDataset object.
starVLA/dataloader/lerobot_datasets.py:247
↓ 3 callersFunctionis_module_trainable
check if a module is trainable: if the module itself has parameters, then all its parameters require_grad must be True; if the module itself
starVLA/model/tools.py:24
↓ 3 callersFunctionload_policy_from_checkpoint
( ckpt_path: str, *, use_bf16: bool = False, device: str = "cuda", )
deployment/model_server/server_policy.py:25
↓ 3 callersFunctionoverride_delta
(cfg: ModalityConfig, keys: Optional[Sequence[str]])
latent_action_model/data_loader/lerobot_dataset.py:44
↓ 3 callersMethodpredict_action
(self, examples: Sequence[LatentWorldPolicyInferExample], **kwargs)
starVLA/model/framework/lawam_framework.py:49
↓ 3 callersMethodpredict_action
(self, query_info: Dict)
deployment/model_server/tools/websocket_policy_client.py:79
↓ 3 callersMethodreset_node_count
(self)
latent_action_model/core/vq.py:672
↓ 3 callersFunctionresolve_task_scheduler_lease_timeout_sec
(heartbeat_sec: float)
examples/Robotwin/eval_files/batched_eval_runner.py:219
↓ 3 callersMethodrun
(self)
deployment/model_server/tools/websocket_policy_server.py:43
↓ 3 callersFunctionsafe_segment
(name: str)
examples/LIBERO/eval_files/libero_benchmark_adapters.py:36
↓ 3 callersMethodsample_noise
( self, shape: Tuple[int, ...], device: torch.device, dtype: torch.dtype )
starVLA/model/framework/vlas/flowmatching_expert.py:351
↓ 3 callersMethodserve_forever
(self)
deployment/model_server/tools/websocket_policy_server.py:40
↓ 3 callersMethodset_metadata
(self, dataset_metadata: DatasetMetadata)
starVLA/dataloader/gr00t_lerobot/transform/video.py:128
↓ 3 callersMethodsync_training_modes
(self, mode: bool = True)
starVLA/model/framework/vlas/lawam.py:526
↓ 3 callersFunctiontask_scheduler_state_path
(run_dir: str | os.PathLike[str])
examples/Robotwin/eval_files/batched_eval_runner.py:224
↓ 3 callersFunctionterminate_process
(proc: subprocess.Popen | None)
examples/Robotwin/eval_files/batched_eval_runner.py:96
↓ 3 callersMethodtrain
(self)
starVLA/dataloader/gr00t_lerobot/transform/base.py:80
↓ 3 callersMethodtrain
(self, mode: bool = True)
latent_action_model/core/vjepa_encoder.py:200
↓ 3 callersFunctiontrunc_normal_
(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0)
latent_action_model/core/utils/utils.py:36
↓ 2 callersMethod__init__
( self, dim: int = 1024, num_heads: int = 16, ffn_expansion_factor: int = 2,
latent_action_model/core/lam_lightinng.py:28
↓ 2 callersMethod__init__
( self, model_id: str = "/mnt/xx/xx/weights/vjepa2_1_vitl_dist_vitG_384.pt", norm_lay
latent_action_model/core/vjepa_encoder.py:107
↓ 2 callersMethod__init__
( self, in_channels: Union[int, List[int]], features=256, use_bn=False,
latent_action_model/core/utils/dpt_head.py:74
↓ 2 callersMethod__init__
(self, dim: int, num_heads: int, ffn_expansion_factor: float = 2.0, dropout: float = 0.0)
latent_action_model/core/utils/lam_decoder.py:25
↓ 2 callersMethod__init__
(self, model_dim: int, max_len: int = 5000)
latent_action_model/core/utils/pos_embs.py:13
↓ 2 callersFunction_accumulate_eval_scalar
Accumulate an optional scalar metric as float64 sum/count.
starVLA/training/train_starvla.py:74
↓ 2 callersMethod_apply_funcs
(x: torch.Tensor, funcs: list)
starVLA/dataloader/gr00t_lerobot/transform/state_action.py:79
↓ 2 callersMethod_apply_latent_norm
(self, encoded_features: torch.Tensor, use_norm: bool)
latent_action_model/core/vjepa_encoder.py:205
↓ 2 callersFunction_as_valid_fps
(value: Any)
starVLA/dataloader/lerobot_datasets.py:79
↓ 2 callersFunction_binarize_gripper_open
(open_val: np.ndarray | float)
examples/LIBERO/eval_files/libero_eval_core.py:142
↓ 2 callersMethod_build_1d_pos_embed
(self, dim: int, length: int)
latent_action_model/core/utils/pos_embs.py:97
↓ 2 callersFunction_build_dino_pixel_decoder_collator
(cfg, *, training: bool)
starVLA/dataloader/__init__.py:153
↓ 2 callersFunction_build_episode_frame_dir
(rollout_video_path: Path)
examples/LIBERO/eval_files/libero_eval_core.py:354
↓ 2 callersFunction_build_episode_video_path
( run_output_dir: Path, task_id: int, task_name: str, episode_idx: int, success: bool, )
examples/LIBERO/eval_files/libero_eval_core.py:342
↓ 2 callersMethod_build_hidden_positional_embeddings
( self, *, action_time_emb: torch.Tensor, batch_size: int, device: tor
starVLA/model/framework/vlas/flowmatching_expert.py:288
↓ 2 callersFunction_build_latent_world_collator
(cfg, *, policy_cfg, training: bool)
starVLA/dataloader/__init__.py:42
↓ 2 callersFunction_build_model_client
(args: EvalArgs)
examples/LIBERO/eval_files/libero_eval_core.py:318
↓ 2 callersMethod_build_observation
(self, payload: dict[str, Any])
deployment/model_server/server_policy_http_isaac_groot.py:214
↓ 2 callersMethod_build_raw_state_from_payload
(self, payload: dict[str, Any])
deployment/model_server/server_policy_http_franka.py:228
↓ 2 callersMethod_build_video_disk_cache_if_needed
(self, cache_inputs: dict[str, Any])
starVLA/dataloader/gr00t_lerobot/datasets.py:1980
↓ 2 callersFunction_cfg_get
(cfg, key, default=None)
deployment/model_server/server_policy.py:14
↓ 2 callersFunction_check_worker_processes
(processes: list[Any], slot_states: dict[int, WorkerSlotState])
examples/LIBERO/eval_files/libero_eval_core.py:1099
↓ 2 callersMethod_chw_uint8_to_pil
(frame: torch.Tensor)
starVLA/model/framework/latent_world/batch_builder.py:192
↓ 2 callersMethod_coerce_image_list
(raw_images: Any, *, field_name: str)
deployment/model_server/server_policy_http_isaac_groot.py:167
↓ 2 callersMethod_coerce_image_list
(raw_images: Any, *, field_name: str)
deployment/model_server/server_policy_http_franka.py:143
↓ 2 callersMethod_collect_all_paths
Recursively collect all accessed paths
starVLA/training/trainer_utils/config_tracker.py:303
↓ 2 callersMethod_compute_dtype
(self)
starVLA/model/framework/vlas/flowmatching_expert.py:264
↓ 2 callersMethod_compute_step
(self, batch: Dict, vq_training: bool)
latent_action_model/core/lam_lightinng.py:582
↓ 2 callersMethod_concat_no_padding
( self, data: dict, concat_order: list[str], modality_name: str, )
starVLA/dataloader/gr00t_lerobot/transform/concat.py:88
↓ 2 callersMethod_copy_external_log
(self, log_dir: str, pl_module)
latent_action_model/core/lam_lightinng.py:1047
↓ 2 callersFunction_crop_size_for_area_scale
(height: int, width: int, area_scale: float)
starVLA/model/framework/latent_world/batch_utils.py:95
↓ 2 callersFunction_decode_video_for_cache_task
( video_path: str, trajectory_id: int, trajectory_index: int, expected_length: int, video_
starVLA/dataloader/gr00t_lerobot/datasets.py:227
↓ 2 callersMethod_denormalize_imagenet_to_unit
(self, images: torch.Tensor)
latent_action_model/core/vjepa_encoder.py:507
↓ 2 callersMethod_encode
(self, nodes: Tensor)
latent_action_model/core/vq.py:49
↓ 2 callersMethod_ensure_sequence_2d_tensor
(value: torch.Tensor)
starVLA/dataloader/gr00t_lerobot/datasets.py:2670
↓ 2 callersMethod_expand_future_tokens
( self, *, batch_size: int, device: torch.device, dtype: torch.dtype,
starVLA/model/framework/vlas/flowmatching_expert.py:267
↓ 2 callersMethod_extract_features
(self, pixel_values_videos: torch.Tensor)
latent_action_model/core/vjepa_encoder.py:226
↓ 2 callersMethod_filter_leaf_paths
Filter to only leaf paths (no sub-paths)
starVLA/training/trainer_utils/config_tracker.py:316
↓ 2 callersFunction_finalize_results
( *, args: EvalArgs, adapter: BenchmarkAdapter, run_output_dir: Path, log_output_dir: Path
examples/LIBERO/eval_files/libero_eval_core.py:641
↓ 2 callersFunction_format_prompt
(prompt: str, *, instruction: str)
starVLA/model/framework/latent_world/vlm_adapter.py:17
↓ 2 callersFunction_get_benchmark_dict
()
examples/LIBERO/eval_files/libero_eval_core.py:136
↓ 2 callersMethod_get_frame_pos
(self, ids, H_patches, W_patches)
latent_action_model/core/utils/modules.py:215
↓ 2 callersMethod_get_frame_pos
(self, ids, H_patches=None, W_patches=None)
latent_action_model/core/utils/modules.py:379
↓ 2 callersMethod_get_interpolation
Get the interpolation mode for the given backend. Args: interpolation (str): The interpolation mode. backend
starVLA/dataloader/gr00t_lerobot/transform/video.py:265
↓ 2 callersMethod_get_nested_value
Get nested value through dot-separated path with bracket notation support
starVLA/training/trainer_utils/config_tracker.py:335
↓ 2 callersMethod_get_retry_index
(self, original_index: int, attempt: int, reason: str)
starVLA/dataloader/gr00t_lerobot/datasets.py:2591
↓ 2 callersMethod_get_temperature
Exponentially decayed temperature: start -> end over decay steps.
latent_action_model/core/vq.py:529
↓ 2 callersMethod_get_video_episode_from_timestamp
(self, trajectory_id: int, video_key: str)
starVLA/dataloader/gr00t_lerobot/datasets.py:1929
↓ 2 callersMethod_infer_binary_indices
(state_norm_stats: Optional[dict])
examples/Robotwin/starvla_policy/deploy_policy.py:210
↓ 2 callersMethod_infer_binary_indices
(state_norm_stats: Optional[dict])
examples/Robotwin/eval_files/model2robotwin_interface.py:563
↓ 2 callersFunction_infer_vjepa_hub_name
(model_id: str)
latent_action_model/core/vjepa_encoder.py:43
↓ 2 callersMethod_inject_queries
( self, *, inputs_embeds: torch.Tensor, placeholder_mask: torch.BoolTensor,
starVLA/model/framework/vlas/lawam.py:326
↓ 2 callersMethod_invert_normalized_value
(x: torch.Tensor, mode: str)
starVLA/dataloader/gr00t_lerobot/transform/state_action.py:501
↓ 2 callersMethod_is_latent_world_framework
(cls, framework_name: str)
examples/LIBERO/eval_files/model2libero_interface.py:80
↓ 2 callersFunction_is_main_process
()
starVLA/dataloader/gr00t_lerobot/datasets.py:80
↓ 2 callersMethod_load_hf_model
(self, model_id: str)
latent_action_model/core/vjepa_encoder.py:147
↓ 2 callersMethod_local_to_abs_index
Map mode-local step index to absolute hf_dataset index.
starVLA/dataloader/gr00t_lerobot/datasets.py:833
↓ 2 callersFunction_normalize_cached_video_frames_static
( frames: np.ndarray, expected_length: int, video_key: str, trajectory_id: int, video_path
starVLA/dataloader/gr00t_lerobot/datasets.py:201
↓ 2 callersMethod_normalize_metric_aliases
Collapse legacy metric spellings onto the canonical W&B keys.
starVLA/training/train_starvla.py:369
← previousnext →101–200 of 1,214, ranked by callers