Functions — github.com/Ke-Wang1017/openpi

hub / github.com/Ke-Wang1017/openpi_subtask / functions

Functions901 in github.com/Ke-Wang1017/openpi_subtask

⨍ Functions 901◇ Types & classes 226

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:158

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:189

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:198

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:212

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:234

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:252

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:274

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:292

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:329

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:353

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:372

Method__call__

(self, data: DataDict)

src/openpi/transforms.py:389

Method__call__

(self, data: dict)

src/openpi/policies/droid_policy.py:35

Method__call__

(self, data: dict)

src/openpi/policies/droid_policy.py:79

Method__call__

(self, data: dict)

src/openpi/policies/flexiv_new_policy.py:56

Method__call__

(self, data: dict)

src/openpi/policies/flexiv_new_policy.py:109

Method__call__

(self, data: dict)

src/openpi/policies/arx_policy.py:40

Method__call__

(self, data: dict)

src/openpi/policies/arx_policy.py:87

Method__call__

(self, data: dict)

src/openpi/policies/arx_policy.py:108

Method__call__

(self, data: dict)

src/openpi/policies/arx_policy.py:155

Method__call__

(self, data: dict)

src/openpi/policies/libero_policy.py:85

Method__call__

(self, data: dict)

src/openpi/policies/libero_policy.py:154

Method__call__

(self, data: dict)

src/openpi/policies/libero_subtask_policy.py:58

Method__call__

(self, data: dict)

src/openpi/policies/libero_subtask_policy.py:123

Method__call__

(self, data: dict)

src/openpi/policies/aloha_policy.py:42

Method__call__

(self, data: dict)

src/openpi/policies/aloha_policy.py:98

Method__call__

(self, data: dict)

src/openpi/policies/flexiv_subtask_policy.py:58

Method__call__

(self, data: dict)

src/openpi/policies/flexiv_subtask_policy.py:117

Method__call__

Create a group.

src/openpi/training/config.py:104

Method__call__

(self, model_config: _model.BaseModelConfig)

src/openpi/training/config.py:115

Method__call__

(self, model_config: _model.BaseModelConfig)

src/openpi/training/config.py:172

Method__call__

(self, directory: epath.Path)

src/openpi/training/checkpoints.py:135

Method__call__

(self, path: nnx.filterlib.PathParts, x: Any)

src/openpi/shared/nnx_utils.py:60

Method__call__

(self, x)

src/openpi/models/vit.py:35

Method__call__

