MCPcopy Create free account

hub / github.com/NVlabs/edm / functions

Functions193 in github.com/NVlabs/edm

↓ 13 callersMethodrandn
(self, size, **kwargs)
generate.py:187
↓ 13 callersMethodupdate
r"""Copies current values of the internal counters to the user-visible state and resets them for the next round. If `keep_previous=Tr
torch_utils/training_stats.py:146
↓ 11 callersMethod__init__
(self, in_features, out_features, bias=True, init_mode='kaiming_normal', init_weight=1, init_bias=0)
training/networks.py:31
↓ 9 callersMethodrandint
(self, *args, size, **kwargs)
generate.py:194
↓ 7 callersFunctionmatrix
(*rows, device=None)
training/augment.py:43
↓ 6 callersMethodround_sigma
(self, sigma)
training/networks.py:519
↓ 5 callersFunctioniterate_images
()
dataset_tool.py:78
↓ 5 callersFunctionmaybe_min
(a: int, b: Optional[int])
dataset_tool.py:39
↓ 4 callersMethod_get_raw_labels
(self)
training/dataset.py:56
↓ 4 callersFunctiongenerate_image_grid
( network_pkl, dest_path, seed=0, gridw=8, gridh=8, device=torch.device('cuda'), num_steps=18, sig
example.py:20
↓ 4 callersMethodrandn_like
(self, input)
generate.py:191
↓ 4 callersFunctionscale2d_inv
(sx, sy, **kwargs)
training/augment.py:103
↓ 4 callersFunctionweight_init
(shape, mode, fan_in, fan_out)
training/networks.py:19
↓ 3 callersMethod_file_ext
(fname)
training/dataset.py:198
↓ 3 callersMethod_get_delta
r"""Returns the raw moments that were accumulated for the given statistic between the last two calls to `update()`, or zero if no scal
torch_utils/training_stats.py:169
↓ 3 callersFunctionfile_ext
(name: Union[str, Path])
dataset_tool.py:46
↓ 3 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:92
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:276
↓ 3 callersMethodmean
r"""Returns the mean of the scalars that were accumulated for the given statistic between the last two calls to `update()`, or NaN if
torch_utils/training_stats.py:187
↓ 3 callersFunctionnamed_params_and_buffers
(module)
torch_utils/misc.py:150
↓ 3 callersFunctionrotate2d_inv
(theta, **kwargs)
training/augment.py:106
↓ 3 callersFunctionscale2d
(sx, sy, **kwargs)
training/augment.py:68
↓ 3 callersFunctiontranslate2d
(tx, ty, **kwargs)
training/augment.py:53
↓ 3 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:77
↓ 2 callersMethod_get_zipfile
(self)
training/dataset.py:201
↓ 2 callersMethod_open_file
(self, fname)
training/dataset.py:207
↓ 2 callersMethodalpha_bar
(self, j)
training/networks.py:617
↓ 2 callersFunctioncalculate_inception_stats
( image_path, num_expected=None, seed=0, max_batch_size=64, num_workers=3, prefetch_factor=2, device=t
fid.py:23
↓ 2 callersMethodclose
(self)
training/dataset.py:69
↓ 2 callersFunctionget_module_from_obj_name
Searches for the underlying module behind the name to some python object. Returns the module and the object name (original name with module part r
dnnlib/util.py:235
↓ 2 callersFunctionget_rank
()
torch_utils/distributed.py:35
↓ 2 callersFunctionis_image_ext
(fname: Union[str, Path])
dataset_tool.py:51
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
torch_utils/persistence.py:140
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
torch_utils/training_stats.py:140
↓ 2 callersMethodsigma
(self, t)
training/networks.py:511
↓ 2 callersFunctionsuppress_tracer_warnings
()
torch_utils/misc.py:70
↓ 2 callersFunctiontranslate2d_inv
(tx, ty, **kwargs)
training/augment.py:100
↓ 1 callersMethod__getstate__
(self)
training/dataset.py:78
↓ 1 callersMethod__init__
(self, name, # Name of the dataset. raw_shape, # Shape of the r
training/dataset.py:27
↓ 1 callersMethod__reduce__
(self)
torch_utils/persistence.py:123
↓ 1 callersFunction_check_pickleable
r"""Check that the given object is pickleable, raising an exception if it is not. This function is expected to be considerably more efficient
torch_utils/persistence.py:237
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
training/dataset.py:72
↓ 1 callersMethod_load_raw_image
(self, raw_idx)
training/dataset.py:224
↓ 1 callersMethod_load_raw_labels
(self)
training/dataset.py:75
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:212
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:222
↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:233
↓ 1 callersMethodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
torch_utils/training_stats.py:211
↓ 1 callersMethodbackward
(ctx, dw)
training/networks.py:121
↓ 1 callersFunctioncalculate_fid_from_inception_stats
(mu, sigma, mu_ref, sigma_ref)
fid.py:82
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:292
↓ 1 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:99
↓ 1 callersMethodget_label
(self, idx)
training/dataset.py:105
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:286
↓ 1 callersFunctionget_world_size
()
torch_utils/distributed.py:40
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:311
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:377
↓ 1 callersFunctionmain
Train diffusion-based generative model using the techniques described in the paper "Elucidating the Design Space of Diffusion-Based Generative Mod
train.py:80
↓ 1 callersFunctionmain
()
example.py:82
↓ 1 callersFunctionmain
Convert an image dataset into a dataset archive usable with StyleGAN2 ADA PyTorch. The input dataset format is guessed from the --source argument
dataset_tool.py:317
↓ 1 callersFunctionmain
Generate random images using the techniques described in the paper "Elucidating the Design Space of Diffusion-Based Generative Models". Examp
generate.py:238
↓ 1 callersFunctionmain
Calculate Frechet Inception Distance (FID). Examples: \b # Generate 50000 images and save them as fid-tmp/*/*.png torchrun --standal
fid.py:91
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:123
↓ 1 callersFunctionmake_transform
( transform: Optional[str], output_width: Optional[int], output_height: Optional[int] )
dataset_tool.py:203
↓ 1 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
torch_utils/misc.py:48
↓ 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
torch_utils/training_stats.py:179
↓ 1 callersFunctionopen_cifar10
(tarball: str, *, max_images: Optional[int])
dataset_tool.py:141
↓ 1 callersFunctionopen_dataset
(source, *, max_images: Optional[int])
dataset_tool.py:258
↓ 1 callersFunctionopen_dest
(dest: str)
dataset_tool.py:278
↓ 1 callersFunctionopen_image_folder
(source_dir, *, max_images: Optional[int])
dataset_tool.py:57
↓ 1 callersFunctionopen_image_zip
(source, *, max_images: Optional[int])
dataset_tool.py:88
↓ 1 callersFunctionopen_lmdb
(lmdb_dir: str, *, max_images: Optional[int])
dataset_tool.py:113
↓ 1 callersFunctionopen_mnist
(images_gz: str, *, max_images: Optional[int])
dataset_tool.py:173
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
torch_utils/persistence.py:34
↓ 1 callersFunctionrecurse
(obj)
torch_utils/persistence.py:242
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
torch_utils/training_stats.py:55
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
training/augment.py:83
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
training/augment.py:90
↓ 1 callersMethodround_sigma
(self, sigma, return_index=False)
training/networks.py:621
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
training/augment.py:75
↓ 1 callersMethodsigma
(self, t)
training/loss.py:36
↓ 1 callersMethodsigma_inv
(self, sigma)
training/networks.py:515
↓ 1 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
torch_utils/training_stats.py:197
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
training/augment.py:60
Method__call__
(self, net, images, labels, augment_pipe=None)
training/loss.py:26
Method__call__
(self, net, images, labels, augment_pipe=None)
training/loss.py:51
Method__call__
(self, net, images, labels=None, augment_pipe=None)
training/loss.py:72
Method__call__
(self, images)
training/augment.py:153
Method__del__
(self)
training/dataset.py:81
Method__delattr__
(self, name: str)
dnnlib/util.py:51
Method__enter__
(self)
dnnlib/util.py:71
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
dnnlib/util.py:74
Method__getattr__
(self, name: str)
dnnlib/util.py:42
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
torch_utils/training_stats.py:225
Method__getitem__
(self, idx)
training/dataset.py:90
Method__getstate__
(self)
training/dataset.py:221
Method__init__
(self, device, seeds)
generate.py:183
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
torch_utils/misc.py:111
Method__init__
(self, *args, **kwargs)
torch_utils/persistence.py:105
Method__init__
(self, regex='.*', keep_previous=True)
torch_utils/training_stats.py:132
next →1–100 of 193, ranked by callers