MCPcopy Create free account

hub / github.com/Little-Podi/AdaWorld / functions

Functions471 in github.com/Little-Podi/AdaWorld

⨍ Functions 471◇ Types & classes 120↳ Endpoints 3
↓ 1 callersFunctionunpatchify
(patches: Tensor, size: int, h_out: int, w_out: int)
lam/lam/modules/blocks.py:19
↓ 1 callersFunctionunpatchify
(patches: Tensor, size: int, h_out: int, w_out: int)
worldmodel/external/lam/modules/blocks.py:17
Method__call__
(self, n, **kwargs)
worldmodel/vwm/lr_scheduler.py:32
Method__call__
(self, n, **kwargs)
worldmodel/vwm/lr_scheduler.py:78
Method__call__
(self, sigma: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/loss_weighting.py:5
Method__call__
(self, sigma: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/loss_weighting.py:13
Method__call__
(self, sigma: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/loss_weighting.py:23
Method__call__
(self, sigma: Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser_scaling.py:11
Method__call__
(self, sigma: Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser_scaling.py:20
Method__call__
(self, sigma: Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser_scaling.py:29
Method__call__
(self, sigma: Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser_scaling.py:38
Method__call__
(self, n, do_append_zero=True, device="cpu", flip=False)
worldmodel/vwm/modules/diffusionmodules/discretizer.py:13
Method__call__
(self, denoiser, x, cond, x_ori, uc=None, num_steps=None)
worldmodel/vwm/modules/diffusionmodules/sampling.py:99
Method__call__
(self, n_samples, bs, num_frames, rand=None)
worldmodel/vwm/modules/diffusionmodules/sigma_sampling.py:12
Method__call__
(self, x: torch.Tensor, sigma: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/guiders.py:8
Method__call__
(self, x: torch.Tensor, sigma: float)
worldmodel/vwm/modules/diffusionmodules/guiders.py:25
Method__getitem__
(self, idx: int)
lam/lam/dataset.py:152
Method__getitem__
(self, idx: int)
lam/lam/dataset.py:285
Method__getitem__
Args: index (int): Index (ignored since we sample randomly). Returns: TensorDict: Dict containing all the data block
lam/lam/dataset.py:428
Method__getitem__
(self, idx: int)
worldmodel/vwm/data/dataset.py:44
Method__getitem__
(self, idx: int)
worldmodel/vwm/data/dataset.py:167
Method__getitem__
(self, idx: int)
worldmodel/vwm/data/dataset.py:281
Method__getitem__
Args: index (int): Index (ignored since we sample randomly). Returns: TensorDict: Dict containing all the data block
worldmodel/vwm/data/dataset.py:418
Method__init__
( self, *args, batch_size: int = 8, num_workers: int = 16,
lam/lam/dataset.py:46
Method__init__
( self, data_root: str, env_source: str = "libero", split: str
lam/lam/dataset.py:235
Method__init__
( self, data_root: str, env_source: str = "libero", split: str
lam/lam/dataset.py:369
Method__init__
( self, data_root: str, env_source: str = "libero", padding: s
lam/lam/dataset.py:447
Method__init__
( self, image_channels: int = 3, # Latent action autoencoder l
lam/lam/model.py:20
Method__init__
(self, model_dim: int, max_len: int = 5000)
lam/lam/modules/blocks.py:27
Method__init__
(self, model_dim: int, num_heads: int, dropout: float = 0.0, rot_emb: bool = False)
lam/lam/modules/blocks.py:42
Method__init__
(self, model_dim: int, num_heads: int, dropout: float = 0.0)
lam/lam/modules/blocks.py:125
Method__init__
( self, in_dim: int, model_dim: int, out_dim: int,
lam/lam/modules/blocks.py:168
Method__init__
( self, in_dim: int, model_dim: int, out_dim: int,
lam/lam/modules/blocks.py:205
Method__init__
(self, num_latents: int, latent_dim: int, code_restart: bool = False)
lam/lam/modules/blocks.py:244
Method__init__
(self, num_latents: int, latent_dim: int)
lam/lam/modules/blocks.py:305
Method__init__
( self, dim, custom_freqs: Union[Tensor, None] = None, freqs_f
lam/lam/modules/embeddings.py:69
Method__init__
( self, in_dim: int, model_dim: int, latent_dim: int,
lam/lam/modules/lam.py:15
Method__init__
( self, resume, now, logdir, ckptdir,
worldmodel/train.py:221
Method__init__
( self, resume, now, logdir, ckptdir,
worldmodel/train_adapt.py:221
Method__init__
(self, num_actions: int = 4, action_dim: int = 32)
worldmodel/fast_init_mlp.py:7
Method__init__
(self, in_channels, out_channels, name)
worldmodel/fvd_utils/pytorch_i3d.py:101
Method__init__
Initializes I3D model instance. Args: num_classes: The number of outputs in the logit layer (default 400, which
worldmodel/fvd_utils/pytorch_i3d.py:173
Method__init__
( self, image_channels: int = 3, # Latent action autoencoder l
worldmodel/external/lam/model.py:18
Method__init__
(self, model_dim: int, max_len: int = 5000)
worldmodel/external/lam/modules/blocks.py:25
Method__init__
(self, model_dim: int, num_heads: int, dropout: float = 0.0, rot_emb: bool = False)
worldmodel/external/lam/modules/blocks.py:40
Method__init__
(self, model_dim: int, num_heads: int, dropout: float = 0.0)
worldmodel/external/lam/modules/blocks.py:123
Method__init__
( self, in_dim: int, model_dim: int, out_dim: int,
worldmodel/external/lam/modules/blocks.py:166
Method__init__
( self, in_dim: int, model_dim: int, out_dim: int,
worldmodel/external/lam/modules/blocks.py:203
Method__init__
( self, dim, custom_freqs: Union[Tensor, None] = None, freqs_f
worldmodel/external/lam/modules/embeddings.py:69
Method__init__
( self, in_dim: int, model_dim: int, latent_dim: int,
worldmodel/external/lam/modules/lam.py:15
Method__init__
(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_steps, verbosity_interval=0)
worldmodel/vwm/lr_scheduler.py:9
Method__init__
(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths, verbosity_interval=0)
worldmodel/vwm/lr_scheduler.py:43
Method__init__
( self, in_channels, n_heads, d_head, depth=1,
worldmodel/vwm/modules/video_attention.py:141
Method__init__
(self, model, decay=0.9999, use_num_upates=True)
worldmodel/vwm/modules/ema.py:6
Method__init__
( self, dim, dim_out=None, mult=4, glu=False,
worldmodel/vwm/modules/attention.py:81
Method__init__
( self, query_dim, context_dim=None, heads=8, dim_
worldmodel/vwm/modules/attention.py:130
Method__init__
( self, query_dim, context_dim=None, heads=8, dim_
worldmodel/vwm/modules/attention.py:197
Method__init__
( self, dim, n_heads, d_head, dropout=0.0,
worldmodel/vwm/modules/attention.py:297
Method__init__
( self, in_channels, n_heads, d_head, depth=1,
worldmodel/vwm/modules/attention.py:388
Method__init__
( self, sigma_sampler_config: Dict, loss_weighting_config: Dict,
worldmodel/vwm/modules/diffusionmodules/loss.py:13
Method__init__
(self, sigma_data: float = 0.5)
worldmodel/vwm/modules/diffusionmodules/loss_weighting.py:10
Method__init__
(self)
worldmodel/vwm/modules/diffusionmodules/loss_weighting.py:18
Method__init__
( self, channels: int, use_conv: bool, dims: int = 2,
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:104
Method__init__
( self, channels: int, emb_channels: int, dropout: float,
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:149
Method__init__
(self, dim: int)
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:272
Method__init__
(self, sigma_data: float = 0.5)
worldmodel/vwm/modules/diffusionmodules/denoiser_scaling.py:8
Method__init__
(self, sigma_min=0.002, sigma_max=700.0, rho=7.0)
worldmodel/vwm/modules/diffusionmodules/discretizer.py:24
Method__init__
(self, sigma_min=0.002, sigma_max=700.0, rho=7.0, shifting_factor=17.0)
worldmodel/vwm/modules/diffusionmodules/discretizer.py:38
Method__init__
( self, discretization_config: Union[Dict, ListConfig, OmegaConf], num_ste
worldmodel/vwm/modules/diffusionmodules/sampling.py:16
Method__init__
(self, s_churn=0.0, s_tmin=0.0, s_tmax=float("inf"), s_noise=1.0, *args, **kwargs)
worldmodel/vwm/modules/diffusionmodules/sampling.py:73
Method__init__
( self, alpha: float, merge_strategy: str, rearrange_pattern:
worldmodel/vwm/modules/diffusionmodules/util.py:171
Method__init__
(self, p_mean=-1.2, p_std=1.2)
worldmodel/vwm/modules/diffusionmodules/sigma_sampling.py:8
Method__init__
( self, channels: int, emb_channels: int, dropout: float,
worldmodel/vwm/modules/diffusionmodules/video_model.py:11
Method__init__
(self, scaling_config: Dict)
worldmodel/vwm/modules/diffusionmodules/denoiser.py:9
Method__init__
(self, diffusion_model, compile_model: bool = False)
worldmodel/vwm/modules/diffusionmodules/wrappers.py:11
Method__init__
(self, in_channels, with_conv)
worldmodel/vwm/modules/diffusionmodules/model.py:51
Method__init__
(self, in_channels, with_conv)
worldmodel/vwm/modules/diffusionmodules/model.py:65
Method__init__
( self, *, in_channels, out_channels=None, conv_sh
worldmodel/vwm/modules/diffusionmodules/model.py:83
Method__init__
(self, in_channels)
worldmodel/vwm/modules/diffusionmodules/model.py:134
Method__init__
(self, in_channels)
worldmodel/vwm/modules/diffusionmodules/model.py:170
Method__init__
( self, *, ch, out_channels, ch_mult=(1, 2, 4, 8),
worldmodel/vwm/modules/diffusionmodules/model.py:359
Method__init__
(self, scale: float)
worldmodel/vwm/modules/diffusionmodules/guiders.py:5
Method__init__
(self, sample: bool = True)
worldmodel/vwm/modules/autoencoding/regularizer.py:23
Method__init__
( self, out_channels, *args, dropout=0.0, video_ke
worldmodel/vwm/modules/autoencoding/temporal_ae.py:11
Method__init__
( self, *args, video_kernel_size: Union[int, list] = 3, alpha:
worldmodel/vwm/modules/autoencoding/temporal_ae.py:105
Method__init__
(self, parameters, deterministic=False)
worldmodel/vwm/modules/distributions/distributions.py:6
Method__init__
(self)
worldmodel/vwm/modules/encoders/modules.py:23
Method__init__
(self, emb_models: Union[List, ListConfig])
worldmodel/vwm/modules/encoders/modules.py:70
Method__init__
( self, arch="ViT-H-14", # version="path_to/huggingface/laion/CLIP-ViT-H-1
worldmodel/vwm/modules/encoders/modules.py:175
Method__init__
(self, num_actions: int = 4, action_dim: int = 32)
worldmodel/vwm/modules/encoders/modules.py:319
Method__init__
(self, lam_config: Dict)
worldmodel/vwm/modules/encoders/modules.py:358
Method__init__
( self, encoder_config: Dict, n_context_frames: int = 5, disab
worldmodel/vwm/modules/encoders/modules.py:372
Method__init__
(self, open_clip_config: Dict)
worldmodel/vwm/modules/encoders/modules.py:401
Method__init__
(self, output_dim: int)
worldmodel/vwm/modules/encoders/modules.py:416
Method__init__
( self, split_path: str, randomize: bool = False, resolution:
worldmodel/vwm/data/dataset.py:142
Method__init__
( self, split_path: str, randomize: bool = False, resolution:
worldmodel/vwm/data/dataset.py:249
Method__init__
( self, data_root: str, env_source: str = "libero", split: str
worldmodel/vwm/data/dataset.py:362
Method__init__
( self, data_root: str, env_source: str = "libero", batch_size
worldmodel/vwm/data/dataset.py:437
Method__init__
( self, network_config, denoiser_config, first_stage_config,
worldmodel/vwm/models/diffusion.py:17
Method__init__
( self, ema_decay: Union[None, float] = None, monitor: Union[None, str] =
worldmodel/vwm/models/autoencoder.py:23
← previousnext →201–300 of 471, ranked by callers