MCPcopy Create free account

hub / github.com/ALRhub/X_IL / functions

Functions1,241 in github.com/ALRhub/X_IL

⨍ Functions 1,241◇ Types & classes 225↳ Endpoints 7
Method__init__
( self, cam_names: list[str], env_name: list[str], data_directory: os.PathLike
environments/dataset/robocasa_pc_dataset_memory.py:14
Method__init__
(self, env, pc_sampler: BasePointCloudSampler, num_points: int)
environments/wrappers/point_cloud_sampling_wrapper.py:7
Method__init__
(self, env)
environments/wrappers/point_cloud_wrapper.py:5
Method__init__
(self, env)
environments/wrappers/robosuite_wrapper.py:19
Method__init__
(self)
utils/visualizer/visualizer/pointcloud.py:15
Method__init__
Initialize.
trainers/base_trainer.py:29
Method__init__
( self, model: DictConfig, obs_encoders: DictConfig, language_
agents/ddpm_agent.py:18
Method__init__
( self, model: DictConfig, obs_encoders: DictConfig, language_encoders: DictCo
agents/base_agent.py:22
Method__init__
( self, model: DictConfig, language_encoders: DictConfig, obs_
agents/beso_agent.py:24
Method__init__
( self, model: DictConfig, obs_encoders: DictConfig, language_encoders: DictCo
agents/fm_agent.py:19
Method__init__
( self, model: DictConfig, obs_encoders: DictConfig, language_encoders: DictCo
agents/bc_agent.py:19
Method__init__
( self, encoder: DictConfig, state_dim: int, action_dim: int,
agents/backbones/decoder_only.py:128
Method__init__
( self, encoder: DictConfig, decoder: DictConfig, state_dim: i
agents/backbones/encoder_decoder.py:202
Method__init__
( self, embedding_dim, num_heads, num_blocks, voca
agents/backbones/xlstm/blocks.py:112
Method__init__
(self, config: xLSTMLargeConfig)
agents/backbones/xlstm/blocks.py:187
Method__init__
(self, config: mLSTMLayerConfig)
agents/backbones/xlstm/blocks.py:265
Method__init__
(self, config: xLSTMLargeConfig)
agents/backbones/xlstm/blocks.py:410
Method__init__
(self, config: xLSTMLargeConfig)
agents/backbones/xlstm/blocks.py:494
Method__init__
( self, num_heads: int, head_dim: int, eps: float = 1e-6, use_weight:
agents/backbones/xlstm/components.py:123
Method__init__
( self, num_heads: int, head_dim: int, eps: float = 1e-6, use_weight:
agents/backbones/xlstm/components.py:212
Method__init__
(self, config: mLSTMBackendConfig)
agents/backbones/xlstm/mlstm_kernels/torch/backend_module.py:96
Method__init__
(self, time_provider=None)
agents/backbones/xlstm/mlstm_kernels/utils/time.py:35
Method__init__
(self, hidden_size)
agents/backbones/mamba/blocks.py:99
Method__init__
( self, dim, mixer_cls, mlp_cls, norm_cls=nn.LayerNorm, fused_add_norm=False, residual_in_fp32=False
agents/backbones/mamba/blocks.py:121
Method__init__
( self, config: MambaConfig, initializer_cfg=None, device=None, dtype=
agents/backbones/mamba/mamba.py:247
Method__init__
(self, ndim, bias)
agents/backbones/transformer/blocks.py:24
Method__init__
(self, dim: int, eps: float = 1e-8)
agents/backbones/transformer/blocks.py:35
Method__init__
(self, in_dim: int, out_dim: int)
agents/backbones/transformer/blocks.py:47
Method__init__
( self, n_embd: int, n_head: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:57
Method__init__
( self, n_embd: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:163
Method__init__
(self, hidden_size)
agents/backbones/transformer/blocks.py:201
Method__init__
( self, n_embd: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:224
Method__init__
( self, embed_dim: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:263
Method__init__
( self, embed_dim: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:303
Method__init__
( self, embed_dim: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:343
Method__init__
( self, embed_dim: int, n_heads: int, attn_pdrop: float,
agents/backbones/transformer/blocks.py:383
Method__init__
Args: parameters: Iterable of `torch.nn.Parameter`; usually the result of `model.parameters()`. decay: The ex
agents/utils/ema.py:15
Method__init__
(self, inv_gamma=1., power=1., min_value=0., max_value=1., start_at=0, last_epoch=0)
agents/utils/ema.py:121
Method__init__
Visualizer is used to generate images of samples, reconstructions, latent traversals and so on of the trained model. Paramet
agents/utils/visualizer.py:29
Method__init__
(self, embed_dim)
agents/utils/time_embedding.py:24
Method__init__
(self, embed_dim)
agents/utils/time_embedding.py:44
Method__init__
(self, hidden_size, frequency_embedding_size=256)
agents/utils/time_embedding.py:59
Method__init__
( self, y_data: np.ndarray, scale_data: bool, device: str
agents/utils/scaler.py:15
Method__init__
(self, x_data: np.ndarray, y_data: np.ndarray, scale_data: bool, device: str)
agents/utils/scaler.py:97
Method__init__
(self, y_data: np.ndarray, scale_data: bool, device: str)
agents/utils/scaler.py:218
Method__init__
(self, model_name: str, freeze_backbone: bool, device: str, camera_names: list[str] = None)
agents/encoders/dino_img_encoder.py:9
Method__init__
(self, freeze_backbone: bool = True, model_name: str = "RN50")
agents/encoders/clip_lang_encoder.py:10
Method__init__
(self, model_name: str, freeze_backbone: bool, device: str, camera_names: list[str] = None)
agents/encoders/clip_img_global_encoder.py:9
Method__init__
( self, point_mlp_encoder: DictConfig, resnet_encoder: DictConfig, )
agents/encoders/point_img_encoder.py:8
Method__init__
Assumes rgb input: B,C,H,W Assumes low_dim input: B,D
agents/encoders/multi_image_obs_encoder.py:25
Method__init__
( self, use_pc_color: bool, out_channels: int, use_layer_norm:
agents/encoders/point_mlp_encoder.py:42
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
agents/encoders/film_resnets.py:50
Method__init__
(self, cond_dim, latent_dim=128, pretrained=False, hidden_size=512)
agents/encoders/film_resnets.py:92
Method__init__
(self, cond_dim, latent_dim=128, pretrained=False)
agents/encoders/film_resnets.py:204
Method__init__
(self, vision_tower, args, delay_load=False)
agents/encoders/siglip_encoder.py:7
Method__init__
(self, condition_dim)
agents/encoders/pretrained_resnets.py:28
Method__init__
(self, condition_dim)
agents/encoders/pretrained_resnets.py:66
Method__init__
(self, condition_dim)
agents/encoders/pretrained_resnets.py:104
Method__init__
Args: input_shape (tuple, list): shape of input (not including batch dimension) crop_height (int): crop height
agents/encoders/crop_randomizer.py:12
Method__init__
( self, use_pc_color: bool, out_channels: int, n_layers: int = 4, use_
agents/encoders/point_attn_encoder.py:7
Method__init__
( self, latent_dim: int = 128, pretrained: bool = False, freeze_backbone: bool
agents/encoders/resnets.py:102
Method__init__
(self, dim)
agents/models/ddpm/utils.py:11
Method__init__
( self, state_dim: int, action_dim: int, inner_model: DictConf
agents/models/ddpm/ddpm.py:21
Method__init__
( self, backbones: DictConfig, device: str = 'cpu', )
agents/models/bc/bc.py:10
Method__init__
( self, backbones: DictConfig, ln=False, device: str = 'cpu',
agents/models/flow_matching/rf.py:8
Method__init__
( self, model: DictConfig, # language_goal: DictConfig, obs_en
agents/models/beso/beso.py:15
Method__init__
(self, mean=0.0, std=1.0)
agents/models/beso/utils/transforms.py:24
Method__init__
(self, mean=0.0, std=1.0)
agents/models/beso/utils/transforms.py:38
Method__init__
(self, pad)
agents/models/beso/utils/transforms.py:74
Method__init__
(self, max_pos, max_orn)
agents/models/beso/utils/transforms.py:100
Method__init__
Args: parameters: Iterable of `torch.nn.Parameter`; usually the result of `model.parameters()`. decay: The ex
agents/models/beso/utils/ema.py:15
Method__init__
(self, inv_gamma=1., power=1., min_value=0., max_value=1., start_at=0, last_epoch=0)
agents/models/beso/utils/ema.py:121
Method__init__
(self, bpe_path: str = default_bpe())
agents/models/beso/utils/clip_tokenizer.py:63
Method__init__
(self)
agents/models/beso/utils/automatic_lang_annotator_mp.py:322
Method__init__
( self, optimizer: Optimizer, configs: DictConfig, )
agents/models/beso/utils/lr_schedulers/warmup_lr_scheduler.py:55
Method__init__
( self, optimizer: Optimizer, configs: DictConfig, )
agents/models/beso/utils/lr_schedulers/tri_stage_scheduler.py:89
Method__init__
(self, optimizer, init_lr)
agents/models/beso/utils/lr_schedulers/lr_scheduler.py:34
Method__init__
( self, resolution: int, patch_size: int, decoder_depth: int, decoder_
agents/models/beso/models/img_generation/masked_transformer_decoder.py:71
Method__init__
(self, inplanes, planes, stride=1)
agents/models/beso/models/networks/clip.py:46
Method__init__
(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)
agents/models/beso/models/networks/clip.py:94
Method__init__
(self, layers, output_dim, heads, input_resolution=224, width=64)
agents/models/beso/models/networks/clip.py:140
Method__init__
(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None)
agents/models/beso/models/networks/clip.py:228
Method__init__
(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None)
agents/models/beso/models/networks/clip.py:256
Method__init__
(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, output_dim: int)
agents/models/beso/models/networks/clip.py:267
Method__init__
(self, dim)
agents/models/beso/models/networks/mdtv_transformer.py:14
Method__init__
(self, dim)
agents/models/beso/models/networks/mdt_transformer.py:16
Method__init__
(self)
agents/models/beso/models/networks/transformers/utils.py:8
Method__init__
(self, in_dim: int, out_dim: int)
agents/models/beso/models/networks/transformers/utils.py:43
Method__init__
(self, dim: int, dim_head: int = 64, heads: int = 8)
agents/models/beso/models/networks/transformers/perceiver_resampler.py:14
Method__init__
(self, scale, causal = False, num_buckets = 32, max_distance = 128, heads = 8)
agents/models/beso/models/networks/transformers/position_embeddings.py:193
Method__init__
(self, dim, *, heads, depth, log_distance = False, norm = False)
agents/models/beso/models/networks/transformers/position_embeddings.py:239
Method__init__
(self, ndim, bias)
agents/models/beso/models/networks/transformers/transformer_blocks.py:32
Method__init__
(self, dim: int, eps: float = 1e-8)
agents/models/beso/models/networks/transformers/transformer_blocks.py:44
Method__init__
(self, in_dim: int, out_dim: int)
agents/models/beso/models/networks/transformers/transformer_blocks.py:56
Method__init__
( self, n_embd: int, n_head: int, attn_pdrop: float,
agents/models/beso/models/networks/transformers/transformer_blocks.py:68
Method__init__
( self, n_embd: int, n_heads: int, attn_pdrop: float,
agents/models/beso/models/networks/transformers/transformer_blocks.py:185
Method__init__
( self, n_embd: int, n_heads: int, attn_pdrop: float,
agents/models/beso/models/networks/transformers/transformer_blocks.py:220
Method__init__
(self, hidden_size)
agents/models/beso/models/networks/transformers/transformer_blocks.py:249
Method__init__
( self, n_embd, n_heads, attn_pdrop, resid_pdr
agents/models/beso/models/networks/transformers/transformer_blocks.py:270
Method__init__
( self, n_embd, n_heads, attn_pdrop, resid_pdr
agents/models/beso/models/networks/transformers/transformer_blocks.py:315
← previousnext →501–600 of 1,241, ranked by callers