Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/edm
/ functions
Functions
193 in github.com/NVlabs/edm
⨍
Functions
193
◇
Types & classes
25
↳
Endpoints
2
↓ 13 callers
Method
randn
(self, size, **kwargs)
generate.py:187
↓ 13 callers
Method
update
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 callers
Method
__init__
(self, in_features, out_features, bias=True, init_mode='kaiming_normal', init_weight=1, init_bias=0)
training/networks.py:31
↓ 9 callers
Method
randint
(self, *args, size, **kwargs)
generate.py:194
↓ 7 callers
Function
matrix
(*rows, device=None)
training/augment.py:43
↓ 6 callers
Method
round_sigma
(self, sigma)
training/networks.py:519
↓ 5 callers
Function
iterate_images
()
dataset_tool.py:78
↓ 5 callers
Function
maybe_min
(a: int, b: Optional[int])
dataset_tool.py:39
↓ 4 callers
Method
_get_raw_labels
(self)
training/dataset.py:56
↓ 4 callers
Function
generate_image_grid
( network_pkl, dest_path, seed=0, gridw=8, gridh=8, device=torch.device('cuda'), num_steps=18, sig
example.py:20
↓ 4 callers
Method
randn_like
(self, input)
generate.py:191
↓ 4 callers
Function
scale2d_inv
(sx, sy, **kwargs)
training/augment.py:103
↓ 4 callers
Function
weight_init
(shape, mode, fan_in, fan_out)
training/networks.py:19
↓ 3 callers
Method
_file_ext
(fname)
training/dataset.py:198
↓ 3 callers
Method
_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 callers
Function
file_ext
(name: Union[str, Path])
dataset_tool.py:46
↓ 3 callers
Method
flush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:92
↓ 3 callers
Function
get_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:276
↓ 3 callers
Method
mean
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 callers
Function
named_params_and_buffers
(module)
torch_utils/misc.py:150
↓ 3 callers
Function
rotate2d_inv
(theta, **kwargs)
training/augment.py:106
↓ 3 callers
Function
scale2d
(sx, sy, **kwargs)
training/augment.py:68
↓ 3 callers
Function
translate2d
(tx, ty, **kwargs)
training/augment.py:53
↓ 3 callers
Method
write
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:77
↓ 2 callers
Method
_get_zipfile
(self)
training/dataset.py:201
↓ 2 callers
Method
_open_file
(self, fname)
training/dataset.py:207
↓ 2 callers
Method
alpha_bar
(self, j)
training/networks.py:617
↓ 2 callers
Function
calculate_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 callers
Method
close
(self)
training/dataset.py:69
↓ 2 callers
Function
get_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 callers
Function
get_rank
()
torch_utils/distributed.py:35
↓ 2 callers
Function
is_image_ext
(fname: Union[str, Path])
dataset_tool.py:51
↓ 2 callers
Function
is_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 callers
Method
names
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 callers
Method
sigma
(self, t)
training/networks.py:511
↓ 2 callers
Function
suppress_tracer_warnings
()
torch_utils/misc.py:70
↓ 2 callers
Function
translate2d_inv
(tx, ty, **kwargs)
training/augment.py:100
↓ 1 callers
Method
__getstate__
(self)
training/dataset.py:78
↓ 1 callers
Method
__init__
(self, name, # Name of the dataset. raw_shape, # Shape of the r
training/dataset.py:27
↓ 1 callers
Method
__reduce__
(self)
torch_utils/persistence.py:123
↓ 1 callers
Function
_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 callers
Method
_load_raw_image
(self, raw_idx)
training/dataset.py:72
↓ 1 callers
Method
_load_raw_image
(self, raw_idx)
training/dataset.py:224
↓ 1 callers
Method
_load_raw_labels
(self)
training/dataset.py:75
↓ 1 callers
Function
_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:212
↓ 1 callers
Function
_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:222
↓ 1 callers
Function
_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:233
↓ 1 callers
Method
as_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 callers
Method
backward
(ctx, dw)
training/networks.py:121
↓ 1 callers
Function
calculate_fid_from_inception_stats
(mu, sigma, mu_ref, sigma_ref)
fid.py:82
↓ 1 callers
Function
call_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:292
↓ 1 callers
Method
close
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:99
↓ 1 callers
Method
get_label
(self, idx)
training/dataset.py:105
↓ 1 callers
Function
get_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:286
↓ 1 callers
Function
get_world_size
()
torch_utils/distributed.py:40
↓ 1 callers
Function
is_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 callers
Function
is_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:377
↓ 1 callers
Function
main
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 callers
Function
main
()
example.py:82
↓ 1 callers
Function
main
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 callers
Function
main
Generate random images using the techniques described in the paper "Elucidating the Design Space of Diffusion-Based Generative Models". Examp
generate.py:238
↓ 1 callers
Function
main
Calculate Frechet Inception Distance (FID). Examples: \b # Generate 50000 images and save them as fid-tmp/*/*.png torchrun --standal
fid.py:91
↓ 1 callers
Function
make_cache_dir_path
(*paths: str)
dnnlib/util.py:123
↓ 1 callers
Function
make_transform
( transform: Optional[str], output_width: Optional[int], output_height: Optional[int] )
dataset_tool.py:203
↓ 1 callers
Function
nan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
torch_utils/misc.py:48
↓ 1 callers
Method
num
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 callers
Function
open_cifar10
(tarball: str, *, max_images: Optional[int])
dataset_tool.py:141
↓ 1 callers
Function
open_dataset
(source, *, max_images: Optional[int])
dataset_tool.py:258
↓ 1 callers
Function
open_dest
(dest: str)
dataset_tool.py:278
↓ 1 callers
Function
open_image_folder
(source_dir, *, max_images: Optional[int])
dataset_tool.py:57
↓ 1 callers
Function
open_image_zip
(source, *, max_images: Optional[int])
dataset_tool.py:88
↓ 1 callers
Function
open_lmdb
(lmdb_dir: str, *, max_images: Optional[int])
dataset_tool.py:113
↓ 1 callers
Function
open_mnist
(images_gz: str, *, max_images: Optional[int])
dataset_tool.py:173
↓ 1 callers
Function
persistent_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 callers
Function
recurse
(obj)
torch_utils/persistence.py:242
↓ 1 callers
Function
report
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 callers
Function
rotate2d
(theta, **kwargs)
training/augment.py:83
↓ 1 callers
Function
rotate3d
(v, theta, **kwargs)
training/augment.py:90
↓ 1 callers
Method
round_sigma
(self, sigma, return_index=False)
training/networks.py:621
↓ 1 callers
Function
scale3d
(sx, sy, sz, **kwargs)
training/augment.py:75
↓ 1 callers
Method
sigma
(self, t)
training/loss.py:36
↓ 1 callers
Method
sigma_inv
(self, sigma)
training/networks.py:515
↓ 1 callers
Method
std
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 callers
Function
translate3d
(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