Applies the AddPositionEmbs module. Args: inputs: Inputs to the layer. Returns: Output tensor with shape `(bs, t

src/openpi/models/vit.py:50

Method__call__

Applies Transformer MlpBlock module.

src/openpi/models/vit.py:78

Method__call__

Applies Encoder1DBlock module. Args: inputs: Inputs to the layer. deterministic: Dropout will not be applied when set to

src/openpi/models/vit.py:124

Method__call__

Applies Transformer model on the inputs. Args: x: Inputs to the layer. train: Set to `True` when training. Retur

src/openpi/models/vit.py:180

Method__call__

(self, inputs, *, train)

src/openpi/models/vit.py:235

Method__call__

(self, x, cond)

src/openpi/models/gemma.py:115

Method__call__

(self, xs, positions, attn_mask, kv_cache)

src/openpi/models/gemma.py:164

Method__call__

(self, x)

src/openpi/models/gemma.py:260

Method__call__

(self, xs, kv_cache, positions, attn_mask, adarms_cond, deterministic=True)

src/openpi/models/gemma.py:293

Method__call__

( self, # list of token arrays, one for each expert, or None if that expert should not be run

src/openpi/models/gemma.py:389

Method__call__

Applies Transformer MlpBlock module.

src/openpi/models/siglip.py:61

Method__call__

(self, x, deterministic=True)

src/openpi/models/siglip.py:84

Method__call__

(self, x, deterministic=True)

src/openpi/models/siglip.py:123

Method__call__

(self, x)

src/openpi/models/siglip.py:172

Method__call__

(self, image, *, train=False)

src/openpi/models/siglip.py:208

Method__call__

(self, x, cond)

src/openpi/models/gemma_05.py:115

Method__call__

(self, xs, positions, attn_mask, kv_cache)

src/openpi/models/gemma_05.py:184

Method__call__

(self, x)

src/openpi/models/gemma_05.py:289

Method__call__

(self, xs, kv_cache, positions, attn_mask, adarms_cond, deterministic=True)

src/openpi/models/gemma_05.py:322

Method__call__

( self, # list of token arrays, one for each expert, or None if that expert should not be run

src/openpi/models/gemma_05.py:425

Method__call__

(self, eqn, x)

src/openpi/models/gemma_fast.py:81

Method__call__

(self, x)

src/openpi/models/gemma_fast.py:90

Method__call__

(self, x, positions, attn_mask, kv_cache, decode, deterministic=True)

src/openpi/models/gemma_fast.py:186

Method__call__

(self, x, kv_cache, positions, attn_mask, decode, deterministic=True)

src/openpi/models/gemma_fast.py:261

Method__call__

Embed only, or complete forward pass. Args: tokens: Embedded, then and appended to `embedded_prefix`. Can be None. embedd

src/openpi/models/gemma_fast.py:303

Method__call__

(self, eqn: str, x)

src/openpi/models/lora.py:55

Method__call__

(self, x)

src/openpi/models/lora.py:124

Method__call__

(self, inputs: jnp.ndarray)

src/openpi/models/utils/fsq_tokenizer.py:88

Method__call__

(self, x: jnp.ndarray, *, train: bool = True)

src/openpi/models/utils/fsq_tokenizer.py:135

Method__call__

(self, x: jnp.ndarray, *, train: bool = True)

src/openpi/models/utils/fsq_tokenizer.py:157

Method__call__

Applies the GeGLU activation to the inputs. Args: inputs: the nd-array to apply the GeGLU activation function to. Returns:

src/openpi/models/utils/fsq_tokenizer.py:255

Method__call__

( self, x: jnp.ndarray, y: jnp.ndarray, *, mask_self: jnp.ndarray | No

src/openpi/models/utils/fsq_tokenizer.py:276

Method__call__

( self, y: jnp.ndarray, *, train: bool = True, state_conditioning: jnp

src/openpi/models/utils/fsq_tokenizer.py:351

Method__call__

Dummy for .init

src/openpi/models/utils/fsq_tokenizer.py:468

Method__call__

(self, x: dict)

scripts/compute_norm_stats.py:20

Method__getitem__

(self, index: SupportsIndex)

src/openpi/training/data_loader.py:25

Method__getitem__

(self, index: SupportsIndex)

src/openpi/training/data_loader.py:58

Method__getitem__

(self, index: SupportsIndex)

src/openpi/training/data_loader.py:104

Method__init__

(self, policy: _base_policy.BasePolicy, action_horizon: int)

packages/openpi-client/src/openpi_client/action_chunk_broker.py:19

Method__init__

(self, host: str = "0.0.0.0", port: Optional[int] = None, api_key: Optional[str] = None)

packages/openpi-client/src/openpi_client/websocket_client_policy.py:18

Method__init__

( self, environment: _environment.Environment, agent: _agent.Agent, subscriber

packages/openpi-client/src/openpi_client/runtime/runtime.py:13

Method__init__

(self, policy: _base_policy.BasePolicy)

packages/openpi-client/src/openpi_client/runtime/agents/policy_agent.py:10

Method__init__

Initialize the Policy. Args: model: The model to use for action sampling. rng: Random number generator key for JAX mo

src/openpi/policies/policy.py:26

Method__init__

(self, policy: _base_policy.BasePolicy, record_dir: str)

src/openpi/policies/policy.py:165

Method__init__

( self, data_dir: str, batch_size: int, *, # Force keyword-only arguments

src/openpi/training/droid_rlds_dataset.py:27

Method__init__

(self, dataset: Dataset, transforms: Sequence[_transforms.DataTransformFn])

src/openpi/training/data_loader.py:54

Method__init__

( self, dataset: IterableDataset, transforms: Sequence[_transforms.DataTransformFn],

src/openpi/training/data_loader.py:66

Method__init__

(self, model_config: _model.BaseModelConfig, num_samples: int)

src/openpi/training/data_loader.py:100

Method__init__

Create a PyTorch data loader. Args: dataset: The dataset to load. local_batch_size: The local batch size for each pro

src/openpi/training/data_loader.py:388

Method__init__

( self, dataset: DroidRldsDataset, *, sharding: jax.sharding.Sharding | None =

src/openpi/training/data_loader.py:496

Method__init__

(self, data_config: _config.DataConfig, data_loader: TorchDataLoader | RLDSDataLoader)

src/openpi/training/data_loader.py:535

Method__init__

(self)

src/openpi/shared/normalize.py:20

Method__init__

( self, policy: _base_policy.BasePolicy, host: str = "0.0.0.0", port: int | No

src/openpi/serving/websocket_policy_server.py:21

Method__init__

(self, **kwargs)

src/openpi/models_pytorch/preprocessing_pytorch.py:161

Method__init__

(self, config)

src/openpi/models_pytorch/pi0_pytorch.py:85

Method__init__

( self, vlm_config, action_expert_config, use_adarms=None, precision:

src/openpi/models_pytorch/gemma_pytorch.py:17

Method__init__

( self, vocab_size=256000, hidden_size=3072, intermediate_size=24576,

src/openpi/models_pytorch/transformers_replace/models/gemma/configuration_gemma.py:115

Method__init__

(self, dim: int, eps: float = 1e-6, cond_dim: Optional[int] = None)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:50

Method__init__

(self, config: GemmaConfig, device=None)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:130

Method__init__

(self, config: GemmaConfig, layer_idx: int)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:259

Method__init__

(self, config: GemmaConfig, layer_idx: int)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:333

Method__init__

(self, config: GemmaConfig)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:420

Method__init__

(self, config)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:567

Method__init__

(self, config)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:690

Method__init__

(self, config)

src/openpi/models_pytorch/transformers_replace/models/gemma/modeling_gemma.py:782

Method__init__

(self, config: PaliGemmaConfig)

src/openpi/models_pytorch/transformers_replace/models/paligemma/modeling_paligemma.py:92

← previous next →401–500 of 901, ranked by callers