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
Functiongenerate_roughly_equally_spaced_steps
(num_substeps: int, max_step: int)
worldmodel/vwm/modules/diffusionmodules/discretizer.py:8
Functionget_ancestral_step
(sigma_from, sigma_to, eta=1.0)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:22
Methodget_attention_layer
( ch, num_heads, dim_head, depth=1,
worldmodel/vwm/modules/diffusionmodules/video_model.py:192
Methodget_axial_freqs
(self, *dims)
lam/lam/modules/embeddings.py:199
Methodget_axial_freqs
(self, *dims)
worldmodel/external/lam/modules/embeddings.py:199
Methodget_input
(self, batch)
worldmodel/vwm/models/autoencoder.py:44
Methodget_input
(self, batch: Dict)
worldmodel/vwm/models/autoencoder.py:105
Methodget_last_layer
(self)
worldmodel/vwm/modules/video_attention.py:136
Methodget_last_layer
(self, **kwargs)
worldmodel/vwm/modules/diffusionmodules/model.py:456
Methodget_last_layer
(self, skip_time_mix=False, **kwargs)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:123
Methodget_last_layer
(self)
worldmodel/vwm/models/autoencoder.py:109
Functionget_model_state_file
(checkpoint_dir, zero_stage)
worldmodel/zero_to_fp32.py:61
Methodget_resblock
( merge_factor, merge_strategy, video_kernel_size,
worldmodel/vwm/modules/diffusionmodules/video_model.py:221
Methodget_sigmas
(self, n, device="cpu")
worldmodel/vwm/modules/diffusionmodules/discretizer.py:29
Methodget_sigmas
(self, n, device="cpu")
worldmodel/vwm/modules/diffusionmodules/discretizer.py:44
Functionget_timestep_embedding
Build sinusoidal embeddings.
worldmodel/vwm/modules/diffusionmodules/model.py:23
Methodget_trainable_parameters
(self)
worldmodel/vwm/modules/autoencoding/regularizer.py:18
Methodget_trainable_parameters
(self)
worldmodel/vwm/modules/autoencoding/regularizer.py:27
Methodinput_key
(self)
worldmodel/vwm/modules/encoders/modules.py:38
Methodinstantiate_optimizer_from_config
(self, params, lr, cfg)
worldmodel/vwm/models/autoencoder.py:75
Methodis_trainable
(self)
worldmodel/vwm/modules/encoders/modules.py:30
Functionlinear_multistep_coeff
(order, t, i, j, epsrel=1e-4)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:7
Functionload_state_dict_from_zero_checkpoint
1. Put the provided model to cpu 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` 3. Load it into the pro
worldmodel/zero_to_fp32.py:519
Methodlog_images
( self, batch: Dict, N: int = 10, sample: bool = True,
worldmodel/vwm/models/diffusion.py:234
Functionmelk
(*args, **kwargs)
worldmodel/train.py:744
Functionmelk
(*args, **kwargs)
worldmodel/train_adapt.py:746
Functionnatural_keys
alist.sort(key=natural_keys) sorts in human order http://nedbatchelder.com/blog/200712/human_sorting.html (See Toothy's implementation in
worldmodel/zero_to_fp32.py:51
Methodnll
(self, sample, dims=[1, 2, 3])
worldmodel/vwm/modules/distributions/distributions.py:36
Methodon_exception
(self, trainer: pl.Trainer, pl_module, exception)
worldmodel/train.py:244
Methodon_exception
(self, trainer: pl.Trainer, pl_module, exception)
worldmodel/train_adapt.py:244
Methodon_fit_start
(self, trainer, pl_module)
worldmodel/train.py:248
Methodon_fit_start
(self, trainer, pl_module)
worldmodel/train_adapt.py:248
Methodon_test_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
worldmodel/train.py:419
Methodon_test_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
worldmodel/train_adapt.py:419
Methodon_train_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
worldmodel/train.py:403
Methodon_train_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
worldmodel/train_adapt.py:403
Methodon_train_batch_end
(self, *args, **kwargs)
worldmodel/vwm/models/diffusion.py:152
Methodon_train_batch_end
(self, *args, **kwargs)
worldmodel/vwm/models/autoencoder.py:47
Methodon_train_batch_start
(self, trainer, pl_module, batch, batch_idx)
worldmodel/train.py:408
Methodon_train_batch_start
(self, trainer, pl_module, batch, batch_idx)
worldmodel/train_adapt.py:408
Methodon_train_start
(self, *args, **kwargs)
worldmodel/vwm/models/diffusion.py:148
Methodon_validation_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx, *args, **kwargs)
worldmodel/train.py:414
Methodon_validation_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx, *args, **kwargs)
worldmodel/train_adapt.py:414
Functionpolynomial_mmd
(X, Y)
worldmodel/fvd_utils/fvd_utils.py:109
Methodprepare_inputs
(self, x, s, c, uc)
worldmodel/vwm/modules/diffusionmodules/guiders.py:28
Methodrandom_restart
(self)
lam/lam/modules/blocks.py:259
Methodreplace_logits
(self, num_classes)
worldmodel/fvd_utils/pytorch_i3d.py:299
Methodreset_num_updates
(self)
worldmodel/vwm/modules/ema.py:29
Methodreset_usage
(self)
lam/lam/modules/blocks.py:271
Methodrotate_queries_and_keys
(self, q, k, freqs, seq_dim=None)
lam/lam/modules/embeddings.py:158
Methodrotate_queries_and_keys
(self, q, k, freqs, seq_dim=None)
worldmodel/external/lam/modules/embeddings.py:158
Methodsampler_step
(self, sigma, next_sigma, denoiser, x, cond, uc, *args, **kwargs)
worldmodel/vwm/modules/diffusionmodules/sampling.py:65
Functionscale_module
Scale the parameters of a module and return it.
worldmodel/vwm/modules/diffusionmodules/util.py:107
Methodschedule
(self, n, **kwargs)
worldmodel/vwm/lr_scheduler.py:83
Methodsetup
(self, stage: str)
lam/lam/dataset.py:473
Functionstr2bool
(v)
worldmodel/train.py:41
Functionstr2bool
(v)
worldmodel/train_adapt.py:41
Methodtest_dataloader
(self)
lam/lam/dataset.py:108
Methodtest_dataloader
(self)
worldmodel/vwm/data/dataset.py:468
Methodtest_step
(self, batch: Dict, batch_idx: int)
lam/lam/model.py:122
Methodtest_step
(self, batch: Dict, batch_idx: int)
worldmodel/external/lam/model.py:118
Functionto_neg_log_sigma
(sigma)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:38
Functionto_sigma
(neg_log_sigma)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:42
Methodtrain_dataloader
(self)
lam/lam/dataset.py:78
Methodtrain_dataloader
(self)
worldmodel/vwm/data/dataset.py:459
Methodtraining_step
(self, batch: Dict, batch_idx: int)
lam/lam/model.py:75
Methodtraining_step
(self, batch: Dict, batch_idx: int)
worldmodel/external/lam/model.py:80
Methodtraining_step
(self, batch, batch_idx: int)
worldmodel/vwm/models/diffusion.py:136
Methoducg_rate
(self)
worldmodel/vwm/modules/encoders/modules.py:34
Methodval_dataloader
(self)
lam/lam/dataset.py:93
Methodval_dataloader
(self)
worldmodel/vwm/data/dataset.py:477
← previous401–471 of 471, ranked by callers