MCPcopy Create free account

hub / github.com/NVlabs/stylegan2-ada / functions

Functions336 in github.com/NVlabs/stylegan2-ada

↓ 32 callersMethoddevice
Name of the TensorFlow device that the weights of this network reside on. Determined by the current device at construction time.
dnnlib/tflib/network.py:195
↓ 32 callersMethodrun
(self)
dataset_tool.py:223
↓ 24 callersMethod_get_vars
(self)
dnnlib/tflib/network.py:295
↓ 23 callersMethodget_output_for
Construct TensorFlow expression(s) for the output(s) of this network, given the input expression(s). The graph is placed on the current Tensor
dnnlib/tflib/network.py:345
↓ 19 callersFunctionadd_command
(cmd, desc, example=None)
dataset_tool.py:870
↓ 18 callersFunctionreport_stat
(aug, name, value)
training/loss.py:20
↓ 16 callersFunction_shape
(tf_expr, dim_idx)
dnnlib/tflib/ops/upfirdn_2d.py:411
↓ 16 callersFunctiongate_augment_params
(probability, params, disabled_val)
training/augment.py:169
↓ 15 callersMethodclone
Create a clone of this network with its own copy of the variables.
dnnlib/tflib/network.py:451
↓ 14 callersFunctionapply_bias_act
(x, act='linear', gain=None, lrmul=1, clamp=None, bias_var='bias', trainable=True)
training/networks.py:48
↓ 14 callersMethodrun
Run this network for the given NumPy array(s), and return the output(s) as NumPy array(s). Args: input_transform: A dict speci
dnnlib/tflib/network.py:540
↓ 13 callersFunctionnf
(stage)
training/networks.py:373
↓ 12 callersMethodadd_image
(self, img)
dataset_tool.py:165
↓ 12 callersFunctioneval_D
(D, aug, images, labels, report=None, augment_inputs=True, return_aux=0)
training/loss.py:57
↓ 11 callersFunctionautosummary
Create a new autosummary. Args: name: Name to use in TensorBoard value: TensorFlow expression or python value to track
dnnlib/tflib/autosummary.py:79
↓ 11 callersMethodget_minibatch_np
(self, minibatch_size, lod=0)
training/dataset.py:158
↓ 9 callersMethod_info
(self, *args)
projector.py:58
↓ 9 callersMethod_report_result
(self, value, suffix='', fmt='%-10.4f')
metrics/metric_base.py:93
↓ 9 callersFunctiondense_layer
(x, fmaps, lrmul=1, weight_var='weight', trainable=True, use_spectral_norm=False)
training/networks.py:55
↓ 9 callersFunctionerror
(msg)
dataset_tool.py:32
↓ 9 callersMethodrun
(self, network_pkl, num_gpus=1, G_kwargs=dict(is_validation=True))
metrics/metric_base.py:49
↓ 8 callersMethod_get_own_vars
(self)
dnnlib/tflib/network.py:284
↓ 8 callersMethod_report_progress
(self, cur, total)
metrics/metric_base.py:96
↓ 7 callersMethodadd_labels
(self, labels)
dataset_tool.py:189
↓ 7 callersMethodchoose_shuffled_order
(self)
dataset_tool.py:70
↓ 6 callersMethod_get_random_labels_tf
(self, minibatch_size)
metrics/metric_base.py:134
↓ 6 callersFunctionadrop
(x)
training/networks.py:526
↓ 6 callersFunctionconstruct_batch_of_matrices
(*rows)
training/augment.py:178
↓ 6 callersFunctionconv2d_layer
(x, fmaps, kernel, up=False, down=False, resample_kernel=None, lrmul=1, trainable=True, use_spectral_norm=Fals
training/networks.py:84
↓ 6 callersFunctioneval_G
(G, latents, labels, return_dlatents=False)
training/loss.py:41
↓ 6 callersFunctionis_next_layer_trainable
()
training/networks.py:534
↓ 6 callersFunctionref
(tensor, name)
dnnlib/tflib/ops/fused_bias_act.py:140
↓ 5 callersMethod_get_trainables
(self)
dnnlib/tflib/network.py:307
↓ 5 callersFunction_simple_upfirdn_2d
(x, k, up=1, down=1, pad0=0, pad1=0, data_format='NCHW', impl='cuda')
dnnlib/tflib/ops/upfirdn_2d.py:394
↓ 5 callersMethodapply_updates
Construct training op to update the registered variables based on their gradients.
dnnlib/tflib/optimizer.py:156
↓ 5 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:98
↓ 5 callersMethodcopy_vars_from
Copy the values of all variables from the given network, including sub-networks.
dnnlib/tflib/network.py:485
↓ 5 callersMethodregister_gradients
Register the gradients of the given loss function with respect to the given variables. Intended to be called once per GPU.
dnnlib/tflib/optimizer.py:114
↓ 4 callersMethod_init_graph
(self)
dnnlib/tflib/network.py:124
↓ 4 callersFunctionassert_tf_initialized
Check that TensorFlow session has been initialized.
dnnlib/tflib/tfutil.py:145
↓ 4 callersMethodconfigure
(self, minibatch_size, lod=0)
training/dataset.py:140
↓ 4 callersFunctionnormalize
(v)
metrics/perceptual_path_length.py:23
↓ 4 callersFunctionscale_2d_inv
(sx, sy)
training/augment.py:230
↓ 3 callersMethod_get_cache_file_for_reals
(self, extension='pkl', **kwargs)
metrics/metric_base.py:100
↓ 3 callersMethod_init_fields
(self, name: str, static_kwargs: dict, build_func: Callable, build_func_name: str, build_module_src: str)
dnnlib/tflib/network.py:76
↓ 3 callersMethod_iterate_reals
(self, minibatch_size)
metrics/metric_base.py:113
↓ 3 callersFunctionfunc
(x)
dnnlib/tflib/ops/upfirdn_2d.py:134
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:260
↓ 3 callersFunctionget_weight
(shape, gain=1, equalized_lr=True, lrmul=1, weight_var='weight', trainable=True, use_spectral_norm=False)
training/networks.py:25
↓ 3 callersFunctionlayer
(x, layer_idx, fmaps, kernel, up=False)
training/networks.py:391
↓ 3 callersFunctionnormalize_2nd_moment
(x, axis=1, eps=1e-8)
training/networks.py:135
↓ 3 callersMethodprint_layers
Print a summary table of the network structure.
dnnlib/tflib/network.py:697
↓ 3 callersFunctionprob_normalize
(p)
metrics/linear_separability.py:69
↓ 3 callersFunctionreport_loss
(aug, G_loss, D_loss, G_reg=None, D_reg=None)
training/loss.py:29
↓ 3 callersFunctionrotate_2d_inv
(theta)
training/augment.py:233
↓ 3 callersFunctionrun
Run the specified ops in the default session.
dnnlib/tflib/tfutil.py:30
↓ 3 callersFunctionsave_image_grid
(images, filename, drange, grid_size)
training/training_loop.py:67
↓ 3 callersFunctiontranslate_2d_inv
(tx, ty)
training/augment.py:227
↓ 3 callersFunctionupfirdn_2d
r"""Pad, upsample, FIR filter, and downsample a batch of 2D images. Accepts a batch of 2D images of the shape `[majorDim, inH, inW, minorDim]`
dnnlib/tflib/ops/upfirdn_2d.py:21
↓ 3 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:78
↓ 2 callersMethod_broadcast_fallback
Sum gradients across devices using TensorFlow collective ops (slow fallback path).
dnnlib/tflib/optimizer.py:299
↓ 2 callersFunction_create_var
Internal helper for creating autosummary accumulators.
dnnlib/tflib/autosummary.py:47
↓ 2 callersMethod_flush_lod
(self, lod)
dataset_tool.py:196
↓ 2 callersMethod_get_components
(self)
dnnlib/tflib/network.py:209
↓ 2 callersMethod_get_dataset_obj
(self)
metrics/metric_base.py:108
↓ 2 callersMethod_get_device
Get internal state for the given TensorFlow device.
dnnlib/tflib/optimizer.py:84
↓ 2 callersMethod_get_var_global_to_local
(self)
dnnlib/tflib/network.py:317
↓ 2 callersFunction_prepare_nvcc_cli
(opts)
dnnlib/tflib/custom_ops.py:71
↓ 2 callersFunction_run_cmd
(cmd)
dnnlib/tflib/custom_ops.py:64
↓ 2 callersFunction_sanitize_tf_config
(config_dict: dict = None)
dnnlib/tflib/tfutil.py:93
↓ 2 callersFunctionabsolute_name_scope
Forcefully enter the specified name scope, ignoring any surrounding scopes.
dnnlib/tflib/tfutil.py:83
↓ 2 callersMethodadd_images
(self, img)
dataset_tool.py:168
↓ 2 callersFunctionblock
(x, res)
training/networks.py:403
↓ 2 callersMethodclose
(self)
dataset_tool.py:153
↓ 2 callersMethodclose
(self)
training/dataset.py:136
↓ 2 callersFunctionconv2d
(x, w, up=False, down=False, resample_kernel=None, padding=0)
training/networks.py:65
↓ 2 callersFunctiondownsample_2d
r"""Downsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and dow
dnnlib/tflib/ops/upfirdn_2d.py:211
↓ 2 callersMethodevaluate
Evaluate if new feature vectors are in the estimated manifold.
metrics/precision_recall.py:99
↓ 2 callersFunctionexport_samples
(source_idx, tfr_prefix)
dataset_tool.py:782
↓ 2 callersMethodfind_var
Find variable by local or global name.
dnnlib/tflib/network.py:388
↓ 2 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:91
↓ 2 callersFunctionfromrgb
(x, y, res)
training/networks.py:553
↓ 2 callersFunctionfunc_y
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:144
↓ 2 callersMethodget_loss_scaling_var
Get or create variable representing log2 of the current dynamic loss scaling factor.
dnnlib/tflib/optimizer.py:271
↓ 2 callersMethodget_minibatch_tf
(self)
training/dataset.py:149
↓ 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:219
↓ 2 callersMethodget_random_labels_tf
(self, minibatch_size)
training/dataset.py:169
↓ 2 callersMethodget_strength_var
(self)
training/augment.py:97
↓ 2 callersFunctiongrad2_d_dy
(d_dx, d_db, x, y)
dnnlib/tflib/ops/fused_bias_act.py:166
↓ 2 callersFunctiongrad_db
(dx)
dnnlib/tflib/ops/fused_bias_act.py:154
↓ 2 callersFunctiongrad_dx
(dy, x, y)
dnnlib/tflib/ops/fused_bias_act.py:150
↓ 2 callersFunctionis_tf_expression
Check whether the input is a valid Tensorflow expression, i.e., Tensorflow Tensor, Variable, or Operation.
dnnlib/tflib/tfutil.py:36
↓ 2 callersFunctionmake_png_path
(outdir, idx)
dataset_tool.py:742
↓ 2 callersFunctionmodulated_conv2d_layer
(x, y, fmaps, kernel, up=False, down=False, demodulate=True, resample_kernel=None, lrmul=1, fused_modconv=Fals
training/networks.py:92
↓ 2 callersMethodpairwise_distances
Evaluate pairwise distances between two batches of feature vectors.
metrics/precision_recall.py:58
↓ 2 callersMethodreset_optimizer_state
Reset internal state of the underlying optimizer.
dnnlib/tflib/optimizer.py:266
↓ 2 callersFunctionscale_2d
(sx, sy)
training/augment.py:199
↓ 2 callersFunctionslerp
(a, b, t)
metrics/perceptual_path_length.py:27
↓ 2 callersMethodstart
(self, target_images)
projector.py:142
↓ 2 callersFunctiontorgb
(x, y, res)
training/networks.py:422
next →1–100 of 336, ranked by callers