MCPcopy Create free account

hub / github.com/SwayStar123/SpeedrunDiT / functions

Functions254 in github.com/SwayStar123/SpeedrunDiT

↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
preprocessing/dnnlib/util.py:299
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
preprocessing/dnnlib/util.py:365
↓ 1 callersMethodless_thans
(self, batch_1, radii_1, batch_2, radii_2)
evaluations/evaluator.py:428
↓ 1 callersFunctionmain
(args)
train.py:99
↓ 1 callersFunctionmain
Run sampling.
generate.py:45
↓ 1 callersFunctionmain
()
evaluations/evaluator.py:27
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
preprocessing/dnnlib/util.py:121
↓ 1 callersMethodnum
r"""Returns the number of scalars that were accumulated for the given statistic between the last two calls to `update()`, or zero if n
preprocessing/torch_utils/training_stats.py:182
↓ 1 callersFunctionopen_image_folder
(source_dir, *, max_images: Optional[int])
preprocessing/dataset_tools.py:65
↓ 1 callersFunctionopen_image_zip
(source, *, max_images: Optional[int])
preprocessing/dataset_tools.py:106
↓ 1 callersFunctionopen_npz_array
(path: str, arr_name: str)
evaluations/evaluator.py:543
↓ 1 callersFunctionparse_args
(input_args=None)
train.py:405
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
preprocessing/torch_utils/persistence.py:38
↓ 1 callersFunctionpreprocess_raw_image
(x, enc_type)
train.py:38
↓ 1 callersMethodread_batch
(self, batch_size: int)
evaluations/evaluator.py:461
↓ 1 callersMethodread_batches
(self, batch_size: int)
evaluations/evaluator.py:468
↓ 1 callersFunctionrecurse
(obj)
preprocessing/torch_utils/persistence.py:245
↓ 1 callersMethodremaining
(self)
evaluations/evaluator.py:465
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. NaNs and Infs are ig
preprocessing/torch_utils/training_stats.py:55
↓ 1 callersFunctionrequires_grad
Set requires_grad flag for all parameters in a model.
train.py:88
↓ 1 callersFunctionrotate_half
(x)
models/pos_embed.py:35
↓ 1 callersFunctionscale_image
Scale image to specified dimensions.
preprocessing/dataset_tools.py:180
↓ 1 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
preprocessing/torch_utils/training_stats.py:200
↓ 1 callersMethodtoken_drop
Drops labels to enable classifier-free guidance.
models/sit.py:87
↓ 1 callersMethodunpatchify
x: (N, T, patch_size**2 * C) imgs: (N, C, H, W)
models/sit.py:406
↓ 1 callersMethodwarmup
(self)
evaluations/evaluator.py:161
FunctionSiT_B_1
(**kwargs)
models/sit.py:682
FunctionSiT_B_2
(**kwargs)
models/sit.py:685
FunctionSiT_B_4
(**kwargs)
models/sit.py:688
FunctionSiT_L_1
(**kwargs)
models/sit.py:673
FunctionSiT_L_2
(**kwargs)
models/sit.py:676
FunctionSiT_L_4
(**kwargs)
models/sit.py:679
FunctionSiT_S_1
(**kwargs)
models/sit.py:691
FunctionSiT_S_2
(**kwargs)
models/sit.py:694
FunctionSiT_S_4
(**kwargs)
models/sit.py:697
FunctionSiT_XL_1
(**kwargs)
models/sit.py:664
FunctionSiT_XL_2
(**kwargs)
models/sit.py:667
FunctionSiT_XL_4
(**kwargs)
models/sit.py:670
FunctionVAE_F8D4
(**kwargs)
models/invae.py:487
Method__call__
(self, model, images, model_kwargs=None, zs=None, cls_token=None, time_input=None, noises=Non
loss.py:54
Method__delattr__
(self, name: str)
preprocessing/dnnlib/util.py:49
Method__enter__
(self)
preprocessing/dnnlib/util.py:69
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
preprocessing/dnnlib/util.py:72
Method__getattr__
(self, name: str)
preprocessing/dnnlib/util.py:40
Method__getitem__
(self, idx)
dataset.py:63
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
preprocessing/torch_utils/training_stats.py:228
Method__getstate__
(self)
preprocessing/encoders.py:72
Method__init__
( self, prediction='v', path_type="linear", weighting="uniform
loss.py:18
Method__init__
(self, data_dir)
dataset.py:17
Method__init__
(self, mu: np.ndarray, sigma: np.ndarray)
evaluations/evaluator.py:94
Method__init__
( self, session, batch_size=64, softmax_batch_size=512, )
evaluations/evaluator.py:145
Method__init__
Estimate the manifold of given feature vectors. :param session: the TensorFlow session. :param row_batch_size: row batch siz
evaluations/evaluator.py:238
Method__init__
(self, session)
evaluations/evaluator.py:395
Method__init__
(self, gen_fn, length)
evaluations/evaluator.py:482
Method__init__
(self, arr_f, shape, dtype)
evaluations/evaluator.py:494
Method__init__
(self, arr)
evaluations/evaluator.py:520
Method__init__
(self, vae_name = 'REPA-E/e2e-invae', # Name of the VAE to use. batch_size = 8,
preprocessing/encoders.py:56
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, start_idx=0)
preprocessing/torch_utils/misc.py:123
Method__init__
(self, **kwargs)
preprocessing/torch_utils/distributed.py:85
Method__init__
(self, *args, **kwargs)
preprocessing/torch_utils/persistence.py:110
Method__init__
(self, regex='.*', keep_previous=False)
preprocessing/torch_utils/training_stats.py:135
Method__init__
(self, file_name: Optional[str] = None, file_mode: str = "w", should_flush: bool = True)
preprocessing/dnnlib/util.py:56
Method__init__
( self, dim, pt_seq_len=16, ft_seq_len=None, custom_freqs=None,
models/pos_embed.py:49
Method__init__
(self, in_channels, with_conv)
models/invae.py:24
Method__init__
(self, in_channels, with_conv)
models/invae.py:40
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
models/invae.py:60
Method__init__
(self, in_channels)
models/invae.py:120
Method__init__
( self, *, ch=128, out_ch=3, ch_mult=(1, 1, 2, 2, 4), num_res_
models/invae.py:280
Method__init__
(self, parameters, deterministic=False)
models/invae.py:404
Method__init__
(self, embed_dim, ch_mult, use_variational=True)
models/invae.py:453
Method__init__
(self, hidden_size, frequency_embedding_size=256)
models/sit.py:39
Method__init__
(self, num_classes, hidden_size, dropout_prob)
models/sit.py:80
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, qk_norm:
models/sit.py:111
Method__init__
(self, hidden_size, num_heads, mlp_ratio=4.0, use_v1_residual: bool = True, **block_kwargs)
models/sit.py:191
Method__init__
(self, hidden_size, patch_size, out_channels, cls_token_dim)
models/sit.py:242
Method__iter__
(self)
evaluations/evaluator.py:489
Method__iter__
(self)
preprocessing/torch_utils/misc.py:135
Method__len__
(self)
dataset.py:59
Method__len__
(self)
evaluations/evaluator.py:486
Method__setattr__
(self, name: str, value: Any)
preprocessing/dnnlib/util.py:46
Method_basic_init
(module)
models/sit.py:368
Function_reconstruct_persistent_obj
r"""Hook that is called internally by the `pickle` module to unpickle a persistent object.
preprocessing/torch_utils/persistence.py:188
Methodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
preprocessing/torch_utils/training_stats.py:214
Functionassert_shape
(tensor, ref_shape)
preprocessing/torch_utils/misc.py:93
Functioncheck_ddp_consistency
(module, ignore_regex=None)
preprocessing/torch_utils/misc.py:184
Functionconst_like
(ref, value, shape=None, dtype=None, device=None, memory_format=None)
preprocessing/torch_utils/misc.py:55
Functionconstruct_class_by_name
Finds the python class with the given name and constructs it with the given arguments.
preprocessing/dnnlib/util.py:288
Functionconvert
Convert an image dataset into archive format for training. Specifying the input images: \b --source path/ Recursive
preprocessing/dataset_tools.py:296
Functioncopy_files_and_create_dirs
Takes in a list of tuples of (src, dst) paths and copies files. Will create all necessary directories.
preprocessing/dnnlib/util.py:351
Functioncopy_params_and_buffers
(src_module, dst_module, require_all=False)
preprocessing/torch_utils/misc.py:159
Functionddp_sync
(module, sync)
preprocessing/torch_utils/misc.py:173
Functiondecorator
(*args, **kwargs)
preprocessing/torch_utils/misc.py:112
Functionencode
Encode pixel data to VAE latents. Parallelization and Memory Management: Use --gpus to control the number of GPUs used for parallel
preprocessing/dataset_tools.py:451
Functionencode_image_worker
Worker function for parallel VAE encoding.
preprocessing/dataset_tools.py:260
Methodencode_pixels
(self, x)
preprocessing/encoders.py:49
Methodevaluate
Evaluate if new feature vectors are at the manifold.
evaluations/evaluator.py:309
Functionfolder_write_bytes
(fname: str, data: Union[bytes, str])
preprocessing/dataset_tools.py:170
Functionformat_time
Convert the seconds to human readable string with days, hours, minutes and seconds.
preprocessing/dnnlib/util.py:136
Functionformat_time_brief
Convert the seconds to human readable string with days, hours, minutes and seconds.
preprocessing/dnnlib/util.py:150
Methodforward
t: (B, Hh, N, D_rot), D_rot == self.rot_dim rope_ids: None | (N,) | (B,N), indexing flattened HW positions for the
models/pos_embed.py:111
← previousnext →101–200 of 254, ranked by callers